#import "JsonElement.h"

@interface JsonElement ()

- (id)initWithDictionary:(NSDictionary *)dictionary;

- (id)initWithArray:(NSArray *)array;

- (id)initWithTerminal:(id)object;

- (NSString *)descriptionWithDepth:(NSInteger)depth;

@interface JsonElement (OutlineDescription)

- (NSString *)outlineItemDescription:(id)item;

- (NSString *)outlineArrayItems;

- (NSString *)outlineDictionaryItems;

@interface NSNumber (JsonElement)

- (NSString *)jsonRepresentation;

@implementation NSNumber (JsonElement)

- (NSString *)jsonRepresentation {

if ([self.className isEqualToString:@"__NSCFBoolean"]) {

return [self boolValue] ? @"true" : @"false";

}

return [self typeFormedDescription];

}

@implementation JsonElement

@synthesize parent=_parent;

@synthesize key=_key, keys=_keys, object=_object, children=_children;

NSDictionary *JsonElementInitializers = nil;

- (id)initWithObject:(id)object {

if ([object isKindOfClass:[NSDictionary class]]) {

return [self initWithDictionary:object];

}

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

return [self initWithArray:object];

}

return [self initWithTerminal:object];

}

+ (id)elementWithObject:(id)object {

return [[[self alloc] initWithObject:object] autorelease];

}

- (id)initWithDictionary:(NSDictionary *)object {

self = [super init];

if (self != nil) {

self.object = object;

self.keys = [object.allKeys sortedArrayUsingSelector:@selector(localizedCaseInsensitiveCompare:)];

}

return self;

}

- (id)initWithArray:(NSArray *)object {

self = [super init];

if (self != nil) {

self.object = object;

NSMutableArray *keys = [NSMutableArray array];

for (NSInteger i = 0; i < object.count; i++) {

[keys addObject:[NSNumber numberWithInteger:i]];

}

self.keys = [NSArray arrayWithArray:keys];

}

return self;

}

- (id)initWithTerminal:(id)object {

self = [super init];

if (self != nil) {

self.object = object;

}

return self;

}

- (void)dealloc {

self.parent = nil;

self.object = nil;

self.key = nil;

self.keys = nil;

self.children = nil;

[super dealloc];

}

- (id)childAtIndex:(NSInteger)index {

if (self.keys == nil) return nil;

if (self.children == nil) self.children = [NSMutableDictionary dictionary];

id key = [self.keys objectAtIndex:index];

JsonElement *child = [self.children objectForKey:key];

if (child == nil) {

id json = [key isKindOfClass:[NSNumber class]] ? [self.object objectAtIndex:[key integerValue]] : [self.object objectForKey:key];

child = [[self class] elementWithObject:json];

child.parent = self;

child.key = key;

[self.children setObject:child forKey:key];

}

return child;

}

- (NSString *)outlineDescription {

NSString *result = nil;

if (self.keys == nil) {

if ([self.object isKindOfClass:[NSNumber class]]) {

result = [self.object jsonRepresentation];

} else if ([self.object isKindOfClass:[NSNull class]]) {

result = @"null";

}

} else {

NSUserDefaults *userDefaults = [NSUserDefaults standardUserDefaults];

NSNumber *value = [userDefaults objectForKey:@"ShowBriefDescription"];

BOOL breif = value.boolValue;

if ([self.object isKindOfClass:[NSArray class]]) {

if (breif) {

result = [NSString stringWithFormat:@"Array(%lu): [%@]", [self.keys count], [self outlineArrayItems]];

} else {

result = [NSString stringWithFormat:@"Array(%lu)", [self.keys count]];

}

} else if ([self.object isKindOfClass:[NSDictionary class]]) {

if (breif) {

result = [NSString stringWithFormat:@"Dict(%lu): {%@}", [self.keys count], [self outlineDictionaryItems]];

} else {

result = [NSString stringWithFormat:@"Dict(%lu)", [self.keys count]];

}

}

}

if (result == nil) {

result = [self.object description];

}

result = [result stringByReplacingOccurrencesOfString:@"\n" withString:@"⏎ "];

return result;

}

- (NSString *)description {

return [self descriptionWithDepth:0];

}

- (NSString *)descriptionWithDepth:(NSInteger)depth {

if (self.object == nil) return @"";

NSMutableString *indent = [NSMutableString string];

for (NSInteger i = 0; i < depth; i++) {

[indent appendString:@"\t"];

}

NSString *indent2 = [indent stringByAppendingString:@"\t"];

NSMutableString *desc = [NSMutableString string];

if ([self.object isKindOfClass:[NSArray class]]) {

[desc appendString:@"[\n"];

for (NSInteger i = 0; i < self.keys.count; i++) {

[desc appendString:indent2];

NSString *append = [[self childAtIndex:i] descriptionWithDepth:depth + 1];

[desc appendString:append ? append : @"null"];

[desc appendString:@",\n"];

}

NSInteger deleteCount = 1 + (self.keys.count != 0);

[desc deleteCharactersInRange:NSMakeRange(desc.length - deleteCount, deleteCount)];

[desc appendString:@"\n"];

[desc appendString:indent];

[desc appendString:@"]"];

} else if ([self.object isKindOfClass:[NSDictionary class]]) {

[desc appendString:@"{\n"];

for (NSInteger i = 0; i < self.keys.count; i++) {

[desc appendString:indent2];

id desc2 = [NSString stringWithFormat:@"\"%@\": %@", [self.keys objectAtIndex:i], [[self childAtIndex:i] descriptionWithDepth:depth + 1]];

[desc appendString:desc2];

[desc appendString:@",\n"];

}

NSInteger deleteCount = 1 + (self.keys.count != 0);

[desc deleteCharactersInRange:NSMakeRange(desc.length - deleteCount, deleteCount)];

[desc appendString:@"\n"];

[desc appendString:indent];

[desc appendString:@"}"];

} else if ([self.object isKindOfClass:[NSString class]]) {

[desc appendString:@"\""];

[desc appendString:[self.object stringByReplacingOccurrencesOfString:@"\"" withString:@"\\\""]];

[desc appendString:@"\""];

} else if ([self.object isKindOfClass:[NSNumber class]]) {

[desc appendString:[self.object jsonRepresentation]];

} else if ([self.object isKindOfClass:[NSNull class]]) {

[desc appendString:@"null"];

} else {

[desc appendString:[self.object description]];

}

return desc;

}

@implementation JsonElement (OutlineDescription)

- (NSString *)outlineItemDescription:(id)item {

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

return [NSString stringWithFormat:@"Array(%lu)", [item count]];

} else if ([item isKindOfClass:[NSDictionary class]]) {

return [NSString stringWithFormat:@"Dict(%lu)", [[item allKeys] count]];

} else if ([item isKindOfClass:[NSString class]]) {

return [NSString stringWithFormat:@"\"%@\"", [item stringByReplacingOccurrencesOfString:@"\"" withString:@"\\\""]];

} else if ([item isKindOfClass:[NSNumber class]]) {

return [item jsonRepresentation];

} else {

return [item description];

}

}

- (NSString *)outlineArrayItems {

NSArray *item = self.object;

NSMutableArray *children = [NSMutableArray array];

for (NSInteger i = 0; i < item.count; i++) {

NSString *desc = [self outlineItemDescription:[item objectAtIndex:i]];

[children addObject:desc];

}

return [children componentsJoinedByString:@","];

}

- (NSString *)outlineDictionaryItems {

NSDictionary *item = self.object;

NSMutableArray *children = [NSMutableArray array];

for (id key in self.keys) {

NSString *desc = [self outlineItemDescription:[item objectForKey:key]];

[children addObject:[NSString stringWithFormat:@"\"%@\":%@", key, desc]];

}

return [children componentsJoinedByString:@","];

}