ladill-mini/apps/ios/LadillMini/Features/Home/HomeViewModel.swift

import Foundation

@MainActor
final class HomeViewModel: ObservableObject {
    @Published var overviewState: LoadState<Overview> = .idle
    @Published var qrsState: LoadState<[PaymentQr]> = .idle
    @Published var paymentsState: LoadState<[Payment]> = .idle
    @Published var payoutsState: LoadState<Payouts> = .idle
    @Published var hasPayout: Bool?
    @Published var unreadCount = 0
    @Published var overviewRefreshing = false

    func loadUnreadCount() async {
        if let count = try? await APIClient.shared.unreadNotificationCount() {
            unreadCount = count
        }
    }

    func loadOverview() async {
        overviewState = .loading
        do {
            let data = try await APIClient.shared.overview()
            overviewState = .success(data)
        } catch {
            overviewState = .failure((error as? LocalizedError)?.errorDescription ?? error.localizedDescription)
        }
    }

    func refreshOverview() async {
        guard !overviewRefreshing else { return }
        overviewRefreshing = true
        defer { overviewRefreshing = false }
        await loadOverview()
        await loadPayoutStatus()
    }

    func loadQrs() async {
        qrsState = .loading
        do {
            qrsState = .success(try await APIClient.shared.paymentQrs())
        } catch {
            qrsState = .failure((error as? LocalizedError)?.errorDescription ?? error.localizedDescription)
        }
    }

    func loadPayments() async {
        paymentsState = .loading
        do {
            paymentsState = .success(try await APIClient.shared.payments())
        } catch {
            paymentsState = .failure((error as? LocalizedError)?.errorDescription ?? error.localizedDescription)
        }
    }

    func loadPayouts() async {
        payoutsState = .loading
        do {
            payoutsState = .success(try await APIClient.shared.payouts())
        } catch {
            payoutsState = .failure((error as? LocalizedError)?.errorDescription ?? error.localizedDescription)
        }
    }

    func loadPayoutStatus() async {
        do {
            hasPayout = try await APIClient.shared.payoutAccount() != nil
        } catch {
            // Leave hasPayout unchanged when the check fails.
        }
    }

    func logout(onComplete: @escaping () -> Void) async {
        await PushTokenRegistrar.unregister()
        await APIClient.shared.logout()
        SessionStore.shared.clear()
        onComplete()
    }
}