2 // NJMappingsController.m
5 // Created by Sam McCall on 4/05/09.
8 #import "NJMappingsController.h"
11 #import "NJMappingsController.h"
13 #import "NJOutputController.h"
16 @implementation NJMappingsController {
17 NSMutableArray *_mappings;
18 NJMapping *manualMapping;
22 if ((self = [super init])) {
23 _mappings = [[NSMutableArray alloc] init];
24 _currentMapping = [[NJMapping alloc] initWithName:@"(default)"];
25 manualMapping = _currentMapping;
26 [_mappings addObject:_currentMapping];
31 - (NJMapping *)objectForKeyedSubscript:(NSString *)name {
32 for (NJMapping *mapping in _mappings)
33 if ([name isEqualToString:mapping.name])
38 - (NJMapping *)objectAtIndexedSubscript:(NSUInteger)idx {
39 return idx < _mappings.count ? _mappings[idx] : nil;
42 - (void)mappingsChanged {
44 [tableView reloadData];
45 [NSNotificationCenter.defaultCenter
46 postNotificationName:NJEventMappingListChanged
50 - (NSUInteger)countByEnumeratingWithState:(NSFastEnumerationState *)state
51 objects:(__unsafe_unretained id [])buffer
52 count:(NSUInteger)len {
53 return [_mappings countByEnumeratingWithState:state
59 - (void)activateMappingForProcess:(NSString *)processName {
60 NJMapping *oldMapping = manualMapping;
61 NJMapping *newMapping = self[processName];
63 newMapping = oldMapping;
64 if (newMapping != _currentMapping)
65 [self activateMapping:newMapping];
66 manualMapping = oldMapping;
69 - (void)activateMapping:(NJMapping *)mapping {
71 mapping = manualMapping;
72 NSLog(@"Switching to mapping %@.", mapping.name);
73 manualMapping = mapping;
74 _currentMapping = mapping;
75 [removeButton setEnabled:_mappings[0] != mapping];
76 [outputController loadCurrent];
77 popoverActivate.title = _currentMapping.name;
78 [NSNotificationCenter.defaultCenter postNotificationName:NJEventMappingChanged
79 object:_currentMapping];
80 [tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:[_mappings indexOfObject:mapping]] byExtendingSelection:NO];
83 - (IBAction)addPressed:(id)sender {
84 NJMapping *newMapping = [[NJMapping alloc] initWithName:@"Untitled"];
85 [_mappings addObject:newMapping];
86 [self mappingsChanged];
87 [self activateMapping:newMapping];
88 [tableView editColumn:0 row:_mappings.count - 1 withEvent:nil select:YES];
91 - (IBAction)removePressed:(id)sender {
92 if (tableView.selectedRow == 0)
95 [_mappings removeObjectAtIndex:tableView.selectedRow];
96 [self mappingsChanged];
97 [self activateMapping:_mappings[0]];
100 -(void)tableViewSelectionDidChange:(NSNotification *)notify {
101 [self activateMapping:self[tableView.selectedRow]];
104 - (id)tableView:(NSTableView *)view objectValueForTableColumn:(NSTableColumn *)column row:(NSInteger)index {
105 return self[index].name;
108 - (void)tableView:(NSTableView *)view
109 setObjectValue:(NSString *)obj
110 forTableColumn:(NSTableColumn *)col
111 row:(NSInteger)index {
112 self[index].name = obj;
113 [self 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 [self mappingsChanged];
168 [self activateMapping:_mappings[current]];
172 - (NJMapping *)mappingWithURL:(NSURL *)url error:(NSError **)error {
173 NSInputStream *stream = [NSInputStream inputStreamWithURL:url];
175 NSDictionary *serialization = !*error
176 ? [NSJSONSerialization JSONObjectWithStream:stream options:0 error:error]
180 if (!([serialization isKindOfClass:NSDictionary.class]
181 && [serialization[@"name"] isKindOfClass:NSString.class]
182 && [serialization[@"entries"] isKindOfClass:NSDictionary.class])) {
183 *error = [NSError errorWithDomain:@"Enjoyable"
185 description:@"This isn't a valid mapping file."];
189 NSDictionary *entries = serialization[@"entries"];
190 NJMapping *mapping = [[NJMapping alloc] initWithName:serialization[@"name"]];
191 for (id key in entries) {
192 NSDictionary *value = entries[key];
193 if ([key isKindOfClass:NSString.class]) {
194 NJOutput *output = [NJOutput outputDeserialize:value
195 withMappings:_mappings];
197 mapping.entries[key] = output;
203 - (void)importPressed:(id)sender {
204 NSOpenPanel *panel = [NSOpenPanel openPanel];
205 panel.allowedFileTypes = @[ @"enjoyable", @"json", @"txt" ];
206 NSWindow *window = NSApplication.sharedApplication.keyWindow;
207 [panel beginSheetModalForWindow:window
208 completionHandler:^(NSInteger result) {
209 if (result != NSFileHandlingPanelOKButton)
214 NJMapping *mapping = [self mappingWithURL:panel.URL error:&error];
218 NJMapping *mergeInto = self[mapping.name];
219 for (id key in mapping.entries) {
220 if (mergeInto.entries[key]
221 && ![mergeInto.entries[key] isEqual:mapping.entries[key]]) {
228 NSAlert *conflictAlert = [[NSAlert alloc] init];
229 conflictAlert.messageText = @"Replace existing mappings?";
230 conflictAlert.informativeText =
231 [NSString stringWithFormat:
232 @"This file contains inputs you've already mapped in \"%@\". Do you "
233 @"want to merge them and replace your existing mappings, or import this "
234 @"as a separate mapping?", mapping.name];
235 [conflictAlert addButtonWithTitle:@"Merge"];
236 [conflictAlert addButtonWithTitle:@"Cancel"];
237 [conflictAlert addButtonWithTitle:@"New Mapping"];
238 NSInteger res = [conflictAlert runModal];
239 if (res == NSAlertSecondButtonReturn)
241 else if (res == NSAlertThirdButtonReturn)
246 [mergeInto.entries addEntriesFromDictionary:mapping.entries];
249 [_mappings addObject:mapping];
252 [self mappingsChanged];
253 [self activateMapping:mapping];
254 [outputController loadCurrent];
256 if (conflict && !mergeInto) {
257 [tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:_mappings.count - 1] byExtendingSelection:NO];
258 [tableView editColumn:0 row:_mappings.count - 1 withEvent:nil select:YES];
263 [window presentError:error
264 modalForWindow:window
266 didPresentSelector:nil
273 - (void)exportPressed:(id)sender {
274 NSSavePanel *panel = [NSSavePanel savePanel];
275 panel.allowedFileTypes = @[ @"enjoyable" ];
276 NJMapping *mapping = _currentMapping;
277 panel.nameFieldStringValue = mapping.name;
278 NSWindow *window = NSApplication.sharedApplication.keyWindow;
279 [panel beginSheetModalForWindow:window
280 completionHandler:^(NSInteger result) {
281 if (result != NSFileHandlingPanelOKButton)
285 NSDictionary *serialization = [mapping serialize];
286 NSData *json = [NSJSONSerialization dataWithJSONObject:serialization
287 options:NSJSONWritingPrettyPrinted
290 [json writeToURL:panel.URL options:NSDataWritingAtomic error:&error];
293 [window presentError:error
294 modalForWindow:window
296 didPresentSelector:nil
302 - (IBAction)mappingPressed:(id)sender {
303 [popover showRelativeToRect:popoverActivate.bounds ofView:popoverActivate preferredEdge:NSMinXEdge];
306 - (void)popoverWillShow:(NSNotification *)notification {
307 popoverActivate.state = NSOnState;
310 - (void)popoverWillClose:(NSNotification *)notification {
311 popoverActivate.state = NSOffState;