99 lines · 3.7 KB · Swift
RawDownload
import CodexBarCoreimport Foundationimport SwiftUI struct FactoryProviderImplementation: ProviderImplementation {    let id: UsageProvider = .factory    let supportsLoginFlow: Bool = true     @MainActor    func observeSettings(_ settings: SettingsStore) {        _ = settings.factoryCookieSource        _ = settings.factoryCookieHeader    }     @MainActor    func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {        .factory(context.settings.factorySettingsSnapshot(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.factoryCookieSource == .manual    }     @MainActor    func applyTokenAccountCookieSource(settings: SettingsStore) {        if settings.factoryCookieSource != .manual {            settings.factoryCookieSource = .manual        }    }     @MainActor    func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {        let cookieBinding = Binding(            get: { context.settings.factoryCookieSource.rawValue },            set: { raw in                context.settings.factoryCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto            })        let cookieOptions = ProviderCookieSourceUI.options(            allowsOff: false,            keychainDisabled: context.settings.debugDisableKeychainAccess)         let cookieSubtitle: () -> String? = {            ProviderCookieSourceUI.subtitle(                source: context.settings.factoryCookieSource,                keychainDisabled: context.settings.debugDisableKeychainAccess,                auto: "Automatic imports browser cookies and WorkOS tokens.",                manual: "Paste a Cookie or Authorization header from app.factory.ai.",                off: "Factory cookies are disabled.")        }         return [            ProviderSettingsPickerDescriptor(                id: "factory-cookie-source",                title: "Cookie source",                subtitle: "Automatic imports browser cookies and WorkOS tokens.",                dynamicSubtitle: cookieSubtitle,                binding: cookieBinding,                options: cookieOptions,                isVisible: nil,                onChange: nil,                trailingText: {                    ProviderCookieSourceUI.cachedTrailingText(provider: .factory)                }),        ]    }     @MainActor    func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {        _ = context        return []    }     @MainActor    func runLoginFlow(context: ProviderLoginContext) async -> Bool {        await context.controller.runFactoryLoginFlow()        return true    }     @MainActor    func loginMenuAction(context _: ProviderMenuLoginContext)        -> (label: String, action: MenuDescriptor.MenuAction)?    {        ("Open Droid in Browser...", .loginToProvider(url: "https://app.factory.ai"))    }     @MainActor    func appendUsageMenuEntries(context: ProviderMenuUsageContext, entries: inout [ProviderMenuEntry]) {        guard context.settings.showOptionalCreditsAndExtraUsage,              let cost = context.snapshot?.providerCost,              cost.period == "Extra usage balance"        else { return }         let balance = UsageFormatter.currencyString(cost.used, currencyCode: cost.currencyCode)        entries.append(.text(L("Extra usage balance: %@", balance), .primary))    }}