steipete--codexbar/Sources/CodexBar/Providers/OpenCodeGo/OpenCodeGoProviderImplementation.swift

import AppKit
import CodexBarCore
import Foundation
import SwiftUI

struct OpenCodeGoProviderImplementation: ProviderImplementation {
    let id: UsageProvider = .opencodego

    @MainActor
    func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {
        ProviderPresentation { _ in "web" }
    }

    @MainActor
    func observeSettings(_ settings: SettingsStore) {
        _ = settings.opencodegoCookieSource
        _ = settings.opencodegoCookieHeader
        _ = settings.opencodegoWorkspaceID
        _ = settings[providerConfig: .opencodego, field: .apiKey]
    }

    @MainActor
    func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {
        .opencodego(context.settings.opencodegoSettingsSnapshot(tokenOverride: context.tokenOverride))
    }

    @MainActor
    func tokenAccountsVisibility(context: ProviderSettingsContext, support: TokenAccountSupport) -> Bool {
        guard support.requiresManualCookieSource else { return true }
        if !context.settings.tokenAccounts(for: context.provider).isEmpty {
            return true
        }
        return context.settings.opencodegoCookieSource == .manual
    }

    @MainActor
    func applyTokenAccountCookieSource(settings: SettingsStore) {
        if settings.opencodegoCookieSource != .manual {
            settings.opencodegoCookieSource = .manual
        }
    }

    @MainActor
    func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {
        let cookieBinding = Binding(
            get: { context.settings.opencodegoCookieSource.rawValue },
            set: { raw in
                context.settings.opencodegoCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto
            })
        let cookieOptions = ProviderCookieSourceUI.options(
            allowsOff: false,
            keychainDisabled: context.settings.debugDisableKeychainAccess)

        let cookieSubtitle: () -> String? = {
            ProviderCookieSourceUI.subtitle(
                source: context.settings.opencodegoCookieSource,
                keychainDisabled: context.settings.debugDisableKeychainAccess,
                auto: "Automatic imports browser cookies from opencode.ai.",
                manual: "Paste a Cookie header captured from the billing page.",
                off: "OpenCode Go cookies are disabled.")
        }

        return [
            ProviderSettingsPickerDescriptor(
                id: "opencodego-cookie-source",
                title: "Cookie source",
                subtitle: "Automatic imports browser cookies from opencode.ai.",
                dynamicSubtitle: cookieSubtitle,
                binding: cookieBinding,
                options: cookieOptions,
                isVisible: nil,
                onChange: nil,
                trailingText: {
                    ProviderCookieRefreshAction.trailingText(
                        provider: .opencodego,
                        cookieSource: context.settings.opencodegoCookieSource,
                        context: context)
                },
                trailingActions: [
                    ProviderCookieRefreshAction.descriptor(
                        provider: .opencodego,
                        cookieSource: { context.settings.opencodegoCookieSource },
                        context: context),
                ]),
        ]
    }

    @MainActor
    func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {
        [
            ProviderSettingsFieldDescriptor(
                id: "opencodego-api-key",
                title: "API key",
                subtitle: "Preferred for Go usage limits. Also reads OPENCODE_API_KEY.",
                kind: .secure,
                placeholder: "OpenCode API key",
                binding: context.providerConfigBinding(.apiKey),
                actions: [],
                isVisible: nil,
                onActivate: nil),
            ProviderSettingsFieldDescriptor(
                id: "opencodego-workspace-id",
                title: "Workspace ID",
                subtitle: "Optional override if workspace lookup fails.",
                kind: .plain,
                placeholder: "wrk_…",
                binding: context.stringBinding(\.opencodegoWorkspaceID),
                actions: [],
                isVisible: nil,
                onActivate: nil),
        ]
    }
}