#if !__has_feature(objc_arc)

#error The JSONMOdel framework is ARC only, you can enable ARC on per file basis.

#endif

#import <objc/runtime.h>

#import <objc/message.h>

#import "JSONModel.h"

#import "JSONModelClassProperty.h"

#pragma mark - associated objects names

static const char * kMapperObjectKey;

static const char * kClassPropertiesKey;

static const char * kClassRequiredPropertyNamesKey;

static const char * kIndexPropertyNameKey;

#pragma mark - class static variables

static NSArray* allowedJSONTypes = nil;

static NSArray* allowedPrimitiveTypes = nil;

static JSONValueTransformer* valueTransformer = nil;

static Class JSONModelClass = NULL;

#pragma mark - model cache

static JSONKeyMapper* globalKeyMapper = nil;

#pragma mark - JSONModel implementation

@implementation JSONModel

{

NSString* _description;

}

#pragma mark - initialization methods

+(void)load

{

static dispatch_once_t once;

dispatch_once(&once, ^{

// initialize all class static objects,

// which are common for ALL JSONModel subclasses

@autoreleasepool {

allowedJSONTypes = @[

[NSString class], [NSNumber class], [NSDecimalNumber class], [NSArray class], [NSDictionary class], [NSNull class], //immutable JSON classes

[NSMutableString class], [NSMutableArray class], [NSMutableDictionary class] //mutable JSON classes

];

allowedPrimitiveTypes = @[

@"BOOL", @"float", @"int", @"long", @"double", @"short",

@"unsigned int", @"usigned long", @"long long", @"unsigned long long", @"unsigned short", @"char", @"unsigned char",

//and some famous aliases

@"NSInteger", @"NSUInteger",

@"Block"

];

valueTransformer = [[JSONValueTransformer alloc] init];

// This is quite strange, but I found the test isSubclassOfClass: (line ~291) to fail if using [JSONModel class].

// somewhat related: https://stackoverflow.com/questions/6524165/nsclassfromstring-vs-classnamednsstring

// //; seems to break the unit tests

// Using NSClassFromString instead of [JSONModel class], as this was breaking unit tests, see below

//http://stackoverflow.com/questions/21394919/xcode-5-unit-test-seeing-wrong-class

JSONModelClass = NSClassFromString(NSStringFromClass(self));

}

});

}

-(void)__setup__

{

//if first instance of this model, generate the property list

if (!objc_getAssociatedObject(self.class, &kClassPropertiesKey)) {

[self __inspectProperties];

}

//if there's a custom key mapper, store it in the associated object

id mapper = [[self class] keyMapper];

if ( mapper && !objc_getAssociatedObject(self.class, &kMapperObjectKey) ) {

objc_setAssociatedObject(

self.class,

&kMapperObjectKey,

mapper,

OBJC_ASSOCIATION_RETAIN // This is atomic

);

}

}

-(id)init

{

self = [super init];

if (self) {

//do initial class setup

[self __setup__];

}

return self;

}

-(instancetype)initWithData:(NSData *)data error:(NSError *__autoreleasing *)err

{

//check for nil input

if (!data) {

if (err) *err = [JSONModelError errorInputIsNil];

return nil;

}

//read the json

JSONModelError* initError = nil;

id obj = [NSJSONSerialization JSONObjectWithData:data

options:kNilOptions

error:&initError];

if (initError) {

if (err) *err = [JSONModelError errorBadJSON];

return nil;

}

//init with dictionary

id objModel = [self initWithDictionary:obj error:&initError];

if (initError && err) *err = initError;

return objModel;

}

-(id)initWithString:(NSString*)string error:(JSONModelError**)err

{

JSONModelError* initError = nil;

id objModel = [self initWithString:string usingEncoding:NSUTF8StringEncoding error:&initError];

if (initError && err) *err = initError;

return objModel;

}

-(id)initWithString:(NSString *)string usingEncoding:(NSStringEncoding)encoding error:(JSONModelError**)err

{

//check for nil input

if (!string) {

if (err) *err = [JSONModelError errorInputIsNil];

return nil;

}

JSONModelError* initError = nil;

id objModel = [self initWithData:[string dataUsingEncoding:encoding] error:&initError];

if (initError && err) *err = initError;

return objModel;

}

-(id)initWithDictionary:(NSDictionary*)dict error:(NSError**)err

{

//check for nil input

if (!dict) {

if (err) *err = [JSONModelError errorInputIsNil];

return nil;

}

//invalid input, just create empty instance

if (![dict isKindOfClass:[NSDictionary class]]) {

if (err) *err = [JSONModelError errorInvalidDataWithMessage:@"Attempt to initialize JSONModel object using initWithDictionary:error: but the dictionary parameter was not an 'NSDictionary'."];

return nil;

}

//create a class instance

self = [self init];

if (!self) {

//super init didn't succeed

if (err) *err = [JSONModelError errorModelIsInvalid];

return nil;

}

//check incoming data structure

if (![self __doesDictionary:dict matchModelWithKeyMapper:self.__keyMapper error:err]) {

return nil;

}

//import the data from a dictionary

if (![self __importDictionary:dict withKeyMapper:self.__keyMapper validation:YES error:err]) {

return nil;

}

//run any custom model validation

if (![self validate:err]) {

return nil;

}

//model is valid! yay!

return self;

}

-(JSONKeyMapper*)__keyMapper

{

//get the model key mapper

return objc_getAssociatedObject(self.class, &kMapperObjectKey);

}

-(BOOL)__doesDictionary:(NSDictionary*)dict matchModelWithKeyMapper:(JSONKeyMapper*)keyMapper error:(NSError**)err

{

//check if all required properties are present

NSArray* incomingKeysArray = [dict allKeys];

NSMutableSet* requiredProperties = [self __requiredPropertyNames].mutableCopy;

NSSet* incomingKeys = [NSSet setWithArray: incomingKeysArray];

//transform the key names, if necessary

if (keyMapper || globalKeyMapper) {

NSMutableSet* transformedIncomingKeys = [NSMutableSet setWithCapacity: requiredProperties.count];

NSString* transformedName = nil;

//loop over the required properties list

for (JSONModelClassProperty* property in [self __properties__]) {

transformedName = (keyMapper||globalKeyMapper) ? [self __mapString:property.name withKeyMapper:keyMapper] : property.name;

//check if exists and if so, add to incoming keys

id value;

@try {

value = [dict valueForKeyPath:transformedName];

}

@catch (NSException *exception) {

value = dict[transformedName];

}

if (value) {

[transformedIncomingKeys addObject: property.name];

}

}

//overwrite the raw incoming list with the mapped key names

incomingKeys = transformedIncomingKeys;

}

//check for missing input keys

if (![requiredProperties isSubsetOfSet:incomingKeys]) {

//get a list of the missing properties

[requiredProperties minusSet:incomingKeys];

//not all required properties are in - invalid input

JMLog(@"Incoming data was invalid [%@ initWithDictionary:]. Keys missing: %@", self.class, requiredProperties);

if (err) *err = [JSONModelError errorInvalidDataWithMissingKeys:requiredProperties];

return NO;

}

//not needed anymore

incomingKeys= nil;

requiredProperties= nil;

return YES;

}

-(NSString*)__mapString:(NSString*)string withKeyMapper:(JSONKeyMapper*)keyMapper

{

if (keyMapper) {

//custom mapper

NSString* mappedName = [keyMapper convertValue:string];

if (globalKeyMapper && [mappedName isEqualToString: string]) {

mappedName = [globalKeyMapper convertValue:string];

}

string = mappedName;

} else if (globalKeyMapper) {

//global keymapper

string = [globalKeyMapper convertValue:string];

}

return string;

}

-(BOOL)__importDictionary:(NSDictionary*)dict withKeyMapper:(JSONKeyMapper*)keyMapper validation:(BOOL)validation error:(NSError**)err

{

//loop over the incoming keys and set self's properties

for (JSONModelClassProperty* property in [self __properties__]) {

//convert key name to model keys, if a mapper is provided

NSString* jsonKeyPath = (keyMapper||globalKeyMapper) ? [self __mapString:property.name withKeyMapper:keyMapper] : property.name;

//JMLog(@"keyPath: %@", jsonKeyPath);

//general check for data type compliance

id jsonValue;

@try {

jsonValue = [dict valueForKeyPath: jsonKeyPath];

}

@catch (NSException *exception) {

jsonValue = dict[jsonKeyPath];

}

//check for Optional properties

if (isNull(jsonValue)) {

//skip this property, continue with next property

if (property.isOptional || !validation) continue;

if (err) {

//null value for required property

NSString* msg = [NSString stringWithFormat:@"Value of required model key %@ is null", property.name];

JSONModelError* dataErr = [JSONModelError errorInvalidDataWithMessage:msg];

*err = [dataErr errorByPrependingKeyPathComponent:property.name];

}

return NO;

}

Class jsonValueClass = [jsonValue class];

BOOL isValueOfAllowedType = NO;

for (Class allowedType in allowedJSONTypes) {

if ( [jsonValueClass isSubclassOfClass: allowedType] ) {

isValueOfAllowedType = YES;

break;

}

}

if (isValueOfAllowedType==NO) {

//type not allowed

JMLog(@"Type %@ is not allowed in JSON.", NSStringFromClass(jsonValueClass));

if (err) {

NSString* msg = [NSString stringWithFormat:@"Type %@ is not allowed in JSON.", NSStringFromClass(jsonValueClass)];

JSONModelError* dataErr = [JSONModelError errorInvalidDataWithMessage:msg];

*err = [dataErr errorByPrependingKeyPathComponent:property.name];

}

return NO;

}

//check if there's matching property in the model

if (property) {

// check for custom setter, than the model doesn't need to do any guessing

// how to read the property's value from JSON

if ([self __customSetValue:jsonValue forProperty:property]) {

//skip to next JSON key

continue;

};

// 0) handle primitives

if (property.type == nil && property.structName==nil) {

//generic setter

if (jsonValue != [self valueForKey:property.name]) {

[self setValue:jsonValue forKey: property.name];

}

//skip directly to the next key

continue;

}

// 0.5) handle nils

if (isNull(jsonValue)) {

if ([self valueForKey:property.name] != nil) {

[self setValue:nil forKey: property.name];

}

continue;

}

// 1) check if property is itself a JSONModel

if ([self __isJSONModelSubClass:property.type]) {

//initialize the property's model, store it

JSONModelError* initErr = nil;

id value = [[property.type alloc] initWithDictionary: jsonValue error:&initErr];

if (!value) {

//skip this property, continue with next property

if (property.isOptional || !validation) continue;

// Propagate the error, including the property name as the key-path component

if((err != nil) && (initErr != nil))

{

*err = [initErr errorByPrependingKeyPathComponent:property.name];

}

return NO;

}

if (![value isEqual:[self valueForKey:property.name]]) {

[self setValue:value forKey: property.name];

}

//for clarity, does the same without continue

continue;

} else {

// 2) check if there's a protocol to the property

// ) might or not be the case there's a built in transform for it

if (property.protocol) {

//JMLog(@"proto: %@", p.protocol);

jsonValue = [self __transform:jsonValue forProperty:property error:err];

if (!jsonValue) {

if ((err != nil) && (*err == nil)) {

NSString* msg = [NSString stringWithFormat:@"Failed to transform value, but no error was set during transformation. (%@)", property];

JSONModelError* dataErr = [JSONModelError errorInvalidDataWithMessage:msg];

*err = [dataErr errorByPrependingKeyPathComponent:property.name];

}

return NO;

}

}

// 3.1) handle matching standard JSON types

if (property.isStandardJSONType && [jsonValue isKindOfClass: property.type]) {

//mutable properties

if (property.isMutable) {

jsonValue = [jsonValue mutableCopy];

}

//set the property value

if (![jsonValue isEqual:[self valueForKey:property.name]]) {

[self setValue:jsonValue forKey: property.name];

}

continue;

}

// 3.3) handle values to transform

if (

(![jsonValue isKindOfClass:property.type] && !isNull(jsonValue))

||

//the property is mutable

property.isMutable

||

//custom struct property

property.structName

) {

// searched around the web how to do this better

// but did not find any solution, maybe that's the best idea? (hardly)

Class sourceClass = [JSONValueTransformer classByResolvingClusterClasses:[jsonValue class]];

//JMLog(@"to type: [%@] from type: [%@] transformer: [%@]", p.type, sourceClass, selectorName);

//build a method selector for the property and json object classes

NSString* selectorName = [NSString stringWithFormat:@"%@From%@:",

(property.structName? property.structName : property.type), //target name

sourceClass]; //source name

SEL selector = NSSelectorFromString(selectorName);

//check for custom transformer

BOOL foundCustomTransformer = NO;

if ([valueTransformer respondsToSelector:selector]) {

foundCustomTransformer = YES;

} else {

//try for hidden custom transformer

selectorName = [NSString stringWithFormat:@"__%@",selectorName];

selector = NSSelectorFromString(selectorName);

if ([valueTransformer respondsToSelector:selector]) {

foundCustomTransformer = YES;

}

}

//check if there's a transformer with that name

if (foundCustomTransformer) {

IMP imp = [valueTransformer methodForSelector:selector];

id (*func)(id, SEL, id) = (void *)imp;

jsonValue = func(valueTransformer, selector, jsonValue);

if (![jsonValue isEqual:[self valueForKey:property.name]])

[self setValue:jsonValue forKey:property.name];

} else {

if (err) {

NSString* msg = [NSString stringWithFormat:@"%@ type not supported for %@.%@", property.type, [self class], property.name];

JSONModelError* dataErr = [JSONModelError errorInvalidDataWithTypeMismatch:msg];

*err = [dataErr errorByPrependingKeyPathComponent:property.name];

}

return NO;

}

} else {

// 3.4) handle "all other" cases (if any)

if (![jsonValue isEqual:[self valueForKey:property.name]])

[self setValue:jsonValue forKey:property.name];

}

}

}

}

return YES;

}

#pragma mark - property inspection methods

-(BOOL)__isJSONModelSubClass:(Class)class

{

#ifdef UNIT_TESTING

return [@"JSONModel" isEqualToString: NSStringFromClass([class superclass])];

#else

return [class isSubclassOfClass:JSONModelClass];

#endif

}

-(NSMutableSet*)__requiredPropertyNames

{

//fetch the associated property names

NSMutableSet* classRequiredPropertyNames = objc_getAssociatedObject(self.class, &kClassRequiredPropertyNamesKey);

if (!classRequiredPropertyNames) {

classRequiredPropertyNames = [NSMutableSet set];

[[self __properties__] enumerateObjectsUsingBlock:^(JSONModelClassProperty* p, NSUInteger idx, BOOL *stop) {

if (!p.isOptional) [classRequiredPropertyNames addObject:p.name];

}];

//persist the list

objc_setAssociatedObject(

self.class,

&kClassRequiredPropertyNamesKey,

classRequiredPropertyNames,

OBJC_ASSOCIATION_RETAIN // This is atomic

);

}

return classRequiredPropertyNames;

}

-(NSArray*)__properties__

{

//fetch the associated object

NSDictionary* classProperties = objc_getAssociatedObject(self.class, &kClassPropertiesKey);

if (classProperties) return [classProperties allValues];

//if here, the class needs to inspect itself

[self __setup__];

//return the property list

classProperties = objc_getAssociatedObject(self.class, &kClassPropertiesKey);

return [classProperties allValues];

}

-(void)__inspectProperties

{

//JMLog(@"Inspect class: %@", [self class]);

NSMutableDictionary* propertyIndex = [NSMutableDictionary dictionary];

//temp variables for the loops

Class class = [self class];

NSScanner* scanner = nil;

NSString* propertyType = nil;

// inspect inherited properties up to the JSONModel class

while (class != [JSONModel class]) {

//JMLog(@"inspecting: %@", NSStringFromClass(class));

unsigned int propertyCount;

objc_property_t *properties = class_copyPropertyList(class, &propertyCount);

//loop over the class properties

for (unsigned int i = 0; i < propertyCount; i++) {

JSONModelClassProperty* p = [[JSONModelClassProperty alloc] init];

//get property name

objc_property_t property = properties[i];

const char *propertyName = property_getName(property);

p.name = @(propertyName);

//JMLog(@"property: %@", p.name);

//get property attributes

const char *attrs = property_getAttributes(property);

NSString* propertyAttributes = @(attrs);

NSArray* attributeItems = [propertyAttributes componentsSeparatedByString:@","];

//ignore read-only properties

if ([attributeItems containsObject:@"R"]) {

continue; //to next property

}

scanner = [NSScanner scannerWithString: propertyAttributes];

//JMLog(@"attr: %@", [NSString stringWithCString:attrs encoding:NSUTF8StringEncoding]);

[scanner scanUpToString:@"T" intoString: nil];

[scanner scanString:@"T" intoString:nil];

//check if the property is an instance of a class

if ([scanner scanString:@"@\"" intoString: &propertyType]) {

[scanner scanUpToCharactersFromSet:[NSCharacterSet characterSetWithCharactersInString:@"\"<"]

intoString:&propertyType];

//JMLog(@"type: %@", propertyClassName);

p.type = NSClassFromString(propertyType);

p.isMutable = ([propertyType rangeOfString:@"Mutable"].location != NSNotFound);

p.isStandardJSONType = [allowedJSONTypes containsObject:p.type];

//read through the property protocols

while ([scanner scanString:@"<" intoString:NULL]) {

NSString* protocolName = nil;

[scanner scanUpToString:@">" intoString: &protocolName];

if ([protocolName isEqualToString:@"Optional"]) {

p.isOptional = YES;

} else if([protocolName isEqualToString:@"Index"]) {

#pragma GCC diagnostic push

#pragma GCC diagnostic ignored "-Wdeprecated-declarations"

p.isIndex = YES;

#pragma GCC diagnostic pop

objc_setAssociatedObject(

self.class,

&kIndexPropertyNameKey,

p.name,

OBJC_ASSOCIATION_RETAIN // This is atomic

);

} else if([protocolName isEqualToString:@"Ignore"]) {

p = nil;

} else {

p.protocol = protocolName;

}

[scanner scanString:@">" intoString:NULL];

}

}

//check if the property is a structure

else if ([scanner scanString:@"{" intoString: &propertyType]) {

[scanner scanCharactersFromSet:[NSCharacterSet alphanumericCharacterSet]

intoString:&propertyType];

p.isStandardJSONType = NO;

p.structName = propertyType;

}

//the property must be a primitive

else {

//the property contains a primitive data type

[scanner scanUpToCharactersFromSet:[NSCharacterSet characterSetWithCharactersInString:@","]

intoString:&propertyType];

//get the full name of the primitive type

propertyType = valueTransformer.primitivesNames[propertyType];

if (![allowedPrimitiveTypes containsObject:propertyType]) {

//type not allowed - programmer mistaken -> exception

@throw [NSException exceptionWithName:@"JSONModelProperty type not allowed"

reason:[NSString stringWithFormat:@"Property type of %@.%@ is not supported by JSONModel.", self.class, p.name]

userInfo:nil];

}

}

NSString *nsPropertyName = @(propertyName);

if([[self class] propertyIsOptional:nsPropertyName]){

p.isOptional = YES;

}

if([[self class] propertyIsIgnored:nsPropertyName]){

p = nil;

}

Class customClass = [[self class] classForCollectionProperty:nsPropertyName];

if (customClass) {

p.protocol = NSStringFromClass(customClass);

}

//few cases where JSONModel will ignore properties automatically

if ([propertyType isEqualToString:@"Block"]) {

p = nil;

}

//add the property object to the temp index

if (p && ![propertyIndex objectForKey:p.name]) {

[propertyIndex setValue:p forKey:p.name];

}

// generate custom setters and getter

if (p)

{

NSString *name = [p.name stringByReplacingCharactersInRange:NSMakeRange(0, 1) withString:[p.name substringToIndex:1].uppercaseString];

// getter

SEL getter = NSSelectorFromString([NSString stringWithFormat:@"JSONObjectFor%@", name]);

if ([self respondsToSelector:getter])

p.customGetter = getter;

// setters

p.customSetters = [NSMutableDictionary new];

SEL genericSetter = NSSelectorFromString([NSString stringWithFormat:@"set%@WithJSONObject:", name]);

if ([self respondsToSelector:genericSetter])

p.customSetters[@"generic"] = [NSValue valueWithBytes:&genericSetter objCType:@encode(SEL)];

for (Class type in allowedJSONTypes)

{

NSString *class = NSStringFromClass([JSONValueTransformer classByResolvingClusterClasses:type]);

if (p.customSetters[class])

continue;

SEL setter = NSSelectorFromString([NSString stringWithFormat:@"set%@With%@:", name, class]);

if ([self respondsToSelector:setter])

p.customSetters[class] = [NSValue valueWithBytes:&setter objCType:@encode(SEL)];

}

}

}

free(properties);

//ascend to the super of the class

//(will do that until it reaches the root class - JSONModel)

class = [class superclass];

}

//finally store the property index in the static property index

objc_setAssociatedObject(

self.class,

&kClassPropertiesKey,

[propertyIndex copy],

OBJC_ASSOCIATION_RETAIN // This is atomic

);

}

#pragma mark - built-in transformer methods

-(id)__transform:(id)value forProperty:(JSONModelClassProperty*)property error:(NSError**)err

{

Class protocolClass = NSClassFromString(property.protocol);

if (!protocolClass) {

//no other protocols on arrays and dictionaries

//except JSONModel classes

if ([value isKindOfClass:[NSArray class]]) {

@throw [NSException exceptionWithName:@"Bad property protocol declaration"

reason:[NSString stringWithFormat:@"<%@> is not allowed JSONModel property protocol, and not a JSONModel class.", property.protocol]

userInfo:nil];

}

return value;

}

//if the protocol is actually a JSONModel class

if ([self __isJSONModelSubClass:protocolClass]) {

//check if it's a list of models

if ([property.type isSubclassOfClass:[NSArray class]]) {

// Expecting an array, make sure 'value' is an array

if(![[value class] isSubclassOfClass:[NSArray class]])

{

if(err != nil)

{

NSString* mismatch = [NSString stringWithFormat:@"Property '%@' is declared as NSArray<%@>* but the corresponding JSON value is not a JSON Array.", property.name, property.protocol];

JSONModelError* typeErr = [JSONModelError errorInvalidDataWithTypeMismatch:mismatch];

*err = [typeErr errorByPrependingKeyPathComponent:property.name];

}

return nil;

}

//one shot conversion

JSONModelError* arrayErr = nil;

value = [[protocolClass class] arrayOfModelsFromDictionaries:value error:&arrayErr];

if((err != nil) && (arrayErr != nil))

{

*err = [arrayErr errorByPrependingKeyPathComponent:property.name];

return nil;

}

}

//check if it's a dictionary of models

if ([property.type isSubclassOfClass:[NSDictionary class]]) {

// Expecting a dictionary, make sure 'value' is a dictionary

if(![[value class] isSubclassOfClass:[NSDictionary class]])

{

if(err != nil)

{

NSString* mismatch = [NSString stringWithFormat:@"Property '%@' is declared as NSDictionary<%@>* but the corresponding JSON value is not a JSON Object.", property.name, property.protocol];

JSONModelError* typeErr = [JSONModelError errorInvalidDataWithTypeMismatch:mismatch];

*err = [typeErr errorByPrependingKeyPathComponent:property.name];

}

return nil;

}

NSMutableDictionary* res = [NSMutableDictionary dictionary];

for (NSString* key in [value allKeys]) {

JSONModelError* initErr = nil;

id obj = [[[protocolClass class] alloc] initWithDictionary:value[key] error:&initErr];

if (obj == nil)

{

// Propagate the error, including the property name as the key-path component

if((err != nil) && (initErr != nil))

{

initErr = [initErr errorByPrependingKeyPathComponent:key];

*err = [initErr errorByPrependingKeyPathComponent:property.name];

}

return nil;

}

[res setValue:obj forKey:key];

}

value = [NSDictionary dictionaryWithDictionary:res];

}

}

return value;

}

-(id)__reverseTransform:(id)value forProperty:(JSONModelClassProperty*)property

{

Class protocolClass = NSClassFromString(property.protocol);

if (!protocolClass) return value;

//if the protocol is actually a JSONModel class

if ([self __isJSONModelSubClass:protocolClass]) {

//check if should export list of dictionaries

if (property.type == [NSArray class] || property.type == [NSMutableArray class]) {

NSMutableArray* tempArray = [NSMutableArray arrayWithCapacity: [(NSArray*)value count] ];

for (NSObject<AbstractJSONModelProtocol>* model in (NSArray*)value) {

if ([model respondsToSelector:@selector(toDictionary)]) {

[tempArray addObject: [model toDictionary]];

} else

[tempArray addObject: model];

}

return [tempArray copy];

}

//check if should export dictionary of dictionaries

if (property.type == [NSDictionary class] || property.type == [NSMutableDictionary class]) {

NSMutableDictionary* res = [NSMutableDictionary dictionary];

for (NSString* key in [(NSDictionary*)value allKeys]) {

id<AbstractJSONModelProtocol> model = value[key];

[res setValue: [model toDictionary] forKey: key];

}

return [NSDictionary dictionaryWithDictionary:res];

}

}

return value;

}

#pragma mark - custom transformations

- (BOOL)__customSetValue:(id <NSObject>)value forProperty:(JSONModelClassProperty *)property

{

NSString *class = NSStringFromClass([JSONValueTransformer classByResolvingClusterClasses:[value class]]);

SEL setter = nil;

[property.customSetters[class] getValue:&setter];

if (!setter)

[property.customSetters[@"generic"] getValue:&setter];

if (!setter)

return NO;

IMP imp = [self methodForSelector:setter];

void (*func)(id, SEL, id <NSObject>) = (void *)imp;

func(self, setter, value);

return YES;

}

- (BOOL)__customGetValue:(id *)value forProperty:(JSONModelClassProperty *)property

{

SEL getter = property.customGetter;

if (!getter)

return NO;

IMP imp = [self methodForSelector:getter];

id (*func)(id, SEL) = (void *)imp;

*value = func(self, getter);

return YES;

}

#pragma mark - persistance

-(void)__createDictionariesForKeyPath:(NSString*)keyPath inDictionary:(NSMutableDictionary**)dict

{

//find if there's a dot left in the keyPath

NSUInteger dotLocation = [keyPath rangeOfString:@"."].location;

if (dotLocation==NSNotFound) return;

//inspect next level

NSString* nextHierarchyLevelKeyName = [keyPath substringToIndex: dotLocation];

NSDictionary* nextLevelDictionary = (*dict)[nextHierarchyLevelKeyName];

if (nextLevelDictionary==nil) {

//create non-existing next level here

nextLevelDictionary = [NSMutableDictionary dictionary];

}

//recurse levels

[self __createDictionariesForKeyPath:[keyPath substringFromIndex: dotLocation+1]

inDictionary:&nextLevelDictionary ];

//create the hierarchy level

[*dict setValue:nextLevelDictionary forKeyPath: nextHierarchyLevelKeyName];

}

-(NSDictionary*)toDictionary

{

return [self toDictionaryWithKeys:nil];

}

-(NSString*)toJSONString

{

return [self toJSONStringWithKeys:nil];

}

-(NSData*)toJSONData

{

return [self toJSONDataWithKeys:nil];

}

- (NSDictionary *)toDictionaryWithKeys:(NSArray <NSString *> *)propertyNames

{

NSArray* properties = [self __properties__];

NSMutableDictionary* tempDictionary = [NSMutableDictionary dictionaryWithCapacity:properties.count];

id value;

//loop over all properties

for (JSONModelClassProperty* p in properties) {

//skip if unwanted

if (propertyNames != nil && ![propertyNames containsObject:p.name])

continue;

//fetch key and value

NSString* keyPath = (self.__keyMapper||globalKeyMapper) ? [self __mapString:p.name withKeyMapper:self.__keyMapper] : p.name;

value = [self valueForKey: p.name];

//JMLog(@"toDictionary[%@]->[%@] = '%@'", p.name, keyPath, value);

if ([keyPath rangeOfString:@"."].location != NSNotFound) {

//there are sub-keys, introduce dictionaries for them

[self __createDictionariesForKeyPath:keyPath inDictionary:&tempDictionary];

}

//check for custom getter

if ([self __customGetValue:&value forProperty:p]) {

//custom getter, all done

[tempDictionary setValue:value forKeyPath:keyPath];

continue;

}

//export nil when they are not optional values as JSON null, so that the structure of the exported data

//is still valid if it's to be imported as a model again

if (isNull(value)) {

if (value == nil)

{

[tempDictionary removeObjectForKey:keyPath];

}

else

{

[tempDictionary setValue:[NSNull null] forKeyPath:keyPath];

}

continue;

}

//check if the property is another model

if ([value isKindOfClass:JSONModelClass]) {

//recurse models

value = [(JSONModel*)value toDictionary];

[tempDictionary setValue:value forKeyPath: keyPath];

//for clarity

continue;

} else {

// 1) check for built-in transformation

if (p.protocol) {

value = [self __reverseTransform:value forProperty:p];

}

// 2) check for standard types OR 2.1) primitives

if (p.structName==nil && (p.isStandardJSONType || p.type==nil)) {

//generic get value

[tempDictionary setValue:value forKeyPath: keyPath];

continue;

}

// 3) try to apply a value transformer

if (YES) {

//create selector from the property's class name

NSString* selectorName = [NSString stringWithFormat:@"%@From%@:", @"JSONObject", p.type?p.type:p.structName];

SEL selector = NSSelectorFromString(selectorName);

BOOL foundCustomTransformer = NO;

if ([valueTransformer respondsToSelector:selector]) {

foundCustomTransformer = YES;

} else {

//try for hidden transformer

selectorName = [NSString stringWithFormat:@"__%@",selectorName];