Add support for reading accounts from Element Classic.

2026-11-04 01:20:12 +00:00 · 2025-10-10 16:44:26 +02:00
parent ac56a0ba43
commit 95b3a8fbb1
18 changed files with 775 additions and 3 deletions
@@ -0,0 +1,62 @@
+//
+// Copyright 2025 New Vector Ltd.
+//
+// SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-Element-Commercial
+// Please see LICENSE files in the repository root for full details.
+//
+
+import Foundation
+import KeychainAccess
+import MatrixRustSDK
+
+protocol ClassicAppManagerProtocol {
+    func loadAccounts() throws -> [ClassicAppAccount]
+}
+
+enum ClassicAppManagerError: Error {
+    case invalidAppGroupIdentifier(String)
+    case missingAccountKeys
+    case missingCryptoStorePassphrase
+}
+
+final class ClassicAppManager: ClassicAppManagerProtocol {
+    private enum KeychainKeys: String {
+        case cryptoSDKStoreKey
+        case accountIV = "accountIv"
+        case accountAESKey = "accountAesKey"
+    }
+    
+    private let classicAppGroupIdentifier: String
+    private let keychain: Keychain
+    
+    init(classicAppGroupIdentifier: String, classicAppKeychainServiceIdentifier: String, classicAppKeychainAccessGroupIdentifier: String) {
+        self.classicAppGroupIdentifier = classicAppGroupIdentifier
+        keychain = Keychain(service: classicAppKeychainServiceIdentifier, accessGroup: classicAppKeychainAccessGroupIdentifier)
+    }
+    
+    func loadAccounts() throws -> [ClassicAppAccount] {
+        guard let url = FileManager.default.containerURL(forSecurityApplicationGroupIdentifier: classicAppGroupIdentifier) else {
+            throw ClassicAppManagerError.invalidAppGroupIdentifier(classicAppGroupIdentifier)
+        }
+        
+        guard let accountIV = try keychain.getData(KeychainKeys.accountIV.rawValue),
+              let accountAESKey = try keychain.getData(KeychainKeys.accountAESKey.rawValue) else {
+            throw ClassicAppManagerError.missingAccountKeys
+        }
+        
+        guard let cryptoStorePassphrase = try keychain.getData(KeychainKeys.cryptoSDKStoreKey.rawValue) else {
+            throw ClassicAppManagerError.missingCryptoStorePassphrase
+        }
+        
+        let accountManager = ClassicAppAccountManager(cacheFolder: url, iv: accountIV, aesKey: accountAESKey)
+        accountManager.loadAccounts()
+        let activeAccounts = accountManager.activeAccounts
+        
+        MXLog.info("Loaded \(accountManager.accounts.count) accounts")
+        MXLog.verbose("Loaded accounts: \(accountManager.accounts.compactMap(\.credentials.userID).formatted(.list(type: .and)))")
+        MXLog.info("Found \(activeAccounts.count) active accounts")
+        MXLog.verbose("Active accounts: \(activeAccounts.compactMap(\.userID).formatted(.list(type: .and)))")
+        
+        return activeAccounts
+    }
+}