wehub-resource-sync
99 lines
3.9 KiB
Swift
99 lines
3.9 KiB
Swift
import Foundation
|
|
|
|
public enum CommandCodeProviderDescriptor {
|
|
public static let descriptor: ProviderDescriptor = Self.makeDescriptor()
|
|
|
|
static func makeDescriptor() -> ProviderDescriptor {
|
|
ProviderDescriptor(
|
|
id: .commandcode,
|
|
metadata: ProviderMetadata(
|
|
id: .commandcode,
|
|
displayName: "Command Code",
|
|
sessionLabel: "Monthly credits",
|
|
weeklyLabel: "Monthly",
|
|
opusLabel: nil,
|
|
supportsOpus: false,
|
|
supportsCredits: true,
|
|
creditsHint: "Monthly USD credits from Command Code billing.",
|
|
toggleTitle: "Show Command Code usage",
|
|
cliName: "commandcode",
|
|
defaultEnabled: false,
|
|
isPrimaryProvider: false,
|
|
usesAccountFallback: false,
|
|
browserCookieOrder: ProviderBrowserCookieDefaults.defaultImportOrder,
|
|
dashboardURL: "https://commandcode.ai/studio",
|
|
subscriptionDashboardURL: "https://commandcode.ai/sixhobbits/settings/billing",
|
|
statusPageURL: nil,
|
|
statusLinkURL: nil),
|
|
branding: ProviderBranding(
|
|
iconStyle: .commandcode,
|
|
iconResourceName: "ProviderIcon-commandcode",
|
|
color: ProviderColor(red: 0 / 255, green: 0 / 255, blue: 0 / 255)),
|
|
tokenCost: ProviderTokenCostConfig(
|
|
supportsTokenCost: false,
|
|
noDataMessage: { "Command Code cost summary is not yet supported." }),
|
|
fetchPlan: ProviderFetchPlan(
|
|
sourceModes: [.auto, .web],
|
|
pipeline: ProviderFetchPipeline(resolveStrategies: { _ in [CommandCodeWebFetchStrategy()] })),
|
|
cli: ProviderCLIConfig(
|
|
name: "commandcode",
|
|
aliases: ["command-code"],
|
|
versionDetector: nil))
|
|
}
|
|
}
|
|
|
|
struct CommandCodeWebFetchStrategy: ProviderFetchStrategy {
|
|
let id: String = "commandcode.web"
|
|
let kind: ProviderFetchKind = .web
|
|
|
|
func isAvailable(_ context: ProviderFetchContext) async -> Bool {
|
|
guard context.settings?.commandcode?.cookieSource != .off else { return false }
|
|
if Self.manualCookieHeader(from: context) != nil {
|
|
return true
|
|
}
|
|
#if os(macOS)
|
|
return true
|
|
#else
|
|
return false
|
|
#endif
|
|
}
|
|
|
|
func fetch(_ context: ProviderFetchContext) async throws -> ProviderFetchResult {
|
|
let cookieHeader: String
|
|
let sourceLabel: String
|
|
if let manual = Self.manualCookieHeader(from: context) {
|
|
cookieHeader = manual
|
|
sourceLabel = "manual"
|
|
} else {
|
|
#if os(macOS)
|
|
let session: CommandCodeCookieImporter.SessionInfo
|
|
do {
|
|
session = try CommandCodeCookieImporter.importSession()
|
|
} catch {
|
|
throw CommandCodeUsageError.missingCredentials
|
|
}
|
|
guard !session.cookies.isEmpty else {
|
|
throw CommandCodeUsageError.missingCredentials
|
|
}
|
|
cookieHeader = session.cookieHeader
|
|
sourceLabel = session.sourceLabel
|
|
#else
|
|
throw CommandCodeUsageError.missingCredentials
|
|
#endif
|
|
}
|
|
let snapshot = try await CommandCodeUsageFetcher.fetchUsage(cookieHeader: cookieHeader)
|
|
return self.makeResult(
|
|
usage: snapshot.toUsageSnapshot(),
|
|
sourceLabel: sourceLabel)
|
|
}
|
|
|
|
private static func manualCookieHeader(from context: ProviderFetchContext) -> String? {
|
|
guard context.settings?.commandcode?.cookieSource == .manual else { return nil }
|
|
return CookieHeaderNormalizer.normalize(context.settings?.commandcode?.manualCookieHeader)
|
|
}
|
|
|
|
func shouldFallback(on _: Error, context _: ProviderFetchContext) -> Bool {
|
|
false
|
|
}
|
|
}
|