iOS/Sources/Thread/CredentialsSharing/ToHomeAssistant/ThreadTransferCredentialToHAViewModel.swift

import Foundation
import HAKit
import Shared

final class ThreadTransferCredentialToHAViewModel: ThreadCredentialsSharingViewModelProtocol {
    @Published var credentials: [ThreadCredential] = []
    @Published var showAlert = false
    @Published var alertType: ThreadCredentialsAlertType?
    @Published var showOperationSuccess = false

    private let threadClient: ThreadClientProtocol
    private var credentialsToImport: [String] = []
    private let server: Server

    init(server: Server, threadClient: ThreadClientProtocol) {
        self.threadClient = threadClient
        self.server = server
    }

    @MainActor
    func mainOperation() async {
        do {
            credentials = try await threadClient.retrieveAllCredentials()

            if credentials.isEmpty {
                showAlert(type: .empty(
                    title: L10n.Thread.Credentials.ShareCredentials.noCredentialsTitle,
                    message: L10n.Thread.Credentials.ShareCredentials.noCredentialsMessage
                ))
            } else {
                credentialsToImport = credentials.map(\.activeOperationalDataSet)
                processImport()
            }
        } catch {
            Current.Log.error("Failed to retrieve all thread credentials, error: \(error.localizedDescription)")
            showAlert(type: .error(title: L10n.errorLabel, message: error.localizedDescription))
        }
    }

    @MainActor
    private func processImport() {
        guard let first = credentialsToImport.first else {
            showOperationSuccess = true
            return
        }

        shareCredentialWithHomeAssistant(credential: first) { [weak self] success in
            if success {
                self?.credentialsToImport.removeFirst()
                self?.processImport()
            }
        }
    }

    @MainActor
    private func shareCredentialWithHomeAssistant(credential: String, completion: @escaping (Bool) -> Void) {
        let request = HARequest(type: .webSocket("thread/add_dataset_tlv"), data: [
            "tlv": credential,
            "source": "iOS-app",
        ])
        Current.api(for: server)?.connection.send(request).promise.pipe { [weak self] result in
            guard let self else { return }
            switch result {
            case .fulfilled:
                completion(true)
            case let .rejected(error):
                showAlert(type: .error(
                    title: L10n.errorLabel,
                    message: error.localizedDescription
                ))
                completion(false)
            }
        }
    }

    private func showAlert(type: ThreadCredentialsAlertType) {
        alertType = type
        showAlert = true
    }
}