mirror of
https://github.com/AgentSeal/codeburn.git
synced 2026-08-29 18:33:01 +00:00
287791e2a9
Extends the existing borrowed-credential quota module (claude/codex) with three providers, per-provider settings toggles, and a provider display map: - gemini: Code Assist loadCodeAssist/retrieveUserQuota via the Gemini CLI's on-disk OAuth creds; optional env-gated token refresh; retired-tier responses degrade to terminalFailure with migration guidance - copilot: copilot_internal/user with editor-plugin headers; token read-only from hosts.json/apps.json with one re-read on 401; marked as an internal API that may drift, failures degrade honestly - antigravity: loopback-only Connect-RPC probe of the local language server (ps/lsof discovery via execFile, csrf token for the app variant, agy CLI accepted tokenless); TLS relaxation scoped to 127.0.0.1; no OAuth fallback - settings: per-provider quota switches (default on) persisted renderer-side, honored in the main-process service so disabled fetchers never run; IPC disabled-list filtered against the provider allowlist - sanitizeError extended for Google (ya29.) and GitHub (gh*_) token shapes Kimi audited, not duplicated: the macOS menubar already fetches Kimi quota live (KimiSubscriptionService.swift); the electron app not surfacing it is a pre-existing gap, unchanged here.
76 lines
5.1 KiB
TypeScript
76 lines
5.1 KiB
TypeScript
import { contextBridge, ipcRenderer } from 'electron'
|
|
|
|
// Handlers resolve with { ok, value } | { ok, error } so the structured error
|
|
// `kind` survives the contextBridge boundary. `import type` is erased at build,
|
|
// so this shares main.ts's declaration without pulling its runtime in.
|
|
import type { Envelope } from './main'
|
|
|
|
type DateRange = { from: string; to: string }
|
|
type PriceRates = { input?: number; output?: number; cacheRead?: number; cacheCreation?: number }
|
|
|
|
async function invoke<T>(channel: string, ...args: unknown[]): Promise<T> {
|
|
const res = (await ipcRenderer.invoke(channel, ...args)) as Envelope<T>
|
|
if (res.ok) return res.value
|
|
// Reject with a plain object so `kind` is preserved (Error subclasses lose
|
|
// custom fields when cloned across worlds).
|
|
return Promise.reject(res.error)
|
|
}
|
|
|
|
// Shape matches CodeburnBridge (app/renderer/lib/types.ts); typing is enforced
|
|
// renderer-side where `window.codeburn` is declared as CodeburnBridge.
|
|
const bridge = {
|
|
getQuota: (force?: boolean, disabled?: string[]) => invoke('codeburn:getQuota', force, disabled),
|
|
getOverview: (period: string, provider: string, range?: DateRange, configSource?: string | null, background?: boolean, scope?: string) => invoke('codeburn:getOverview', period, provider, range, configSource, background, scope),
|
|
getTimeline: (period: string, provider: string, range?: DateRange) => invoke('codeburn:getTimeline', period, provider, range),
|
|
getPlans: (period: string) => invoke('codeburn:getPlans', period),
|
|
getActReport: () => invoke('codeburn:getActReport'),
|
|
getModels: (period: string, provider: string, byTask: boolean, range?: DateRange) => invoke('codeburn:getModels', period, provider, byTask, range),
|
|
getSessions: (period: string, provider: string, range?: DateRange) => invoke('codeburn:getSessions', period, provider, range),
|
|
getCompareModels: (period: string, provider: string) => invoke('codeburn:getCompareModels', period, provider),
|
|
getCompare: (period: string, provider: string, modelA: string, modelB: string) => invoke('codeburn:getCompare', period, provider, modelA, modelB),
|
|
getYield: (period: string, provider: string, range?: DateRange) => invoke('codeburn:getYield', period, provider, range),
|
|
getSpendFlow: (period: string, provider: string, range?: DateRange) => invoke('codeburn:getSpendFlow', period, provider, range),
|
|
getOptimizeReport: (period: string, provider: string, range?: DateRange) => invoke('codeburn:getOptimizeReport', period, provider, range),
|
|
getDevices: (period: string) => invoke('codeburn:getDevices', period),
|
|
getDevicesScan: () => invoke('codeburn:getDevicesScan'),
|
|
getShareStatus: () => invoke('codeburn:getShareStatus'),
|
|
getIdentity: () => invoke('codeburn:getIdentity'),
|
|
getAliases: () => invoke('codeburn:getAliases'),
|
|
getProxyPaths: () => invoke('codeburn:getProxyPaths'),
|
|
getAudit: (period: string, provider: string, range?: DateRange) => invoke('codeburn:getAudit', period, provider, range),
|
|
getPriceOverrides: () => invoke('codeburn:getPriceOverrides'),
|
|
setPriceOverride: (model: string, rates: PriceRates) => invoke('codeburn:setPriceOverride', model, rates),
|
|
removePriceOverride: (model: string) => invoke('codeburn:removePriceOverride', model),
|
|
setCurrency: (code: string) => invoke('codeburn:setCurrency', code),
|
|
resetCurrency: () => invoke('codeburn:resetCurrency'),
|
|
addAlias: (from: string, to: string) => invoke('codeburn:addAlias', from, to),
|
|
removeAlias: (from: string) => invoke('codeburn:removeAlias', from),
|
|
removeDevice: (name: string) => invoke('codeburn:removeDevice', name),
|
|
setPlan: (id: string, provider: string) => invoke('codeburn:setPlan', id, provider),
|
|
resetPlan: (provider: string) => invoke('codeburn:resetPlan', provider),
|
|
exportData: (format: string, provider: string, outPath: string) => invoke('codeburn:exportData', format, provider, outPath),
|
|
chooseDirectory: () => invoke('codeburn:chooseDirectory'),
|
|
cliStatus: () => invoke('codeburn:cliStatus'),
|
|
telemetryStatus: () => invoke('codeburn:telemetryStatus'),
|
|
setTelemetryEnabled: (enabled: boolean) => invoke('codeburn:telemetrySetEnabled', enabled),
|
|
completeOnboarding: (enabled: boolean) => invoke('codeburn:telemetryOnboarded', enabled),
|
|
telemetryTrack: (name: string, props?: Record<string, unknown>) => invoke('codeburn:telemetryTrack', name, props),
|
|
openExternal: (url: string) => ipcRenderer.invoke('open-external', url),
|
|
// Cold-start scan progress (main → renderer). Returns an unsubscribe fn.
|
|
onProgress: (cb: (event: unknown) => void) => {
|
|
const listener = (_e: unknown, event: unknown) => cb(event)
|
|
ipcRenderer.on('codeburn:progress', listener)
|
|
return () => { ipcRenderer.removeListener('codeburn:progress', listener) }
|
|
},
|
|
// Update availability: a one-shot read plus a push for the launch + 24h checks.
|
|
getUpdateStatus: () => invoke('codeburn:getUpdateStatus'),
|
|
onUpdateStatus: (cb: (status: unknown) => void) => {
|
|
const listener = (_e: unknown, status: unknown) => cb(status)
|
|
ipcRenderer.on('codeburn:update', listener)
|
|
return () => { ipcRenderer.removeListener('codeburn:update', listener) }
|
|
},
|
|
platform: process.platform,
|
|
arch: process.arch,
|
|
}
|
|
|
|
contextBridge.exposeInMainWorld('codeburn', bridge)
|