acetheking987/bsky-social-app
1
0
Fork 0
Code Issues Actions Packages Projects Releases 1 Wiki Activity

tweak

Browse Source
This commit is contained in:
Hailey
2024-07-12 16:47:22 -07:00
parent 62aa42f073
commit 8e44de0168
1 changed files with 19 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/BlueskyNSE/NotificationService.swift
+19 -23
View File
@@ -13,6 +13,7 @@ enum NotificationType: String {
case ChatMessage = "chat-message" case ChatMessage = "chat-message"
case MarkReadGeneric = "mark-read-generic" case MarkReadGeneric = "mark-read-generic"
case MarkReadMessages = "mark-read-messages" case MarkReadMessages = "mark-read-messages"
case StarterPackJoined = "starterpack-joined"
} }
enum BadgeType: String { enum BadgeType: String {
@@ -25,9 +26,11 @@ enum BadgeOperation {
case Decrement case Decrement
} }
class NotificationService: UNNotificationServiceExtension { class Defaults {
var prefs = UserDefaults(suiteName: APP_GROUP) static let shared = UserDefaults(suiteName: APP_GROUP)
}
class NotificationService: UNNotificationServiceExtension {
override func didReceive(_ request: UNNotificationRequest, withContentHandler contentHandler: @escaping (UNNotificationContent) -> Void) { override func didReceive(_ request: UNNotificationRequest, withContentHandler contentHandler: @escaping (UNNotificationContent) -> Void) {
guard let bestAttempt = createCopy(request.content), guard let bestAttempt = createCopy(request.content),
let reasonString = request.content.userInfo["reason"] as? String, let reasonString = request.content.userInfo["reason"] as? String,
@@ -38,7 +41,7 @@ class NotificationService: UNNotificationServiceExtension {
} }
switch reason { switch reason {
case .Like, .Repost, .Follow, .Reply, .Quote: case .Like, .Repost, .Follow, .Reply, .Quote, .StarterPackJoined:
mutateWithBadge(bestAttempt, badgeType: .Generic, operation: .Increment) mutateWithBadge(bestAttempt, badgeType: .Generic, operation: .Increment)
case .ChatMessage: case .ChatMessage:
@@ -72,38 +75,31 @@ class NotificationService: UNNotificationServiceExtension {
} }
func mutateWithBadge(_ content: UNMutableNotificationContent, badgeType type: BadgeType, operation: BadgeOperation) { func mutateWithBadge(_ content: UNMutableNotificationContent, badgeType type: BadgeType, operation: BadgeOperation) {
var genericCount = prefs?.integer(forKey: BadgeType.Generic.rawValue) ?? 0 var genericCount = Defaults.shared?.integer(forKey: BadgeType.Generic.rawValue) ?? 0
var messagesCount = prefs?.integer(forKey: BadgeType.Messages.rawValue) ?? 0 var messagesCount = Defaults.shared?.integer(forKey: BadgeType.Messages.rawValue) ?? 0
if type == .Generic { if type == .Generic {
var genericCount = Defaults.shared?.integer(forKey: BadgeType.Generic.rawValue) ?? 0
if operation == .Decrement { if operation == .Decrement {
if let decrementBy = content.userInfo["decrementBy"] as? Int { genericCount = 0
genericCount = getDecrementedBadgeCount(current: genericCount, decrementBy: decrementBy)
} else {
genericCount = 0
}
} else { } else {
genericCount += 1 genericCount += 1
} }
prefs?.setValue(genericCount, forKey: BadgeType.Generic.rawValue) Defaults.shared?.setValue(genericCount, forKey: BadgeType.Generic.rawValue)
} else if type == .Messages { } else if type == .Messages {
if operation == .Decrement { if operation == .Decrement, shouldDecrementForConvo(content) {
if let convoId = content.userInfo["convoId"] as? String, shouldDecrementForConvo(content) { messagesCount = getDecrementedBadgeCount(current: messagesCount, decrementBy: 1)
messagesCount = getDecrementedBadgeCount(current: messagesCount, decrementBy: 1)
} else if let decrementBy = content.userInfo["decrementBy"] as? Int {
messagesCount = getDecrementedBadgeCount(current: messagesCount, decrementBy: decrementBy)
}
} else if shouldIncrementForConvo(content) { } else if shouldIncrementForConvo(content) {
messagesCount += 1 messagesCount += 1
} }
prefs?.setValue(messagesCount, forKey: BadgeType.Generic.rawValue) Defaults.shared?.setValue(messagesCount, forKey: BadgeType.Generic.rawValue)
} }
content.badge = NSNumber(value: genericCount + messagesCount) content.badge = NSNumber(value: genericCount + messagesCount)
} }
func mutateWithChatMessage(_ content: UNMutableNotificationContent) { func mutateWithChatMessage(_ content: UNMutableNotificationContent) {
if self.prefs?.bool(forKey: "playSoundChat") == true { if Defaults.shared?.bool(forKey: "playSoundChat") == true {
mutateWithDmSound(content) mutateWithDmSound(content)
} }
} }
@@ -118,7 +114,7 @@ class NotificationService: UNNotificationServiceExtension {
func shouldIncrementForConvo(_ content: UNMutableNotificationContent) -> Bool { func shouldIncrementForConvo(_ content: UNMutableNotificationContent) -> Bool {
guard let convoId = content.userInfo["convoId"] as? String, guard let convoId = content.userInfo["convoId"] as? String,
var dict = self.prefs?.dictionary(forKey: INCREMENTED_FOR_KEY) as? [String: Bool] var dict = Defaults.shared?.dictionary(forKey: INCREMENTED_FOR_KEY) as? [String: Bool]
else { else {
return false return false
} }
@@ -128,13 +124,13 @@ class NotificationService: UNNotificationServiceExtension {
} }
dict[convoId] = true dict[convoId] = true
self.prefs?.set(dict, forKey: INCREMENTED_FOR_KEY) Defaults.shared?.set(dict, forKey: INCREMENTED_FOR_KEY)
return true return true
} }
func shouldDecrementForConvo(_ content: UNMutableNotificationContent) -> Bool { func shouldDecrementForConvo(_ content: UNMutableNotificationContent) -> Bool {
guard let convoId = content.userInfo["convoId"] as? String, guard let convoId = content.userInfo["convoId"] as? String,
var dict = self.prefs?.dictionary(forKey: INCREMENTED_FOR_KEY) as? [String: Bool] var dict = Defaults.shared?.dictionary(forKey: INCREMENTED_FOR_KEY) as? [String: Bool]
else { else {
return false return false
} }
@@ -144,7 +140,7 @@ class NotificationService: UNNotificationServiceExtension {
} }
dict.removeValue(forKey: convoId) dict.removeValue(forKey: convoId)
self.prefs?.set(dict, forKey: INCREMENTED_FOR_KEY) Defaults.shared?.set(dict, forKey: INCREMENTED_FOR_KEY)
return true return true
} }
} }