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
|
//
// AccountExpiryInAppNotificationProvider.swift
// MullvadVPN
//
// Created by pronebird on 12/12/2022.
// Copyright © 2022 Mullvad VPN AB. All rights reserved.
//
import Foundation
import MullvadSettings
import MullvadTypes
final class AccountExpiryInAppNotificationProvider: NotificationProvider, InAppNotificationProvider {
private var accountExpiry = AccountExpiry()
private var tunnelObserver: TunnelBlockObserver?
private var timer: DispatchSourceTimer?
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)
}
)
self.tunnelObserver = tunnelObserver
tunnelManager.addObserver(tunnelObserver)
}
override var identifier: NotificationProviderIdentifier {
.accountExpiryInAppNotification
}
// MARK: - InAppNotificationProvider
var notificationDescriptor: InAppNotificationDescriptor? {
guard let duration = accountExpiry.formattedDuration else {
return nil
}
return InAppNotificationDescriptor(
identifier: identifier,
style: .warning,
title: NSLocalizedString(
"ACCOUNT_EXPIRY_INAPP_NOTIFICATION_TITLE",
value: "ACCOUNT CREDIT EXPIRES SOON",
comment: "Title for in-app notification, displayed within the last 3 days until account expiry."
),
body: .init(string: String(
format: NSLocalizedString(
"ACCOUNT_EXPIRY_INAPP_NOTIFICATION_BODY",
value: "%@ left. Buy more credit.",
comment: "Message for in-app notification, displayed within the last 3 days until account expiry."
), duration
))
)
}
// MARK: - Private
private func invalidate(deviceState: DeviceState) {
updateExpiry(deviceState: deviceState)
updateTimer()
invalidate()
}
private func updateExpiry(deviceState: DeviceState) {
accountExpiry.expiryDate = deviceState.accountData?.expiry
}
private func updateTimer() {
timer?.cancel()
guard let triggerDate = accountExpiry.triggerDate else {
return
}
let now = Date()
let fireDate = max(now, triggerDate)
let timer = DispatchSource.makeTimerSource(queue: .main)
timer.setEventHandler { [weak self] in
self?.timerDidFire()
}
timer.schedule(
wallDeadline: .now() + fireDate.timeIntervalSince(now),
repeating: .seconds(NotificationConfiguration.closeToExpiryInAppNotificationRefreshInterval)
)
timer.activate()
self.timer = timer
}
private func timerDidFire() {
let shouldCancelTimer = accountExpiry.expiryDate.map { $0 <= Date() } ?? true
if shouldCancelTimer {
timer?.cancel()
}
invalidate()
}
}
|
