5 // Created by Sam McCall on 4/05/09.
13 @implementation NJMapping {
14 NSMutableDictionary *_entries;
17 // Extra checks during initialization because the data is often loaded
18 // from untrusted serializations.
20 - (id)initWithName:(NSString *)name {
21 if ((self = [super init])) {
22 self.name = [name isKindOfClass:NSString.class] ? name : @"Untitled";
23 _entries = [[NSMutableDictionary alloc] init];
28 - (id)initWithSerialization:(NSDictionary *)serialization
29 mappings:(NSArray *)mappings {
30 if ((self = [self initWithName:serialization[@"name"]])) {
31 NSDictionary *entries = serialization[@"entries"];
32 if ([entries isKindOfClass:NSDictionary.class]) {
33 for (id key in entries) {
34 if ([key isKindOfClass:NSString.class]) {
35 NJOutput *output = [NJOutput outputDeserialize:entries[key]
36 withMappings:mappings];
38 _entries[key] = output;
46 - (NJOutput *)objectForKeyedSubscript:(NJInput *)input {
47 return input ? _entries[input.uid] : nil;
50 - (void)setObject:(NJOutput *)output forKeyedSubscript:(NJInput *)input {
53 _entries[input.uid] = output;
55 [_entries removeObjectForKey:input.uid];
59 - (NSDictionary *)serialize {
60 NSMutableDictionary *entries = [[NSMutableDictionary alloc] initWithCapacity:_entries.count];
61 for (id key in _entries) {
62 id serialized = [_entries[key] serialize];
64 entries[key] = serialized;
66 return @{ @"name": _name, @"entries": entries };
69 - (BOOL)writeToURL:(NSURL *)url error:(NSError **)error {
70 [NSProcessInfo.processInfo disableSuddenTermination];
71 NSDictionary *serialization = [self serialize];
72 NSData *json = [NSJSONSerialization dataWithJSONObject:serialization
73 options:NSJSONWritingPrettyPrinted
75 BOOL success = json && [json writeToURL:url options:NSDataWritingAtomic error:error];
76 [NSProcessInfo.processInfo enableSuddenTermination];
81 return _entries.count;
84 - (BOOL)hasConflictWith:(NJMapping *)other {
85 if (other.count < self.count)
86 return [other hasConflictWith:self];
87 for (NSString *uid in _entries) {
88 NJOutput *output = other->_entries[uid];
89 if (output && ![output isEqual:_entries[uid]])
95 + (id)mappingWithContentsOfURL:(NSURL *)url mappings:(NSArray *)mappings error:(NSError **)error {
96 NSInputStream *stream = [NSInputStream inputStreamWithURL:url];
98 NSDictionary *serialization = stream && !*error
99 ? [NSJSONSerialization JSONObjectWithStream:stream options:0 error:error]
103 if (!serialization && error)
106 if (!([serialization isKindOfClass:NSDictionary.class]
107 && [serialization[@"name"] isKindOfClass:NSString.class]
108 && [serialization[@"entries"] isKindOfClass:NSDictionary.class])) {
109 *error = [NSError errorWithDomain:@"Enjoyable"
111 description:@"This isn't a valid mapping file."];
115 return [[NJMapping alloc] initWithSerialization:serialization
119 - (void)mergeEntriesFrom:(NJMapping *)other {
121 [_entries addEntriesFromDictionary:other->_entries];