PassengerPref.m

#import "PassengerPref.h"

@implementation PassengerPref

@synthesize authorized, dirty;

@synthesize applications;
@synthesize textStateColor;

- (void) mainViewDidLoad {
  CLI *cli = [CLI sharedInstance];
  [cli setPathToCLI:[[self bundle] pathForResource:@"ppane" ofType:nil inDirectory:@"bin"]];
  [cli setAppDelegate:self];
  
  [self setupUI];
  [self setupAuthorizationView];
  [self setupApplicationView];
  
  
  [[NSNotificationCenter defaultCenter] addObserver:self
                                           selector:@selector(paneWillBecomeActive:)
                                               name:NSApplicationWillBecomeActiveNotification
                                             object:NULL];
}

- (void) setupUI {
  NSImage *browserButtonImage;

  [passengerIconView setImage:[[NSImage alloc] initByReferencingFile:[[self bundle] pathForImageResource:@"label"]]];
  browserButtonImage = [[NSImage alloc] initByReferencingFile:[[self bundle] pathForImageResource:@"OpenInBrowserTemplate"]];
  [browserButtonImage setTemplate:YES];
  [openInBrowserButton setImage:browserButtonImage];
  
  [self setTextStateColor:NSColor.disabledControlTextColor];
}

- (void) setupAuthorizationView {
  self.authorized = NO;
  [authorizationView setString:kAuthorizationRightExecute];
  [authorizationView setDelegate:self];
  [authorizationView setAutoupdate:YES];
  [authorizationView updateStatus:self];
}

- (void) setupApplicationView {
  [self loadApplications];
  if ([applications count] > 0) {
    [applicationsController setSelectedObjects:[NSArray arrayWithObjects:[applications objectAtIndex:0], nil]];
  }
  [applicationsTableView registerForDraggedTypes:[NSArray arrayWithObject:NSFilenamesPboardType]];
  [applicationsTableView setDraggingSourceOperationMask:NSDragOperationGeneric forLocal:NO];
}

#pragma SFAuthorizationView delegate methods

- (void) authorizationViewDidAuthorize:(SFAuthorizationView *)view {
  [self setTextStateColor:NSColor.blackColor];
  [[CLI sharedInstance] setAuthorizationRef:[[view authorization] authorizationRef]];
  self.authorized = YES;
  NSLog(@"Pane is now authorized");
}

- (void) authorizationViewDidDeauthorize:(SFAuthorizationView *)authorizationView {
  [self setTextStateColor:NSColor.disabledControlTextColor];
  [[CLI sharedInstance] deauthorize];
  self.authorized = NO;
  NSLog(@"Pane is now deauthorized");
}

#pragma NSTableViewDataSource protocol methods

- (NSDragOperation) tableView:(NSTableView *)aTableView validateDrop:(id)info proposedRow:(NSInteger)row proposedDropOperation:(NSTableViewDropOperation)operation {
  id items, path;
  NSFileManager *fileManager = [[NSFileManager alloc] init];
  BOOL isDir;
  
  if (!authorized) {
    return NSDragOperationNone;
  }
  
  items = [[info draggingPasteboard] propertyListForType:NSFilenamesPboardType];
  for (path in items) {
    if (![fileManager fileExistsAtPath:path isDirectory:&isDir] || !isDir) {
      NSLog(@"%@ %d", path, isDir);
      return NSDragOperationNone;
    }
  }
  
  return NSDragOperationGeneric;
}

- (BOOL) tableView:(NSTableView *)aTableView acceptDrop:(id)info row:(NSInteger)row dropOperation:(NSTableViewDropOperation)operation {
  id items, path;
  NSMutableArray *droppedApplications;
  Application *application;
  
  items = [[info draggingPasteboard] propertyListForType:NSFilenamesPboardType];
  droppedApplications = [NSMutableArray arrayWithCapacity:[items count]];
  for (path in items) {
    application = [[Application alloc] initWithDirectory:path];
    [application setDelegate:self];
    [droppedApplications addObject:application];
  }
  [applicationsController addObjects:droppedApplications];
  [applicationsController setSelectedObjects:[NSArray arrayWithObjects:[droppedApplications lastObject], nil]];
  [self apply:self];
    
  return YES;
}

- (BOOL) tableView:(NSTableView *)aTableView writeRowsWithIndexes:(NSIndexSet *)rowIndexes toPasteboard:(NSPasteboard *)pboard {
  Application *application;
  NSMutableArray *paths = [NSMutableArray arrayWithCapacity:[rowIndexes count]];
  
  for (application in [applications objectsAtIndexes:rowIndexes]) {
    [paths addObject:[application configFilename]];
  }
  [pboard declareTypes:[NSArray arrayWithObjects:NSFilenamesPboardType, nil] owner:self];
  [pboard setPropertyList:paths forType:NSFilenamesPboardType];
  
  return YES;
}

#pragma KeyValueObserving protocol methods

- (void)didChangeValueForKey:(NSString *)key {
  [super didChangeValueForKey:key];
  if (key == @"dirty") {
    [self checkForDirtyApplications];
  }
}

#pragma Notifications

- (void) paneWillBecomeActive:(id)sender {
  [self reloadApplications];
}

#pragma Actions

- (IBAction) add:(id)sender {
  Application *application = [[Application alloc] init];
  [application setDelegate:self];
  [applicationsController addObject:application];
}

- (IBAction) remove:(id)sender {
  Application *application = [self selectedApplication];
  if (![application isFresh]) {
    [[CLI sharedInstance] delete:application];
  }
  [applicationsController removeObject:application];
  [self checkForDirtyApplications];
}

- (IBAction) browse:(id)sender {
  NSOpenPanel *panel = [NSOpenPanel openPanel];
  [panel setCanChooseDirectories:YES];
  [panel setCanChooseFiles:NO];
  [panel setDirectory:[self pathForDirectoryBrowser]];
  [panel beginSheetModalForWindow:[mainView window] completionHandler:^(NSInteger button) {
    Application *application = [self selectedApplication];
    if (button == NSFileHandlingPanelOKButton) {
      [application setValue:[panel filename] forKey:@"path"];
    } else if (button == NSFileHandlingPanelCancelButton) {
      if ([application isFresh] && ![application isDirty]) {
        [self remove:sender];
      }
    }
  }];
}

- (IBAction) apply:(id)sender {
  Application *application;
  BOOL isChanged = NO;
  
  if ([self requestAuthorization]) {
    for (application in applications) {
      if ([application isDirty]) {
        if ([application isFresh]) {
          NSLog(@"Found new application: %@", application.host);
          [[CLI sharedInstance] add:application];
        } else {
          NSLog(@"Found dirty application: %@", application.host);
          [[CLI sharedInstance] update:application];
        }
        isChanged = YES;
      }
    }
    if (isChanged) {
      [[CLI sharedInstance] restart];
    }
  } else {
    NSLog(@"Unable to apply because authorization failed.");
  }
  [self checkForDirtyApplications];
}

- (IBAction) revert:(id)sender {
  Application *application;
  for (application in applications) {
    [application revert];
  }
}

- (IBAction) restart:(id)sender {
  Application *application = [self selectedApplication];
  [[CLI sharedInstance] restart:application];
}

- (IBAction) openAddressInBrowser:(id)sender {
  Application *application = [self selectedApplication];
  NSURL *url = [NSURL URLWithString:[NSString stringWithFormat:@"http://%@", application.host]];
  [[NSWorkspace sharedWorkspace] openURL:url];
}

- (IBAction) showPassengerHelp:(id)sender {}

#pragma Properties

- (void) loadApplications {
  [self setApplications:[[CLI sharedInstance] listApplications]];
}

- (void) reloadApplications {
  NSUInteger index;
  Application *existingApplication, *loadedApplication;
  NSArray *loadedApplications = [[CLI sharedInstance] listApplications];
  for (loadedApplication in loadedApplications) {
    index = [applications indexOfObjectPassingTest:^ BOOL (id object, NSUInteger index, BOOL *stop) {
      return [[object host] isEqualToString:[loadedApplication host]];
    }];
    // Application is already in the list
    if (index != -1) {
      existingApplication = [applications objectAtIndex:index];
      if (![existingApplication isDirty]) {
        [existingApplication updateAttributes:[loadedApplication toDictionary]];
        [existingApplication didApplyChanges];
      }
    // Someone added a new applications we didn't know about
    } else {
      [applicationsController addObject:loadedApplication];
    }
  }
}

- (Application *) selectedApplication {
  if (applications && ([applications count] > 0)) {
    return [[applicationsController selectedObjects] objectAtIndex:0];
  } else {
    return nil;
  }
}

- (NSString *) pathForDirectoryBrowser {
  Application *application = [self selectedApplication];
  if (application) {
    return application.path;
  } else {
    return NSHomeDirectory();
  }
}

- (BOOL) requestAuthorization {
  NSError *error;
  if ([[authorizationView authorization] obtainWithRight:kAuthorizationRightExecute
                                               flags:(kAuthorizationFlagPreAuthorize ||
                                                      kAuthorizationFlagExtendRights ||
                                                      kAuthorizationFlagInteractionAllowed)
                                               error:&error]
      ) {
    [self authorizationViewDidAuthorize:authorizationView];
    return YES;
  } else {
    return NO;
  }
}

- (void)checkForDirtyApplications {
  Application *application;
  for (application in applications) {
    if ([application isDirty]) {
      [self setDirty:YES];
      return;
    }
  }
  [self setDirty:NO];
}

@end