//
//  NSString+Additions.m
//  Perforce
//
//  Created by Adam Czubernat on 16.05.2013.
//  Copyright (c) 2013 Perforce Software, Inc. All rights reserved.
//

#import "NSString+Additions.h"

@implementation NSString (Additions)

+ (NSString *)stringWithTimeInterval:(NSTimeInterval)timeInterval {
	if (timeInterval == INFINITY)
		return @"∞";
	NSInteger time = (NSInteger)timeInterval;
	return [NSString stringWithFormat:@"%@%@%@",
			time < 3600 ? @"" : [NSString stringWithFormat:@"%02ld:", time/3600],
			time < 60 ? @"" : [NSString stringWithFormat:@"%02ld:", (time/60)%60],
			[NSString stringWithFormat:(time < 60 ? @"%lds" : @"%02ld"), time%60]];
}

+ (NSString *)stringWithByteCount:(NSInteger)bytes {
	NSString *units;
	int decimals = 0;
	CGFloat size = bytes;
	if (size < 1024.0f)
		units = @"bytes";
	else if ((size /= 1024.0f) < 1024.0f)
		units = @"KB";
	else if ((size /= 1024.0f) < 1024.0f)
		units = @"MB", decimals = 1;
	else
		size /= 1024.0f, units = @"GB", decimals = 2;
	
	return [NSString stringWithFormat:@"%.*f %@", decimals, size, units];
}

- (NSString *)stringByRemovingPrefix:(NSString *)prefix {
	return ([self hasPrefix:prefix] ?
			[self substringFromIndex:prefix.length] :
			self);
}

- (NSString *)stringByRemovingSuffix:(NSString *)suffix {
	return ([self hasSuffix:suffix] ?
			[self substringToIndex:self.length - suffix.length] :
			self);
}

- (NSArray *)arrayOfArguments {

	CFStringRef string = (__bridge CFStringRef)self;
	CFIndex length = CFStringGetLength(string);
	
	NSMutableArray *array = [NSMutableArray arrayWithCapacity:length];

	unichar quoteChar;
	CFIndex bufferLength = 0;
	unichar *buffer = malloc(sizeof(unichar) * length);
	BOOL inQuote = NO;

	for (CFIndex i=0; i<length; i++) {
		unichar c = CFStringGetCharacterAtIndex(string, i);
		if (!inQuote && (c==0x0020 || c==0x0009 || c==0x000A || c==0x000D || c==0x0085)) {
			if (bufferLength) {
				CFStringRef argument = CFStringCreateWithCharacters(NULL, buffer, bufferLength);
				[array addObject:CFBridgingRelease(argument)];
				bufferLength = 0;
			}
			continue;
		}
		
		if ((c=='\'' || c=='"') && (!inQuote || c==quoteChar)) {
			inQuote = !inQuote;
			quoteChar = c;
			continue;
		}
		
		buffer[bufferLength++] = c;
	}
	
	if (bufferLength) {
		CFStringRef argument = CFStringCreateWithCharacters(NULL, buffer, bufferLength);
		[array addObject:CFBridgingRelease(argument)];
	}
	
	free(buffer);	
	
	return array;
}

- (CGSize)sizeWithFont:(NSFont *)font {
	return [self sizeWithAttributes:@{ NSFontAttributeName : font }];
}

#pragma mark - Path support

- (NSString *)stringByDeletingPath {
	NSRange range = { 0, self.length-1 };
	range = [self rangeOfString:@"/" options:NSBackwardsSearch range:range];
	NSUInteger loc = range.location;
	if (loc != NSNotFound)
		return [self substringToIndex:loc+1];
	return self;
}

- (NSString *)stringByAppendingPath:(NSString *)path {
	return ([self hasSuffix:@"/"] ?
			[self stringByAppendingString:path] :
			[self stringByAppendingFormat:@"/%@", path]);
}

- (NSString *)stringByRenamingPath:(NSString *)newName {
	NSString *newPath = [[self stringByDeletingPath] stringByAppendingPath:newName];
	return [self hasSuffix:@"/"] ? [newPath stringByAppendingString:@"/"] : newPath;
}

- (NSString *)directoryPath {
	if (self.length)
		return [self hasSuffix:@"/"] ? self : [self stringByAppendingString:@"/"];
	return nil;
}

- (BOOL)isSubpath:(NSString *)base {
	NSString *path = [base directoryPath];
	return [self hasPrefixCaseInsensitive:path];
}

- (NSString *)relativePath:(NSString *)base {
	NSString *path = [base directoryPath];
	if ([self hasPrefixCaseInsensitive:path])
		return [self substringFromIndex:path.length];
	return nil;
}

#pragma mark -

- (BOOL)isEqualCaseInsensitive:(NSString *)string {
	return string && [self caseInsensitiveCompare:string] == NSOrderedSame;
}

- (BOOL)isEmptyString {
	NSCharacterSet *set = [NSCharacterSet whitespaceAndNewlineCharacterSet];
	return [self stringByTrimmingCharactersInSet:set].length == 0;
}

- (BOOL)hasPrefixCaseInsensitive:(NSString *)prefix {
	return prefix && [self
					  rangeOfString:prefix
					  options:NSAnchoredSearch |
					  NSCaseInsensitiveSearch
					  ].location != NSNotFound;
}

- (BOOL)hasSuffixCaseInsensitive:(NSString *)suffix {
	return suffix && [self
					  rangeOfString:suffix
					  options:NSAnchoredSearch |
					  NSCaseInsensitiveSearch |
					  NSBackwardsSearch
					  ].location != NSNotFound;
}

@end