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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
//
// AccountExpirySystemNotificationProvider.swift
// MullvadVPN
//
// Created by pronebird on 03/06/2021.
// Copyright © 2021 Mullvad VPN AB. All rights reserved.
//
import Foundation
import MullvadSettings
import UserNotifications
final class AccountExpirySystemNotificationProvider: NotificationProvider, SystemNotificationProvider {
private var accountExpiry: Date?
private var tunnelObserver: TunnelBlockObserver?
init(tunnelManager: TunnelManager) {
super.init()
let tunnelObserver = TunnelBlockObserver(
didLoadConfiguration: { [weak self] tunnelManager in
self?.invalidate(deviceState: tunnelManager.deviceState)
},
didUpdateDeviceState: { [weak self] _, deviceState, _ in
self?.invalidate(deviceState: deviceState)
}
)
tunnelManager.addObserver(tunnelObserver)
self.tunnelObserver = tunnelObserver
}
override var identifier: NotificationProviderIdentifier {
.accountExpirySystemNotification
}
// MARK: - SystemNotificationProvider
var notificationRequest: UNNotificationRequest? {
guard let trigger else { return nil }
_ = NSLocalizedString(
"ACCOUNT_EXPIRY_SYSTEM_NOTIFICATION_TITLE",
comment: "Title for system account expiry notification, fired 3 days prior to account expiry."
)
_ = NSLocalizedString(
"ACCOUNT_EXPIRY_SYSTEM_NOTIFICATION_BODY",
comment: "Message for system account expiry notification, fired 3 days prior to account expiry."
)
let content = UNMutableNotificationContent()
content.title = NSString.localizedUserNotificationString(
forKey: "ACCOUNT_EXPIRY_SYSTEM_NOTIFICATION_TITLE",
arguments: nil
)
content.body = NSString.localizedUserNotificationString(
forKey: "ACCOUNT_EXPIRY_SYSTEM_NOTIFICATION_BODY",
arguments: nil
)
content.sound = UNNotificationSound.default
return UNNotificationRequest(
identifier: identifier.domainIdentifier,
content: content,
trigger: trigger
)
}
var shouldRemovePendingRequests: Bool {
// Remove pending notifications when account expiry is not set (user logged out)
shouldRemovePendingOrDeliveredRequests
}
var shouldRemoveDeliveredRequests: Bool {
// Remove delivered notifications when account expiry is not set (user logged out)
shouldRemovePendingOrDeliveredRequests
}
// MARK: - Private
private var trigger: UNNotificationTrigger? {
guard let accountExpiry else { return nil }
guard let triggerDate = Calendar.current.date(
byAdding: .day,
value: -NotificationConfiguration.closeToExpiryTriggerInterval,
to: accountExpiry
) else { return nil }
// Do not produce notification if less than 3 days left till expiry
guard triggerDate > Date() else { return nil }
// Create date components for calendar trigger
let dateComponents = Calendar.current.dateComponents(
[.second, .minute, .hour, .day, .month, .year],
from: triggerDate
)
return UNCalendarNotificationTrigger(dateMatching: dateComponents, repeats: false)
}
private var shouldRemovePendingOrDeliveredRequests: Bool {
accountExpiry == nil
}
private func invalidate(deviceState: DeviceState) {
accountExpiry = deviceState.accountData?.expiry
invalidate()
}
}
|
