@slkiser/opencode-quota 3.11.2 → 4.0.0

package/dist/tui.tsx

@@ -8,6 +8,8 @@ import type {
 } from "@opencode-ai/plugin/tui";
 import { Show, createEffect, createSignal, onCleanup } from "solid-js";
 
+import type { SessionTokenError } from "./lib/quota-status.js";
+import type { TuiCommandDisplay } from "./lib/types.js";
 import type {
   CompactStatusState,
   HomeBottomState,
@@ -25,11 +27,21 @@ import {
 } from "./lib/tui-panel-state.js";
 import { getSidebarBodyLineColor } from "./lib/tui-line-style.js";
 import {
+  createTuiQuotaClient,
+  getTuiRuntimeRootHints,
+  getTuiSessionModelMeta,
   loadTuiHomeBottomStatus,
   loadTuiSessionQuotaSurfaces,
+  normalizeTuiSessionID,
   resolveTuiSurfaceRegistration,
   writeTuiQuotaExportIfEnabled,
 } from "./lib/tui-runtime.js";
+import {
+  QUOTA_DIALOG_COMMANDS,
+  buildQuotaDialogCommandOutput,
+  type QuotaDialogCommandId,
+  type QuotaDialogCommandSpec,
+} from "./lib/quota-dialog-commands.js";
 
 const id = "@slkiser/opencode-quota";
 // Place Quota near the top so variable-height built-in sections
@@ -41,6 +53,11 @@ const EVENT_REFRESH_DELAYS_MS = [150, 600] as const;
 const MOUNT_RECOVERY_DELAYS_MS = [500, 1_500, 4_000] as const;
 
 type TuiPromptRefCallback = (ref: TuiPromptRef | undefined) => void;
+type DialogSize = "medium" | "large" | "xlarge";
+
+type QuotaDialogCommandState = {
+  lastSessionTokenError?: SessionTokenError;
+};
 type SessionQuotaResource = {
   sessionID: string;
   sidebar: () => SidebarPanelState;
@@ -197,10 +214,13 @@ function acquireSessionQuotaResource(api: TuiPluginApi, sessionID: string): Sess
   return next;
 }
 
-function createHomeBottomResource(api: TuiPluginApi): HomeBottomResource {
+function createHomeBottomResource(
+  api: TuiPluginApi,
+  compactHomeBottomEnabled: boolean,
+): HomeBottomResource {
   const [bottom, setBottom] = createSignal<HomeBottomState>({
     status: "loading",
-    compact: { status: "loading" },
+    compact: compactHomeBottomEnabled ? { status: "loading" } : { status: "disabled" },
   });
 
   let refCount = 0;
@@ -295,11 +315,14 @@ function createHomeBottomResource(api: TuiPluginApi): HomeBottomResource {
   return resource;
 }
 
-function acquireHomeBottomResource(api: TuiPluginApi): HomeBottomResource {
+function acquireHomeBottomResource(
+  api: TuiPluginApi,
+  compactHomeBottomEnabled: boolean,
+): HomeBottomResource {
   const existing = homeResources.get(api);
   if (existing) return existing.retain();
 
-  const next = createHomeBottomResource(api).retain();
+  const next = createHomeBottomResource(api, compactHomeBottomEnabled).retain();
   homeResources.set(api, next);
   return next;
 }
@@ -435,30 +458,251 @@ function SessionPromptWithCompactStatus(props: {
   );
 }
 
-function HomeBottomView(props: { api: TuiPluginApi }) {
-  const resource = acquireHomeBottomResource(props.api);
+function HomeBottomView(props: { api: TuiPluginApi; compactHomeBottomEnabled: boolean }) {
+  const resource = acquireHomeBottomResource(props.api, props.compactHomeBottomEnabled);
   onCleanup(() => resource.release());
 
   const announcement = () => getHomeBottomAnnouncementText(resource.bottom());
   const compact = () => resource.bottom().compact;
+  const visible = () => shouldRenderHomeBottom(resource.bottom());
 
   return (
-    <Show when={shouldRenderHomeBottom(resource.bottom())}>
-      <box gap={0}>
+    <box gap={0}>
+      <Show when={visible()}>
         <text> </text>
-        <Show when={announcement()}>
-          <box flexDirection="row" justifyContent="center">
-            <text fg={props.api.theme.current.textMuted} wrapMode="none">
-              {announcement()}
-            </text>
-          </box>
-        </Show>
+      </Show>
+      <Show when={visible() && announcement()}>
+        <box flexDirection="row" justifyContent="center">
+          <text fg={props.api.theme.current.textMuted} wrapMode="none">
+            {announcement()}
+          </text>
+        </box>
+      </Show>
+      <Show when={visible()}>
         <CompactStatusLine api={props.api} panel={compact} justifyContent="center" />
-      </box>
-    </Show>
+      </Show>
+    </box>
   );
 }
 
+function getActiveTuiSessionID(api: TuiPluginApi): string | undefined {
+  if (api.route.current.name !== "session") return undefined;
+  return normalizeTuiSessionID(api.route.current.params?.sessionID);
+}
+
+function getTuiCommandArguments(input: unknown): string | undefined {
+  if (!input || typeof input !== "object") return undefined;
+  const record = input as Record<string, unknown>;
+  for (const key of ["arguments", "args", "query"] as const) {
+    const value = record[key];
+    if (typeof value === "string" && value.trim()) return value.trim();
+  }
+  return undefined;
+}
+
+function CommandLoadingDialog(props: { api: TuiPluginApi; title: string }) {
+  return (
+    <box gap={1} paddingLeft={2} paddingRight={2} paddingBottom={1}>
+      <text fg={props.api.theme.current.text}>
+        <b>{props.title}</b>
+      </text>
+      <text fg={props.api.theme.current.textMuted}>Loading deterministic local output…</text>
+    </box>
+  );
+}
+
+function CommandOutputDialog(props: { api: TuiPluginApi; title: string; output: string }) {
+  const lines = () => props.output.split("\n");
+  const bodyHeight = () => Math.min(28, Math.max(6, lines().length));
+  return (
+    <box gap={1} width="100%" flexGrow={1} paddingLeft={2} paddingRight={2} paddingBottom={1}>
+      <text fg={props.api.theme.current.text}>
+        <b>{props.title}</b>
+      </text>
+      <scrollbox width="100%" flexGrow={1} minHeight={bodyHeight()} maxHeight={28}>
+        <box gap={0} width="100%" minWidth={0}>
+          {lines().map((line) => (
+            <text fg={props.api.theme.current.text} wrapMode="word" width="100%">
+              {line || " "}
+            </text>
+          ))}
+        </box>
+      </scrollbox>
+      <text fg={props.api.theme.current.textMuted}>esc closes</text>
+    </box>
+  );
+}
+
+function CommandErrorDialog(props: { api: TuiPluginApi; title: string; error: unknown }) {
+  const message = props.error instanceof Error ? props.error.message : String(props.error);
+  return (
+    <box gap={1} paddingLeft={2} paddingRight={2} paddingBottom={1}>
+      <text fg={props.api.theme.current.text}>
+        <b>{props.title}</b>
+      </text>
+      <text fg={props.api.theme.current.text}>OpenCode Quota command failed.</text>
+      <text fg={props.api.theme.current.textMuted} wrapMode="none">
+        {message || "Unknown error"}
+      </text>
+      <text fg={props.api.theme.current.textMuted}>esc closes</text>
+    </box>
+  );
+}
+
+function getCommandPromptCopy(spec: QuotaDialogCommandSpec): {
+  title: string;
+  placeholder: string;
+  description: string;
+} {
+  switch (spec.id) {
+    case "tokens_between":
+      return {
+        title: "OpenCode Quota Token Range",
+        placeholder: "YYYY-MM-DD YYYY-MM-DD",
+        description: "Enter start and end dates, for example: 2026-01-01 2026-01-15",
+      };
+    case "quota_status":
+      return {
+        title: "OpenCode Quota Status Options",
+        placeholder: 'Optional JSON, e.g. {"refreshGoogleTokens":true}',
+        description: "Leave blank for normal diagnostics, or enter one JSON options object.",
+      };
+    default:
+      return {
+        title: spec.title,
+        placeholder: "Optional arguments",
+        description: "Leave blank to run with no arguments.",
+      };
+  }
+}
+
+function replaceDialog(api: TuiPluginApi, size: DialogSize, render: () => JSX.Element): void {
+  api.ui.dialog.replace(render);
+  // OpenCode dialog.replace() resets size to medium.
+  api.ui.dialog.setSize(size);
+}
+
+async function runQuotaDialogCommandAsync(
+  api: TuiPluginApi,
+  command: QuotaDialogCommandId,
+  commandDisplay: TuiCommandDisplay,
+  rawInput?: unknown,
+  state?: QuotaDialogCommandState,
+): Promise<void> {
+  const spec = QUOTA_DIALOG_COMMANDS.find((item) => item.id === command)!;
+  const argumentsText = getTuiCommandArguments(rawInput);
+  const sessionID = getActiveTuiSessionID(api);
+
+  if (spec.acceptsArguments && rawInput === undefined) {
+    const prompt = getCommandPromptCopy(spec);
+    replaceDialog(api, "medium", () => (
+      <api.ui.DialogPrompt
+        title={prompt.title}
+        placeholder={prompt.placeholder}
+        description={() => (
+          <text fg={api.theme.current.textMuted} wrapMode="word">
+            {prompt.description}
+          </text>
+        )}
+        onCancel={() => api.ui.dialog.clear()}
+        onConfirm={(value) => {
+          void runQuotaDialogCommandAsync(
+            api,
+            command,
+            commandDisplay,
+            { arguments: value.trim() },
+            state,
+          );
+        }}
+      />
+    ));
+    return;
+  }
+
+  const destination =
+    commandDisplay === "inline" && sessionID
+      ? { type: "inline" as const, sessionID }
+      : { type: "dialog" as const };
+  if (destination.type === "dialog") {
+    replaceDialog(api, spec.dialogSize, () => (
+      <CommandLoadingDialog api={api} title={spec.title} />
+    ));
+  }
+
+  try {
+    const result = await buildQuotaDialogCommandOutput({
+      command,
+      arguments: argumentsText,
+      client: createTuiQuotaClient(api),
+      roots: getTuiRuntimeRootHints(api),
+      sessionID,
+      resolveSessionMeta: (id) => getTuiSessionModelMeta(api, id),
+      lastSessionTokenError: state?.lastSessionTokenError,
+      setLastSessionTokenError: state
+        ? (error) => {
+            state.lastSessionTokenError = error;
+          }
+        : undefined,
+      log: async (message, extra) => {
+        await api.client.app.log({
+          body: {
+            service: "quota-toast",
+            level: "debug",
+            message,
+            extra,
+          },
+        });
+      },
+    });
+
+    if (result.state === "noop") {
+      if (destination.type === "dialog") api.ui.dialog.clear();
+      return;
+    }
+
+    if (destination.type === "inline") {
+      await api.client.session.prompt({
+        sessionID: destination.sessionID,
+        noReply: true,
+        parts: [{ type: "text", text: result.output, ignored: true }],
+      });
+      return;
+    }
+
+    replaceDialog(api, result.dialogSize, () => (
+      <CommandOutputDialog api={api} title={result.title} output={result.output} />
+    ));
+  } catch (error) {
+    replaceDialog(api, "large", () => (
+      <CommandErrorDialog api={api} title={spec.title} error={error} />
+    ));
+    api.ui.toast({
+      variant: "error",
+      message: "OpenCode Quota command failed",
+    });
+  }
+}
+
+function registerQuotaDialogCommands(api: TuiPluginApi, commandDisplay: TuiCommandDisplay): void {
+  const commandState: QuotaDialogCommandState = {};
+  const dispose = api.keymap.registerLayer({
+    commands: QUOTA_DIALOG_COMMANDS.map((spec) => ({
+      namespace: "palette",
+      name: `opencode-quota.${spec.id}`,
+      title: spec.title,
+      desc: spec.description,
+      category: "OpenCode Quota",
+      slashName: spec.slashName,
+      run(input?: unknown) {
+        void runQuotaDialogCommandAsync(api, spec.id, commandDisplay, input, commandState);
+      },
+    })),
+    bindings: [],
+  });
+
+  api.lifecycle.onDispose(dispose);
+}
+
 function registerSidebarSlots(api: TuiPluginApi): void {
   api.slots.register({
     order: SIDEBAR_ORDER,
@@ -475,10 +719,13 @@ const tui: TuiPlugin = async (api) => {
   try {
     surfaceRegistration = await resolveTuiSurfaceRegistration(api);
   } catch {
+    registerQuotaDialogCommands(api, "inline");
     registerSidebarSlots(api);
     return;
   }
 
+  registerQuotaDialogCommands(api, surfaceRegistration.commandDisplay);
+
   if (surfaceRegistration.sidebar.enabled) {
     registerSidebarSlots(api);
   }
@@ -511,7 +758,9 @@ const tui: TuiPlugin = async (api) => {
   }
 
   if (surfaceRegistration.homeBottom) {
-    compactSlots.home_bottom = () => <HomeBottomView api={api} />;
+    compactSlots.home_bottom = () => (
+      <HomeBottomView api={api} compactHomeBottomEnabled={surfaceRegistration.compact.homeBottom} />
+    );
   }
 
   if (Object.keys(compactSlots).length > 0) {

package/package.json

@@ -1,7 +1,6 @@
 {
   "name": "@slkiser/opencode-quota",
-  "version": "3.11.2",
-  "packageManager": "pnpm@11.0.0",
+  "version": "4.0.0",
   "description": "OpenCode quota & tokens usage with zero context window pollution. Supports GitHub Copilot, OpenAI (Plus/Pro), Qwen Code, Chutes AI, Synthetic, Google Antigravity, Z.ai coding plan and more.",
   "type": "module",
   "main": "./dist/index.js",
@@ -32,21 +31,6 @@
     "README.md",
     "LICENSE"
   ],
-  "scripts": {
-    "build": "node scripts/clean-dist.mjs && tsc && node scripts/copy-data.mjs && node scripts/prepare-tui-dist.mjs",
-    "build:check": "pnpm run build && pnpm pack --dry-run",
-    "pricing:refresh": "node scripts/refresh-modelsdev-pricing.mjs",
-    "pricing:refresh:if-stale": "node scripts/refresh-modelsdev-pricing-if-stale.mjs",
-    "upstream:check": "node scripts/check-upstream-plugin-updates.mjs",
-    "upstream:prepare-review": "node scripts/prepare-upstream-plugin-review.mjs",
-    "upstream:sync": "node scripts/sync-upstream-plugin-references.mjs",
-    "verify:release-version": "node scripts/verify-release-version.mjs",
-    "typecheck": "tsc --noEmit",
-    "test": "vitest run",
-    "test:watch": "vitest",
-    "prepublishOnly": "pnpm run build",
-    "prepare": "husky"
-  },
   "keywords": [
     "opencode",
     "plugin",
@@ -72,14 +56,15 @@
   "devDependencies": {
     "@babel/core": "^7.28.0",
     "@babel/preset-typescript": "^7.27.1",
-    "@opencode-ai/plugin": "^1.4.3",
+    "@opencode-ai/plugin": "1.18.1",
     "@types/node": "^22.0.0",
     "babel-preset-solid": "^1.9.12",
     "husky": "^9.1.7",
     "lint-staged": "^16.2.7",
     "prettier": "^3.8.0",
     "typescript": "^5.9.3",
-    "vitest": "^4.0.17"
+    "vitest": "^4.0.17",
+    "yaml": "^2.8.3"
   },
   "dependencies": {
     "@clack/prompts": "^1.2.0",
@@ -97,6 +82,28 @@
     "@opencode-ai/plugin": "^1.4.3"
   },
   "engines": {
-    "node": ">=20.0.0"
+    "node": ">=22.0.0"
+  },
+  "scripts": {
+    "build": "node scripts/clean-dist.mjs && tsc && node scripts/copy-data.mjs && node scripts/prepare-tui-dist.mjs",
+    "build:check": "pnpm run build && pnpm pack --dry-run",
+    "pack:release-package": "node scripts/pack-release-package.mjs",
+    "format:check": "node scripts/verify-v4-formatting.mjs",
+    "pricing:refresh": "node scripts/refresh-modelsdev-pricing.mjs",
+    "pricing:refresh:if-stale": "node scripts/refresh-modelsdev-pricing-if-stale.mjs",
+    "upstream:check": "node scripts/check-upstream-plugin-updates.mjs",
+    "upstream:prepare-review": "node scripts/prepare-upstream-plugin-review.mjs",
+    "upstream:sync": "node scripts/sync-upstream-plugin-references.mjs",
+    "release:check": "pnpm run format:check && pnpm run verify:typescript-version && pnpm run verify:v4-history && pnpm run typecheck && pnpm run build && pnpm test && pnpm run test:four-surfaces && pnpm run verify:release-package && pnpm run verify:release-version",
+    "smoke:package": "node scripts/smoke-packed-package.mjs",
+    "test:four-surfaces": "vitest run tests/v4-phase5-cross-surface.test.ts",
+    "verify:package-contents": "node scripts/verify-package-contents.mjs",
+    "verify:release-package": "node scripts/verify-release-package.mjs",
+    "verify:release-version": "node scripts/verify-release-version.mjs",
+    "verify:typescript-version": "node scripts/verify-typescript-version.mjs",
+    "verify:v4-history": "node scripts/verify-v4-history.mjs",
+    "typecheck": "tsc --noEmit",
+    "test": "vitest run",
+    "test:watch": "vitest"
   }
-}
+}