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.
21 if ((self = [super init])) {
22 self.name = NSLocalizedString(@"Untitled", @"name for new mappings");
23 _entries = [[NSMutableDictionary alloc] init];
28 - (id)initWithName:(NSString *)name {
29 if ((self = [self init])) {
30 if ([name isKindOfClass:NSString.class])
36 - (id)initWithSerialization:(NSDictionary *)serialization
37 mappings:(NSArray *)mappings {
38 if ((self = [self initWithName:serialization[@"name"]])) {
39 NSDictionary *entries = serialization[@"entries"];
40 if ([entries isKindOfClass:NSDictionary.class]) {
41 for (id key in entries) {
42 if ([key isKindOfClass:NSString.class]) {
43 NJOutput *output = [NJOutput outputDeserialize:entries[key]
44 withMappings:mappings];
46 _entries[key] = output;
54 - (NJOutput *)objectForKeyedSubscript:(NJInput *)input {
55 return input ? _entries[input.uid] : nil;
58 - (void)setObject:(NJOutput *)output forKeyedSubscript:(NJInput *)input {
61 _entries[input.uid] = output;
63 [_entries removeObjectForKey:input.uid];
67 - (NSDictionary *)serialize {
68 NSMutableDictionary *entries = [[NSMutableDictionary alloc] initWithCapacity:_entries.count];
69 for (id key in _entries) {
70 id serialized = [_entries[key] serialize];
72 entries[key] = serialized;
74 return @{ @"name": _name, @"entries": entries };
77 - (BOOL)writeToURL:(NSURL *)url error:(NSError **)error {
78 [NSProcessInfo.processInfo disableSuddenTermination];
79 NSDictionary *serialization = [self serialize];
80 NSData *json = [NSJSONSerialization dataWithJSONObject:serialization
81 options:NSJSONWritingPrettyPrinted
83 BOOL success = json && [json writeToURL:url options:NSDataWritingAtomic error:error];
84 [NSProcessInfo.processInfo enableSuddenTermination];
89 return _entries.count;
92 - (BOOL)hasConflictWith:(NJMapping *)other {
93 if (other.count < self.count)
94 return [other hasConflictWith:self];
95 for (NSString *uid in _entries) {
96 NJOutput *output = other->_entries[uid];
97 if (output && ![output isEqual:_entries[uid]])
103 + (id)mappingWithContentsOfURL:(NSURL *)url mappings:(NSArray *)mappings error:(NSError **)error {
104 NSInputStream *stream = [NSInputStream inputStreamWithURL:url];
106 NSDictionary *serialization = stream && !*error
107 ? [NSJSONSerialization JSONObjectWithStream:stream options:0 error:error]
111 if (!serialization && error)
114 if (!([serialization isKindOfClass:NSDictionary.class]
115 && [serialization[@"name"] isKindOfClass:NSString.class]
116 && [serialization[@"entries"] isKindOfClass:NSDictionary.class])) {
117 *error = [NSError errorWithDomain:@"Enjoyable"
119 description:NSLocalizedString(@"invalid mapping file",
120 @"error when imported file was JSON but not a mapping")];
124 return [[NJMapping alloc] initWithSerialization:serialization
128 - (void)mergeEntriesFrom:(NJMapping *)other {
130 [_entries addEntriesFromDictionary:other->_entries];