source: branches/CoreData/SapphireFrappliance/FRAppliance/SapphireImportHelper.m @ 659

/*
 * SapphireImportHelper.m
 * Sapphire
 *
 * Created by Graham Booker on Dec. 8, 2007.
 * Copyright 2007 Sapphire Development Team and/or www.nanopi.net
 * All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify it under the terms of the GNU
 * General Public License as published by the Free Software Foundation; either version 3 of the License,
 * or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even
 * the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
 * Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License along with this program; if not,
 * write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#import <Security/Security.h>
#include <sys/stat.h>
#include <sys/mount.h>

#import "SapphireImportHelper.h"
#import "SapphireMetaData.h"
#import "SapphireAllImporter.h"
#import "SapphireFileDataImporter.h"
#import "SapphireTVShowImporter.h"
#import "SapphireMovieImporter.h"
#import "SapphireFileMetaData.h"
#import "SapphireMetaDataSupport.h"

#define CONNECTION_NAME @"Sapphire Server"

@interface SapphireImportFile : NSObject <SapphireImportFileProtocol>{
        NSString                                                                *path;
        id <SapphireImporterBackgroundProtocol> informer;
        ImportType                                                              type;
}
- (id)initWithPath:(NSString *)aPath informer:(id <SapphireImporterBackgroundProtocol>)aInformer type:(ImportType)aType;
@end

@interface SapphireImportHelperServer (private)
- (void)startClient;
@end

@implementation SapphireImportHelper

static SapphireImportHelper *shared = nil;

+ (SapphireImportHelper *)sharedHelperForContext:(NSManagedObjectContext *)moc
{
        if(shared == nil)
                shared = [[SapphireImportHelperServer alloc] initWithContext:moc];

        return shared;
}

+ (void)relinquishHelper
{
        if(shared != nil)
                [shared relinquishHelper];
}

- (void)relinquishHelper
{
}

- (void)importFileData:(SapphireFileMetaData *)file inform:(id <SapphireImporterBackgroundProtocol>)inform;
{
}

- (void)importAllData:(SapphireFileMetaData *)file inform:(id <SapphireImporterBackgroundProtocol>)inform;
{
}

- (void)removeObjectsWithInform:(id <SapphireImporterBackgroundProtocol>)inform
{
}

@end

@implementation SapphireImportHelperClient

- (id)initWithContext:(NSManagedObjectContext *)context
{
        self = [super init];
        if(!self)
                return nil;
        
        moc = [context retain];
        SapphireFileDataImporter *fileImp = [[SapphireFileDataImporter alloc] init];
        SapphireTVShowImporter *tvImp = [[SapphireTVShowImporter alloc] initWithContext:moc];
        SapphireMovieImporter *movImp = [[SapphireMovieImporter alloc] initWithContext:moc];
        allImporter = [[SapphireAllImporter alloc] initWithImporters:[NSArray arrayWithObjects:tvImp,movImp,fileImp,nil]];
        [fileImp release];
        [tvImp release];
        [movImp release];
        keepRunning = YES;
        
        return self;
}
- (void) dealloc
{
        [server release];
        [allImporter release];
        [moc release];
        [super dealloc];
}

- (void)importFileData:(SapphireFileMetaData *)file inform:(id <SapphireImporterBackgroundProtocol>)inform;
{
        updateMetaData(file);
}

- (void)startChild
{
        /*Child here*/
        @try {
                NSConnection *connection = [NSConnection connectionWithRegisteredName:CONNECTION_NAME host:nil];
                id serverobj = [[connection rootProxy] retain];
                [serverobj setProtocolForProxy:@protocol(SapphireImportServer)];
                shared = self;
                [serverobj setClient:(SapphireImportHelperClient *)shared];
                server = serverobj;     
                [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(connectionDidDie:) name:NSConnectionDidDieNotification object:nil];           
        }
        @catch (NSException * e) {
                keepRunning = NO;
        }
}

- (BOOL)keepRunning
{
        return keepRunning;
}

- (void)connectionDidDie:(NSNotification *)note
{
        [self exitChild];
}

- (oneway void)exitChild
{
        keepRunning = NO;
}

- (oneway void)startQueue
{
        id <SapphireImportFileProtocol> file = nil;
        while((file = [server nextFile]) != nil)
        {
                NSAutoreleasePool *singleImportPool = [[NSAutoreleasePool alloc] init];
                ImportType type = [file importType];
                BOOL ret;
                NSString *path = [file path];
                SapphireFileMetaData *file = [SapphireFileMetaData fileWithPath:path inContext:moc];
                [moc refreshObject:file mergeChanges:YES];
                if(type == IMPORT_TYPE_FILE_DATA)
                        ret = updateMetaData(file);
                else
                        ret = ([allImporter importMetaData:file] == IMPORT_STATE_UPDATED);
                NSError *error = nil;
                [moc save:&error];
                [server importComplete:ret];
                [singleImportPool release];
        }
}
@end

@implementation SapphireImportHelperServer

- (id)initWithContext:(NSManagedObjectContext *)context
{
        self = [super init];
        if (self == nil)
                return nil;
        
        queue = [[NSMutableArray alloc] init];
        queueSuspended = NO;
        
        serverConnection = [NSConnection defaultConnection];
        [serverConnection setRootObject:self];
        if([serverConnection registerName:CONNECTION_NAME] == NO)
                NSLog(@"Register failed");
        
        [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(connectionDidDie:) name:NSConnectionDidDieNotification object:nil];
        moc = [context retain];
        
        [self startClient];
        
        return self;
}

- (void) dealloc
{
        [[NSNotificationCenter defaultCenter] removeObserver:self];
        [client release];
        [moc release];
        [queue release];
        [currentImporting release];
        [super dealloc];
}

- (void)relinquishHelper
{
        [client exitChild];
        [serverConnection registerName:nil];
        [serverConnection setRootObject:nil];
        [shared autorelease];
        shared = nil;
}

- (BOOL)isSlashReadOnly
{
        struct statfs *mntbufp;
        
    int i, mountCount = getmntinfo(&mntbufp, MNT_NOWAIT);
        for(i=0; i<mountCount; i++)
        {
                if(!strcmp(mntbufp[i].f_mntonname, "/"))
                        return (mntbufp[i].f_flags & MNT_RDONLY) ? YES : NO;
        }
        
        return NO;
}

- (BOOL)fixClientPermissions:(NSString *)path
{
        /* Permissions are incorrect */
        AuthorizationItem authItems[2] = {
                {kAuthorizationEnvironmentUsername, strlen("frontrow"), "frontrow", 0},
                {kAuthorizationEnvironmentPassword, strlen("frontrow"), "frontrow", 0},
        };
        AuthorizationEnvironment environ = {2, authItems};
        AuthorizationItem rightSet[] = {{kAuthorizationRightExecute, 0, NULL, 0}};
        AuthorizationRights rights = {1, rightSet};
        AuthorizationRef auth;
        OSStatus result = AuthorizationCreate(&rights, &environ, kAuthorizationFlagPreAuthorize | kAuthorizationFlagExtendRights, &auth);
        if(result == errAuthorizationSuccess)
        {
                BOOL roslash = [self isSlashReadOnly];
                if(roslash)
                {
                        char *command = "mount -uw /";
                        char *arguments[] = {"-c", command, NULL};
                        AuthorizationExecuteWithPrivileges(auth, "/bin/sh", kAuthorizationFlagDefaults, arguments, NULL);
                }
                char *command = "chmod +rx \"$HELP\"";
                setenv("HELP", [path fileSystemRepresentation], 1);
                char *arguments[] = {"-c", command, NULL};
                result = AuthorizationExecuteWithPrivileges(auth, "/bin/sh", kAuthorizationFlagDefaults, arguments, NULL);
                unsetenv("HELP");
                if(roslash)
                {
                        char *command = "mount -ur /";
                        char *arguments[] = {"-c", command, NULL};
                        AuthorizationExecuteWithPrivileges(auth, "/bin/sh", kAuthorizationFlagDefaults, arguments, NULL);
                }
        }
        if(result != errAuthorizationSuccess)
                return NO;
        
        return YES;
}

- (void)startClient
{
        NSString *path = [[NSBundle bundleForClass:[SapphireImportHelper class]] pathForResource:@"ImportHelper" ofType:@""];
        NSDictionary *attrs = [[NSFileManager defaultManager] fileAttributesAtPath:path traverseLink:YES];
        if(([[attrs objectForKey:NSFilePosixPermissions] intValue] | S_IXOTH) || [self fixClientPermissions:path])
        {
                @try {
                        [NSTask launchedTaskWithLaunchPath:path arguments:[NSArray array]];
                }
                @catch (NSException * e) {
                        NSLog(@"Could not launch helper because of exception %@ launching %@.  Make this file executable", e, path);
                }               
        }
        else
                NSLog(@"Could not correct helper permissions on %@.  Make this file executable!", path);
}

- (void)connectionDidDie:(NSNotification *)note
{
        [client release];
        client = nil;
        /*Inform that import completed (since it crashed, no update done)*/
        [self importComplete:NO];
        if(shared != nil)
                /* Don't start it again if we are shutting down*/
                [self startClient];
}

- (void)itemAdded
{
        if(!queueSuspended)
                return;
        queueSuspended = NO;
        [SapphireMetaDataSupport save:moc];
        [client startQueue];
}

- (void)importFileData:(SapphireFileMetaData *)file inform:(id <SapphireImporterBackgroundProtocol>)inform;
{
        SapphireImportFile *item = [[SapphireImportFile alloc] initWithPath:[file path] informer:inform  type:IMPORT_TYPE_FILE_DATA];
        [queue addObject:item];
        [item release];
        [self itemAdded];
}

- (void)importAllData:(SapphireFileMetaData *)file inform:(id <SapphireImporterBackgroundProtocol>)inform;
{
        SapphireImportFile *item = [[SapphireImportFile alloc] initWithPath:[file path] informer:inform  type:IMPORT_TYPE_ALL_DATA];
        [queue addObject:item];
        [item release];
        [self itemAdded];
}

- (void)removeObjectsWithInform:(id <SapphireImporterBackgroundProtocol>)inform
{
        if(inform == nil)
                return;
        
        int i, count=[queue count];
        for(i=0; i<count; i++)
        {
                id <SapphireImportFileProtocol> file = [queue objectAtIndex:i];
                if([file informer] == inform)
                {
                        [queue removeObjectAtIndex:i];
                        i--;
                        count--;
                }
        }
}

- (id <SapphireImportFileProtocol>)nextFile
{
        if(![queue count])
        {
                queueSuspended = YES;
                return nil;
        }
        [currentImporting release];
        currentImporting = [[queue objectAtIndex:0] retain];
        [queue removeObjectAtIndex:0];
        return currentImporting;
}

- (oneway void)setClient:(id <SapphireImportClient>)aClient
{
        if(shared == nil)
        {
                [aClient exitChild];
                return;
        }
        client = [aClient retain];
        if([queue count])
        {
                queueSuspended = NO;
                [client startQueue];
        }
        else
                queueSuspended = YES;
}

- (oneway void)importComplete:(BOOL)updated
{
        if(currentImporting == nil)
                return;
        NSString *path = [currentImporting path];
        SapphireFileMetaData *file = [SapphireFileMetaData fileWithPath:path inContext:moc];
        if(file != nil)
                [moc refreshObject:file mergeChanges:YES];
        [[currentImporting informer] informComplete:updated];
        [currentImporting release];
        currentImporting = nil;
}

@end

@implementation SapphireImportFile
- (id)initWithPath:(NSString *)aPath informer:(id <SapphireImporterBackgroundProtocol>)aInformer type:(ImportType)aType;
{
        self = [super init];
        if(!self)
                return nil;
        
        path = [aPath retain];
        informer = [aInformer retain];
        type = aType;
        
        return self;
}
- (void) dealloc
{
        [path release];
        [informer release];
        [super dealloc];
}

- (bycopy NSString *)path
{
        return path;
}
- (id <SapphireImporterBackgroundProtocol>)informer
{
        return informer;
}

- (ImportType)importType
{
        return type;
}

@end