Impressia/Vernissage/Services/AuthorizationService.swift

//
//  https://mczachurski.dev
//  Copyright © 2023 Marcin Czachurski and the repository contributors.
//  Licensed under the MIT License.
//
    
import Foundation
import MastodonSwift

public class AuthorizationService {
    public static let shared = AuthorizationService()
    
    public func verifyAccount(_ result: @escaping (AccountData?) -> Void) async {
        let accountDataHandler = AccountDataHandler()
        let currentAccount = accountDataHandler.getCurrentAccountData()
        
        // When we dont have even one account stored in database then we have to ask user to enter server and sign in.
        guard let accountData = currentAccount, let accessToken = accountData.accessToken else {
            result(nil)
            return
        }
        
        // When we have at least one account then we have to verify access token.
        let client = MastodonClient(baseURL: accountData.serverUrl).getAuthenticated(token: accessToken)

        do {
            let account = try await client.verifyCredentials()
            try await self.updateAccount(accountData: accountData, account: account)
            result(accountData)
        } catch {
            do {
                try await self.refreshCredentials(accountData: accountData)
                result(accountData)
            } catch {
                // TODO: show information to the user.
                print("Cannot refresh credentials!!!")
            }
        }
    }
    
    public func signIn(serverAddress: String, _ result: @escaping (AccountData?) -> Void) async throws {
        let baseUrl = URL(string: serverAddress)!
        let client = MastodonClient(baseURL: baseUrl)
        
        // Verify address.
        let instanceInformation = try await client.readInstanceInformation()
        print(instanceInformation)

        // Create application (we will get clientId amd clientSecret).
        let oAuthApp = try await client.createApp(
            named: "Photofed",
            redirectUri: "oauth-vernissage://oauth-callback/mastodon",
            scopes: Scopes(["read", "write", "follow", "push"]),
            website: baseUrl)
        
        // Authorize a user (browser, we will get clientCode).
        let oAuthSwiftCredential = try await client.authenticate(
            app: oAuthApp,
            scope: Scopes(["read", "write", "follow", "push"]))
        
        // Get authenticated client.
        let authenticatedClient = client.getAuthenticated(token: oAuthSwiftCredential.oauthToken)
        
        // Get account information from server.
        let account = try await authenticatedClient.verifyCredentials()
        
        // Create account object in database.
        let accountDataHandler = AccountDataHandler()
        let accountData = accountDataHandler.createAccountDataEntity()

        accountData.id = account.id
        accountData.username = account.username
        accountData.acct = account.acct
        accountData.displayName = account.displayName
        accountData.note = account.note
        accountData.url = account.url
        accountData.avatar = account.avatar
        accountData.header = account.header
        accountData.locked = account.locked
        accountData.createdAt = account.createdAt
        accountData.followersCount = Int32(account.followersCount)
        accountData.followingCount = Int32(account.followingCount)
        accountData.statusesCount = Int32(account.statusesCount)
        
        accountData.serverUrl = baseUrl
        accountData.clientId = oAuthApp.clientId
        accountData.clientSecret = oAuthApp.clientSecret
        accountData.clientVapidKey = oAuthApp.vapidKey ?? ""
        accountData.accessToken = oAuthSwiftCredential.oauthToken
        
        // Download avatar image.
        if let avatarUrl = account.avatar {
            do {
                let avatarData = try await RemoteFileService.shared.fetchData(url: avatarUrl)
                accountData.avatarData = avatarData
            }
            catch {
                print("Avatar has not been downloaded")
            }
        }
        
        // Set newly created account as current.
        let applicationSettingsHandler = ApplicationSettingsHandler()
        let defaultSettings = applicationSettingsHandler.getDefaultSettings()
        defaultSettings.currentAccount = accountData.id
        
        // Save account data in database and in application state.
        CoreDataHandler.shared.save()
        
        // Return account data.
        result(accountData)
    }
    
    private func refreshCredentials(accountData: AccountData) async throws {
        let client = MastodonClient(baseURL: accountData.serverUrl)

        // Create application (we will get clientId amd clientSecret).
        let oAuthApp = App(clientId: accountData.clientId, clientSecret: accountData.clientSecret)
        
        // Authorize a user (browser, we will get clientCode).
        let oAuthSwiftCredential = try await client.authenticate(app: oAuthApp, scope: Scopes(["read", "write", "follow", "push"]))
        
        // Get authenticated client.
        let authenticatedClient = client.getAuthenticated(token: oAuthSwiftCredential.oauthToken)
        
        // Get account information from server.
        let account = try await authenticatedClient.verifyCredentials()
        try await self.updateAccount(accountData: accountData, account: account, accessToken: oAuthSwiftCredential.oauthToken)
    }
    
    private func updateAccount(accountData: AccountData, account: Account, accessToken: String? = nil) async throws {
        accountData.username = account.username
        accountData.acct = account.acct
        accountData.displayName = account.displayName
        accountData.note = account.note
        accountData.url = account.url
        accountData.avatar = account.avatar
        accountData.header = account.header
        accountData.locked = account.locked
        accountData.createdAt = account.createdAt
        accountData.followersCount = Int32(account.followersCount)
        accountData.followingCount = Int32(account.followingCount)
        accountData.statusesCount = Int32(account.statusesCount)
        
        if accessToken != nil {
            accountData.accessToken = accessToken
        }
        
        // Download avatar image.
        if let avatarUrl = account.avatar {
            do {
                let avatarData = try await RemoteFileService.shared.fetchData(url: avatarUrl)
                accountData.avatarData = avatarData
            }
            catch {
                print("Avatar has not been downloaded")
            }
        }
        
        // We have to be sure that account id is saved as default account.
        let applicationSettingsHandler = ApplicationSettingsHandler()
        let defaultSettings = applicationSettingsHandler.getDefaultSettings()
        defaultSettings.currentAccount = accountData.id
        
        // Save account data in database and in application state.
        CoreDataHandler.shared.save()
    }
}