blob: 2568c153c9c010b27e5b5daebaae994ee3e94f5c (
plain) (
blame)
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
|
// SPDX-License-Identifier: MIT
// Copyright © 2018-2019 WireGuard LLC. All Rights Reserved.
import Cocoa
class StatusItemController {
var currentTunnel: TunnelContainer? {
didSet {
updateStatusItemImage()
}
}
let statusItem = NSStatusBar.system.statusItem(withLength: NSStatusItem.squareLength)
private let statusBarImageWhenActive = NSImage(named: "StatusBarIcon")!
private let statusBarImageWhenInactive = NSImage(named: "StatusBarIconDimmed")!
private let animationImages = [
NSImage(named: "StatusBarIconDot1")!,
NSImage(named: "StatusBarIconDot2")!,
NSImage(named: "StatusBarIconDot3")!
]
private var animationImageIndex: Int = 0
private var animationTimer: Timer?
init() {
updateStatusItemImage()
}
func updateStatusItemImage() {
guard let currentTunnel = currentTunnel else {
stopActivatingAnimation()
statusItem.button?.image = statusBarImageWhenInactive
return
}
switch currentTunnel.status {
case .inactive:
stopActivatingAnimation()
statusItem.button?.image = statusBarImageWhenInactive
case .active:
stopActivatingAnimation()
statusItem.button?.image = statusBarImageWhenActive
case .activating, .waiting, .reasserting, .restarting:
startActivatingAnimation()
case .deactivating:
break
}
}
func startActivatingAnimation() {
guard animationTimer == nil else { return }
let timer = Timer(timeInterval: 0.3, repeats: true) { [weak self] _ in
guard let self = self else { return }
self.statusItem.button?.image = self.animationImages[self.animationImageIndex]
self.animationImageIndex = (self.animationImageIndex + 1) % self.animationImages.count
}
RunLoop.main.add(timer, forMode: .default)
animationTimer = timer
}
func stopActivatingAnimation() {
guard let timer = self.animationTimer else { return }
timer.invalidate()
animationTimer = nil
animationImageIndex = 0
}
}
|
