blob: 45ae1c2c7de2fe6504934636703c0d1e38124b7d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
// Copyright (c) 2009 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#import "keystone_glue.h"
// Provide declarations of the Keystone registration bits needed here. From
// KSRegistration.h.
typedef enum { kKSPathExistenceChecker } KSExistenceCheckerType;
@interface KSRegistration : NSObject
+ (id)registrationWithProductID:(NSString*)productID;
- (BOOL)registerWithVersion:(NSString*)version
existenceCheckerType:(KSExistenceCheckerType)xctype
existenceCheckerString:(NSString*)xc
serverURLString:(NSString*)serverURLString;
- (void)setActive;
@end
@implementation KeystoneGlue
// TODO(mmentovai): Determine if the app is writable, and don't register for
// updates if not - but keep the periodic activity pings.
+ (void)registerWithKeystone {
// Figure out who we are.
NSBundle* mainBundle = [NSBundle mainBundle];
NSDictionary* infoDictionary = [mainBundle infoDictionary];
NSString* url = [infoDictionary objectForKey:@"KSUpdateURL"];
NSString* bundleIdentifier = [infoDictionary objectForKey:@"KSProductID"];
if (bundleIdentifier == nil) {
bundleIdentifier = [mainBundle bundleIdentifier];
}
// TODO(mmentovai): The svn version serves our purposes for now, but it will
// likely be replaced. The key problem is that it does not monotonically
// increase for releases when considering svn branches and tags. For the
// purposes of TestShell, though, which will likely only ever survive in
// auto-updatable form in builds straight from the trunk, this is fine.
NSString* version = [infoDictionary objectForKey:@"SVNRevision"];
if (!bundleIdentifier || !url || !version) {
// If parameters required for Keystone are missing, don't use it.
return;
}
// Load the KeystoneRegistration framework bundle.
NSString* ksrPath =
[[mainBundle privateFrameworksPath]
stringByAppendingPathComponent:@"KeystoneRegistration.framework"];
NSBundle* ksrBundle = [NSBundle bundleWithPath:ksrPath];
[ksrBundle load];
// Harness the KSRegistration class.
Class ksrClass = [ksrBundle classNamed:@"KSRegistration"];
KSRegistration* ksr = [ksrClass registrationWithProductID:bundleIdentifier];
if (!ksr) {
// Strictly speaking, this isn't necessary, because it's harmless to send
// messages to nil. However, if there really isn't a
// KeystoneRegistration.framework or KSRegistration class, bailing out here
// avoids setting up the timer that will only be able to perform no-ops.
return;
}
// Keystone will asynchronously handle installation and registration as
// needed.
[ksr registerWithVersion:version
existenceCheckerType:kKSPathExistenceChecker
existenceCheckerString:[mainBundle bundlePath]
serverURLString:url];
// Set up hourly activity pings.
[NSTimer scheduledTimerWithTimeInterval:60 * 60 // One hour
target:self
selector:@selector(markActive:)
userInfo:ksr
repeats:YES];
}
+ (void)markActive:(NSTimer*)timer {
KSRegistration* ksr = [timer userInfo];
[ksr setActive];
}
@end
|
