2 // NJMappingsController.m
5 // Created by Sam McCall on 4/05/09.
8 #import "NJMappingsController.h"
10 #import "ApplicationController.h"
12 #import "NJMappingsController.h"
14 #import "NJOutputController.h"
17 @implementation NJMappingsController {
18 NSMutableArray *_mappings;
19 NJMapping *manualMapping;
23 if ((self = [super init])) {
24 _mappings = [[NSMutableArray alloc] init];
25 _currentMapping = [[NJMapping alloc] initWithName:@"(default)"];
26 manualMapping = _currentMapping;
27 [_mappings addObject:_currentMapping];
32 - (NJMapping *)objectForKeyedSubscript:(NSString *)name {
33 for (NJMapping *mapping in _mappings)
34 if ([name isEqualToString:mapping.name])
39 - (void)activateMappingForProcess:(NSString *)processName {
40 NJMapping *oldMapping = manualMapping;
41 NJMapping *newMapping = self[processName];
43 newMapping = oldMapping;
44 if (newMapping != _currentMapping)
45 [self activateMapping:newMapping];
46 manualMapping = oldMapping;
49 - (void)activateMapping:(NJMapping *)mapping {
51 mapping = manualMapping;
52 NSLog(@"Switching to mapping %@.", mapping.name);
53 manualMapping = mapping;
54 _currentMapping = mapping;
55 [removeButton setEnabled:_mappings[0] != mapping];
56 [outputController loadCurrent];
57 [NSNotificationCenter.defaultCenter postNotificationName:NJEventMappingChanged
58 object:_currentMapping];
59 [tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:[_mappings indexOfObject:mapping]] byExtendingSelection:NO];
62 - (IBAction)addPressed:(id)sender {
63 NJMapping *newMapping = [[NJMapping alloc] initWithName:@"Untitled"];
64 [_mappings addObject:newMapping];
65 [(ApplicationController *)NSApplication.sharedApplication.delegate mappingsChanged];
66 [tableView reloadData];
67 [tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:_mappings.count - 1] byExtendingSelection:NO];
68 [tableView editColumn:0 row:_mappings.count - 1 withEvent:nil select:YES];
69 [self activateMapping:newMapping];
72 - (IBAction)removePressed:(id)sender {
73 if (tableView.selectedRow == 0)
76 [_mappings removeObjectAtIndex:tableView.selectedRow];
77 [tableView reloadData];
78 [(ApplicationController *)NSApplication.sharedApplication.delegate mappingsChanged];
79 [self activateMapping:_mappings[0]];
83 -(void)tableViewSelectionDidChange:(NSNotification *)notify {
84 if (tableView.selectedRow >= 0)
85 [self activateMapping:_mappings[tableView.selectedRow]];
88 - (id)tableView:(NSTableView *)view objectValueForTableColumn:(NSTableColumn *)column row:(NSInteger)index {
89 return [_mappings[index] name];
92 - (void)tableView:(NSTableView *)view setObjectValue:(NSString *)obj forTableColumn:(NSTableColumn *)col row:(NSInteger)index {
93 [(NJMapping *)_mappings[index] setName:obj];
94 [tableView reloadData];
95 [(ApplicationController *)NSApplication.sharedApplication.delegate mappingsChanged];
98 - (NSInteger)numberOfRowsInTableView:(NSTableView *)tableView {
99 return _mappings.count;
102 - (BOOL)tableView:(NSTableView *)view shouldEditTableColumn:(NSTableColumn *)column row:(NSInteger)index {
107 NSLog(@"Saving mappings to defaults.");
108 [NSUserDefaults.standardUserDefaults setValuesForKeysWithDictionary:[self dumpAll]];
112 [self loadAllFrom:NSUserDefaults.standardUserDefaults.dictionaryRepresentation];
115 - (NSDictionary *)dumpAll {
116 NSMutableArray *ary = [[NSMutableArray alloc] initWithCapacity:_mappings.count];
117 for (NJMapping *mapping in _mappings)
118 [ary addObject:[mapping serialize]];
119 NSUInteger current = _currentMapping ? [_mappings indexOfObject:_currentMapping] : 0;
120 return @{ @"mappings": ary, @"selected": @(current) };
123 - (void)loadAllFrom:(NSDictionary*)envelope {
124 NSArray *storedMappings = envelope[@"mappings"];
125 NSMutableArray* newMappings = [[NSMutableArray alloc] initWithCapacity:storedMappings.count];
127 // have to do two passes in case mapping1 refers to mapping2 via a NJOutputMapping
128 for (NSDictionary *storedMapping in storedMappings) {
129 NJMapping *mapping = [[NJMapping alloc] initWithName:storedMapping[@"name"]];
130 [newMappings addObject:mapping];
133 for (unsigned i = 0; i < storedMappings.count; ++i) {
134 NSDictionary *entries = storedMappings[i][@"entries"];
135 NJMapping *mapping = newMappings[i];
136 for (id key in entries) {
137 NJOutput *output = [NJOutput outputDeserialize:entries[key]
138 withMappings:newMappings];
140 mapping.entries[key] = output;
144 if (newMappings.count) {
145 unsigned current = [envelope[@"selected"] unsignedIntValue];
146 if (current >= newMappings.count)
148 _mappings = newMappings;
149 [tableView reloadData];
150 [(ApplicationController *)NSApplication.sharedApplication.delegate mappingsChanged];
151 [self activateMapping:_mappings[current]];
155 - (NJMapping *)mappingWithURL:(NSURL *)url error:(NSError **)error {
156 NSInputStream *stream = [NSInputStream inputStreamWithURL:url];
158 NSDictionary *serialization = !*error
159 ? [NSJSONSerialization JSONObjectWithStream:stream options:0 error:error]
163 if (!([serialization isKindOfClass:NSDictionary.class]
164 && [serialization[@"name"] isKindOfClass:NSString.class]
165 && [serialization[@"entries"] isKindOfClass:NSDictionary.class])) {
166 *error = [NSError errorWithDomain:@"Enjoyable"
168 description:@"This isn't a valid mapping file."];
172 NSDictionary *entries = serialization[@"entries"];
173 NJMapping *mapping = [[NJMapping alloc] initWithName:serialization[@"name"]];
174 for (id key in entries) {
175 NSDictionary *value = entries[key];
176 if ([key isKindOfClass:NSString.class]) {
177 NJOutput *output = [NJOutput outputDeserialize:value
178 withMappings:_mappings];
180 mapping.entries[key] = output;
186 - (void)importPressed:(id)sender {
187 NSOpenPanel *panel = [NSOpenPanel openPanel];
188 panel.allowedFileTypes = @[ @"enjoyable", @"json", @"txt" ];
189 NSWindow *window = NSApplication.sharedApplication.keyWindow;
190 [panel beginSheetModalForWindow:window
191 completionHandler:^(NSInteger result) {
192 if (result != NSFileHandlingPanelOKButton)
197 NJMapping *mapping = [self mappingWithURL:panel.URL error:&error];
201 NJMapping *mergeInto = self[mapping.name];
202 for (id key in mapping.entries) {
203 if (mergeInto.entries[key]
204 && ![mergeInto.entries[key] isEqual:mapping.entries[key]]) {
211 NSAlert *conflictAlert = [[NSAlert alloc] init];
212 conflictAlert.messageText = @"Replace existing mappings?";
213 conflictAlert.informativeText =
214 [NSString stringWithFormat:
215 @"This file contains inputs you've already mapped in \"%@\". Do you "
216 @"want to merge them and replace your existing mappings, or import this "
217 @"as a separate mapping?", mapping.name];
218 [conflictAlert addButtonWithTitle:@"Merge"];
219 [conflictAlert addButtonWithTitle:@"Cancel"];
220 [conflictAlert addButtonWithTitle:@"New Mapping"];
221 NSInteger res = [conflictAlert runModal];
222 if (res == NSAlertSecondButtonReturn)
224 else if (res == NSAlertThirdButtonReturn)
229 [mergeInto.entries addEntriesFromDictionary:mapping.entries];
232 [_mappings addObject:mapping];
233 [tableView reloadData];
237 [(ApplicationController *)NSApplication.sharedApplication.delegate mappingsChanged];
238 [self activateMapping:mapping];
239 [outputController loadCurrent];
241 if (conflict && !mergeInto) {
242 [tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:_mappings.count - 1] byExtendingSelection:NO];
243 [tableView editColumn:0 row:_mappings.count - 1 withEvent:nil select:YES];
248 [window presentError:error
249 modalForWindow:window
251 didPresentSelector:nil
258 - (void)exportPressed:(id)sender {
259 NSSavePanel *panel = [NSSavePanel savePanel];
260 panel.allowedFileTypes = @[ @"enjoyable" ];
261 NJMapping *mapping = _currentMapping;
262 panel.nameFieldStringValue = mapping.name;
263 NSWindow *window = NSApplication.sharedApplication.keyWindow;
264 [panel beginSheetModalForWindow:window
265 completionHandler:^(NSInteger result) {
266 if (result != NSFileHandlingPanelOKButton)
270 NSDictionary *serialization = [mapping serialize];
271 NSData *json = [NSJSONSerialization dataWithJSONObject:serialization
272 options:NSJSONWritingPrettyPrinted
275 [json writeToURL:panel.URL options:NSDataWritingAtomic error:&error];
278 [window presentError:error
279 modalForWindow:window
281 didPresentSelector:nil