ConditionalJSONModel

ConditionalJsonModel.h

#import <Foundation/Foundation.h>
#import <JSONModel/JSONModel.h>


@interface ConditionalJsonModel : JSONModel

- (NSDictionary*)classMapping;

- (NSString*)classNameProperty;

- (BOOL)requiresExactMatch;

@end

ConditionalJsonModel.m

#import "ConditionalJsonModel.h"


@implementation ConditionalJsonModel

- (id)initWithDictionary:(NSDictionary *)dict error:(NSError *__autoreleasing *)err
{
    NSString* value = dict[[self classNameProperty]];
    NSDictionary* mapping = [self classMapping];
    
    Class class = nil;
    if ([self requiresExactMatch]) {
         class = mapping[value];
    }
    else {
        for (id key in mapping.allKeys) {
            if ([value hasPrefix:key]) {
                class = mapping[key];
                break;
            }
        }
    }
    
    if (!class) {
         NSLog(@"!!Warning!! ConditionalJsonModel did not find a class for value: %@", value);
    }
    
    return [[class alloc] initWithDictionary:dict error:err];
}
        
- (BOOL)requiresExactMatch
{
    return YES;
}

- (NSDictionary*)classMapping
{
    return nil;
}

- (NSString*)classNameProperty
{
    return nil;
}

@end