Fix up copyright notices.
[enjoyable.git] / ConfigsController.m
index 01bd0d3..9693985 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;
-       if([configs indexOfObject:neutralConfig] < 0) {// deleted, keep what we have
-               neutralConfig = NULL;
-               return;
-       }
-       [self activateConfig: neutralConfig forApplication: NULL];
-}
-
--(void) activateConfig: (Config*)config forApplication: (ProcessSerialNumber*) psn {
-       if(currentConfig == config)
-               return;
-
-       if(psn) {
-               if(!neutralConfig)
-                       neutralConfig = currentConfig;
-               attachedApplication = *psn;
-       } else {
-               neutralConfig = NULL;
-       }
-       
-       if(currentConfig != NULL) {
-               [targetController reset];
-       }
-       currentConfig = config;
-       [removeButton setEnabled: ![config protect]];
-       [targetController load];
-       [(ApplicationController *)[[NSApplication sharedApplication] delegate] configChanged];
-       [tableView selectRow: [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 selectRow: ([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 objectAtIndex: [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 objectAtIndex:i] entries];
-               for(id key in entries) {
-                       Target* target = (Target*) [entries objectForKey: 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 {
-       [self activateConfig: (Config*)[configs objectAtIndex:[tableView selectedRow]] forApplication: NULL];
-}
-       
--(id) tableView: (NSTableView*)view objectValueForTableColumn: (NSTableColumn*) column row: (int) index {
-    NSParameterAssert(index >= 0 && index < [configs count]);
-       return [[configs objectAtIndex: 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 objectAtIndex: 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 objectAtIndex: index] protect];
-}      
-
--(Config*) currentConfig {
-       return currentConfig;
-}
-
--(Config*) currentNeutralConfig {
-       if(neutralConfig)
-               return neutralConfig;
-       return currentConfig;
-}
-
--(void) save {
+#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;
+    
+    Config *toRemove = _configs[tableView.selectedRow];
+    [_configs removeObjectAtIndex:tableView.selectedRow];
+    
+    if (toRemove == _currentConfig)
+        _currentConfig = _configs[0];
+    if (toRemove == manualConfig)
+        manualConfig = _configs[0];
+    
+    [(ApplicationController *)[[NSApplication sharedApplication] delegate] configsChanged];
+    [tableView reloadData];
+}
+
+-(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"];
-       [[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 setObject:[config name] forKey:@"name"];
-               NSMutableDictionary* cfgEntries = [[NSMutableDictionary alloc] init];
-               for(id key in [config entries]) {
-                       [cfgEntries setObject:[[[config entries]objectForKey:key]stringify] forKey: key];
-               }
-               [cfgInfo setObject: cfgEntries forKey: @"entries"];
-               [ary addObject: cfgInfo];
-       }
-       [envelope setObject: ary forKey: @"configurationList"];
-       [envelope setObject: [NSNumber numberWithInt: [configs indexOfObject: [self currentNeutralConfig] ] ] forKey: @"selectedIndex"];
-       return envelope;
-}
--(void) loadAllFrom: (NSDictionary*) envelope{
-       if(envelope == NULL)
-               return;
-       NSArray* ary = [envelope objectForKey: @"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 objectAtIndex:i] objectForKey:@"name"]];            
-               [newConfigs addObject: cfg];
-       }
-       [[configs objectAtIndex:0] setProtect: YES];
-       for(int i=0; i<[ary count]; i++) {
-               NSDictionary* dict = [[ary objectAtIndex:i] objectForKey:@"entries"];
-               for(id key in dict) {
-                       [[[newConfigs objectAtIndex:i] entries] 
-                        setObject: [Target unstringify: [dict objectForKey: key] withConfigList: newConfigs]
-                        forKey: key];
-               }
-       }
-       
-       configs = newConfigs;
-       [tableView reloadData];
-       currentConfig = NULL;
-       [(ApplicationController *)[[NSApplication sharedApplication] delegate] configsChanged];
-       
-       int index = [[envelope objectForKey: @"selectedIndex"] intValue];
-       [self activateConfig: [configs objectAtIndex:index] forApplication: NULL];
-}
-
--(void) applicationSwitchedTo: (NSString*) name withPsn: (ProcessSerialNumber) psn {
-       for(int i=0; i<[configs count]; i++) {
-               Config* cfg = [configs objectAtIndex:i];
-               if([[cfg name] isEqualToString: name]) {
-                       [self activateConfig: cfg forApplication: &psn];
-                       return;
-               }
-       }
-       [self restoreNeutralConfig];
-}
-
--(ProcessSerialNumber*) targetApplication {
-       if(neutralConfig)
-               return &attachedApplication;
-       return NULL;
+}
+
+- (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)
+            config.entries[key] = [Target targetDeserialize:entries[key]
+                                                withConfigs:newConfigs];
+    }
+    
+    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]];
+    }
+}
+
+- (void)importPressed:(id)sender {
+    NSOpenPanel *panel = [NSOpenPanel openPanel];
+    panel.allowedFileTypes = @[ @"enjoyable", @"json", @"txt" ];
+    if ([panel runModal] == NSFileHandlingPanelOKButton) {
+        NSError *error;
+        NSInputStream *stream = [NSInputStream inputStreamWithURL:panel.URL];
+        [stream open];
+        NSDictionary *serialization = !error
+            ? [NSJSONSerialization JSONObjectWithStream:stream options:0 error:&error]
+            : nil;
+        [stream close];
+        
+        if (!([serialization isKindOfClass:[NSDictionary class]]
+              && serialization[@"entries"])) {
+            error = [NSError errorWithDomain:@"Enjoyable"
+                                        code:0
+                                 description:@"This isn't a valid mapping file."];
+        }
+        
+        
+        if (!error) {
+            NSDictionary *entries = serialization[@"entries"];
+            Config *cfg = [[Config alloc] initWithName:serialization[@"name"]];
+            Config *mergeInto = self[cfg.name];
+            BOOL conflict = NO;
+            for (id key in entries) {
+                cfg.entries[key] = [Target targetDeserialize:entries[key]
+                                                    withConfigs:_configs];
+                if (mergeInto.entries[key])
+                    conflict = YES;
+            }
+            
+            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)
+            [[NSAlert alertWithError:error] runModal];
+    }
+}
+
+- (void)exportPressed:(id)sender {
+    NSSavePanel *panel = [NSSavePanel savePanel];
+    panel.allowedFileTypes = @[ @"enjoyable" ];
+    if ([panel runModal] == NSFileHandlingPanelOKButton) {
+        NSError *error;
+        NSDictionary *serialization = [_currentConfig serialize];
+        NSData *json = [NSJSONSerialization dataWithJSONObject:serialization
+                                                       options:NSJSONWritingPrettyPrinted
+                                                         error:&error];
+        if (!error)
+            [json writeToURL:panel.URL options:NSDataWritingAtomic error:&error];
+        
+        if (error)
+            [[NSAlert alertWithError:error] runModal];
+    }
 }
 
 @end