Merge pull request #108 from Infomaniak/token-store

feat: TokenStore

Sources/InfomaniakCore/Account/TokenStore.swift

@@ -0,0 +1,71 @@
+/*
+ Infomaniak Core - iOS
+ Copyright (C) 2024 Infomaniak Network SA
+
+ This program is free software: you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+import Foundation
+import InfomaniakDI
+import InfomaniakLogin
+
+@available(macOS 10.15, iOS 13.0, tvOS 13.0, watchOS 6.0, *)
+public class TokenStore {
+    public typealias UserId = Int
+
+    @LazyInjectService var keychainHelper: KeychainHelper
+
+    public enum TokenStoreFetchLocation {
+        case cache
+        case keychain
+    }
+
+    let tokens = SendableDictionary<UserId, ApiToken>()
+
+    public init() {
+        let keychainTokens = keychainHelper.loadTokens()
+        for token in keychainTokens {
+            tokens[token.userId] = token
+        }
+    }
+
+    @discardableResult
+    public func removeTokenFor(userId: UserId) -> ApiToken? {
+        let removedToken = tokens.removeValue(forKey: userId)
+        keychainHelper.deleteToken(for: userId)
+
+        return removedToken
+    }
+
+    @discardableResult
+    public func removeTokenFor(account: Account) -> ApiToken? {
+        return removeTokenFor(userId: account.userId)
+    }
+
+    public func tokenFor(userId: UserId, fetchLocation: TokenStoreFetchLocation = .cache) -> ApiToken? {
+        if fetchLocation == .keychain {
+            let keychainTokens = keychainHelper.loadTokens()
+            for token in keychainTokens {
+                tokens[token.userId] = token
+            }
+        }
+
+        return tokens[userId]
+    }
+
+    public func addToken(newToken: ApiToken) {
+        keychainHelper.storeToken(newToken)
+        tokens[newToken.userId] = newToken
+    }
+}