95 lines · 3.5 KB · Swift
RawDownload
import AppKitimport CodexBarCoreimport CodexBarMacroSupportimport Foundationimport SwiftUI @ProviderImplementationRegistrationstruct PerplexityProviderImplementation: ProviderImplementation {    let id: UsageProvider = .perplexity    let supportsLoginFlow: Bool = true     @MainActor    func presentation(context _: ProviderPresentationContext) -> ProviderPresentation {        ProviderPresentation { _ in "web" }    }     @MainActor    func runLoginFlow(context _: ProviderLoginContext) async -> Bool {        if let url = URL(string: "https://www.perplexity.ai/") {            NSWorkspace.shared.open(url)        }        return false    }     @MainActor    func observeSettings(_ settings: SettingsStore) {        _ = settings.perplexityCookieSource        _ = settings.perplexityManualCookieHeader    }     @MainActor    func settingsSnapshot(context: ProviderSettingsSnapshotContext) -> ProviderSettingsSnapshotContribution? {        .perplexity(context.settings.perplexitySettingsSnapshot(tokenOverride: context.tokenOverride))    }     @MainActor    func settingsPickers(context: ProviderSettingsContext) -> [ProviderSettingsPickerDescriptor] {        let cookieBinding = Binding(            get: { context.settings.perplexityCookieSource.rawValue },            set: { raw in                context.settings.perplexityCookieSource = ProviderCookieSource(rawValue: raw) ?? .auto            })        let options = ProviderCookieSourceUI.options(            allowsOff: true,            keychainDisabled: context.settings.debugDisableKeychainAccess)         let subtitle: () -> String? = {            ProviderCookieSourceUI.subtitle(                source: context.settings.perplexityCookieSource,                keychainDisabled: context.settings.debugDisableKeychainAccess,                auto: "Automatically imports browser session cookie.",                manual: "Paste a full cookie header or the __Secure-next-auth.session-token value.",                off: "Perplexity cookies are disabled.")        }         return [            ProviderSettingsPickerDescriptor(                id: "perplexity-cookie-source",                title: "Cookie source",                subtitle: "Automatically imports browser session cookie.",                dynamicSubtitle: subtitle,                binding: cookieBinding,                options: options,                isVisible: nil,                onChange: nil),        ]    }     @MainActor    func settingsFields(context: ProviderSettingsContext) -> [ProviderSettingsFieldDescriptor] {        [            ProviderSettingsFieldDescriptor(                id: "perplexity-cookie",                title: "",                subtitle: "",                kind: .secure,                placeholder: "Cookie: \u{2026}\n\nor paste the __Secure-next-auth.session-token value",                binding: context.stringBinding(\.perplexityManualCookieHeader),                actions: [                    ProviderSettingsActionDescriptor(                        id: "perplexity-open-usage",                        title: "Open Usage Page",                        style: .link,                        isVisible: nil,                        perform: {                            if let url = URL(string: "https://www.perplexity.ai/account/usage") {                                NSWorkspace.shared.open(url)                            }                        }),                ],                isVisible: { context.settings.perplexityCookieSource == .manual },                onActivate: nil),        ]    }}