Calendarr/Calendarr iOS/Views/LoginView.swift

import Foundation
import SwiftUI

struct LoginView: View {
    @Environment(AppState.self) var appState
    @AppStorage("appLanguage") private var appLang = "system"
    @State private var username = ""
    @State private var password = ""
    @State private var totpCode = ""
    @State private var rememberMe = true
    @State private var needsTOTP = false
    @State private var error = ""
    @State private var isLoading = false
    @State private var ssoProviders: [OIDCProvider] = []
    /// Key of the provider whose flow is running — per-provider, so a second
    /// button does not also spin.
    @State private var ssoBusyKey: String?
    @State private var oidc = OIDCAuthenticator()

    var body: some View {
        NavigationStack {
            ScrollView {
                VStack(spacing: 0) {
                    Spacer().frame(height: 60)

                    VStack(spacing: 8) {
                        Image(systemName: "calendar")
                            .font(.system(size: 48, weight: .light))
                            .foregroundStyle(Color.accentColor)
                        Text("Calendarr")
                            .font(.largeTitle.bold())
                        Text(appState.serverURL
                            .replacingOccurrences(of: "https://", with: "")
                            .replacingOccurrences(of: "http://", with: ""))
                            .font(.caption)
                            .foregroundStyle(.secondary)
                    }
                    .padding(.bottom, 40)

                    VStack(spacing: 16) {
                        VStack(alignment: .leading, spacing: 6) {
                            Text(L10n.t("login.username", appLang))
                                .font(.footnote.weight(.medium))
                                .foregroundStyle(.secondary)
                            TextField(L10n.t("login.username", appLang), text: $username)
                                .textInputAutocapitalization(.never)
                                .autocorrectionDisabled()
                                .padding(12)
                                .background(.quaternary)
                                .clipShape(RoundedRectangle(cornerRadius: 10))
                        }

                        VStack(alignment: .leading, spacing: 6) {
                            Text(L10n.t("login.password", appLang))
                                .font(.footnote.weight(.medium))
                                .foregroundStyle(.secondary)
                            SecureField(L10n.t("login.password", appLang), text: $password)
                                .padding(12)
                                .background(.quaternary)
                                .clipShape(RoundedRectangle(cornerRadius: 10))
                        }

                        if needsTOTP {
                            VStack(alignment: .leading, spacing: 6) {
                                Text(L10n.t("login.totp", appLang))
                                    .font(.footnote.weight(.medium))
                                    .foregroundStyle(.secondary)
                                TextField(L10n.t("login.totp_placeholder", appLang), text: $totpCode)
                                    .keyboardType(.numberPad)
                                    .padding(12)
                                    .background(.quaternary)
                                    .clipShape(RoundedRectangle(cornerRadius: 10))
                            }
                            .transition(.move(edge: .top).combined(with: .opacity))
                        }

                        Toggle(L10n.t("login.remember", appLang), isOn: $rememberMe)
                            .tint(Color.accentColor)

                        if !error.isEmpty {
                            Text(error)
                                .font(.caption)
                                .foregroundStyle(.red)
                                .frame(maxWidth: .infinity, alignment: .leading)
                        }

                        Button {
                            Task { await login() }
                        } label: {
                            HStack {
                                if isLoading {
                                    ProgressView().tint(.white)
                                } else {
                                    Text(L10n.t("login.signin", appLang)).fontWeight(.semibold)
                                }
                            }
                            .frame(maxWidth: .infinity)
                            .padding(14)
                            .background(Color.accentColor)
                            .foregroundStyle(.white)
                            .clipShape(RoundedRectangle(cornerRadius: 12))
                        }
                        .disabled(username.isEmpty || password.isEmpty || isLoading)

                        // Single sign-on — hidden entirely when the server has
                        // no provider configured.
                        if !ssoProviders.isEmpty {
                            HStack(spacing: 12) {
                                Rectangle().frame(height: 1).foregroundStyle(.quaternary)
                                Text(L10n.t("sso.or", appLang))
                                    .font(.caption)
                                    .foregroundStyle(.secondary)
                                Rectangle().frame(height: 1).foregroundStyle(.quaternary)
                            }
                            .padding(.top, 8)

                            ForEach(ssoProviders) { provider in
                                Button {
                                    Task { await signInWithSSO(provider) }
                                } label: {
                                    HStack {
                                        if ssoBusyKey == provider.key {
                                            ProgressView()
                                        } else {
                                            Text(String(format: L10n.t("sso.login_with", appLang),
                                                        provider.name))
                                                .fontWeight(.semibold)
                                        }
                                    }
                                    .frame(maxWidth: .infinity)
                                    .padding(14)
                                    .background(Color.accentColor.opacity(0.12))
                                    .foregroundStyle(Color.accentColor)
                                    .clipShape(RoundedRectangle(cornerRadius: 12))
                                }
                                .disabled(isLoading || ssoBusyKey != nil)
                            }
                        }
                    }
                    .padding(.horizontal, 32)
                    .animation(.easeInOut, value: needsTOTP)
                    .animation(.easeInOut, value: ssoProviders)

                    Spacer().frame(height: 40)

                    Button(L10n.t("login.choose_server", appLang)) {
                        appState.resetServer()
                    }
                    .font(.footnote)
                    .foregroundStyle(.secondary)
                }
            }
            .navigationBarHidden(true)
            .task { await loadSSOProviders() }
        }
    }

    /// Best-effort: an older server or a network hiccup simply means no button.
    private func loadSSOProviders() async {
        guard ssoProviders.isEmpty else { return }
        ssoProviders = (try? await CalendarrAPI.oidcProviders(baseURL: appState.serverURL)) ?? []
    }

    private func signInWithSSO(_ provider: OIDCProvider) async {
        ssoBusyKey = provider.key
        error = ""
        defer { ssoBusyKey = nil }

        do {
            let tokens = try await oidc.authorize(provider: provider)
            let result = try await CalendarrAPI.exchangeOIDC(
                baseURL: appState.serverURL,
                provider: provider.key,
                clientID: provider.mobileClientID,
                idToken: tokens.idToken,
                accessToken: tokens.accessToken,
                nonce: tokens.nonce
            )
            // Keep the provider's refresh token so the session can be renewed
            // later without sending the user through the browser again.
            OIDCAuthenticator.store(state: tokens.serialisedState, providerKey: provider.key)
            appState.saveLogin(token: result.token, user: result.username, admin: result.isAdmin)
        } catch let e as OIDCError {
            // Cancelling the browser sheet is a normal action, not an error.
            if e.slug == "oidc_cancelled" { return }
            let key = "sso.err." + e.slug
            let msg = L10n.t(key, appLang)
            error = msg == key ? L10n.t("sso.err.generic", appLang) : msg
        } catch {
            self.error = error.localizedDescription
        }
    }

    private func login() async {
        isLoading = true
        error = ""
        defer { isLoading = false }

        do {
            let code = needsTOTP ? (totpCode.isEmpty ? nil : totpCode) : nil
            let result = try await CalendarrAPI.login(
                baseURL: appState.serverURL,
                username: username,
                password: password,
                totpCode: code,
                rememberMe: rememberMe
            )
            appState.saveLogin(token: result.token, user: result.username, admin: result.isAdmin)
        } catch APIError.twoFactorRequired {
            withAnimation { needsTOTP = true }
        } catch {
            self.error = error.localizedDescription
        }
    }
}