@qadanm/opencode-smart-router 0.0.0-stage → 0.4.0

package/src/core/ui.ts

@@ -0,0 +1,236 @@
+import { execFile } from "node:child_process"
+import { promisify } from "node:util"
+import type { ProviderQuotaSnapshot } from "./types.js"
+import { percent, usd } from "./util.js"
+
+const run = promisify(execFile)
+
+export type IndicatorState = "AUTO" | "MANUAL" | "PAUSED" | "DRY RUN" | "BLOCKED"
+
+export interface Indicator {
+  readonly state: IndicatorState
+  readonly provider?: string
+  readonly model?: string
+  readonly modelLabel?: string
+  readonly title?: string
+}
+
+export const providerShort = (providerID: string | undefined): string => {
+  if (!providerID) return "?"
+  if (providerID === "opencode-go") return "Go"
+  if (providerID === "commandcode") return "CC"
+  if (providerID === "opendesign") return "OD"
+  return providerID
+}
+
+/**
+ * Human-readable confidence state for a provider quota snapshot.
+ *
+ * `verified` means the provider reported the numbers. `estimated` means the
+ * router modelled them from its own usage and they are a lower bound.
+ * `unknown` means nothing could be established safely.
+ */
+export const quotaStateLabel = (snapshot: ProviderQuotaSnapshot): string => {
+  if (snapshot.stale) return "stale"
+  switch (snapshot.quotaState) {
+    case "verified":
+      return "verified"
+    case "estimated":
+      return "estimated"
+    case "exhausted":
+      return "exhausted"
+    case "unavailable":
+      return "unavailable"
+    case "unknown":
+      return "unknown"
+    case "stale":
+      return "stale"
+    default:
+      return snapshot.source === "provider"
+        ? "verified"
+        : snapshot.source === "estimate"
+          ? "estimated"
+          : "unknown"
+  }
+}
+
+/**
+ * Render a provider quota line.
+ *
+ * OpenDesign is reported in Design Plan credits, which are USD-equivalent.
+ * A plan with more than one funding pool prints one line per pool rather
+ * than a summed balance: the pools have different rules, and adding them
+ * would suggest the total can be spent in one go.
+ *
+ * The purchased wallet balance is always shown separately so a non-zero
+ * balance can never be mistaken for permission to spend it.
+ */
+export const formatQuotaLines = (snapshot: ProviderQuotaSnapshot): string[] => {
+  const pools = snapshot.pools ?? []
+  const lines: string[] = []
+  if (pools.length > 1) {
+    lines.push(`${snapshot.label} (${quotaStateLabel(snapshot)}):`)
+    for (const pool of pools) {
+      const windows: string[] = []
+      if (pool.fiveHourUsd !== undefined && pool.fiveHourUsedPct !== undefined) {
+        windows.push(`5h ${percent(pool.fiveHourUsedPct)} of ${usd(pool.fiveHourUsd)}`)
+      }
+      if (pool.weeklyUsd !== undefined && pool.weeklyUsedPct !== undefined) {
+        windows.push(`7d ${percent(pool.weeklyUsedPct)} of ${usd(pool.weeklyUsd)}`)
+      }
+      const windowText = windows.length > 0 ? ` · ${windows.join(" · ")}` : " · no usage windows"
+      lines.push(
+        `  ${pool.label}: ${usd(pool.remainingUsd)} of ${usd(pool.monthlyUsd)} remaining${windowText}`,
+      )
+    }
+    lines.push("  The pools are separate allowances and are never added together.")
+  } else {
+    const windows =
+      snapshot.rolling.source === "unknown" && snapshot.weekly.source === "unknown"
+        ? "no usage windows"
+        : `5h ${percent(snapshot.rolling.usedPct)} · weekly ${percent(snapshot.weekly.usedPct)}`
+    lines.push(
+      `${snapshot.label}: ${windows} · monthly ${percent(snapshot.monthly.usedPct)} (${quotaStateLabel(snapshot)})`,
+    )
+  }
+  if (snapshot.providerID === "opendesign") {
+    const purchased = snapshot.extraCredits?.purchasedUsd ?? 0
+    lines.push(
+      `  Purchased wallet ${usd(purchased)} is separate from both allowances and is never spent automatically.`,
+    )
+    if (snapshot.monthlyRemainingUsd !== undefined) {
+      lines.push(`  Design Plan allowance remaining ${usd(snapshot.monthlyRemainingUsd)}.`)
+    }
+  }
+  if (snapshot.error) lines.push(`  ${snapshot.error}`)
+  return lines
+}
+
+export const indicatorState = (input: {
+  enabled: boolean
+  runtimePaused: boolean
+  manual: boolean
+  dryRun: boolean
+}): IndicatorState => {
+  if (!input.enabled || input.runtimePaused) return "PAUSED"
+  if (input.manual) return "MANUAL"
+  if (input.dryRun) return "DRY RUN"
+  return "AUTO"
+}
+
+export const stripTitlePrefix = (title: string | undefined): string => {
+  if (!title) return ""
+  const match = /^(AUTO|MANUAL|PAUSED|DRY RUN|BLOCKED) · [^—]*(?: — ([\s\S]*))?$/.exec(title)
+  if (!match) return title.trim()
+  return (match[2] ?? "").trim()
+}
+
+/**
+ * The name a session should keep, independent of whatever status decoration is
+ * currently in front of it.
+ *
+ * The failure this exists to prevent: the router used to derive the base title
+ * from `session.title`, which holds the last title *the router itself wrote*. So
+ * the first time the separator was lost — compact mode, a BLOCKED rewrite, a
+ * title written before the generator named the chat — the prefix became the
+ * name, and every later write read that prefix back as if it were a title.
+ * There was no way out of that state.
+ *
+ * Rules:
+ *   - A visible title the router did not write is the user's name. Adopt it.
+ *   - Otherwise keep the base already on record.
+ *   - Only fall back to stripping a prefix as a last resort, and return
+ *     `undefined` rather than an empty string, so the caller can leave the
+ *     session alone instead of writing a nameless title.
+ */
+export const resolveBaseTitle = (input: {
+  currentTitle: string | undefined
+  /** The exact title this router last wrote, if any. */
+  lastWrittenTitle: string | undefined
+  storedBase: string | undefined
+}): string | undefined => {
+  const visible = input.currentTitle?.trim()
+  const routerWroteIt = Boolean(visible) && visible === input.lastWrittenTitle
+  if (visible && !routerWroteIt) return visible
+  if (input.storedBase) return input.storedBase
+  const stripped = stripTitlePrefix(input.currentTitle)
+  return stripped || undefined
+}
+
+export const buildIndicatorTitle = (input: {
+  state: IndicatorState
+  provider?: string
+  modelLabel?: string
+  baseTitle?: string
+  compact?: boolean
+}): string => {
+  const provider = providerShort(input.provider)
+  const model = input.modelLabel?.trim() || "model"
+  const prefix = input.compact ? `${input.state} · ${model}` : `${input.state} · ${provider} · ${model}`
+  const base = stripTitlePrefix(input.baseTitle)
+  // A prefix on its own is not a name, and neither is the placeholder OpenCode
+  // uses before the title generator has run. Compact shortens the prefix; it
+  // must never delete the name, which is what it used to do.
+  if (!base || base === prefix || base === "Session") return prefix
+  return `${prefix} — ${base.slice(0, 150)}`
+}
+
+export class ChangeNotifier {
+  private readonly last = new Map<string, number>()
+  private backend: "terminal-notifier" | "osascript" | "none" | undefined
+
+  shouldNotify(key: string, ttlMs = 10 * 60_000) {
+    const now = Date.now()
+    const previous = this.last.get(key)
+    if (previous && now - previous < ttlMs) return false
+    this.last.set(key, now)
+    if (this.last.size > 100) {
+      for (const [entry, at] of this.last) {
+        if (now - at > ttlMs) this.last.delete(entry)
+      }
+    }
+    return true
+  }
+
+  async notify(message: string): Promise<boolean> {
+    if (!message.trim()) return false
+    if (this.backend === "none") return false
+    if (this.backend === "terminal-notifier" || this.backend === undefined) {
+      try {
+        await run(
+          "terminal-notifier",
+          ["-title", "OpenCode Smart Router", "-message", message.slice(0, 220), "-group", "opencode-smart-router"],
+          { timeout: 1_500, maxBuffer: 4_096 },
+        )
+        this.backend = "terminal-notifier"
+        return true
+      } catch (error) {
+        if ((error as NodeJS.ErrnoException).code !== "ENOENT") {
+          // terminal-notifier exists but the notification was suppressed or
+          // denied. Fall through to the AppleScript path once.
+        }
+      }
+    }
+    try {
+      const escaped = message.replaceAll("\\", "/").replaceAll('"', "'").slice(0, 220)
+      await run(
+        "osascript",
+        ["-e", `display notification "${escaped}" with title "OpenCode Smart Router"`],
+        { timeout: 1_500, maxBuffer: 4_096 },
+      )
+      this.backend = "osascript"
+      return true
+    } catch {
+      this.backend = "none"
+      return false
+    }
+  }
+}
+
+export const __uiInternals = {
+  stripTitlePrefix,
+  buildIndicatorTitle,
+  resolveBaseTitle,
+  providerShort,
+  quotaStateLabel,
+}

package/src/core/util.ts

@@ -0,0 +1,114 @@
+import type { ModelKey, ModelRef, TaskTier } from "./types.js"
+
+export const clamp = (value: number, min: number, max: number) => Math.min(max, Math.max(min, value))
+
+export const pct = (used: number, total: number) => (total <= 0 ? 0 : clamp((used / total) * 100, 0, 100))
+
+export const monthKey = (at = Date.now()) => {
+  const date = new Date(at)
+  return `${date.getUTCFullYear()}-${String(date.getUTCMonth() + 1).padStart(2, "0")}`
+}
+
+export const usd = (value: number) => {
+  if (!Number.isFinite(value)) return "$0.00"
+  const abs = Math.abs(value)
+  if (abs > 0 && abs < 0.01) return `$${value.toFixed(5)}`
+  if (abs < 1) return `$${value.toFixed(4)}`
+  return `$${value.toFixed(2)}`
+}
+
+export const percent = (value: number) => `${value.toFixed(value >= 10 ? 0 : 1)}%`
+
+/**
+ * Describe a plan id for display when the catalog could not resolve it.
+ *
+ * An unrecognized configured id must not be echoed as if it were a tier this
+ * build understands, and must not be silently replaced by a default: either
+ * would imply knowledge of an allowance that was never established. The raw
+ * value is kept only as a suffix so a user can see what they typed to fix it.
+ */
+export const describePlan = (resolvedPlanID: string | undefined, configuredPlanID: string | undefined): string => {
+  if (resolvedPlanID) return resolvedPlanID
+  const configured = (configuredPlanID ?? "").trim()
+  return configured ? `unresolved (${configured})` : "unresolved"
+}
+
+/** Group an integer with thousands separators, for token counts in prose. */
+export const thousands = (value: number) =>
+  Number.isFinite(value) ? Math.round(value).toLocaleString("en-US") : "0"
+
+export const modelKey = (providerID: string, modelID: string): string => `${providerID}/${modelID}`
+
+export const parseModelKey = (key: string): ModelKey => {
+  const index = key.indexOf("/")
+  if (index <= 0) return { providerID: key, modelID: "" }
+  return { providerID: key.slice(0, index), modelID: key.slice(index + 1) }
+}
+
+export const keyOfRef = (ref: ModelRef | undefined): string | undefined =>
+  ref ? modelKey(ref.providerID, ref.id) : undefined
+
+export const refOfKey = (key: string, variant?: string): ModelRef => {
+  const parsed = parseModelKey(key)
+  return variant ? { providerID: parsed.providerID, id: parsed.modelID, variant } : { providerID: parsed.providerID, id: parsed.modelID }
+}
+
+export const refKey = (ref: ModelRef): string => modelKey(ref.providerID, ref.id)
+
+/**
+ * Small, stable hash used only for local diagnostics. It is not a security
+ * primitive and must not be used to protect secrets.
+ */
+export const stableHash = (input: string): string => {
+  let hash = 0x811c9dc5
+  for (let index = 0; index < input.length; index += 1) {
+    hash ^= input.charCodeAt(index)
+    hash = Math.imul(hash, 0x01000193)
+  }
+  return (hash >>> 0).toString(16).padStart(8, "0")
+}
+
+export const truncate = (value: string, max: number) => (value.length <= max ? value : `${value.slice(0, Math.max(0, max - 1))}…`)
+
+export const tierLabel = (tier: TaskTier) =>
+  tier === 0 ? "Deterministic" : tier === 1 ? "Economy" : tier === 2 ? "Standard" : "Premium"
+
+export const shaLength = (value: string) => ({ hash: stableHash(value), chars: value.length })
+
+const secretPatterns = [
+  /bearer\s+[A-Za-z0-9._~+/=-]+/gi,
+  /(?:sk|user|key|token)-[A-Za-z0-9_-]{12,}/gi,
+  /\b[A-Za-z0-9_-]{32,}\b/g,
+]
+
+/** Remove values that look like credentials from diagnostic text. */
+export const redact = (value: string): string => {
+  let result = value
+  for (const pattern of secretPatterns) result = result.replace(pattern, "[redacted]")
+  return truncate(result, 500)
+}
+
+export const safeError = (error: unknown): string => {
+  if (error instanceof Error) return redact(error.message)
+  if (typeof error === "string") return redact(error)
+  try {
+    return redact(JSON.stringify(error))
+  } catch {
+    return "Unknown error"
+  }
+}
+
+export const unique = <T>(values: ReadonlyArray<T>): T[] => [...new Set(values)]
+
+export const isImageUri = (uri: string): boolean => {
+  const lower = uri.toLowerCase()
+  if (lower.startsWith("data:image/")) return true
+  return /\.(png|jpe?g|gif|webp|bmp|tiff?|heic|heif|avif)$/i.test(lower)
+}
+
+export const isImageName = (name: string | undefined): boolean => {
+  if (!name) return false
+  return /\.(png|jpe?g|gif|webp|bmp|tiff?|heic|heif|avif)$/i.test(name)
+}
+
+export const nowId = (prefix: string) => `${prefix}_${Date.now().toString(36)}_${Math.random().toString(36).slice(2, 8)}`