SparkleManager.m

/*
 
 The MIT License
 
 Copyright (c) 2008-2009 ClickToFlash Developers
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal
 in the Software without restriction, including without limitation the rights
 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 copies of the Software, and to permit persons to whom the Software is
 furnished to do so, subject to the following conditions:
 
 The above copyright notice and this permission notice shall be included in
 all copies or substantial portions of the Software.
 
 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 THE SOFTWARE.
 
 */

#import "SparkleManager.h"
#import <Sparkle/Sparkle.h>

#import "CTFUserDefaultsController.h"
#import "CTFPreferencesDictionary.h"

#import <objc/runtime.h>

// NSUserDefaults keys
static NSString *sAutomaticallyCheckForUpdates = @"checkForUpdatesOnFirstLoad";

@implementation SparkleManager

+ (id)sharedManager {
    static SparkleManager *result = nil;
    if (!result) {
        result = [[SparkleManager alloc] init];
    }
    return result;
}

- (id)init {
    self = [super init];
    if (self) {
        _canUpdate = NO;
    }
    return self;
}

- (void)dealloc {
    [_updater setDelegate:nil];
    [super dealloc];
}

- (BOOL)canUpdate {
    return _canUpdate;
}

- (SUUpdater*)_updater {
    if (_updater)
        return _updater;
    
    NSString *frameworksPath = [[NSBundle bundleForClass:[self class]] privateFrameworksPath];
    NSAssert(frameworksPath, nil);
    
    NSString *sparkleFrameworkPath = [NSBundle pathForResource:@"Sparkle" ofType:@"framework" inDirectory:frameworksPath];
    NSAssert(sparkleFrameworkPath, nil);
    
    NSBundle *sparkleFramework = [NSBundle bundleWithPath:sparkleFrameworkPath];
    NSAssert(sparkleFramework, nil);
    
    NSError *error = nil;
    BOOL loaded;
    if ([sparkleFramework respondsToSelector:@selector(loadAndReturnError:)]) {
        loaded = [sparkleFramework loadAndReturnError:&error];
    } else {
        loaded = [sparkleFramework load];
    }
    if (loaded) {
        NSBundle *clickToFlashBundle = [NSBundle bundleWithIdentifier:@"com.github.rentzsch.clicktoflash"];
        NSAssert(clickToFlashBundle, nil);
        
        Class updaterClass = objc_getClass("SUUpdater");
        NSAssert(updaterClass, nil);
        
		if ([updaterClass respondsToSelector:@selector(updaterForBundle:)]) {
			_canUpdate = YES;
			_updater = [updaterClass updaterForBundle:clickToFlashBundle];
			NSAssert(_updater, nil);
			
			[_updater setDelegate:self];
		}
    }
    
    if (error) NSLog(@"error loading ClickToFlash's Sparkle: %@", error);
    
    return _updater;
}

- (void)startAutomaticallyCheckingForUpdates {
    if (![[CTFUserDefaultsController standardUserDefaults] objectForKey:sAutomaticallyCheckForUpdates]) {
        // If the key isn't set yet, default to YES, automatically check for updates.
        [[CTFUserDefaultsController standardUserDefaults] setBool:YES forKey:sAutomaticallyCheckForUpdates];
    }
    
	SUUpdater *updater = [self _updater];
	if (_canUpdate) {
		if ([[CTFUserDefaultsController standardUserDefaults] boolForKey:sAutomaticallyCheckForUpdates]) {
			[updater setAutomaticallyChecksForUpdates:YES];
            static BOOL calledUpdaterApplicationDidFinishLaunching = NO;
            if (!calledUpdaterApplicationDidFinishLaunching) {
                calledUpdaterApplicationDidFinishLaunching = YES;
                [updater applicationDidFinishLaunching:nil];
            }
		} else {
			[updater setAutomaticallyChecksForUpdates:NO];
		}
	}
}

- (void)checkForUpdates {
    [[self _updater] checkForUpdates:nil];
}

- (void)setAutomaticallyChecksForUpdates:(BOOL)checksForUpdates
{
	[[self _updater] setAutomaticallyChecksForUpdates:checksForUpdates];
}

- (NSString*)pathToRelaunchForUpdater:(SUUpdater*)updater {
    return _pathToRelaunch;
}

- (BOOL)updater:(SUUpdater *)updater
shouldPostponeRelaunchForUpdate:(SUAppcastItem *)update
  untilInvoking:(NSInvocation *)invocation;
{
	NSString *appNameString = [[[NSBundle mainBundle] infoDictionary] objectForKey:@"CFBundleName"];
	int relaunchResult = NSRunAlertPanel([NSString stringWithFormat:@"Relaunch %@ now?",appNameString],
										 [NSString stringWithFormat:@"To use the new features of ClickToFlash, %@ needs to be relaunched.",appNameString],
										 @"Relaunch",
										 @"Do not relaunch",
										 nil);
	
	BOOL shouldPostpone = YES;
	if (relaunchResult == NSAlertDefaultReturn) {
		// we want to relaunch now, so don't postpone the relaunch
		
		shouldPostpone = NO;
	} else {
		// we want to postpone the relaunch and let the user decide when to do so,
		// so we don't even bother with saving the invocation and reinvoking
		// it later
	}
	return shouldPostpone;
}

- (NSString *)pathToRelaunch
{
	return _pathToRelaunch;
}
- (void)setPathToRelaunch:(NSString *)newValue
{
	[newValue retain];
	[_pathToRelaunch release];
	_pathToRelaunch = newValue;
}

@end