/*

 Additions to Cocoa touch classes

 Copyright 2013 Thincast Technologies GmbH, Authors: Dorian Johnson, Martin Fleisz

 This Source Code Form is subject to the terms of the Mozilla Public License, v. 2.0.

 If a copy of the MPL was not distributed with this file, You can obtain one at http://mozilla.org/MPL/2.0/.

 */

#import "TSXAdditions.h"

#include <openssl/bio.h>

#include <openssl/evp.h>

@implementation NSObject (TSXAdditions)

- (void)setValuesForKeyPathsWithDictionary:(NSDictionary *)keyedValues

{

	for (id keyPath in keyedValues)

		[self setValue:[keyedValues objectForKey:keyPath] forKeyPath:keyPath];

}

- mutableDeepCopy

{

    if([self respondsToSelector:@selector(mutableCopyWithZone:)])

        return [self mutableCopy];

    else if([self respondsToSelector:@selector(copyWithZone:)])

        return [self copy];

    else

        return [self retain];

}

@end

#pragma mark -

@implementation NSString (TSXAdditions)

#pragma mark Creation routines

+ (NSString*)stringWithUUID

{

	CFUUIDRef uuidObj = CFUUIDCreate(nil);

	NSString* uuidString = (NSString*)CFUUIDCreateString(nil, uuidObj);

	CFRelease(uuidObj);

	return [uuidString autorelease];

}

/* Code from http://code.google.com/p/google-toolbox-for-mac/source/browse/trunk/Foundation/GTMNSData%2BHex.m?r=344 */ 

- (NSData*)dataFromHexString

{

    NSData *hexData = [self dataUsingEncoding:NSASCIIStringEncoding];

    const char *hexBuf = [hexData bytes];

    NSUInteger hexLen = [hexData length];

    // This indicates an error converting to ASCII.

    if (!hexData)

        return nil;

    if ((hexLen % 2) != 0) {

        return nil;

    }

    NSMutableData *binaryData = [NSMutableData dataWithLength:(hexLen / 2)];

    unsigned char *binaryPtr = [binaryData mutableBytes];

    unsigned char value = 0;

    for (NSUInteger i = 0; i < hexLen; i++) {

        char c = hexBuf[i];

        if (!isxdigit(c)) {

            return nil;

        }

        if (isdigit(c)) {

            value += c - '0';

        } else if (islower(c)) {

            value += 10 + c - 'a';

        } else {

            value += 10 + c - 'A';

        }

        if (i & 1) {

            *binaryPtr++ = value;

            value = 0;

        } else {

            value <<= 4;

        }

    }

    return [NSData dataWithData:binaryData];    

}

+ (NSString*)hexStringFromData:(const unsigned char *)data ofSize:(unsigned int)size withSeparator:(NSString *)sep afterNthChar:(int)sepnth

{

	int i;

	NSMutableString *result;

	NSString *immutableResult;

	result = [[NSMutableString alloc] init];

	for (i = 0; i < size; i++) {

		if(i && sep && sepnth && i%sepnth==0)

			[result appendString:sep];

		[result appendFormat:@"%02X", data[i]];

	}

	immutableResult = [NSString stringWithString:result];

	[result release];

	return immutableResult;

}

@end

#pragma mark Mutable deep copy for dicionary, array and set

@implementation NSDictionary (TSXAdditions)

- mutableDeepCopy

{

    NSMutableDictionary *newDictionary = [[NSMutableDictionary alloc] init];

    NSEnumerator *enumerator = [self keyEnumerator];

    id key;

    while((key = [enumerator nextObject]))

    {

        id obj = [[self objectForKey:key] mutableDeepCopy];

        [newDictionary setObject:obj forKey:key];

        [obj release];

    }

    return newDictionary;

}

@end

@implementation NSArray (TSXAdditions)

- mutableDeepCopy

{

    NSMutableArray *newArray = [[NSMutableArray alloc] init];

    NSEnumerator *enumerator = [self objectEnumerator];

    id obj;

    while((obj = [enumerator nextObject]))

    {

        obj = [obj mutableDeepCopy];

        [newArray addObject:obj];

        [obj release];

    }

    return newArray;

}

@end

@implementation NSSet (TSXAdditions)

- mutableDeepCopy

{

    NSMutableSet *newSet = [[NSMutableSet alloc] init];

    NSEnumerator *enumerator = [self objectEnumerator];

    id obj;

    while((obj = [enumerator nextObject]))

    {

        obj = [obj mutableDeepCopy];

        [newSet addObject:obj];

        [obj release];

    }

    return newSet;

}

@end

#pragma mark -

/* Code from http://stackoverflow.com/questions/1305225/best-way-to-serialize-a-nsdata-into-an-hexadeximal-string */

@implementation NSData (TSXAdditions)

#pragma mark - String Conversion

- (NSString *)hexadecimalString {

    /* Returns hexadecimal string of NSData. Empty string if data is empty.   */

    const unsigned char *dataBuffer = (const unsigned char *)[self bytes];

    if (!dataBuffer)

        return [NSString string];

    NSUInteger          dataLength  = [self length];

    NSMutableString     *hexString  = [NSMutableString stringWithCapacity:(dataLength * 2)];

    for (int i = 0; i < dataLength; ++i)

        [hexString appendString:[NSString stringWithFormat:@"%02lx", (unsigned long)dataBuffer[i]]];

    return [NSString stringWithString:hexString];

}

/* Code from http://cocoawithlove.com/2009/06/base64-encoding-options-on-mac-and.html */

- (NSString *)base64EncodedString

{

    // Construct an OpenSSL context

    BIO *context = BIO_new(BIO_s_mem());

    // Tell the context to encode base64

    BIO *command = BIO_new(BIO_f_base64());

    context = BIO_push(command, context);

    BIO_set_flags(context, BIO_FLAGS_BASE64_NO_NL);

    // Encode all the data

    BIO_write(context, [self bytes], [self length]);

    (void)BIO_flush(context);

    // Get the data out of the context

    char *outputBuffer;

    long outputLength = BIO_get_mem_data(context, &outputBuffer);

    NSString *encodedString = [[NSString alloc] initWithBytes:outputBuffer length:outputLength encoding:NSASCIIStringEncoding];

    BIO_free_all(context);

    return encodedString;

}

@end