TF-2384 Handle refresh token when token expired in NSE

Signed-off-by: dab246 <tdvu@linagora.com> (cherry picked from commit c8a39bb42792770c001ac90d6301d53126225d95)
2023-12-24 21:45:42 +07:00
parent 5bf1b97de3
commit b1b2bb90a5
19 changed files with 416 additions and 37 deletions
@@ -0,0 +1,69 @@
+import Foundation
+import Alamofire
+
+class TokenRefreshManager {
+    private let MOBIE_CLIENT_ID = "teammail-mobile"
+    private let MOBIE_REDIRECT_URL = "teammail.mobile://oauthredirect"
+    private let OIDC_SCOPES = ["openid", "profile", "email", "offline_access"]
+
+    private let GRANT_TYPE = "grant_type"
+    private let REFRESH_TOKEN = "refresh_token"
+    private let CLIENT_ID = "client_id"
+    private let REDIRECT_URI = "redirect_uri"
+    private let SCOPES = "scopes"
+
+    let refreshToken: String
+    let tokenEndpoint: String
+    let scopes: [String]?
+
+    init(refreshToken: String, tokenEndpoint: String, scopes: [String]?) {
+        self.refreshToken = refreshToken
+        self.tokenEndpoint = tokenEndpoint
+        self.scopes = scopes
+    }
+
+    private func getScopes() -> String {
+        if let scopes, !scopes.isEmpty {
+            return scopes.joined(separator: " ")
+        }
+        return OIDC_SCOPES.joined(separator: " ")
+    }
+
+    func handleRefreshAccessToken(
+        onSuccess: @escaping (TokenResponse) -> Void,
+        onFailure: @escaping (Error) -> Void
+    ) {
+        guard let tokenEndpointUrl = URL(string: tokenEndpoint),
+              var request = try? URLRequest(url: tokenEndpointUrl, method: .post) else {
+            return onFailure(NetworkExceptions.requestUrlInvalid)
+        }
+
+        let params = [
+            CLIENT_ID: MOBIE_CLIENT_ID,
+            GRANT_TYPE: REFRESH_TOKEN,
+            REDIRECT_URI: MOBIE_REDIRECT_URL,
+            REFRESH_TOKEN: refreshToken,
+            SCOPES: getScopes(),
+        ]
+
+        let data = params
+            .map { "\($0)=\($1)" }
+            .joined(separator: "&")
+            .data(using: .utf8)
+
+        request.httpBody = data
+
+        AF.request(request).responseDecodable(of: TokenResponse.self) { response in
+            if (response.response?.statusCode != 200) {
+                onFailure(NetworkExceptions(value: "Failed to get token: \(response.error?.localizedDescription ?? "Unknown Error")"))
+            } else {
+                switch(response.result) {
+                case .success(let data):
+                    onSuccess(data)
+                case .failure(let error):
+                    onFailure(NetworkExceptions(value: "Failed to get token \(error.localizedDescription)"))
+                }
+            }
+        }
+    }
+}