From e999118b090feab083609b787fce33fdbb764f63 Mon Sep 17 00:00:00 2001
From: Philippe Weidmann <philippe.weidmann@infomaniak.com>
Date: Tue, 5 Mar 2024 11:00:17 +0100
Subject: [PATCH] feat: TokenStore

---
 .../InfomaniakCore/Account/TokenStore.swift   | 71 +++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 Sources/InfomaniakCore/Account/TokenStore.swift

diff --git a/Sources/InfomaniakCore/Account/TokenStore.swift b/Sources/InfomaniakCore/Account/TokenStore.swift
new file mode 100644
index 0000000..0343920
--- /dev/null
+++ b/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
+    }
+}