oc-plugin-litellm-budget 0.2.0 → 0.4.0

package/package.json

@@ -1,7 +1,7 @@
 {
   "$schema": "https://json.schemastore.org/package.json",
   "name": "oc-plugin-litellm-budget",
-  "version": "0.2.0",
+  "version": "0.4.0",
   "description": "Display LiteLLM budget usage in OpenCode TUI sidebar and footer",
   "type": "module",
   "exports": {

package/tui.tsx

@@ -1,6 +1,9 @@
 // @ts-nocheck
 import type { TuiPlugin, TuiPluginModule } from "@opencode-ai/plugin/tui"
 import { createSignal, onCleanup, Show } from "solid-js"
+import { readFileSync } from "fs"
+import { join } from "path"
+import { homedir } from "os"
 
 const POLL_MS = 60_000
 
@@ -11,9 +14,27 @@ type QuotaData = {
   tpm: number
 }
 
+const readKeyFromConfig = (): string => {
+  const paths = [
+    join(process.env.XDG_CONFIG_HOME || join(homedir(), ".config"), "opencode", "opencode.json"),
+    join(process.env.XDG_CONFIG_HOME || join(homedir(), ".config"), "opencode", "opencode.jsonc"),
+  ]
+  for (const p of paths) {
+    try {
+      const raw = readFileSync(p, "utf8")
+      const clean = raw.replace(/\/\/.*$/gm, "").replace(/\/\*[\s\S]*?\*\//g, "")
+      const d = JSON.parse(clean)
+      const k = d?.provider?.binetz?.options?.apiKey
+      if (k && typeof k === "string" && k !== "{env:BINETZ_REMOTE_CONFIG_TOKEN}") return k
+    } catch {}
+  }
+  return ""
+}
+
 const getConfig = () => {
-  const url = process.env.LITELLM_BASE_URL || process.env.OPENAI_BASE_URL || ""
-  const key = process.env.LITELLM_API_KEY || process.env.OPENAI_API_KEY || process.env.OPENCODE_API_KEY || ""
+  const envKey = process.env.LITELLM_API_KEY || process.env.OPENAI_API_KEY || process.env.BINETZ_REMOTE_CONFIG_TOKEN || ""
+  const key = envKey || readKeyFromConfig()
+  const url = process.env.LITELLM_BASE_URL || process.env.OPENAI_BASE_URL || (key ? "https://litellm.binetz.com" : "")
   return { url: url.replace(/\/+$/, ""), key }
 }