ZSyncDaemon.m

#import "ZSyncDaemon.h"
#import <CoreData/CoreData.h>
#import <SyncServices/SyncServices.h>

#define ZSyncVersionNumber @"ZSyncVersionNumber"

@implementation ZSyncDaemon

+ (NSBundle*)myBundle
{
  NSString *path = [[NSBundle mainBundle] pathForResource:@"ZSyncInstaller" ofType:@"bundle"];
  return [NSBundle bundleWithPath:path];
}

+ (NSBundle*)daemonBundle
{
  NSBundle *myBundle = [self myBundle];
  NSString *path = [myBundle pathForResource:@"ZSyncDaemon" ofType:@"app"];
  return [NSBundle bundleWithPath:path];
}

+ (NSString*)basePath
{
  //Build our standard install path
  NSArray *paths = NSSearchPathForDirectoriesInDomains(NSApplicationSupportDirectory, NSUserDomainMask, YES);
  NSString *basePath = ([paths count] > 0) ? [paths objectAtIndex:0] : NSTemporaryDirectory();
  basePath = [basePath stringByAppendingPathComponent:@"ZSyncDaemon"];
  return basePath;
}

+ (NSString*)applicationPath
{
  NSString *basePath = [self basePath];
  return [basePath stringByAppendingPathComponent:@"ZSyncDaemon.app"];
}

+ (BOOL)checkBasePath:(NSString*)basePath error:(NSError**)error;
{
  NSFileManager *fileManager = [NSFileManager defaultManager];
  BOOL isDirectory = NO;
  if (![fileManager fileExistsAtPath:basePath isDirectory:&isDirectory]) {
    return NO;
  }
  if (!isDirectory) {
    NSString *errorDesc = [NSString stringWithFormat:@"Unknown file at base installation path: %@", basePath];
    NSDictionary *dictionary = [NSDictionary dictionaryWithObject:errorDesc forKey:NSLocalizedDescriptionKey];
    *error = [NSError errorWithDomain:@"ZSync" code:1123 userInfo:dictionary];
    return NO;
  }
  return YES;
}

+ (BOOL)checkApplicationPath:(NSString*)applicationPath error:(NSError**)error;
{
  NSFileManager *fileManager = [NSFileManager defaultManager];
  BOOL isDirectory = NO;
  if (![fileManager fileExistsAtPath:applicationPath isDirectory:&isDirectory]) {
    return NO;
  }
  if (!isDirectory) {
    NSString *errorDesc = [NSString stringWithFormat:@"Unknown file at daemon application installation path: %@", applicationPath];
    NSDictionary *dictionary = [NSDictionary dictionaryWithObject:errorDesc forKey:NSLocalizedDescriptionKey];
    *error = [NSError errorWithDomain:@"ZSync" code:1124 userInfo:dictionary];
    return NO;
  }
  return YES;
}

+ (BOOL)isDaemonInstalled:(NSError**)error;
{
  NSString *basePath = [self basePath];
  if (![self checkBasePath:basePath error:error]) {
    if (error) {
      return NO;
    }
    if (![[NSFileManager defaultManager] createDirectoryAtPath:basePath withIntermediateDirectories:YES attributes:nil error:error]) {
      return NO;
    }
  }
  NSString *applicationPath = [self applicationPath];
  return [self checkApplicationPath:applicationPath error:error];
}

+ (BOOL)installDaemon:(NSError**)error;
{
  NSString *basePath = [self basePath];
  NSFileManager *fileManager = [NSFileManager defaultManager];
  BOOL isDirectory = NO;
  
  if (![fileManager fileExistsAtPath:basePath isDirectory:&isDirectory]) {
    if (![[NSFileManager defaultManager] createDirectoryAtPath:basePath withIntermediateDirectories:YES attributes:nil error:error]) {
      return NO;
    }
  } else if (!isDirectory) {
    NSString *errorDesc = [NSString stringWithFormat:@"Unknown file at base installation path: %@", basePath];
    NSDictionary *dictionary = [NSDictionary dictionaryWithObject:errorDesc forKey:NSLocalizedDescriptionKey];
    *error = [NSError errorWithDomain:@"ZSync" code:1123 userInfo:dictionary];
    return NO;
  }
  
  NSString *myBundlePath = [[self daemonBundle] bundlePath];
  NSString *applicationPath = [self applicationPath];
  return [fileManager copyItemAtPath:myBundlePath toPath:applicationPath error:error];
}

+ (BOOL)stopDaemon:(NSError**)error
{
  NSString *appName = [[[self myBundle] infoDictionary] objectForKey:@"CFBundleExecutable"];
  NSString *command = [NSString stringWithFormat:@"tell application \"%@\" to quit", appName];
  NSAppleScript *quitScript;
  quitScript = [[NSAppleScript alloc] initWithSource:command];
  NSDictionary *errorDict = nil;
  // TODO: This should be turned into an NSError response
  if (![quitScript executeAndReturnError:&errorDict]) {
    NSAssert1(NO, @"Failure. What does it look like: %@", errorDict);
    [quitScript release], quitScript = nil;
    return NO;
  }
  [quitScript release], quitScript = nil;
  return YES;
}

+ (BOOL)updateInstalledApplication:(NSError**)error;
{
  NSString *appName = [[[self myBundle] infoDictionary] objectForKey:@"CFBundleExecutable"];
  
  if ([self isDaemonRunning] && ![self stopDaemon:error]) return NO;
  
  NSString *myBundlePath = [[self myBundle] bundlePath];
  
  NSString *applicationPath = [self applicationPath];
  NSString *basePath = [self basePath];
  NSWorkspace *workspace = [NSWorkspace sharedWorkspace];
  if (![workspace performFileOperation:NSWorkspaceRecycleOperation 
                                source:basePath 
                           destination:@"" 
                                 files:[NSArray arrayWithObject:appName] 
                                   tag:NULL]) {
    NSString *errorDesc = [NSString stringWithFormat:@"Failed to move old app version to the trash"];
    NSDictionary *dictionary = [NSDictionary dictionaryWithObject:errorDesc forKey:NSLocalizedDescriptionKey];
    *error = [NSError errorWithDomain:@"ZSync" code:1126 userInfo:dictionary];
    return NO;
  }
  
  NSFileManager *fileManager = [NSFileManager defaultManager];
  return [fileManager copyItemAtPath:myBundlePath toPath:applicationPath error:error];
}

+ (NSString*)pluginPath
{
  NSString *basePath = [self basePath];
  return [basePath stringByAppendingPathComponent:@"Plugins"];
}

+ (BOOL)isPluginInstalled:(NSString*)path error:(NSError**)error
{
  NSString *pluginPath = [self pluginPath];
  NSFileManager *fileManager = [NSFileManager defaultManager];
  BOOL isDirectory = NO;
  if (![fileManager fileExistsAtPath:pluginPath isDirectory:&isDirectory]) {
    return NO;
  }
  if (!isDirectory) {
    NSString *errorDesc = [NSString stringWithFormat:@"Unknown file at plugin installation path: %@", pluginPath];
    NSDictionary *dictionary = [NSDictionary dictionaryWithObject:errorDesc forKey:NSLocalizedDescriptionKey];
    *error = [NSError errorWithDomain:@"ZSync" code:1123 userInfo:dictionary];
    return NO;
  }
  
  NSString *finalPath = [pluginPath stringByAppendingPathComponent:[path lastPathComponent]];
  if (![fileManager fileExistsAtPath:finalPath isDirectory:&isDirectory]) {
    return NO;
  }
  return YES;
}

+ (BOOL)installPluginAtPath:(NSString*)path intoDaemonWithError:(NSError**)error;
{
  if (![[path pathExtension] isEqualToString:@"zsyncPlugin"]) {
    NSString *errorDesc = [NSString stringWithFormat:@"url is not a valid ZSync plugin: %@", path];
    NSDictionary *dictionary = [NSDictionary dictionaryWithObject:errorDesc forKey:NSLocalizedDescriptionKey];
    *error = [NSError errorWithDomain:@"ZSync" code:1125 userInfo:dictionary];
    return NO;
  }
  
  BOOL isDaemonInstalled = [self isDaemonInstalled:error];
  
  if (!isDaemonInstalled && *error) {
    return NO;
  }
  
  if (!isDaemonInstalled) {
    if (![self installDaemon:error]) return NO;
  } else {
    NSDictionary *infoDictionary = [[self daemonBundle] infoDictionary];
    NSInteger currentVersionNumber = [[infoDictionary objectForKey:ZSyncVersionNumber] integerValue];
  
    NSBundle *installed = [NSBundle bundleWithPath:[self applicationPath]];
    NSInteger installedVersionNumber = [[[installed infoDictionary] objectForKey:ZSyncVersionNumber] integerValue];
    
    if (installedVersionNumber < currentVersionNumber && ![self updateInstalledApplication:error]) return NO;
  }
  
  //Is plugin already installed?
  if ([self isPluginInstalled:path error:error]) {
    return YES;
  }
  
  //Shutdown the daemon to install the plugin
  if ([self isDaemonRunning] && ![self stopDaemon:error]) {
    return NO;
  }
  
  //Install the plugin
  NSString *pluginPath = [self pluginPath];
  NSFileManager *fileManager = [NSFileManager defaultManager];
  BOOL isDirectory = NO;
  
  if (![fileManager fileExistsAtPath:pluginPath isDirectory:&isDirectory]) {
    if (![[NSFileManager defaultManager] createDirectoryAtPath:pluginPath withIntermediateDirectories:YES attributes:nil error:error]) {
      return NO;
    }
  } else if (!isDirectory) {
    NSString *errorDesc = [NSString stringWithFormat:@"Unknown file at base installation path: %@", pluginPath];
    NSDictionary *dictionary = [NSDictionary dictionaryWithObject:errorDesc forKey:NSLocalizedDescriptionKey];
    *error = [NSError errorWithDomain:@"ZSync" code:1123 userInfo:dictionary];
    return NO;
  }
  
  pluginPath = [pluginPath stringByAppendingPathComponent:[path lastPathComponent]];
  if (![fileManager copyItemAtPath:path toPath:pluginPath error:error]) {
    return NO;
  }
  
  //Start the daemon back up
  [self startDaemon];
  
  return YES;
}

+ (BOOL)isDaemonRunning;
{
  NSDictionary *infoDictionary = [[self daemonBundle] infoDictionary];
  NSString *myBundleID = [infoDictionary objectForKey:@"CFBundleIdentifier"];
  
  NSWorkspace *workspace = [NSWorkspace sharedWorkspace];
  NSArray *apps = [workspace launchedApplications];
  for (NSDictionary *appDict in apps) {
    NSString *bundleID = [appDict objectForKey:@"NSApplicationBundleIdentifier"];
    if ([bundleID isEqualToString:myBundleID]) return YES;
  }
  return NO;
}

+ (void)startDaemon;
{
  NSDictionary *infoDictionary = [[self myBundle] infoDictionary];
  NSString *appName = [infoDictionary objectForKey:@"CFBundleExecutable"];
  
  NSWorkspace *workspace = [NSWorkspace sharedWorkspace];
  [workspace launchApplication:appName];
}

+ (NSManagedObjectContext*)managedObjectContext:(NSError**)error
{
  NSBundle *appBundle = [NSBundle bundleWithPath:[self applicationPath]];
  if (!appBundle) {
    NSString *errorDesc = [NSString stringWithFormat:@"ZSyncDaemon is not installed: %@", [self applicationPath]];
    NSDictionary *dictionary = [NSDictionary dictionaryWithObject:errorDesc forKey:NSLocalizedDescriptionKey];
    *error = [NSError errorWithDomain:@"ZSync" code:1129 userInfo:dictionary];
    return nil;
  }
  NSString *path = [appBundle pathForResource:@"ZSyncModel" ofType:@"mom"];
  NSManagedObjectModel *model = [[NSManagedObjectModel alloc] initWithContentsOfURL:[NSURL fileURLWithPath:path]];
  
  if (!model) {
    NSString *errorDesc = [NSString stringWithFormat:@"Failed to find model at path: %@", path];
    NSDictionary *dictionary = [NSDictionary dictionaryWithObject:errorDesc forKey:NSLocalizedDescriptionKey];
    *error = [NSError errorWithDomain:@"ZSync" code:1127 userInfo:dictionary];
    return nil;
  }
  
  NSString *basePath = [ZSyncDaemon basePath];
  if (![ZSyncDaemon checkBasePath:basePath error:error]) return nil;
  
  NSString *filePath = [basePath stringByAppendingPathComponent:@"SyncHistory.sqlite"];
  
  NSPersistentStoreCoordinator *psc = [[NSPersistentStoreCoordinator alloc] initWithManagedObjectModel:model];
  
  if (![psc addPersistentStoreWithType:NSSQLiteStoreType configuration:nil URL:[NSURL fileURLWithPath:filePath] options:nil error:error]) return nil;
  
  NSManagedObjectContext *moc = [[NSManagedObjectContext alloc] init];
  [moc setPersistentStoreCoordinator:psc];
  [psc release], psc = nil;
  [model release], model = nil;
  return [moc autorelease];
}

+ (BOOL)deregisterDeviceForUUID:(NSString*)uuid error:(NSError**)error;
{
  NSArray *components = [uuid componentsSeparatedByString:@":"];
  if ([components count] != 2) {
    NSString *errorDesc = [NSString stringWithFormat:@"Invalid UUID: %@", uuid];
    NSDictionary *dictionary = [NSDictionary dictionaryWithObject:errorDesc forKey:NSLocalizedDescriptionKey];
    *error = [NSError errorWithDomain:@"ZSync" code:1128 userInfo:dictionary];
    return NO;
  }
  NSManagedObjectContext *moc = [self managedObjectContext:error];
  if (!moc) return NO;
  
  NSFetchRequest *request = [[NSFetchRequest alloc] init];
  [request setEntity:[NSEntityDescription entityForName:@"Application" inManagedObjectContext:moc]];
  [request setPredicate:[NSPredicate predicateWithFormat:@"uuid == %@ && device.uuid == %@", [components objectAtIndex:1], [components objectAtIndex:0]]];
  
  id result = [[moc executeFetchRequest:request error:error] lastObject];
  [request release], request = nil;
  if (!result && error) return NO;
  if (!result) return YES;
  
  ISyncClient *client = [[ISyncManager sharedManager] clientWithIdentifier:[components objectAtIndex:1]];
  if (client) {
    [[ISyncManager sharedManager] unregisterClient:client];
  }
  
  [moc deleteObject:result];
  return YES;
}

+ (NSArray*)devicesRegisteredForSchema:(NSString*)schema error:(NSError**)error;
{
  NSManagedObjectContext *moc = [self managedObjectContext:error];
  if (!moc) return nil;
  
  NSFetchRequest *request = [[NSFetchRequest alloc] init];
  [request setEntity:[NSEntityDescription entityForName:@"Application" inManagedObjectContext:moc]];
  [request setPredicate:[NSPredicate predicateWithFormat:@"schema == %@", schema]];
  
  NSArray *applications = [moc executeFetchRequest:request error:error];
  [request release], request = nil;
  
  NSMutableArray *deviceArray = [NSMutableArray array];
  for (NSManagedObject *applicationMO in applications) {
    NSManagedObject *deviceMO = [applicationMO valueForKey:@"device"];
    NSMutableDictionary *dict = [[NSMutableDictionary alloc] init];
    [dict setObject:[deviceMO valueForKey:@"name"] forKey:@"name"];
    [dict setObject:[NSString stringWithFormat:@"%@:%@", [deviceMO valueForKey:@"uuid"], [applicationMO valueForKey:@"uuid"]] forKey:@"uuid"];
    [deviceArray addObject:dict];
    [dict release], dict = nil;
  }
  return deviceArray;
}

@end