#import "SSHKey.h"

@implementation SSHKey

/* Construct a SSHKey object with thePath as private key. */
+ (id)keyWithPath:(NSString *)thePath
{
	thePath = [thePath stringByExpandingTildeInPath];

	return [[[self alloc] initWithPath:thePath] autorelease];
}

+ (SSHKeyType) typeOfKeyAtPath:(NSString *)thePath
{
	/* Read the file and determine what type we're working with. */
	NSMutableData *data = [NSMutableData dataWithContentsOfFile:thePath];
	
	if(data == nil)
		return SSH_KEYTYPE_UNKNOWN;
	
	const char *dataPtr = [data bytes];
	const char *newline = index(dataPtr, '\n');
	
	NSString *header = [NSString string];
	if (newline != NULL)
	{
		[data setLength:newline - dataPtr];
		header = [[[NSString alloc] initWithData:data encoding:NSASCIIStringEncoding] autorelease];
	}
	
	if ([header isEqualToString:@"SSH PRIVATE KEY FILE FORMAT 1.1"])
		return SSH_KEYTYPE_RSA1;
	else if ([header isEqualToString:@"-----BEGIN RSA PRIVATE KEY-----"])
		return SSH_KEYTYPE_RSA;
	else if ([header isEqualToString:@"-----BEGIN DSA PRIVATE KEY-----"])
		return SSH_KEYTYPE_DSA;
	else
		return SSH_KEYTYPE_UNKNOWN;
}

- (id)initWithPath:(NSString *)thePath
{

	if (!(self = [super init]))
		return nil;

	fullPath = [thePath copy];
	type = [[self class] typeOfKeyAtPath:thePath];

	return self;
}

- (void) dealloc
{
	[fullPath release];
	[super dealloc];
}

/* Return the type of the private key as integer. */
- (SSHKeyType)type
{
	return type;
}

/* Return the type of the private key as NSString. */
- (NSString *)typeAsString
{
	switch (type)
	{
		case SSH_KEYTYPE_RSA1:
			return @"RSA1";
		case SSH_KEYTYPE_RSA:
			return @"RSA";
		case SSH_KEYTYPE_DSA:
			return @"DSA";
		case SSH_KEYTYPE_UNKNOWN:
			return nil;
	}
	return nil;
}

/* Return the path of the private key. */
- (NSString *)path
{
	return [[fullPath copy] autorelease];
}

@end