import CodexBarMacroSupportimport Foundation @ProviderDescriptorRegistration@ProviderDescriptorDefinitionpublic enum OpenCodeGoProviderDescriptor { static func makeDescriptor() -> ProviderDescriptor { ProviderDescriptor( id: .opencodego, metadata: ProviderMetadata( id: .opencodego, displayName: "OpenCode Go", sessionLabel: "5-hour", weeklyLabel: "Weekly", opusLabel: "Monthly", supportsOpus: true, supportsCredits: false, creditsHint: "", toggleTitle: "Show OpenCode Go usage", cliName: "opencodego", defaultEnabled: false, isPrimaryProvider: false, usesAccountFallback: false, browserCookieOrder: ProviderBrowserCookieDefaults.defaultImportOrder, dashboardURL: "https://opencode.ai", statusPageURL: nil), branding: ProviderBranding( iconStyle: .opencodego, iconResourceName: "ProviderIcon-opencodego", color: ProviderColor(red: 59 / 255, green: 130 / 255, blue: 246 / 255)), tokenCost: ProviderTokenCostConfig( supportsTokenCost: false, noDataMessage: { "OpenCode Go cost summary is not supported." }), fetchPlan: ProviderFetchPlan( sourceModes: [.auto, .web], pipeline: ProviderFetchPipeline(resolveStrategies: { _ in [OpenCodeGoUsageFetchStrategy()] })), cli: ProviderCLIConfig( name: "opencodego", versionDetector: nil)) }} struct OpenCodeGoUsageFetchStrategy: ProviderFetchStrategy { let id: String = "opencodego.web" let kind: ProviderFetchKind = .web func isAvailable(_ context: ProviderFetchContext) async -> Bool { guard context.settings?.opencodego?.cookieSource != .off else { return false } return true } func fetch(_ context: ProviderFetchContext) async throws -> ProviderFetchResult { let workspaceOverride = context.settings?.opencodego?.workspaceID ?? context.env["CODEXBAR_OPENCODEGO_WORKSPACE_ID"] let cookieSource = context.settings?.opencodego?.cookieSource ?? .auto do { let cookieHeader = try Self.resolveCookieHeader(context: context, allowCached: true) let snapshot = try await OpenCodeGoUsageFetcher.fetchUsage( cookieHeader: cookieHeader, timeout: context.webTimeout, workspaceIDOverride: workspaceOverride) return self.makeResult( usage: snapshot.toUsageSnapshot(), sourceLabel: "web") } catch OpenCodeGoUsageError.invalidCredentials where cookieSource != .manual { #if os(macOS) CookieHeaderCache.clear(provider: .opencodego) let cookieHeader = try Self.resolveCookieHeader(context: context, allowCached: false) let snapshot = try await OpenCodeGoUsageFetcher.fetchUsage( cookieHeader: cookieHeader, timeout: context.webTimeout, workspaceIDOverride: workspaceOverride) return self.makeResult( usage: snapshot.toUsageSnapshot(), sourceLabel: "web") #else throw OpenCodeGoUsageError.invalidCredentials #endif } } func shouldFallback(on _: Error, context _: ProviderFetchContext) -> Bool { false } private static func resolveCookieHeader(context: ProviderFetchContext, allowCached: Bool) throws -> String { try OpenCodeWebCookieSupport.resolveCookieHeader( context: OpenCodeWebCookieSupport.Context( settings: context.settings?.opencodego, provider: .opencodego, browserDetection: context.browserDetection, allowCached: allowCached), invalidCookie: OpenCodeGoSettingsError.invalidCookie, missingCookie: OpenCodeGoSettingsError.missingCookie) }} enum OpenCodeGoSettingsError: LocalizedError { case missingCookie case invalidCookie var errorDescription: String? { switch self { case .missingCookie: "No OpenCode Go session cookies found in browsers." case .invalidCookie: "OpenCode Go cookie header is invalid." } }}
import CodexBarMacroSupport
import Foundation
@ProviderDescriptorRegistration
@ProviderDescriptorDefinition
public enum OpenCodeGoProviderDescriptor {
static func makeDescriptor() -> ProviderDescriptor {
ProviderDescriptor(
id: .opencodego,
metadata: ProviderMetadata(
id: .opencodego,
displayName: "OpenCode Go",
sessionLabel: "5-hour",
weeklyLabel: "Weekly",
opusLabel: "Monthly",
supportsOpus: true,
supportsCredits: false,
creditsHint: "",
toggleTitle: "Show OpenCode Go usage",
cliName: "opencodego",
defaultEnabled: false,
isPrimaryProvider: false,
usesAccountFallback: false,
browserCookieOrder: ProviderBrowserCookieDefaults.defaultImportOrder,
dashboardURL: "https://opencode.ai",
statusPageURL: nil),
branding: ProviderBranding(
iconStyle: .opencodego,
iconResourceName: "ProviderIcon-opencodego",
color: ProviderColor(red: 59 / 255, green: 130 / 255, blue: 246 / 255)),
tokenCost: ProviderTokenCostConfig(
supportsTokenCost: false,
noDataMessage: { "OpenCode Go cost summary is not supported." }),
fetchPlan: ProviderFetchPlan(
sourceModes: [.auto, .web],
pipeline: ProviderFetchPipeline(resolveStrategies: { _ in [OpenCodeGoUsageFetchStrategy()] })),
cli: ProviderCLIConfig(
name: "opencodego",
versionDetector: nil))
}
}
struct OpenCodeGoUsageFetchStrategy: ProviderFetchStrategy {
let id: String = "opencodego.web"
let kind: ProviderFetchKind = .web
func isAvailable(_ context: ProviderFetchContext) async -> Bool {
guard context.settings?.opencodego?.cookieSource != .off else { return false }
return true
}
func fetch(_ context: ProviderFetchContext) async throws -> ProviderFetchResult {
let workspaceOverride = context.settings?.opencodego?.workspaceID
?? context.env["CODEXBAR_OPENCODEGO_WORKSPACE_ID"]
let cookieSource = context.settings?.opencodego?.cookieSource ?? .auto
do {
let cookieHeader = try Self.resolveCookieHeader(context: context, allowCached: true)
let snapshot = try await OpenCodeGoUsageFetcher.fetchUsage(
cookieHeader: cookieHeader,
timeout: context.webTimeout,
workspaceIDOverride: workspaceOverride)
return self.makeResult(
usage: snapshot.toUsageSnapshot(),
sourceLabel: "web")
} catch OpenCodeGoUsageError.invalidCredentials where cookieSource != .manual {
#if os(macOS)
CookieHeaderCache.clear(provider: .opencodego)
let cookieHeader = try Self.resolveCookieHeader(context: context, allowCached: false)
let snapshot = try await OpenCodeGoUsageFetcher.fetchUsage(
cookieHeader: cookieHeader,
timeout: context.webTimeout,
workspaceIDOverride: workspaceOverride)
return self.makeResult(
usage: snapshot.toUsageSnapshot(),
sourceLabel: "web")
#else
throw OpenCodeGoUsageError.invalidCredentials
#endif
}
}
func shouldFallback(on _: Error, context _: ProviderFetchContext) -> Bool {
false
}
private static func resolveCookieHeader(context: ProviderFetchContext, allowCached: Bool) throws -> String {
try OpenCodeWebCookieSupport.resolveCookieHeader(
context: OpenCodeWebCookieSupport.Context(
settings: context.settings?.opencodego,
provider: .opencodego,
browserDetection: context.browserDetection,
allowCached: allowCached),
invalidCookie: OpenCodeGoSettingsError.invalidCookie,
missingCookie: OpenCodeGoSettingsError.missingCookie)
}
}
enum OpenCodeGoSettingsError: LocalizedError {
case missingCookie
case invalidCookie
var errorDescription: String? {
switch self {
case .missingCookie:
"No OpenCode Go session cookies found in browsers."
case .invalidCookie:
"OpenCode Go cookie header is invalid."
}
}
}