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
82
83
84
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#import "chrome/browser/web_applications/os_integration/mac/app_shim_termination_observer.h"
#import <Cocoa/Cocoa.h>
#include "base/check.h"
#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "content/public/browser/browser_thread.h"
@implementation AppShimTerminationObserver {
NSRunningApplication* __strong _app;
base::OnceClosure _callback;
}
+ (NSMutableSet<AppShimTerminationObserver*>*)allObservers {
static NSMutableSet<AppShimTerminationObserver*>* set = [NSMutableSet set];
return set;
}
+ (void)startObservingForRunningApplication:(NSRunningApplication*)app
withCallback:(base::OnceClosure)callback {
AppShimTerminationObserver* observer = [[AppShimTerminationObserver alloc]
initWithRunningApplication:app
callback:std::move(callback)];
if (observer) {
[[AppShimTerminationObserver allObservers] addObject:observer];
}
}
- (instancetype)initWithRunningApplication:(NSRunningApplication*)app
callback:(base::OnceClosure)callback {
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
if (self = [super init]) {
_callback = std::move(callback);
_app = app;
// Note that |observeValueForKeyPath| will be called with the initial value
// within the |addObserver| call.
[_app addObserver:self
forKeyPath:@"isTerminated"
options:NSKeyValueObservingOptionNew |
NSKeyValueObservingOptionInitial
context:nullptr];
}
return self;
}
- (void)observeValueForKeyPath:(NSString*)keyPath
ofObject:(id)object
change:(NSDictionary*)change
context:(void*)context {
NSNumber* newNumberValue = change[NSKeyValueChangeNewKey];
BOOL newValue = newNumberValue.boolValue;
if (newValue) {
// Note that a block is posted, which will hold a retain on `self`.
content::GetUIThreadTaskRunner({})->PostTask(FROM_HERE, base::BindOnce(^{
[self onTerminated];
}));
}
}
- (void)onTerminated {
DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
// If |onTerminated| is called repeatedly (which in theory it should not),
// then ensure that we only call removeObserver and release once by doing an
// early-out if |callback_| has already been made.
if (!_callback) {
return;
}
std::move(_callback).Run();
DCHECK(!_callback);
[_app removeObserver:self forKeyPath:@"isTerminated" context:nullptr];
[[AppShimTerminationObserver allObservers]
performSelector:@selector(removeObject:)
withObject:self
afterDelay:0];
}
@end
|
