2 // NJMappingsController.m
5 // Created by Sam McCall on 4/05/09.
8 #import "NJMappingsController.h"
10 #import "EnjoyableApplicationDelegate.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 - (NJMapping *)objectAtIndexedSubscript:(NSUInteger)idx {
40 return idx < _mappings.count ? _mappings[idx] : nil;
43 - (NSUInteger)countByEnumeratingWithState:(NSFastEnumerationState *)state
44 objects:(__unsafe_unretained id [])buffer
45 count:(NSUInteger)len {
46 return [_mappings countByEnumeratingWithState:state
52 - (void)activateMappingForProcess:(NSString *)processName {
53 NJMapping *oldMapping = manualMapping;
54 NJMapping *newMapping = self[processName];
56 newMapping = oldMapping;
57 if (newMapping != _currentMapping)
58 [self activateMapping:newMapping];
59 manualMapping = oldMapping;
62 - (void)activateMapping:(NJMapping *)mapping {
64 mapping = manualMapping;
65 NSLog(@"Switching to mapping %@.", mapping.name);
66 manualMapping = mapping;
67 _currentMapping = mapping;
68 [removeButton setEnabled:_mappings[0] != mapping];
69 [outputController loadCurrent];
70 popoverActivate.title = _currentMapping.name;
71 [NSNotificationCenter.defaultCenter postNotificationName:NJEventMappingChanged
72 object:_currentMapping];
73 [tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:[_mappings indexOfObject:mapping]] byExtendingSelection:NO];
76 - (IBAction)addPressed:(id)sender {
77 NJMapping *newMapping = [[NJMapping alloc] initWithName:@"Untitled"];
78 [_mappings addObject:newMapping];
79 [(EnjoyableApplicationDelegate *)NSApplication.sharedApplication.delegate mappingsChanged];
80 [tableView reloadData];
81 [tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:_mappings.count - 1] byExtendingSelection:NO];
82 [tableView editColumn:0 row:_mappings.count - 1 withEvent:nil select:YES];
83 [self activateMapping:newMapping];
86 - (IBAction)removePressed:(id)sender {
87 if (tableView.selectedRow == 0)
90 [_mappings removeObjectAtIndex:tableView.selectedRow];
91 [tableView reloadData];
92 [(EnjoyableApplicationDelegate *)NSApplication.sharedApplication.delegate mappingsChanged];
93 [self activateMapping:_mappings[0]];
97 -(void)tableViewSelectionDidChange:(NSNotification *)notify {
98 if (tableView.selectedRow >= 0)
99 [self activateMapping:_mappings[tableView.selectedRow]];
102 - (id)tableView:(NSTableView *)view objectValueForTableColumn:(NSTableColumn *)column row:(NSInteger)index {
103 return self[index].name;
106 - (void)tableView:(NSTableView *)view
107 setObjectValue:(NSString *)obj
108 forTableColumn:(NSTableColumn *)col
109 row:(NSInteger)index {
110 self[index].name = obj;
112 [tableView reloadData];
113 [(EnjoyableApplicationDelegate *)NSApplication.sharedApplication.delegate mappingsChanged];
116 - (NSInteger)numberOfRowsInTableView:(NSTableView *)tableView {
117 return _mappings.count;
120 - (BOOL)tableView:(NSTableView *)view shouldEditTableColumn:(NSTableColumn *)column row:(NSInteger)index {
125 NSLog(@"Saving mappings to defaults.");
126 [NSUserDefaults.standardUserDefaults setValuesForKeysWithDictionary:[self dumpAll]];
130 [self loadAllFrom:NSUserDefaults.standardUserDefaults.dictionaryRepresentation];
133 - (NSDictionary *)dumpAll {
134 NSMutableArray *ary = [[NSMutableArray alloc] initWithCapacity:_mappings.count];
135 for (NJMapping *mapping in _mappings)
136 [ary addObject:[mapping serialize]];
137 NSUInteger current = _currentMapping ? [_mappings indexOfObject:_currentMapping] : 0;
138 return @{ @"mappings": ary, @"selected": @(current) };
141 - (void)loadAllFrom:(NSDictionary*)envelope {
142 NSArray *storedMappings = envelope[@"mappings"];
143 NSMutableArray* newMappings = [[NSMutableArray alloc] initWithCapacity:storedMappings.count];
145 // have to do two passes in case mapping1 refers to mapping2 via a NJOutputMapping
146 for (NSDictionary *storedMapping in storedMappings) {
147 NJMapping *mapping = [[NJMapping alloc] initWithName:storedMapping[@"name"]];
148 [newMappings addObject:mapping];
151 for (unsigned i = 0; i < storedMappings.count; ++i) {
152 NSDictionary *entries = storedMappings[i][@"entries"];
153 NJMapping *mapping = newMappings[i];
154 for (id key in entries) {
155 NJOutput *output = [NJOutput outputDeserialize:entries[key]
156 withMappings:newMappings];
158 mapping.entries[key] = output;
162 if (newMappings.count) {
163 unsigned current = [envelope[@"selected"] unsignedIntValue];
164 if (current >= newMappings.count)
166 _mappings = newMappings;
167 [tableView reloadData];
168 [(EnjoyableApplicationDelegate *)NSApplication.sharedApplication.delegate mappingsChanged];
169 [self activateMapping:_mappings[current]];
173 - (NJMapping *)mappingWithURL:(NSURL *)url error:(NSError **)error {
174 NSInputStream *stream = [NSInputStream inputStreamWithURL:url];
176 NSDictionary *serialization = !*error
177 ? [NSJSONSerialization JSONObjectWithStream:stream options:0 error:error]
181 if (!([serialization isKindOfClass:NSDictionary.class]
182 && [serialization[@"name"] isKindOfClass:NSString.class]
183 && [serialization[@"entries"] isKindOfClass:NSDictionary.class])) {
184 *error = [NSError errorWithDomain:@"Enjoyable"
186 description:@"This isn't a valid mapping file."];
190 NSDictionary *entries = serialization[@"entries"];
191 NJMapping *mapping = [[NJMapping alloc] initWithName:serialization[@"name"]];
192 for (id key in entries) {
193 NSDictionary *value = entries[key];
194 if ([key isKindOfClass:NSString.class]) {
195 NJOutput *output = [NJOutput outputDeserialize:value
196 withMappings:_mappings];
198 mapping.entries[key] = output;
204 - (void)importPressed:(id)sender {
205 NSOpenPanel *panel = [NSOpenPanel openPanel];
206 panel.allowedFileTypes = @[ @"enjoyable", @"json", @"txt" ];
207 NSWindow *window = NSApplication.sharedApplication.keyWindow;
208 [panel beginSheetModalForWindow:window
209 completionHandler:^(NSInteger result) {
210 if (result != NSFileHandlingPanelOKButton)
215 NJMapping *mapping = [self mappingWithURL:panel.URL error:&error];
219 NJMapping *mergeInto = self[mapping.name];
220 for (id key in mapping.entries) {
221 if (mergeInto.entries[key]
222 && ![mergeInto.entries[key] isEqual:mapping.entries[key]]) {
229 NSAlert *conflictAlert = [[NSAlert alloc] init];
230 conflictAlert.messageText = @"Replace existing mappings?";
231 conflictAlert.informativeText =
232 [NSString stringWithFormat:
233 @"This file contains inputs you've already mapped in \"%@\". Do you "
234 @"want to merge them and replace your existing mappings, or import this "
235 @"as a separate mapping?", mapping.name];
236 [conflictAlert addButtonWithTitle:@"Merge"];
237 [conflictAlert addButtonWithTitle:@"Cancel"];
238 [conflictAlert addButtonWithTitle:@"New Mapping"];
239 NSInteger res = [conflictAlert runModal];
240 if (res == NSAlertSecondButtonReturn)
242 else if (res == NSAlertThirdButtonReturn)
247 [mergeInto.entries addEntriesFromDictionary:mapping.entries];
250 [_mappings addObject:mapping];
251 [tableView reloadData];
255 [(EnjoyableApplicationDelegate *)NSApplication.sharedApplication.delegate mappingsChanged];
256 [self activateMapping:mapping];
257 [outputController loadCurrent];
259 if (conflict && !mergeInto) {
260 [tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:_mappings.count - 1] byExtendingSelection:NO];
261 [tableView editColumn:0 row:_mappings.count - 1 withEvent:nil select:YES];
266 [window presentError:error
267 modalForWindow:window
269 didPresentSelector:nil
276 - (void)exportPressed:(id)sender {
277 NSSavePanel *panel = [NSSavePanel savePanel];
278 panel.allowedFileTypes = @[ @"enjoyable" ];
279 NJMapping *mapping = _currentMapping;
280 panel.nameFieldStringValue = mapping.name;
281 NSWindow *window = NSApplication.sharedApplication.keyWindow;
282 [panel beginSheetModalForWindow:window
283 completionHandler:^(NSInteger result) {
284 if (result != NSFileHandlingPanelOKButton)
288 NSDictionary *serialization = [mapping serialize];
289 NSData *json = [NSJSONSerialization dataWithJSONObject:serialization
290 options:NSJSONWritingPrettyPrinted
293 [json writeToURL:panel.URL options:NSDataWritingAtomic error:&error];
296 [window presentError:error
297 modalForWindow:window
299 didPresentSelector:nil
305 - (IBAction)mappingPressed:(id)sender {
306 [popover showRelativeToRect:popoverActivate.bounds ofView:popoverActivate preferredEdge:NSMinXEdge];
309 - (void)popoverWillShow:(NSNotification *)notification {
310 popoverActivate.state = NSOnState;
313 - (void)popoverWillClose:(NSNotification *)notification {
314 popoverActivate.state = NSOffState;