NJMappingsController.m

   1 //
   2 //  NJMappingsController.m
   3 //  Enjoy
   4 //
   5 //  Created by Sam McCall on 4/05/09.
   6 //
   7
   8 #import "NJMappingsController.h"
   9
  10 #import "EnjoyableApplicationDelegate.h"
  11 #import "NJMapping.h"
  12 #import "NJMappingsController.h"
  13 #import "NJOutput.h"
  14 #import "NJOutputController.h"
  15 #import "NJEvents.h"
  16
  17 @implementation NJMappingsController {
  18     NSMutableArray *_mappings;
  19     NJMapping *manualMapping;
  20 }
  21
  22 - (id)init {
  23     if ((self = [super init])) {
  24         _mappings = [[NSMutableArray alloc] init];
  25         _currentMapping = [[NJMapping alloc] initWithName:@"(default)"];
  26         manualMapping = _currentMapping;
  27         [_mappings addObject:_currentMapping];
  28     }
  29     return self;
  30 }
  31
  32 - (NJMapping *)objectForKeyedSubscript:(NSString *)name {
  33     for (NJMapping *mapping in _mappings)
  34         if ([name isEqualToString:mapping.name])
  35             return mapping;
  36     return nil;
  37 }
  38
  39 - (NJMapping *)objectAtIndexedSubscript:(NSUInteger)idx {
  40     return idx < _mappings.count ? _mappings[idx] : nil;
  41 }
  42
  43 - (NSUInteger)countByEnumeratingWithState:(NSFastEnumerationState *)state
  44                                   objects:(__unsafe_unretained id [])buffer
  45                                     count:(NSUInteger)len {
  46     return [_mappings countByEnumeratingWithState:state
  47                                           objects:buffer
  48                                             count:len];
  49 }
  50
  51
  52 - (void)activateMappingForProcess:(NSString *)processName {
  53     NJMapping *oldMapping = manualMapping;
  54     NJMapping *newMapping = self[processName];
  55     if (!newMapping)
  56         newMapping = oldMapping;
  57     if (newMapping != _currentMapping)
  58         [self activateMapping:newMapping];
  59     manualMapping = oldMapping;
  60 }
  61
  62 - (void)activateMapping:(NJMapping *)mapping {
  63     if (!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];
  74 }
  75
  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];
  84 }
  85
  86 - (IBAction)removePressed:(id)sender {
  87     if (tableView.selectedRow == 0)
  88         return;
  89
  90     [_mappings removeObjectAtIndex:tableView.selectedRow];
  91     [tableView reloadData];
  92     [(EnjoyableApplicationDelegate *)NSApplication.sharedApplication.delegate mappingsChanged];
  93     [self activateMapping:_mappings[0]];
  94     [self save];
  95 }
  96
  97 -(void)tableViewSelectionDidChange:(NSNotification *)notify {
  98     if (tableView.selectedRow >= 0)
  99         [self activateMapping:_mappings[tableView.selectedRow]];
 100 }
 101
 102 - (id)tableView:(NSTableView *)view objectValueForTableColumn:(NSTableColumn *)column row:(NSInteger)index {
 103     return self[index].name;
 104 }
 105
 106 - (void)tableView:(NSTableView *)view
 107    setObjectValue:(NSString *)obj
 108    forTableColumn:(NSTableColumn *)col
 109               row:(NSInteger)index {
 110     self[index].name = obj;
 111     [self save];
 112     [tableView reloadData];
 113     [(EnjoyableApplicationDelegate *)NSApplication.sharedApplication.delegate mappingsChanged];
 114 }
 115
 116 - (NSInteger)numberOfRowsInTableView:(NSTableView *)tableView {
 117     return _mappings.count;
 118 }
 119
 120 - (BOOL)tableView:(NSTableView *)view shouldEditTableColumn:(NSTableColumn *)column row:(NSInteger)index {
 121     return index > 0;
 122 }
 123
 124 - (void)save {
 125     NSLog(@"Saving mappings to defaults.");
 126     [NSUserDefaults.standardUserDefaults setValuesForKeysWithDictionary:[self dumpAll]];
 127 }
 128
 129 - (void)load {
 130     [self loadAllFrom:NSUserDefaults.standardUserDefaults.dictionaryRepresentation];
 131 }
 132
 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) };
 139 }
 140
 141 - (void)loadAllFrom:(NSDictionary*)envelope {
 142     NSArray *storedMappings = envelope[@"mappings"];
 143     NSMutableArray* newMappings = [[NSMutableArray alloc] initWithCapacity:storedMappings.count];
 144
 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];
 149     }
 150
 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];
 157             if (output)
 158                 mapping.entries[key] = output;
 159         }
 160     }
 161
 162     if (newMappings.count) {
 163         unsigned current = [envelope[@"selected"] unsignedIntValue];
 164         if (current >= newMappings.count)
 165             current = 0;
 166         _mappings = newMappings;
 167         [tableView reloadData];
 168         [(EnjoyableApplicationDelegate *)NSApplication.sharedApplication.delegate mappingsChanged];
 169         [self activateMapping:_mappings[current]];
 170     }
 171 }
 172
 173 - (NJMapping *)mappingWithURL:(NSURL *)url error:(NSError **)error {
 174     NSInputStream *stream = [NSInputStream inputStreamWithURL:url];
 175     [stream open];
 176     NSDictionary *serialization = !*error
 177         ? [NSJSONSerialization JSONObjectWithStream:stream options:0 error:error]
 178         : nil;
 179     [stream close];
 180
 181     if (!([serialization isKindOfClass:NSDictionary.class]
 182           && [serialization[@"name"] isKindOfClass:NSString.class]
 183           && [serialization[@"entries"] isKindOfClass:NSDictionary.class])) {
 184         *error = [NSError errorWithDomain:@"Enjoyable"
 185                                      code:0
 186                               description:@"This isn't a valid mapping file."];
 187         return nil;
 188     }
 189
 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];
 197             if (output)
 198                 mapping.entries[key] = output;
 199         }
 200     }
 201     return mapping;
 202 }
 203
 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)
 211                           return;
 212
 213                       [panel close];
 214                       NSError *error;
 215                       NJMapping *mapping = [self mappingWithURL:panel.URL error:&error];
 216
 217                       if (!error) {
 218                           BOOL conflict = NO;
 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]]) {
 223                                   conflict = YES;
 224                                   break;
 225                               }
 226                           }
 227
 228                           if (conflict) {
 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)
 241                                   return;
 242                               else if (res == NSAlertThirdButtonReturn)
 243                                   mergeInto = nil;
 244                           }
 245
 246                           if (mergeInto) {
 247                               [mergeInto.entries addEntriesFromDictionary:mapping.entries];
 248                               mapping = mergeInto;
 249                           } else {
 250                               [_mappings addObject:mapping];
 251                               [tableView reloadData];
 252                           }
 253
 254                           [self save];
 255                           [(EnjoyableApplicationDelegate *)NSApplication.sharedApplication.delegate mappingsChanged];
 256                           [self activateMapping:mapping];
 257                           [outputController loadCurrent];
 258
 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];
 262                           }
 263                       }
 264
 265                       if (error) {
 266                           [window presentError:error
 267                                 modalForWindow:window
 268                                       delegate:nil
 269                             didPresentSelector:nil
 270                                    contextInfo:nil];
 271                       }
 272                   }];
 273
 274 }
 275
 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)
 285                           return;
 286                       [panel close];
 287                       NSError *error;
 288                       NSDictionary *serialization = [mapping serialize];
 289                       NSData *json = [NSJSONSerialization dataWithJSONObject:serialization
 290                                                                      options:NSJSONWritingPrettyPrinted
 291                                                                        error:&error];
 292                       if (!error)
 293                           [json writeToURL:panel.URL options:NSDataWritingAtomic error:&error];
 294
 295                       if (error) {
 296                           [window presentError:error
 297                                 modalForWindow:window
 298                                       delegate:nil
 299                             didPresentSelector:nil
 300                                    contextInfo:nil];
 301                       }
 302                   }];
 303 }
 304
 305 - (IBAction)mappingPressed:(id)sender {
 306     [popover showRelativeToRect:popoverActivate.bounds ofView:popoverActivate preferredEdge:NSMinXEdge];
 307 }
 308
 309 @end