openclaw/apps/ios/Sources/Onboarding/GatewayOnboardingView.swift

import Foundation
import OpenClawKit
import SwiftUI

struct GatewayOnboardingView: View {
    var body: some View {
        NavigationStack {
            List {
                Section {
                    Text("Connect to your gateway to get started.")
                        .foregroundStyle(.secondary)
                }

                Section {
                    NavigationLink("Auto detect") {
                        AutoDetectStep()
                    }
                    NavigationLink("Manual entry") {
                        ManualEntryStep()
                    }
                }
            }
            .navigationTitle("Connect Gateway")
        }
        .gatewayTrustPromptAlert()
    }
}

private struct AutoDetectStep: View {
    @Environment(NodeAppModel.self) private var appModel: NodeAppModel
    @Environment(GatewayConnectionController.self) private var gatewayController: GatewayConnectionController
    @AppStorage("gateway.preferredStableID") private var preferredGatewayStableID: String = ""
    @AppStorage("gateway.lastDiscoveredStableID") private var lastDiscoveredGatewayStableID: String = ""

    @State private var connectingGatewayID: String?
    @State private var connectStatusText: String?

    var body: some View {
        Form {
            Section {
                Text("We’ll scan for gateways on your network and connect automatically when we find one.")
                    .foregroundStyle(.secondary)
            }

            gatewayConnectionStatusSection(
                appModel: self.appModel,
                gatewayController: self.gatewayController,
                secondaryLine: self.connectStatusText)

            Section {
                Button("Retry") {
                    resetGatewayConnectionState(
                        appModel: self.appModel,
                        connectStatusText: &self.connectStatusText,
                        connectingGatewayID: &self.connectingGatewayID)
                    self.triggerAutoConnect()
                }
                .disabled(self.connectingGatewayID != nil)
            }
        }
        .navigationTitle("Auto detect")
        .onAppear { self.triggerAutoConnect() }
        .onChange(of: self.gatewayController.gateways) { _, _ in
            self.triggerAutoConnect()
        }
    }

    private func triggerAutoConnect() {
        guard self.appModel.gatewayServerName == nil else { return }
        guard self.connectingGatewayID == nil else { return }
        guard let candidate = self.autoCandidate() else { return }

        self.connectingGatewayID = candidate.id
        Task {
            defer { self.connectingGatewayID = nil }
            await self.gatewayController.connect(candidate)
        }
    }

    private func autoCandidate() -> GatewayDiscoveryModel.DiscoveredGateway? {
        let preferred = self.preferredGatewayStableID.trimmingCharacters(in: .whitespacesAndNewlines)
        let lastDiscovered = self.lastDiscoveredGatewayStableID.trimmingCharacters(in: .whitespacesAndNewlines)

        if !preferred.isEmpty,
           let match = self.gatewayController.gateways.first(where: { $0.stableID == preferred })
        {
            return match
        }
        if !lastDiscovered.isEmpty,
           let match = self.gatewayController.gateways.first(where: { $0.stableID == lastDiscovered })
        {
            return match
        }
        if self.gatewayController.gateways.count == 1 {
            return self.gatewayController.gateways.first
        }
        return nil
    }
}

private struct ManualEntryStep: View {
    @Environment(NodeAppModel.self) private var appModel: NodeAppModel
    @Environment(GatewayConnectionController.self) private var gatewayController: GatewayConnectionController

    @State private var setupCode: String = ""
    @State private var setupStatusText: String?
    @State private var manualHost: String = ""
    @State private var manualPortText: String = ""
    @State private var manualUseTLS: Bool = true
    @State private var manualToken: String = ""
    @State private var manualPassword: String = ""
    @State private var pendingManualAuthOverride: GatewayConnectionController.ManualAuthOverride?

    @State private var connectingGatewayID: String?
    @State private var connectStatusText: String?

    var body: some View {
        Form {
            Section("Setup code") {
                Text("Use /pair in your bot to get a setup code.")
                    .font(.footnote)
                    .foregroundStyle(.secondary)

                TextField("Paste setup code", text: self.$setupCode)
                    .textInputAutocapitalization(.never)
                    .autocorrectionDisabled()

                Button("Apply setup code") {
                    self.applySetupCode()
                }
                .disabled(self.setupCode.trimmingCharacters(in: .whitespacesAndNewlines).isEmpty)

                if let setupStatusText, !setupStatusText.isEmpty {
                    Text(setupStatusText)
                        .font(.footnote)
                        .foregroundStyle(.secondary)
                }
            }

            Section {
                TextField("Host", text: self.$manualHost)
                    .textInputAutocapitalization(.never)
                    .autocorrectionDisabled()

                TextField("Port", text: self.$manualPortText)
                    .keyboardType(.numberPad)

                Toggle("Use TLS", isOn: self.$manualUseTLS)

                TextField("Gateway token", text: self.$manualToken)
                    .textInputAutocapitalization(.never)
                    .autocorrectionDisabled()

                SecureField("Gateway password", text: self.$manualPassword)
                    .textInputAutocapitalization(.never)
                    .autocorrectionDisabled()
            }

            gatewayConnectionStatusSection(
                appModel: self.appModel,
                gatewayController: self.gatewayController,
                secondaryLine: self.connectStatusText)

            Section {
                Button {
                    Task { await self.connectManual() }
                } label: {
                    if self.connectingGatewayID == "manual" {
                        HStack(spacing: 8) {
                            ProgressView()
                                .progressViewStyle(.circular)
                            Text("Connecting…")
                        }
                    } else {
                        Text("Connect")
                    }
                }
                .disabled(self.connectingGatewayID != nil)

                Button("Retry") {
                    resetGatewayConnectionState(
                        appModel: self.appModel,
                        connectStatusText: &self.connectStatusText,
                        connectingGatewayID: &self.connectingGatewayID)
                    self.resetManualForm()
                }
                .disabled(self.connectingGatewayID != nil)
            }
        }
        .navigationTitle("Manual entry")
    }

    private func connectManual() async {
        let host = self.manualHost.trimmingCharacters(in: .whitespacesAndNewlines)
        guard !host.isEmpty else {
            self.connectStatusText = "Failed: host required"
            return
        }

        if let port = self.manualPortValue(), !(1...65535).contains(port) {
            self.connectStatusText = "Failed: invalid port"
            return
        }

        let defaults = UserDefaults.standard
        defaults.set(true, forKey: "gateway.manual.enabled")
        defaults.set(host, forKey: "gateway.manual.host")
        defaults.set(self.manualPortValue() ?? 0, forKey: "gateway.manual.port")
        defaults.set(self.manualUseTLS, forKey: "gateway.manual.tls")

        let instanceId = GatewaySettingsStore.currentInstanceID()
        if !instanceId.isEmpty {
            let trimmedToken = self.manualToken.trimmingCharacters(in: .whitespacesAndNewlines)
            let trimmedPassword = self.manualPassword.trimmingCharacters(in: .whitespacesAndNewlines)
            if !trimmedToken.isEmpty {
                GatewaySettingsStore.saveGatewayToken(trimmedToken, instanceId: instanceId)
            }
            GatewaySettingsStore.saveGatewayPassword(trimmedPassword, instanceId: instanceId)
        }

        self.connectingGatewayID = "manual"
        defer { self.connectingGatewayID = nil }
        let authOverride = GatewayConnectionController.ManualAuthOverride.currentManualInput(
            token: self.manualToken,
            pendingOverride: self.pendingManualAuthOverride,
            password: self.manualPassword)
        self.pendingManualAuthOverride = nil
        await self.gatewayController.connectManual(
            host: host,
            port: self.manualPortValue() ?? 0,
            useTLS: self.manualUseTLS,
            authOverride: authOverride)
    }

    private func manualPortValue() -> Int? {
        let trimmed = self.manualPortText.trimmingCharacters(in: .whitespacesAndNewlines)
        guard !trimmed.isEmpty else { return nil }
        return Int(trimmed.filter(\.isNumber))
    }

    private func resetManualForm() {
        self.setupCode = ""
        self.setupStatusText = nil
        self.manualHost = ""
        self.manualPortText = ""
        self.manualUseTLS = true
        self.manualToken = ""
        self.manualPassword = ""
    }

    private func applySetupCode() {
        let raw = self.setupCode.trimmingCharacters(in: .whitespacesAndNewlines)
        guard !raw.isEmpty else {
            self.setupStatusText = "Paste a setup code to continue."
            return
        }

        guard let link = GatewayConnectDeepLink.fromSetupInput(raw) else {
            self.setupStatusText = "Setup code not recognized or uses an insecure ws:// gateway URL."
            return
        }

        self.manualHost = link.host
        self.manualPortText = String(link.port)
        self.manualUseTLS = link.tls

        let setupAuth = GatewayConnectionController.ManualAuthOverride.setupAuth(from: link)
        if setupAuth.shouldApplyTokenField {
            self.manualToken = setupAuth.token
        }
        if setupAuth.shouldApplyPasswordField {
            self.manualPassword = setupAuth.password
        }

        let trimmedInstanceId = GatewaySettingsStore.currentInstanceID()
        if !trimmedInstanceId.isEmpty {
            if setupAuth.hasBootstrapToken {
                GatewayOnboardingReset.prepareForBootstrapPairing(
                    appModel: self.appModel,
                    instanceId: trimmedInstanceId)
            }
            GatewaySettingsStore.saveGatewayBootstrapToken(setupAuth.bootstrapToken, instanceId: trimmedInstanceId)
        }
        self.pendingManualAuthOverride = setupAuth.manualAuthOverride

        self.setupStatusText = "Setup code applied."
    }
}

@MainActor
private func gatewayConnectionStatusLines(
    appModel: NodeAppModel,
    gatewayController: GatewayConnectionController) -> [String]
{
    ConnectionStatusBox.defaultLines(appModel: appModel, gatewayController: gatewayController)
}

@MainActor
private func resetGatewayConnectionState(
    appModel: NodeAppModel,
    connectStatusText: inout String?,
    connectingGatewayID: inout String?)
{
    appModel.disconnectGateway()
    connectStatusText = nil
    connectingGatewayID = nil
}

@MainActor
private func gatewayConnectionStatusSection(
    appModel: NodeAppModel,
    gatewayController: GatewayConnectionController,
    secondaryLine: String?) -> some View
{
    Section("Connection status") {
        ConnectionStatusBox(
            statusLines: gatewayConnectionStatusLines(
                appModel: appModel,
                gatewayController: gatewayController),
            secondaryLine: secondaryLine)
    }
}

private struct ConnectionStatusBox: View {
    let statusLines: [String]
    let secondaryLine: String?

    var body: some View {
        VStack(alignment: .leading, spacing: 6) {
            ForEach(self.statusLines, id: \.self) { line in
                Text(line)
                    .font(.system(size: 12, weight: .regular, design: .monospaced))
                    .foregroundStyle(.secondary)
            }
            if let secondaryLine, !secondaryLine.isEmpty {
                Text(secondaryLine)
                    .font(.footnote)
                    .foregroundStyle(.secondary)
            }
        }
        .frame(maxWidth: .infinity, alignment: .leading)
        .padding(10)
        .background(.thinMaterial, in: RoundedRectangle(cornerRadius: 10, style: .continuous))
    }

    static func defaultLines(
        appModel: NodeAppModel,
        gatewayController: GatewayConnectionController) -> [String]
    {
        var lines: [String] = [
            "gateway: \(appModel.gatewayDisplayStatusText)",
            "discovery: \(gatewayController.discoveryStatusText)",
        ]
        lines.append("server: \(appModel.gatewayServerName ?? "—")")
        lines.append("address: \(appModel.gatewayRemoteAddress ?? "—")")
        return lines
    }
}