source: trunk/Sparkle/SUAutomaticUpdateDriver.m @ 1046

//
//  SUAutomaticUpdateDriver.m
//  Sparkle
//
//  Created by Andy Matuschak on 5/6/08.
//  Copyright 2008 Andy Matuschak. All rights reserved.
//

#import "SUAutomaticUpdateDriver.h"

#import "SUAutomaticUpdateAlert.h"
#import "SUHost.h"

@implementation SUAutomaticUpdateDriver

- (void)unarchiverDidFinish:(SUUnarchiver *)ua
{
    alert = [[SUAutomaticUpdateAlert alloc] initWithAppcastItem:updateItem host:host delegate:self];
    if ([NSApp isActive])
        [[alert window] makeKeyAndOrderFront:self];
    else
        [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(applicationDidBecomeActive:) name:NSApplicationDidBecomeActiveNotification object:NSApp]; 
}

- (void)applicationDidBecomeActive:(NSNotification *)aNotification
{
    [[alert window] makeKeyAndOrderFront:self];
    [[NSNotificationCenter defaultCenter] removeObserver:self name:@"NSApplicationDidBecomeActiveNotification" object:NSApp];
}

- (void)automaticUpdateAlert:(SUAutomaticUpdateAlert *)aua finishedWithChoice:(SUAutomaticInstallationChoice)choice;
{
    switch (choice)
    {
        case SUInstallNowChoice:
            [self installUpdate];
            break;
            
        case SUInstallLaterChoice:
            postponingInstallation = YES;
            [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(applicationWillTerminate:) name:NSApplicationWillTerminateNotification object:nil];
            break;

        case SUDoNotInstallChoice:
            [host setObject:[updateItem versionString] forUserDefaultsKey:SUSkippedVersionKey];
            [self abortUpdate];
            break;
    }
}

- (BOOL)shouldInstallSynchronously { return postponingInstallation; }

- (void)installUpdate
{
    showErrors = YES;
    [super installUpdate];
}

- (void)applicationWillTerminate:(NSNotification *)note
{
    [self installUpdate];
}

- (void)installerFinishedForHost:(SUHost *)aHost
{
    if (aHost != host) { return; }
    if (!postponingInstallation)
        [self relaunchHostApp];
}

- (void)abortUpdateWithError:(NSError *)error
{
    if (showErrors)
        [super abortUpdateWithError:error];
    else
        [self abortUpdate];
}

@end