302 lines
11 KiB
Swift
302 lines
11 KiB
Swift
//
|
|
// NotificationService.swift
|
|
// Mastodon
|
|
//
|
|
// Created by MainasuK Cirno on 2021-4-22.
|
|
//
|
|
|
|
import UIKit
|
|
import Combine
|
|
import CoreData
|
|
import CoreDataStack
|
|
import MastodonSDK
|
|
import MastodonCommon
|
|
import MastodonLocalization
|
|
|
|
public final class NotificationService {
|
|
|
|
public static let unreadShortcutItemIdentifier = "org.joinmastodon.app.NotificationService.unread-shortcut"
|
|
|
|
var disposeBag = Set<AnyCancellable>()
|
|
|
|
let workingQueue = DispatchQueue(label: "org.joinmastodon.app.NotificationService.working-queue")
|
|
|
|
// input
|
|
weak var apiService: APIService?
|
|
weak var authenticationService: AuthenticationService?
|
|
public let isNotificationPermissionGranted = CurrentValueSubject<Bool, Never>(false)
|
|
public let deviceToken = CurrentValueSubject<Data?, Never>(nil)
|
|
public let applicationIconBadgeNeedsUpdate = CurrentValueSubject<Void, Never>(Void())
|
|
|
|
// output
|
|
/// [Token: NotificationViewModel]
|
|
public let notificationSubscriptionDict: [String: NotificationViewModel] = [:]
|
|
public let unreadNotificationCountDidUpdate = CurrentValueSubject<Void, Never>(Void())
|
|
public let requestRevealNotificationPublisher = PassthroughSubject<MastodonPushNotification, Never>()
|
|
|
|
init(
|
|
apiService: APIService,
|
|
authenticationService: AuthenticationService
|
|
) {
|
|
self.apiService = apiService
|
|
self.authenticationService = authenticationService
|
|
|
|
AuthenticationServiceProvider.shared.$authentications
|
|
.sink(receiveValue: { [weak self] mastodonAuthentications in
|
|
guard let self = self else { return }
|
|
|
|
// request permission when sign-in
|
|
guard !mastodonAuthentications.isEmpty else { return }
|
|
self.requestNotificationPermission()
|
|
})
|
|
.store(in: &disposeBag)
|
|
|
|
Publishers.CombineLatest(
|
|
authenticationService.$mastodonAuthenticationBoxes,
|
|
applicationIconBadgeNeedsUpdate
|
|
)
|
|
.receive(on: DispatchQueue.main)
|
|
.sink { [weak self] mastodonAuthenticationBoxes, _ in
|
|
guard let self = self else { return }
|
|
|
|
var count = 0
|
|
for authenticationBox in mastodonAuthenticationBoxes {
|
|
count += UserDefaults.shared.getNotificationCountWithAccessToken(accessToken: authenticationBox.userAuthorization.accessToken)
|
|
}
|
|
|
|
UserDefaults.shared.notificationBadgeCount = count
|
|
UIApplication.shared.applicationIconBadgeNumber = count
|
|
Task { @MainActor in
|
|
UIApplication.shared.shortcutItems = try? await self.unreadApplicationShortcutItems()
|
|
}
|
|
|
|
self.unreadNotificationCountDidUpdate.send()
|
|
}
|
|
.store(in: &disposeBag)
|
|
}
|
|
|
|
}
|
|
|
|
extension NotificationService {
|
|
private func requestNotificationPermission() {
|
|
let center = UNUserNotificationCenter.current()
|
|
center.requestAuthorization(options: [.alert, .sound, .badge]) { [weak self] granted, error in
|
|
guard let self = self else { return }
|
|
|
|
self.isNotificationPermissionGranted.value = granted
|
|
|
|
if let _ = error {
|
|
// Handle the error here.
|
|
}
|
|
|
|
// Enable or disable features based on the authorization.
|
|
}
|
|
}
|
|
}
|
|
|
|
extension NotificationService {
|
|
public func unreadApplicationShortcutItems() async throws -> [UIApplicationShortcutItem] {
|
|
guard let authenticationService = self.authenticationService else { return [] }
|
|
|
|
var items: [UIApplicationShortcutItem] = []
|
|
for authentication in AuthenticationServiceProvider.shared.authentications {
|
|
guard let account = authentication.account() else { continue }
|
|
let accessToken = authentication.userAccessToken
|
|
let count = UserDefaults.shared.getNotificationCountWithAccessToken(accessToken: accessToken)
|
|
guard count > 0 else { continue }
|
|
|
|
let title = "@\(account.acctWithDomain)"
|
|
let subtitle = L10n.A11y.Plural.Count.Unread.notification(count)
|
|
|
|
let item = UIApplicationShortcutItem(
|
|
type: NotificationService.unreadShortcutItemIdentifier,
|
|
localizedTitle: title,
|
|
localizedSubtitle: subtitle,
|
|
icon: nil,
|
|
userInfo: [
|
|
"accessToken": accessToken as NSSecureCoding
|
|
]
|
|
)
|
|
items.append(item)
|
|
}
|
|
return items
|
|
}}
|
|
|
|
extension NotificationService {
|
|
|
|
public func dequeueNotificationViewModel(
|
|
mastodonAuthenticationBox: MastodonAuthenticationBox
|
|
) -> NotificationViewModel? {
|
|
var _notificationSubscription: NotificationViewModel?
|
|
workingQueue.sync {
|
|
let domain = mastodonAuthenticationBox.domain
|
|
let userID = mastodonAuthenticationBox.userID
|
|
let key = [domain, userID].joined(separator: "@")
|
|
|
|
if let notificationSubscription = notificationSubscriptionDict[key] {
|
|
_notificationSubscription = notificationSubscription
|
|
} else {
|
|
let notificationSubscription = NotificationViewModel(domain: domain, userID: userID)
|
|
_notificationSubscription = notificationSubscription
|
|
}
|
|
}
|
|
return _notificationSubscription
|
|
}
|
|
|
|
public func handle(
|
|
pushNotification: MastodonPushNotification
|
|
) {
|
|
defer {
|
|
unreadNotificationCountDidUpdate.send()
|
|
}
|
|
|
|
Task {
|
|
// trigger notification timeline update
|
|
try? await fetchLatestNotifications(pushNotification: pushNotification)
|
|
|
|
// cancel sign-out account push notification subscription
|
|
try? await cancelSubscriptionForDetachedAccount(pushNotification: pushNotification)
|
|
} // end Task
|
|
}
|
|
|
|
}
|
|
|
|
extension NotificationService {
|
|
public func clearNotificationCountForActiveUser() {
|
|
guard let authenticationService = self.authenticationService else { return }
|
|
if let accessToken = authenticationService.mastodonAuthenticationBoxes.first?.userAuthorization.accessToken {
|
|
UserDefaults.shared.setNotificationCountWithAccessToken(accessToken: accessToken, value: 0)
|
|
}
|
|
|
|
applicationIconBadgeNeedsUpdate.send()
|
|
}
|
|
}
|
|
|
|
extension NotificationService {
|
|
private func fetchLatestNotifications(
|
|
pushNotification: MastodonPushNotification
|
|
) async throws {
|
|
guard let apiService = apiService else { return }
|
|
guard let authenticationBox = try await authenticationBox(for: pushNotification) else { return }
|
|
|
|
_ = try await apiService.notifications(
|
|
maxID: nil,
|
|
scope: .everything,
|
|
authenticationBox: authenticationBox
|
|
)
|
|
}
|
|
|
|
private func cancelSubscriptionForDetachedAccount(
|
|
pushNotification: MastodonPushNotification
|
|
) async throws {
|
|
// Subscription maybe failed to cancel when sign-out
|
|
// Try cancel again if receive that kind push notification
|
|
guard let managedObjectContext = authenticationService?.managedObjectContext else { return }
|
|
guard let apiService = apiService else { return }
|
|
|
|
let userAccessToken = pushNotification.accessToken
|
|
|
|
let needsCancelSubscription: Bool = try await managedObjectContext.perform {
|
|
// check authentication exists
|
|
let results = AuthenticationServiceProvider.shared.authentications.filter { $0.userAccessToken == userAccessToken }
|
|
return results.first == nil
|
|
}
|
|
|
|
guard needsCancelSubscription else {
|
|
return
|
|
}
|
|
|
|
guard let domain = try await domain(for: pushNotification) else { return }
|
|
|
|
do {
|
|
_ = try await apiService.cancelSubscription(
|
|
domain: domain,
|
|
authorization: .init(accessToken: userAccessToken)
|
|
)
|
|
} catch {
|
|
}
|
|
}
|
|
|
|
private func domain(for pushNotification: MastodonPushNotification) async throws -> String? {
|
|
guard let authenticationService = self.authenticationService else { return nil }
|
|
let managedObjectContext = authenticationService.managedObjectContext
|
|
return try await managedObjectContext.perform {
|
|
let subscriptionRequest = NotificationSubscription.sortedFetchRequest
|
|
subscriptionRequest.predicate = NotificationSubscription.predicate(userToken: pushNotification.accessToken)
|
|
let subscriptions = managedObjectContext.safeFetch(subscriptionRequest)
|
|
|
|
// note: assert setting not remove after sign-out
|
|
guard let subscription = subscriptions.first else { return nil }
|
|
guard let setting = subscription.setting else { return nil }
|
|
let domain = setting.domain
|
|
|
|
return domain
|
|
}
|
|
}
|
|
|
|
private func authenticationBox(for pushNotification: MastodonPushNotification) async throws -> MastodonAuthenticationBox? {
|
|
guard self.authenticationService != nil else { return nil }
|
|
let results = AuthenticationServiceProvider.shared.authentications.filter { $0.userAccessToken == pushNotification.accessToken }
|
|
guard let authentication = results.first else { return nil }
|
|
|
|
return MastodonAuthenticationBox(
|
|
authentication: authentication,
|
|
domain: authentication.domain,
|
|
userID: authentication.userID,
|
|
appAuthorization: .init(accessToken: authentication.appAccessToken),
|
|
userAuthorization: .init(accessToken: authentication.userAccessToken),
|
|
inMemoryCache: .sharedCache(for: authentication.userAccessToken)
|
|
)
|
|
}
|
|
|
|
}
|
|
|
|
// MARK: - NotificationViewModel
|
|
|
|
extension NotificationService {
|
|
public final class NotificationViewModel {
|
|
|
|
var disposeBag = Set<AnyCancellable>()
|
|
|
|
// input
|
|
let domain: String
|
|
let userID: Mastodon.Entity.Account.ID
|
|
|
|
// output
|
|
|
|
init(domain: String, userID: Mastodon.Entity.Account.ID) {
|
|
self.domain = domain
|
|
self.userID = userID
|
|
}
|
|
}
|
|
}
|
|
|
|
extension NotificationService.NotificationViewModel {
|
|
public func createSubscribeQuery(
|
|
deviceToken: Data,
|
|
queryData: Mastodon.API.Subscriptions.QueryData,
|
|
mastodonAuthenticationBox: MastodonAuthenticationBox
|
|
) -> Mastodon.API.Subscriptions.CreateSubscriptionQuery {
|
|
let deviceToken = [UInt8](deviceToken).toHexString()
|
|
|
|
let appSecret = AppSecret.default
|
|
let endpoint = appSecret.notificationEndpoint + "/" + deviceToken
|
|
let p256dh = appSecret.notificationPublicKey.x963Representation
|
|
let auth = appSecret.notificationAuth
|
|
|
|
let query = Mastodon.API.Subscriptions.CreateSubscriptionQuery(
|
|
subscription: Mastodon.API.Subscriptions.QuerySubscription(
|
|
endpoint: endpoint,
|
|
keys: Mastodon.API.Subscriptions.QuerySubscription.Keys(
|
|
p256dh: p256dh,
|
|
auth: auth
|
|
)
|
|
),
|
|
data: queryData
|
|
)
|
|
|
|
return query
|
|
}
|
|
|
|
}
|