blob: 8b13ab8ba6c0d91bbd6ca75e8b495632ffa07546 [file] [log] [blame]
/*
* Copyright (C) 2009 Apple Inc. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#if !ENABLE_SPARKLE
void initializeSparkle()
{
// No-op.
}
#else // ENABLE_SPARKLE
#import <Cocoa/Cocoa.h>
#import <Sparkle/SUUpdater.h>
#import <objc/objc-runtime.h>
#import "WebKitNightlyEnabler.h"
// We need to tweak the wording of the prompt to make sense in the context of WebKit and Safari.
static NSString* updatePermissionPromptDescription(id self, SEL _cmd)
{
return @"Should WebKit automatically check for updates? You can always check for updates manually from the Safari menu.";
}
static NSPanel *updateAlertPanel(id updateItem, id host)
{
NSString *hostName = objc_msgSend(host, @selector(name));
NSPanel *panel = NSGetInformationalAlertPanel([NSString stringWithFormat:@"Would you like to download and install %@ %@ now?", hostName, objc_msgSend(updateItem, @selector(displayVersionString))],
[NSString stringWithFormat:@"You are currently running %@ %@.", hostName, objc_msgSend(host, @selector(displayVersion))],
@"Install Update", @"Skip This Version", @"Remind Me Later");
NSArray *subviews = [[panel contentView] subviews];
NSEnumerator *e = [subviews objectEnumerator];
NSView *view;
while ((view = [e nextObject])) {
if (![view isKindOfClass:[NSButton class]])
continue;
NSButton *button = (NSButton *)view;
[button setAction:@selector(webKitHandleButtonPress:)];
if ([button tag] == NSAlertOtherReturn)
[button setKeyEquivalent:@"\033"];
}
[panel center];
return panel;
}
// Sparkle's udpate alert panel looks odd with the release notes hidden, so we
// swap it out with a standard NSAlert-style panel instead.
static id updateAlertInitForAlertPanel(id self, SEL _cmd, id updateItem, id host)
{
NSPanel *panel = updateAlertPanel(updateItem, host);
[panel setDelegate:self];
self = [self initWithWindow:panel];
if (!self)
return nil;
[updateItem retain];
[host retain];
object_setInstanceVariable(self, "updateItem", (void*)updateItem);
object_setInstanceVariable(self, "host", (void*)host);
[self setShouldCascadeWindows:NO];
return self;
}
@implementation NSAlert (WebKitLauncherExtensions)
- (void)webKitHandleButtonPress:(id)sender
{
// We rely on the fact that NSAlertOtherReturn == -1, NSAlertAlternateReturn == 0 and NSAlertDefaultReturn == 1
// to map the button tag to the corresponding selector
SEL selectors[] = { @selector(remindMeLater:), @selector(skipThisVersion:), @selector(installUpdate:) };
SEL selector = selectors[[sender tag] + 1];
id delegate = [[sender window] delegate];
objc_msgSend(delegate, selector, sender);
}
@end
#if __LP64__
#define setMethodImplementation method_setImplementation
#else
static void setMethodImplementation(Method m, IMP imp)
{
m->method_imp = imp;
}
#endif
static NSString *userAgentStringForSparkle()
{
NSBundle *safariBundle = [NSBundle mainBundle];
NSString *safariVersion = [[safariBundle localizedInfoDictionary] valueForKey:@"CFBundleShortVersionString"];
NSString *safariBuild = [[[safariBundle infoDictionary] valueForKey:(NSString *)kCFBundleVersionKey] substringFromIndex:1];
NSString *webKitRevision = [[webKitLauncherBundle() infoDictionary] valueForKey:(NSString *)kCFBundleVersionKey];
NSString *applicationName = [NSString stringWithFormat:@"Version/%@ Safari/%@ WebKitRevision/%@", safariVersion, safariBuild, webKitRevision];
Class WebView = objc_lookUpClass("WebView");
return objc_msgSend(WebView, @selector(_standardUserAgentWithApplicationName:), applicationName);
}
void initializeSparkle()
{
// Override some Sparkle behaviour
Method methodToPatch = class_getInstanceMethod(objc_getRequiredClass("SUUpdatePermissionPrompt"), @selector(promptDescription));
setMethodImplementation(methodToPatch, (IMP)updatePermissionPromptDescription);
methodToPatch = class_getInstanceMethod(objc_getRequiredClass("SUUpdateAlert"), @selector(initWithAppcastItem:host:));
setMethodImplementation(methodToPatch, (IMP)updateAlertInitForAlertPanel);
SUUpdater *updater = [SUUpdater updaterForBundle:webKitLauncherBundle()];
[updater setUserAgentString:userAgentStringForSparkle()];
// Find the first separator on the Safari menu…
NSMenu *applicationSubmenu = [[[NSApp mainMenu] itemAtIndex:0] submenu];
int i = 0;
for (; i < [applicationSubmenu numberOfItems]; i++) {
if ([[applicationSubmenu itemAtIndex:i] isSeparatorItem])
break;
}
// … and insert a menu item that can be used to manually trigger update checks.
NSMenuItem *updateMenuItem = [[NSMenuItem alloc] initWithTitle:@"Check for WebKit Updates…" action:@selector(checkForUpdates:) keyEquivalent:@""];
[updateMenuItem setTarget:updater];
[applicationSubmenu insertItem:updateMenuItem atIndex:i];
[updateMenuItem release];
}
#endif // ENABLE_SPARKLE