Back to Codexbar

Provider authoring guide

docs/provider.md

0.48.014.2 KB
Original Source

Provider authoring guide

Goal: adding a provider should feel like:

  • add one folder
  • define one descriptor + strategies
  • add one implementation (UI hooks only)
  • done (tests + docs)

This doc describes the current provider architecture and the exact steps to add a new provider.

Terms

  • Provider: a source of usage/quota/status data (Codex, Claude, Gemini, Antigravity, Cursor, …).
  • Descriptor: the single source of truth for labels, URLs, defaults, and fetch strategies.
  • Fetch strategy: one concrete way to obtain usage (CLI, web cookies, OAuth API, local probe, etc.).
  • Host APIs: shared capabilities we provide to providers (Keychain, browser cookies, PTY, HTTP, WebView scrape, token-cost).
  • Identity fields: email/org/plan/loginMethod. Must stay siloed per provider.

Architecture overview (now)

  • Sources/CodexBarCore: provider descriptors + fetch strategies + probes + parsing + shared utilities.
  • Sources/CodexBar: UI/state + provider implementations (settings/login/menu hooks only).
  • Provider IDs are compile-time: UsageProvider enum (used for persistence + widgets).
  • Provider wiring is descriptor-driven:
    • ProviderDescriptor owns labels, URLs, default enablement, and fetch pipeline.
    • ProviderFetchStrategy objects implement concrete fetch paths.
    • CLI + app both call the same descriptor/fetch pipeline.

Common building blocks already exist:

  • PTY: TTYCommandRunner
  • subprocess: SubprocessRunner
  • cookie import: BrowserCookieImporter (Safari/Chrome/Firefox adapters)
  • OpenAI dashboard web scrape: OpenAIDashboardFetcher (WKWebView + JS)
  • cost usage: local log scanner (Codex + Claude)

Provider behavior is descriptor-driven. Two flat first-party manifests form the closed bootstrap boundary: ProviderManifest lists core descriptors and ProviderImplementationManifest lists app implementations. The registries retain thread-safe register(_:) methods for future dynamic providers.

Runtime settings follow the same boundary. A provider owns its ProviderSettingsSectionKey and section payload in its core folder, registers that key on its descriptor, and contributes the payload from its app implementation. The generic ProviderSettingsSnapshot container performs the sole type-erased cast behind its constrained subscript; provider-local accessors keep fetch strategies fully typed. Providers that share a payload type still declare a distinct key for each ProviderInstanceID, while providers with no runtime settings receive an empty section from the descriptor default.

Credential and config behavior follows the descriptor boundary too. Providers with credentials register a Sendable ProviderCredentialAdapter that owns config-to-environment projection, token resolution, token-account support, diagnose classification, validation, and missing-credential messaging; a missing adapter means the provider has no credential behavior. Typed settings-section registrations optionally expose cookie settings and a CLI credential contribution, so the app, CLI, and plugin cookie broker consume the same provider-owned settings shape.

Provider architecture gatekeeper threat model

ProviderArchitectureGatekeeperTests is a drift tripwire against honest architecture mistakes by future contributors and AI agents. Its scope is deliberately narrower than a Swift parser's: the lexical scanner detects dotted provider case literals, including qualified, labeled, and multiline statements, and lowercase raw provider-ID string literals in every single-statement position (including assignments, bare function arguments, dictionary keys and values, array elements, and returns). It scans shipped Swift under Sources/** and WidgetExtension/**, with suppressions applied to exact provider tokens rather than whole statements.

The following are out of scope by design:

  • Dotted provider cases whose role requires real expression parsing, including implicit closure returns and closure-body dataflow. A line-and-statement lexical scan cannot model those positions honestly.
  • String concatenation, reflection, and dynamic lookup. Their runtime values are not recoverable from literal-token matching.
  • Provider literals nested inside the arguments of a suppressed call (for example a routing call passed into a logging call). Attributing a literal to the inner rather than the outer call requires expression-tree parsing.
  • Multi-line block comments interleaved with an expression. Single-line /* ... */ comments are blanked before scanning; comments spanning statement lines are treated as ending the scanned code for that line.
  • Tests/**, where fixtures legitimately name providers, and non-Swift files, because this tripwire is scoped to shipped Swift architecture.

This is engineering scoping, not a claim of adversarial completeness: the gatekeeper is a lexical drift tripwire for honest mistakes. If in-the-wild drift is ever observed slipping past it, the concrete upgrade path is to replace the lexical policy scan with a SwiftSyntax-based implementation that can model expressions and dataflow.

Provider descriptor (source of truth)

Introduce a single descriptor per provider:

  • id (stable UsageProvider)
  • display/labels/URLs (menu title, dashboard URL, status URL)
  • UI branding (icon name, primary color, 2–3-color confetti palette)
  • capabilities (supportsCredits, supportsTokenCost, supportsStatusPolling, supportsLogin)
  • fetch plan (allowed --source modes + ordered strategy pipeline)
  • CLI metadata (cliName, aliases, version provider)
  • account behavior (e.g., usesAccountFallback for Codex auth.json)

UI and settings should become descriptor-driven:

  • no provider-specific branching for labels/links/toggle titles
  • minimal provider-specific UI (only when a provider truly needs bespoke UX)

Fetch strategies

A provider declares a pipeline of strategies, in priority order. Each strategy:

  • advertises a kind (cli, web cookies, oauth, api token, local probe, web dashboard)
  • declares availability (checks settings, cookies, env vars, installed CLI)
  • fetches UsageSnapshot (and optional credits/dashboard)
  • can be filtered by CLI --source or app settings

The pipeline resolves to the best available strategy, and falls back on failure when allowed. Each run returns a ProviderFetchOutcome with attempts + errors for debug UI and CLI --verbose.

Host APIs are explicit, small, testable

Expose a narrow set of protocols/structs that provider implementations can use:

  • KeychainAPI: read-only, allowlisted service/account pairs
  • BrowserCookieAPI: import cookies by domain list; returns cookie header + diagnostics
  • BrowserLocalStorageAPI: read origin-scoped key/value snapshots across browser profiles
  • PTYAPI: run CLI interactions with timeouts + “send on substring” + stop rules
  • HTTPAPI: URLSession wrapper with domain allowlist + standard headers + tracing
  • WebViewScrapeAPI: WKWebView lease + evaluateJavaScript + snapshot dumping
  • TokenCostAPI: Cost Usage local-log integration (Codex/Claude today; extend later)
  • StatusAPI: status polling helpers (Statuspage + Workspace incidents)
  • LoggerAPI: scoped logger + redaction helpers

Rule: providers do not talk to FileManager, Security, or “browser internals” directly unless they are the host API implementation.

Provider-specific code layout

  • Sources/CodexBarCore/Providers/<ProviderID>/
    • <ProviderID>Descriptor.swift (descriptor + strategy pipeline)
    • <ProviderID>Strategies.swift (strategy implementations)
    • <ProviderID>Probe.swift / <ProviderID>Fetcher.swift
    • <ProviderID>Models.swift
    • <ProviderID>Parser.swift (if text/HTML parsing)
  • Sources/CodexBar/Providers/<ProviderID>/
    • <ProviderID>ProviderImplementation.swift (settings/login UI hooks only)

Minimal provider example (copy-paste)

swift
import Foundation

public enum ExampleProviderDescriptor {
    public static let descriptor: ProviderDescriptor = Self.makeDescriptor()

    static func makeDescriptor() -> ProviderDescriptor {
        ProviderDescriptor(
            id: .example,
            metadata: ProviderMetadata(
                id: .example,
                displayName: "Example",
                sessionLabel: "Session",
                weeklyLabel: "Weekly",
                opusLabel: nil,
                supportsOpus: false,
                supportsCredits: false,
                creditsHint: "",
                toggleTitle: "Show Example usage",
                cliName: "example",
                defaultEnabled: false,
                isPrimaryProvider: false,
                usesAccountFallback: false,
                dashboardURL: nil,
                statusPageURL: nil),
            branding: ProviderBranding(
                iconStyle: .init(provider: .example),
                iconResourceName: "ProviderIcon-example",
                color: ProviderColor(red: 0.2, green: 0.6, blue: 0.8),
                confettiPalette: [
                    ProviderColor(hex: 0x3399CC),
                    ProviderColor(hex: 0x66C2FF),
                ]),
            tokenCost: ProviderTokenCostConfig(
                supportsTokenCost: false,
                noDataMessage: { "Example cost summary is not supported." }),
            fetchPlan: ProviderFetchPlan(
                sourceModes: [.auto, .cli],
                pipeline: ProviderFetchPipeline(resolveStrategies: { _ in [ExampleFetchStrategy()] })),
            cli: ProviderCLIConfig(
                name: "example",
                versionDetector: nil))
    }
}

struct ExampleFetchStrategy: ProviderFetchStrategy {
    let id: String = "example.cli"
    let kind: ProviderFetchKind = .cli

    func isAvailable(_: ProviderFetchContext) async -> Bool { true }

    func fetch(_: ProviderFetchContext) async throws -> ProviderFetchResult {
        let usage = UsageSnapshot(
            primary: .init(usedPercent: 0, windowMinutes: nil, resetsAt: nil, resetDescription: nil),
            secondary: nil,
            updatedAt: Date(),
            identity: nil)
        return self.makeResult(usage: usage, sourceLabel: "cli")
    }

    func shouldFallback(on _: Error, context _: ProviderFetchContext) -> Bool { false }
}

Guardrails (non-negotiable)

  • Identity silo: never display identity/plan fields from provider A inside provider B UI.
  • Privacy: default to on-device parsing; browser cookies are opt-in and never persisted by us beyond WebKit stores.
  • Reliability: providers must be timeout-bounded; no unbounded waits on network/PTY/UI.
  • Degradation: prefer cached data over flapping; show clear errors when stale.

Hosted relay eligibility

Hosted relays and upstream aggregators need enough public evidence for maintainers and users to evaluate the trust boundary:

  • An identifiable legal operator and jurisdiction.
  • Verifiable authorization to resell or provide the advertised upstream access; operator self-assertion alone is not sufficient.
  • A public operating track record that supports ongoing reliability, security, and maintenance review.

An integration can be restored when missing operator or authorization evidence becomes available.

Adding a new provider

Adding a first-party provider currently requires all of these registration points:

  1. Create Sources/CodexBarCore/Providers/<Name>/ with the descriptor, fetch strategies, and core settings or credential types.
  2. Create Sources/CodexBar/Providers/<Name>/ with the app implementation and any app settings contribution or UI.
  3. Add one stable case, in the intended bootstrap order, to UsageProvider in Sources/CodexBarCore/Providers/Providers.swift.
  4. Run Scripts/regenerate-provider-manifests.sh. Do not edit ProviderManifest.swift, ProviderImplementationManifest.swift, or ProviderInstanceIDAliases.generated.swift directly. The generator also refreshes docs/provider-ids.md, which is linked from docs/configuration.md.
  5. Add Sources/CodexBar/Resources/ProviderIcon-<id>.svg and reference it from the descriptor's branding.
  6. Unless the descriptor sets widgetSelectable: false, add the matching case and literal caseDisplayRepresentations entry to the WidgetKit ProviderChoice AppEnum. AppIntents extracts this table statically, so widget display representations cannot be derived at runtime. WidgetProviderChoiceTests keeps the literal table synchronized with selectable descriptor metadata and display names.
  7. Add focused tests for the provider's parser/snapshot mapping, strategy availability and fallback, credential or settings projection, and CLI aliases/source validation as applicable.
  8. Add or update the user-facing provider entry in docs/providers.md, including authentication and data-source guidance. Add a dedicated provider document when the integration needs more detail.

If the provider has runtime settings, add its section key and payload beside the descriptor, pass the key as the descriptor's settingsSection, and return a typed contribution from the app implementation. No central settings file or builder switch changes are needed.

If the provider has credential behavior, define its credential adapter beside the descriptor. Register token-account metadata and config validation there, and register any cookie/settings projection through the descriptor's typed settings section; do not add provider cases to the generic config, diagnose, CLI, or plugin broker consumers.

Descriptor-owned metadata derives icon-style identity, log-category construction, display and compact labels, default enablement, fetch/CLI metadata, config capabilities, menu-bar metric capabilities, and icon validation. Generated manifests derive their order from UsageProvider; the provider architecture gatekeeper reports missing descriptor, implementation, icon, settings-section, or widget registrations by provider ID. The WidgetKit case and display table remain deliberate literal exceptions because AppIntents requires statically extractable declarations.

UI notes (Providers settings)

Current: checkboxes per provider.

Preferred direction: table/list rows (like a “sessions” table):

  • Provider (name + short auth hint)
  • Enabled toggle
  • Status (ok/stale/error + last updated)
  • Auth source (CLI / cookies / web / oauth) when applicable
  • Actions (Login / Diagnose / Copy debug log)

This keeps the pane scannable once we have >5 providers.