Use keychain instead of DB in notification service

DB/Sources/DB/Identity/IdentityDatabase.swift

@@ -228,16 +228,6 @@ public extension IdentityDatabase {
 
         return Identity(info: info)
     }
-
-    // Only for use in notification extension
-    func identity(id: Identity.Id) throws -> Identity? {
-        guard let info = try databaseWriter.read(
-                IdentityInfo.request(IdentityRecord.filter(IdentityRecord.Columns.id == id))
-                    .fetchOne)
-        else { return nil }
-
-        return Identity(info: info)
-    }
 }
 
 private extension IdentityDatabase {

Notification Service Extension/NotificationService.swift

@@ -49,23 +49,16 @@ final class NotificationService: UNNotificationServiceExtension {
             bestAttemptContent.sound = .default
         }
 
-        var identity: Identity?
+        if appPreferences.notificationAccountName,
-
+           case let .success(handle) = parsingService.handle(identityId: identityId) {
-        if appPreferences.notificationAccountName {
-            identity = try? AllIdentitiesService(environment: Self.environment).identity(id: identityId)
-
-            if let handle = identity?.handle {
             bestAttemptContent.subtitle = handle
         }
-        }
 
         Self.attachment(imageURL: pushNotification.icon)
             .map { [$0] }
             .replaceError(with: [])
             .handleEvents(receiveOutput: { bestAttemptContent.attachments = $0 })
-            .zip(parsingService.title(pushNotification: pushNotification,
+            .zip(parsingService.title(pushNotification: pushNotification, identityId: identityId)
-                                      identityId: identityId,
-                                      accountId: identity?.account?.id)
                     .replaceError(with: pushNotification.title)
                     .handleEvents(receiveOutput: { bestAttemptContent.title = $0 }))
             .sink { _ in contentHandler(bestAttemptContent) }

Secrets/Sources/Secrets/Secrets.swift

@@ -35,6 +35,8 @@ public extension Secrets {
         case databaseKey
         case imageCacheKey
         case identityDatabaseName
+        case accountId
+        case username
     }
 }
 
@@ -165,6 +167,22 @@ public extension Secrets {
         try set(accessToken, forItem: .accessToken)
     }
 
+    func getAccountId() throws -> String {
+        try item(.accountId)
+    }
+
+    func setAccountId(_ accountId: String) throws {
+        try set(accountId, forItem: .accountId)
+    }
+
+    func getUsername() throws -> String {
+        try item(.username)
+    }
+
+    func setUsername(_ username: String) throws {
+        try set(username, forItem: .username)
+    }
+
     func generatePushKeyAndReturnPublicKey() throws -> Data {
         try keychain.generateKeyAndReturnPublicKey(
             applicationTag: scopedKey(item: .pushKey),

ServiceLayer/Sources/ServiceLayer/Services/AllIdentitiesService.swift

@@ -136,11 +136,6 @@ public extension AllIdentitiesService {
             .ignoreOutput()
             .eraseToAnyPublisher()
     }
-
-    // Only for use in notification extension
-    func identity(id: Identity.Id) throws -> Identity? {
-        try database.identity(id: id)
-    }
 }
 
 private extension AllIdentitiesService.IdentityCreation {

ServiceLayer/Sources/ServiceLayer/Services/IdentityService.swift

@@ -51,6 +51,10 @@ public extension IdentityService {
 
     func verifyCredentials() -> AnyPublisher<Never, Error> {
         mastodonAPIClient.request(AccountEndpoint.verifyCredentials)
+            .handleEvents(receiveOutput: {
+                try? secrets.setAccountId($0.id)
+                try? secrets.setUsername($0.username)
+            })
             .flatMap { identityDatabase.updateAccount($0, id: id) }
             .eraseToAnyPublisher()
     }

ServiceLayer/Sources/ServiceLayer/Services/PushNotificationParsingService.swift

@@ -51,9 +51,13 @@ public extension PushNotificationParsingService {
                 identityId)
     }
 
-    func title(pushNotification: PushNotification,
+    func handle(identityId: Identity.Id) -> Result<String, Error> {
-               identityId: Identity.Id,
+        let secrets = Secrets(identityId: identityId, keychain: environment.keychain)
-               accountId: Account.Id?) -> AnyPublisher<String, Error> {
+
+        return Result { try secrets.getUsername().appending("@").appending(secrets.getInstanceURL().host ?? "") }
+    }
+
+    func title(pushNotification: PushNotification, identityId: Identity.Id) -> AnyPublisher<String, Error> {
         switch pushNotification.notificationType {
         case .poll, .status:
             let secrets = Secrets(identityId: identityId, keychain: environment.keychain)
@@ -79,15 +83,15 @@ public extension PushNotificationParsingService {
                             NSLocalizedString("notification.status-%@", comment: ""),
                             $0.account.displayName)
                     case .poll:
-                        switch accountId ?? (try? AllIdentitiesService(environment: environment)
+                        guard let accountId = try? secrets.getAccountId() else {
-                                                .identity(id: identityId)?.account?.id) {
-                        case .some($0.account.id):
-                            return NSLocalizedString("notification.poll.own", comment: "")
-                        case .some:
-                            return NSLocalizedString("notification.poll", comment: "")
-                        default:
                             return NSLocalizedString("notification.poll.unknown", comment: "")
                         }
+
+                        if $0.account.id == accountId {
+                            return NSLocalizedString("notification.poll.own", comment: "")
+                        } else {
+                            return NSLocalizedString("notification.poll", comment: "")
+                        }
                     default:
                         return pushNotification.title
                     }