Harden importing.
[enjoyable.git] / ConfigsController.m
index 63211ae..d38ada8 100644 (file)
 //  Created by Sam McCall on 4/05/09.
 //
 
-@implementation ConfigsController
-
-@synthesize configs;
-
--(id) init {
-       if(self = [super init]) {
-               configs = [[NSMutableArray alloc] init];
-               currentConfig = [[Config alloc] init];
-               [currentConfig setName: @"(default)"];
-               [currentConfig setProtect: YES];
-               [configs addObject: currentConfig];             
-       }
-       return self;
-}
-
--(void) restoreNeutralConfig {
-       if(!neutralConfig)
-               return;
-       [self activateConfig: neutralConfig forApplication: NULL];
-}
-
-// TODO: Not an appropriate way to track 'neutral configs', it should just
-// always be the first config and be unremovable.
-
--(void) activateConfig: (Config*)config forApplication: (ProcessSerialNumber*) psn {
-       if(currentConfig == config)
-               return;
-
-       if(psn) {
-               if(!neutralConfig)
-                       neutralConfig = currentConfig;
-       } else {
-               neutralConfig = NULL;
-       }
-       
-       if(currentConfig != NULL) {
-               [targetController reset];
-       }
-       currentConfig = config;
-       [removeButton setEnabled: ![config protect]];
-       [targetController load];
-       [(ApplicationController *)[[NSApplication sharedApplication] delegate] configChanged];
-    [tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:[configs indexOfObject:config]] byExtendingSelection:NO];
-}
-
--(IBAction) addPressed: (id)sender {
-       Config* newConfig = [[Config alloc] init];
-       [newConfig setName: @"untitled"];
-       [configs addObject: newConfig];
-       [(ApplicationController *)[[NSApplication sharedApplication] delegate] configsChanged];
-       [tableView reloadData];
-    [tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:configs.count - 1] byExtendingSelection:NO];
-       [tableView editColumn: 0 row:([configs count]-1) withEvent:nil select:YES];
-}
--(IBAction) removePressed: (id)sender {
-       // save changes first
-       [tableView reloadData];
-       Config* current_config = configs[[tableView selectedRow]];
-       if([current_config protect])
-               return;
-       [configs removeObjectAtIndex: [tableView selectedRow]];
-       
-       // remove all "switch to configuration" actions
-       for(int i=0; i<[configs count]; i++) {
-               NSMutableDictionary* entries = [(Config*)configs[i] entries];
-               for(id key in entries) {
-                       Target* target = (Target*) entries[key];
-                       if([target isKindOfClass: [TargetConfig class]] && [(TargetConfig*)target config] == current_config)
-                               [entries removeObjectForKey: key];
-               }
-       }
-       [(ApplicationController *)[[NSApplication sharedApplication] delegate] configsChanged];
-       
-       [tableView reloadData];
-}
-
--(void)tableViewSelectionDidChange:(NSNotification*) notify {
-    if (tableView.selectedRow < configs.count)
-        [self activateConfig: (Config*)configs[[tableView selectedRow]] forApplication: NULL];
-}
-       
--(id) tableView: (NSTableView*)view objectValueForTableColumn: (NSTableColumn*) column row: (int) index {
-    NSParameterAssert(index >= 0 && index < [configs count]);
-       return [configs[index] name];
-}
-
--(void) tableView: (NSTableView*) view setObjectValue:obj forTableColumn:(NSTableColumn*) col row: (int)index {
-    NSParameterAssert(index >= 0 && index < [configs count]);
-       /* ugly hack so stringification doesn't fail */
-       NSString* newName = [(NSString*)obj stringByReplacingOccurrencesOfString: @"~" withString: @""];
-       [(Config*)configs[index] setName: newName];
-       [targetController refreshConfigsPreservingSelection:YES];
-       [tableView reloadData];
-       [(ApplicationController *)[[NSApplication sharedApplication] delegate] configsChanged];
-}
-
--(int)numberOfRowsInTableView: (NSTableView*)table {
-       return [configs count];
-}
-
--(BOOL)tableView: (NSTableView*)view shouldEditTableColumn: (NSTableColumn*) column row: (int) index {
-       return ![configs[index] protect];
-}      
-
--(Config*) currentConfig {
-       return currentConfig;
-}
-
--(Config*) currentNeutralConfig {
-       if(neutralConfig)
-               return neutralConfig;
-       return currentConfig;
-}
-
--(void) save {
-    [[NSUserDefaults standardUserDefaults] setObject:[self dumpAll] forKey:@"configurations"];
-       [[NSUserDefaults standardUserDefaults] synchronize];
-}
--(void) load {
-       [self loadAllFrom: [[NSUserDefaults standardUserDefaults] objectForKey:@"configurations"]];
-}
-
--(NSDictionary*) dumpAll {
-       NSMutableDictionary *envelope = [[NSMutableDictionary alloc] init];
-       NSMutableArray* ary = [[NSMutableArray alloc] init];
-       for(Config* config in configs) {
-               NSMutableDictionary* cfgInfo = [[NSMutableDictionary alloc] init];
-               cfgInfo[@"name"] = [config name];
-               NSMutableDictionary* cfgEntries = [[NSMutableDictionary alloc] init];
-               for(id key in [config entries]) {
-                       cfgEntries[key] = [[config entries][key]stringify];
-               }
-               cfgInfo[@"entries"] = cfgEntries;
-               [ary addObject: cfgInfo];
-       }
-       envelope[@"configurationList"] = ary;
-       envelope[@"selectedIndex"] = @([configs indexOfObject: [self currentNeutralConfig] ]);
-       return envelope;
-}
--(void) loadAllFrom: (NSDictionary*) envelope{
-       if(envelope == NULL)
-               return;
-       NSArray* ary = envelope[@"configurationList"];
-       
-       NSMutableArray* newConfigs = [[NSMutableArray alloc] init];
-       // have to do two passes in case config1 refers to config2 via a TargetConfig
-       for(int i=0; i<[ary count]; i++) {
-               Config* cfg = [[Config alloc] init];
-               [cfg setName: ary[i][@"name"]];         
-               [newConfigs addObject: cfg];
-       }
-       [configs[0] setProtect: YES];
-       for(int i=0; i<[ary count]; i++) {
-               NSDictionary* dict = ary[i][@"entries"];
-               for(id key in dict) {
-                       [newConfigs[i] entries][key] = [Target unstringify: dict[key] withConfigList: newConfigs];
-               }
-       }
-       
-       configs = newConfigs;
-       [tableView reloadData];
-       currentConfig = NULL;
-       [(ApplicationController *)[[NSApplication sharedApplication] delegate] configsChanged];
-       
-       int index = [envelope[@"selectedIndex"] intValue];
-    if (index < configs.count)
-        [self activateConfig: configs[index] forApplication: NULL];
-}
-
--(void) applicationSwitchedTo: (NSString*) name withPsn: (ProcessSerialNumber) psn {
-       for(int i=0; i<[configs count]; i++) {
-               Config* cfg = configs[i];
-               if([[cfg name] isEqualToString: name]) {
-                       [self activateConfig: cfg forApplication: &psn];
-                       return;
-               }
-       }
-       [self restoreNeutralConfig];
+#import "ConfigsController.h"
+
+#import "ApplicationController.h"
+#import "Config.h"
+#import "ConfigsController.h"
+#import "Target.h"
+#import "TargetController.h"
+
+@implementation ConfigsController {
+    NSMutableArray *_configs;
+    Config *manualConfig;
+}
+
+- (id)init {
+    if ((self = [super init])) {
+        _configs = [[NSMutableArray alloc] init];
+        _currentConfig = [[Config alloc] initWithName:@"(default)"];
+        manualConfig = _currentConfig;
+        [_configs addObject:_currentConfig];
+    }
+    return self;
+}
+
+- (Config *)objectForKeyedSubscript:(NSString *)name {
+    for (Config *config in _configs)
+        if ([name isEqualToString:config.name])
+            return config;
+    return nil;
+}
+
+- (void)activateConfigForProcess:(NSString *)processName {
+    Config *oldConfig = manualConfig;
+    [self activateConfig:self[processName]];
+    manualConfig = oldConfig;
+}
+
+- (void)activateConfig:(Config *)config {
+    if (!config)
+        config = manualConfig;
+    if (_currentConfig == config)
+        return;
+    manualConfig = config;
+    _currentConfig = config;
+    [removeButton setEnabled:_configs[0] != config];
+    [targetController loadCurrent];
+    [(ApplicationController *)NSApplication.sharedApplication.delegate configChanged];
+    [tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:[_configs indexOfObject:config]] byExtendingSelection:NO];
+}
+
+- (IBAction)addPressed:(id)sender {
+    Config *newConfig = [[Config alloc] initWithName:@"Untitled"];
+    [_configs addObject:newConfig];
+    [(ApplicationController *)NSApplication.sharedApplication.delegate configsChanged];
+    [tableView reloadData];
+    [tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:_configs.count - 1] byExtendingSelection:NO];
+    [tableView editColumn:0 row:_configs.count - 1 withEvent:nil select:YES];
+}
+
+- (IBAction)removePressed:(id)sender {
+    if (tableView.selectedRow == 0)
+        return;
+    
+    [_configs removeObjectAtIndex:tableView.selectedRow];
+    [tableView reloadData];
+    [(ApplicationController *)NSApplication.sharedApplication.delegate configsChanged];
+    [self activateConfig:_configs[0]];
+    [self save];
+}
+
+-(void)tableViewSelectionDidChange:(NSNotification *)notify {
+    if (tableView.selectedRow >= 0)
+        [self activateConfig:_configs[tableView.selectedRow]];
+}
+
+- (id)tableView:(NSTableView *)view objectValueForTableColumn:(NSTableColumn *)column row:(NSInteger)index {
+    return [_configs[index] name];
+}
+
+- (void)tableView:(NSTableView *)view setObjectValue:(NSString *)obj forTableColumn:(NSTableColumn *)col row:(NSInteger)index {
+    [(Config *)_configs[index] setName:obj];
+    [tableView reloadData];
+    [(ApplicationController *)NSApplication.sharedApplication.delegate configsChanged];
+}
+
+- (NSInteger)numberOfRowsInTableView:(NSTableView *)tableView {
+    return _configs.count;
+}
+
+- (BOOL)tableView:(NSTableView *)view shouldEditTableColumn:(NSTableColumn *)column row:(NSInteger)index {
+    return index > 0;
+}
+
+- (void)save {
+    NSLog(@"Saving defaults.");
+    [NSUserDefaults.standardUserDefaults setObject:[self dumpAll] forKey:@"configurations"];
+}
+
+- (void)load {
+    [self loadAllFrom:[NSUserDefaults.standardUserDefaults objectForKey:@"configurations"]];
+}
+
+- (NSDictionary *)dumpAll {
+    NSMutableArray *ary = [[NSMutableArray alloc] initWithCapacity:_configs.count];
+    for (Config *config in _configs)
+        [ary addObject:[config serialize]];
+    NSUInteger current = _currentConfig ? [_configs indexOfObject:_currentConfig] : 0;
+    return @{ @"configurations": ary, @"selected": @(current) };
+}
+
+- (void)loadAllFrom:(NSDictionary*) envelope{
+    NSArray *storedConfigs = envelope[@"configurations"];
+    NSMutableArray* newConfigs = [[NSMutableArray alloc] initWithCapacity:storedConfigs.count];
+
+    // have to do two passes in case config1 refers to config2 via a TargetConfig
+    for (NSDictionary *storedConfig in storedConfigs) {
+        Config *cfg = [[Config alloc] initWithName:storedConfig[@"name"]];
+        [newConfigs addObject:cfg];
+    }
+
+    for (unsigned i = 0; i < storedConfigs.count; ++i) {
+        NSDictionary *entries = storedConfigs[i][@"entries"];
+        Config *config = newConfigs[i];
+        for (id key in entries) {
+            Target *target = [Target targetDeserialize:entries[key]
+                                            withConfigs:newConfigs];
+            if (target)
+                config.entries[key] = target;
+        }
+    }
+    
+    if (newConfigs.count) {
+        unsigned current = [envelope[@"selected"] unsignedIntValue];
+        if (current >= newConfigs.count)
+            current = 0;
+        _configs = newConfigs;
+        [tableView reloadData];
+        [(ApplicationController *)NSApplication.sharedApplication.delegate configsChanged];
+        [self activateConfig:_configs[current]];
+    }
+}
+
+- (Config *)configWithURL:(NSURL *)url error:(NSError **)error {
+    NSInputStream *stream = [NSInputStream inputStreamWithURL:url];
+    [stream open];
+    NSDictionary *serialization = !*error
+        ? [NSJSONSerialization JSONObjectWithStream:stream options:0 error:error]
+        : nil;
+    [stream close];
+    
+    if (!([serialization isKindOfClass:NSDictionary.class]
+          && [serialization[@"name"] isKindOfClass:NSString.class]
+          && [serialization[@"entries"] isKindOfClass:NSDictionary.class])) {
+        *error = [NSError errorWithDomain:@"Enjoyable"
+                                     code:0
+                              description:@"This isn't a valid mapping file."];
+        return nil;
+    }
+
+    NSDictionary *entries = serialization[@"entries"];
+    Config *cfg = [[Config alloc] initWithName:serialization[@"name"]];
+    for (id key in entries) {
+        NSDictionary *value = entries[key];
+        if ([key isKindOfClass:NSString.class]) {
+            Target *target = [Target targetDeserialize:value
+                                           withConfigs:_configs];
+            if (target)
+                cfg.entries[key] = target;
+        }
+    }
+    return cfg;
+}
+
+- (void)importPressed:(id)sender {
+    NSOpenPanel *panel = [NSOpenPanel openPanel];
+    panel.allowedFileTypes = @[ @"enjoyable", @"json", @"txt" ];
+    NSWindow *window = NSApplication.sharedApplication.keyWindow;
+    [panel beginSheetModalForWindow:window
+                  completionHandler:^(NSInteger result) {
+                      if (result != NSFileHandlingPanelOKButton)
+                          return;
+
+                      [panel close];
+                      NSError *error;
+                      Config *cfg = [self configWithURL:panel.URL error:&error];
+                      
+                      if (!error) {
+                          BOOL conflict;
+                          Config *mergeInto = self[cfg.name];
+                          for (id key in cfg.entries) {
+                              if (mergeInto.entries[key]) {
+                                  conflict = YES;
+                                  break;
+                              }
+                          }
+                          
+                          if (conflict) {
+                              NSAlert *conflictAlert = [[NSAlert alloc] init];
+                              conflictAlert.messageText = @"Replace existing mappings?";
+                              conflictAlert.informativeText =
+                                  [NSString stringWithFormat:
+                                   @"This file contains inputs you've already mapped in \"%@\". Do you "
+                                   @"want to merge them and replace your existing mappings, or import this "
+                                   @"as a separate mapping?", cfg.name];
+                              [conflictAlert addButtonWithTitle:@"Merge"];
+                              [conflictAlert addButtonWithTitle:@"Cancel"];
+                              [conflictAlert addButtonWithTitle:@"New Mapping"];
+                              NSInteger res = [conflictAlert runModal];
+                              if (res == NSAlertSecondButtonReturn)
+                                  return;
+                              else if (res == NSAlertThirdButtonReturn)
+                                  mergeInto = nil;
+                          }
+                          
+                          if (mergeInto) {
+                              [mergeInto.entries addEntriesFromDictionary:cfg.entries];
+                              cfg = mergeInto;
+                          } else {
+                              [_configs addObject:cfg];
+                              [tableView reloadData];
+                          }
+                          
+                          [self save];
+                          [(ApplicationController *)NSApplication.sharedApplication.delegate configsChanged];
+                          [self activateConfig:cfg];
+                          [targetController loadCurrent];
+                          
+                          if (conflict && !mergeInto) {
+                              [tableView selectRowIndexes:[NSIndexSet indexSetWithIndex:_configs.count - 1] byExtendingSelection:NO];
+                              [tableView editColumn:0 row:_configs.count - 1 withEvent:nil select:YES];
+                          }
+                      }
+                      
+                      if (error) {
+                          [window presentError:error
+                                modalForWindow:window
+                                      delegate:nil
+                            didPresentSelector:nil
+                                   contextInfo:nil];
+                      }
+                  }];
+     
+}
+
+- (void)exportPressed:(id)sender {
+    NSSavePanel *panel = [NSSavePanel savePanel];
+    panel.allowedFileTypes = @[ @"enjoyable" ];
+    Config *cfg = _currentConfig;
+    panel.nameFieldStringValue = cfg.name;
+    NSWindow *window = NSApplication.sharedApplication.keyWindow;
+    [panel beginSheetModalForWindow:window
+                  completionHandler:^(NSInteger result) {
+                      if (result != NSFileHandlingPanelOKButton)
+                          return;
+                      [panel close];
+                      NSError *error;
+                      NSDictionary *serialization = [cfg serialize];
+                      NSData *json = [NSJSONSerialization dataWithJSONObject:serialization
+                                                                     options:NSJSONWritingPrettyPrinted
+                                                                       error:&error];
+                      if (!error)
+                          [json writeToURL:panel.URL options:NSDataWritingAtomic error:&error];
+                      
+                      if (error) {
+                          [window presentError:error
+                                modalForWindow:window
+                                      delegate:nil
+                            didPresentSelector:nil
+                                   contextInfo:nil];
+                      }
+                  }];
 }
 
 @end