oc-plugin-litellm-budget 0.3.0 → 0.5.0

package/package.json

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

package/tui.tsx

@@ -11,10 +11,36 @@ type QuotaData = {
   tpm: number
 }
 
-const getConfig = () => {
-  const binetzToken = process.env.BINETZ_REMOTE_CONFIG_TOKEN || ""
-  const url = process.env.LITELLM_BASE_URL || process.env.OPENAI_BASE_URL || (binetzToken ? "https://litellm.binetz.com" : "")
-  const key = process.env.LITELLM_API_KEY || process.env.OPENAI_API_KEY || binetzToken || ""
+const readKeyFromConfig = async (): Promise<string> => {
+  try {
+    const home = process.env.HOME || process.env.USERPROFILE || ""
+    if (!home) return ""
+    const base = process.env.XDG_CONFIG_HOME || `${home}/.config`
+    for (const name of ["opencode.json", "opencode.jsonc"]) {
+      try {
+        const raw = await Bun.file(`${base}/opencode/${name}`).text()
+        if (!raw) continue
+        const d = JSON.parse(raw)
+        const k = d?.provider?.binetz?.options?.apiKey
+        if (k && typeof k === "string" && !k.startsWith("{env:")) return k
+      } catch {
+        try {
+          const raw = await Bun.file(`${base}/opencode/${name}`).text()
+          const clean = raw.replace(/^\s*\/\/.*$/gm, "").replace(/,(\s*[}\]])/g, "$1")
+          const d = JSON.parse(clean)
+          const k = d?.provider?.binetz?.options?.apiKey
+          if (k && typeof k === "string" && !k.startsWith("{env:")) return k
+        } catch {}
+      }
+    }
+  } catch {}
+  return ""
+}
+
+const getConfig = async () => {
+  const envKey = process.env.LITELLM_API_KEY || process.env.OPENAI_API_KEY || process.env.BINETZ_REMOTE_CONFIG_TOKEN || ""
+  const key = envKey || await readKeyFromConfig()
+  const url = process.env.LITELLM_BASE_URL || process.env.OPENAI_BASE_URL || (key ? "https://litellm.binetz.com" : "")
   return { url: url.replace(/\/+$/, ""), key }
 }
 
@@ -56,12 +82,14 @@ const fmtTpm = (n: number): string =>
   n >= 1_000_000 ? `${(n / 1_000_000).toFixed(0)}M` : n >= 1_000 ? `${(n / 1_000).toFixed(0)}K` : `${n}`
 
 const useQuota = () => {
-  const { url, key } = getConfig()
   const [data, setData] = createSignal<QuotaData | null>(null)
-  const [err, setErr] = createSignal(!url || !key)
+  const [err, setErr] = createSignal(false)
+  const [configured, setConfigured] = createSignal(true)
 
   const poll = async () => {
-    if (!url || !key) { setErr(true); return }
+    const { url, key } = await getConfig()
+    if (!url || !key) { setConfigured(false); setErr(true); return }
+    setConfigured(true)
     const q = await fetchQuota(url, key)
     if (q) { setData(q); setErr(false) }
     else setErr(true)
@@ -71,7 +99,7 @@ const useQuota = () => {
   const timer = setInterval(poll, POLL_MS)
   onCleanup(() => clearInterval(timer))
 
-  return { data, err, configured: !!url && !!key }
+  return { data, err, configured }
 }
 
 const colorForPct = (pct: number, theme: any) =>
@@ -83,7 +111,7 @@ const Sidebar = (props: { theme: any }) => {
   return (
     <box paddingTop={1} width="100%" flexDirection="column">
       <text bold fg={props.theme.current.primary}>Quota</text>
-      <Show when={configured} fallback={
+      <Show when={configured()} fallback={
         <text fg={props.theme.current.textMuted}>Not configured</text>
       }>
         <Show when={!err()} fallback={
@@ -118,7 +146,7 @@ const FooterQuota = (props: { theme: any }) => {
 
   return (
     <box>
-      <Show when={configured} fallback={
+      <Show when={configured()} fallback={
         <text fg={props.theme.current.textMuted}>Quota: not configured</text>
       }>
         <Show when={!err() && data()} fallback={