wehub-resource-sync
94 lines
3.5 KiB
Swift
94 lines
3.5 KiB
Swift
import AppKit
|
|
import CodexBarCore
|
|
import Foundation
|
|
import SwiftUI
|
|
|
|
struct 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),
|
|
]
|
|
}
|
|
}
|