1/* 2 * Copyright (C) 2009 Apple Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 1. Redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer. 9 * 2. Redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution. 12 * 13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY 14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR 17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, 18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, 19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY 21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26#if !ENABLE_SPARKLE 27 28void initializeSparkle() 29{ 30 // No-op. 31} 32 33#else // ENABLE_SPARKLE 34 35#import <Cocoa/Cocoa.h> 36#import <Sparkle/SUUpdater.h> 37#import <objc/objc-runtime.h> 38#import "WebKitNightlyEnabler.h" 39 40// We need to tweak the wording of the prompt to make sense in the context of WebKit and Safari. 41static NSString* updatePermissionPromptDescription(id self, SEL _cmd) 42{ 43 return @"Should WebKit automatically check for updates? You can always check for updates manually from the Safari menu."; 44} 45 46static NSPanel *updateAlertPanel(id updateItem, id host) 47{ 48 NSString *hostName = objc_msgSend(host, @selector(name)); 49 NSPanel *panel = NSGetInformationalAlertPanel([NSString stringWithFormat:@"Would you like to download and install %@ %@ now?", hostName, objc_msgSend(updateItem, @selector(displayVersionString))], 50 [NSString stringWithFormat:@"You are currently running %@ %@.", hostName, objc_msgSend(host, @selector(displayVersion))], 51 @"Install Update", @"Skip This Version", @"Remind Me Later"); 52 NSArray *subviews = [[panel contentView] subviews]; 53 NSEnumerator *e = [subviews objectEnumerator]; 54 NSView *view; 55 while ((view = [e nextObject])) { 56 if (![view isKindOfClass:[NSButton class]]) 57 continue; 58 59 NSButton *button = (NSButton *)view; 60 [button setAction:@selector(webKitHandleButtonPress:)]; 61 if ([button tag] == NSAlertOtherReturn) 62 [button setKeyEquivalent:@"\033"]; 63 } 64 [panel center]; 65 return panel; 66} 67 68// Sparkle's udpate alert panel looks odd with the release notes hidden, so we 69// swap it out with a standard NSAlert-style panel instead. 70static id updateAlertInitForAlertPanel(id self, SEL _cmd, id updateItem, id host) 71{ 72 NSPanel *panel = updateAlertPanel(updateItem, host); 73 [panel setDelegate:self]; 74 75 self = [self initWithWindow:panel]; 76 if (!self) 77 return nil; 78 79 [updateItem retain]; 80 [host retain]; 81 82 object_setInstanceVariable(self, "updateItem", (void*)updateItem); 83 object_setInstanceVariable(self, "host", (void*)host); 84 85 [self setShouldCascadeWindows:NO]; 86 87 return self; 88} 89 90@implementation NSAlert (WebKitLauncherExtensions) 91 92- (void)webKitHandleButtonPress:(id)sender 93{ 94 // We rely on the fact that NSAlertOtherReturn == -1, NSAlertAlternateReturn == 0 and NSAlertDefaultReturn == 1 95 // to map the button tag to the corresponding selector 96 SEL selectors[] = { @selector(remindMeLater:), @selector(skipThisVersion:), @selector(installUpdate:) }; 97 SEL selector = selectors[[sender tag] + 1]; 98 99 id delegate = [[sender window] delegate]; 100 objc_msgSend(delegate, selector, sender); 101} 102 103@end 104 105#if __LP64__ 106 107#define setMethodImplementation method_setImplementation 108 109#else 110 111static void setMethodImplementation(Method m, IMP imp) 112{ 113 m->method_imp = imp; 114} 115 116#endif 117 118static NSString *userAgentStringForSparkle() 119{ 120 NSBundle *safariBundle = [NSBundle mainBundle]; 121 NSString *safariVersion = [[safariBundle localizedInfoDictionary] valueForKey:@"CFBundleShortVersionString"]; 122 NSString *safariBuild = [[[safariBundle infoDictionary] valueForKey:(NSString *)kCFBundleVersionKey] substringFromIndex:1]; 123 NSString *webKitRevision = [[webKitLauncherBundle() infoDictionary] valueForKey:(NSString *)kCFBundleVersionKey]; 124 NSString *applicationName = [NSString stringWithFormat:@"Version/%@ Safari/%@ WebKitRevision/%@", safariVersion, safariBuild, webKitRevision]; 125 Class WebView = objc_lookUpClass("WebView"); 126 return objc_msgSend(WebView, @selector(_standardUserAgentWithApplicationName:), applicationName); 127} 128 129void initializeSparkle() 130{ 131 // Override some Sparkle behaviour 132 Method methodToPatch = class_getInstanceMethod(objc_getRequiredClass("SUUpdatePermissionPrompt"), @selector(promptDescription)); 133 setMethodImplementation(methodToPatch, (IMP)updatePermissionPromptDescription); 134 135 methodToPatch = class_getInstanceMethod(objc_getRequiredClass("SUUpdateAlert"), @selector(initWithAppcastItem:host:)); 136 setMethodImplementation(methodToPatch, (IMP)updateAlertInitForAlertPanel); 137 138 SUUpdater *updater = [SUUpdater updaterForBundle:webKitLauncherBundle()]; 139 [updater setUserAgentString:userAgentStringForSparkle()]; 140 141 // Find the first separator on the Safari menu… 142 NSMenu *applicationSubmenu = [[[NSApp mainMenu] itemAtIndex:0] submenu]; 143 int i = 0; 144 for (; i < [applicationSubmenu numberOfItems]; i++) { 145 if ([[applicationSubmenu itemAtIndex:i] isSeparatorItem]) 146 break; 147 } 148 149 // … and insert a menu item that can be used to manually trigger update checks. 150 NSMenuItem *updateMenuItem = [[NSMenuItem alloc] initWithTitle:@"Check for WebKit Updates…" action:@selector(checkForUpdates:) keyEquivalent:@""]; 151 [updateMenuItem setTarget:updater]; 152 [applicationSubmenu insertItem:updateMenuItem atIndex:i]; 153 [updateMenuItem release]; 154} 155 156#endif // ENABLE_SPARKLE 157