oc-plugin-litellm-budget 0.4.0 → 0.6.0

package/package.json

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

package/tui.tsx

@@ -1,9 +1,6 @@
 // @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
 
@@ -12,32 +9,53 @@ type QuotaData = {
   resetAt: string
   rpm: number
   tpm: number
+  periodDays: number
+  daysIn: 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 {}
-  }
+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 = () => {
+const getConfig = async () => {
   const envKey = process.env.LITELLM_API_KEY || process.env.OPENAI_API_KEY || process.env.BINETZ_REMOTE_CONFIG_TOKEN || ""
-  const key = envKey || readKeyFromConfig()
+  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 }
 }
 
+const parseDurationDays = (d: string): number => {
+  if (!d) return 7
+  const m = d.match(/^(\d+)([dhms])$/)
+  if (!m) return 7
+  const n = parseInt(m[1], 10)
+  if (m[2] === "d") return n
+  if (m[2] === "h") return Math.max(1, Math.round(n / 24))
+  return 7
+}
+
 const fetchQuota = async (url: string, key: string): Promise<QuotaData | null> => {
   if (!url || !key) return null
   try {
@@ -52,11 +70,17 @@ const fetchQuota = async (url: string, key: string): Promise<QuotaData | null> =
     const bt = info.litellm_budget_table || {}
     const max = bt.max_budget ?? 0
     const spend = info.spend ?? 0
+    const resetAt = bt.budget_reset_at ?? ""
+    const periodDays = bt.budget_duration ? parseDurationDays(bt.budget_duration) : 7
+    const daysLeft = resetAt ? Math.max(0, Math.ceil((new Date(resetAt).getTime() - Date.now()) / 86_400_000)) : 0
+    const daysIn = Math.max(1, periodDays - daysLeft + 1)
     return {
       usedPct: max > 0 ? Math.min(100, Math.round((spend / max) * 100)) : 0,
-      resetAt: bt.budget_reset_at ?? "",
+      resetAt,
       rpm: info.rpm_limit ?? 0,
       tpm: info.tpm_limit ?? 0,
+      periodDays,
+      daysIn: Math.min(daysIn, periodDays),
     }
   } catch {
     return null
@@ -76,12 +100,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)
@@ -91,7 +117,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) =>
@@ -103,7 +129,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={
@@ -114,16 +140,16 @@ const Sidebar = (props: { theme: any }) => {
               const free = () => 100 - q().usedPct
               const days = () => daysUntil(q().resetAt)
               const color = () => colorForPct(q().usedPct, props.theme)
-              return (
-                <box flexDirection="column">
-                  <text fg={color()}>{bar(q().usedPct, 10)} {free()}% libre</text>
-                  <text fg={props.theme.current.textMuted}>Reset {days()}d</text>
-                  <Show when={q().rpm > 0 || q().tpm > 0}>
-                    <text fg={props.theme.current.textMuted}>
-                      {q().rpm > 0 ? `${q().rpm} rpm` : ""}{q().rpm > 0 && q().tpm > 0 ? " | " : ""}{q().tpm > 0 ? `${fmtTpm(q().tpm)} tpm` : ""}
-                    </text>
-                  </Show>
-                </box>
+               return (
+                 <box flexDirection="column">
+                   <text fg={color()}>{bar(q().usedPct, 10)} {free()}% libre</text>
+                   <text fg={props.theme.current.textMuted}>Dia {q().daysIn}/{q().periodDays} | Reset {days()}d</text>
+                   <Show when={q().rpm > 0 || q().tpm > 0}>
+                     <text fg={props.theme.current.textMuted}>
+                       {q().rpm > 0 ? `${q().rpm} rpm` : ""}{q().rpm > 0 && q().tpm > 0 ? " | " : ""}{q().tpm > 0 ? `${fmtTpm(q().tpm)} tpm` : ""}
+                     </text>
+                   </Show>
+                 </box>
               )
             }}
           </Show>
@@ -138,7 +164,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={
@@ -156,7 +182,7 @@ const FooterQuota = (props: { theme: any }) => {
             }
             return (
               <text fg={color()}>
-                Quota: {free()}% libre [{bar(q().usedPct, 10)}] Reset {days()}d{limits()}
+                Quota: {free()}% libre [{bar(q().usedPct, 10)}] Dia {q().daysIn}/{q().periodDays} | Reset {days()}d{limits()}
               </text>
             )
           }}