@slkiser/opencode-quota 3.11.2 → 4.0.0

package/dist/tui.js

@@ -11,7 +11,8 @@ import { createElement as _$createElement } from "@opentui/solid";
 import { Show, createEffect, createSignal, onCleanup } from "solid-js";
 import { getCompactStatusText, getHomeBottomAnnouncementText, getSidebarPanelLines, getSidebarPanelLinesExpanded, shouldRenderCompactStatus, shouldRenderHomeBottom, shouldRenderSidebarPanel } from "./lib/tui-panel-state.js";
 import { getSidebarBodyLineColor } from "./lib/tui-line-style.js";
-import { loadTuiHomeBottomStatus, loadTuiSessionQuotaSurfaces, resolveTuiSurfaceRegistration, writeTuiQuotaExportIfEnabled } from "./lib/tui-runtime.js";
+import { createTuiQuotaClient, getTuiRuntimeRootHints, getTuiSessionModelMeta, loadTuiHomeBottomStatus, loadTuiSessionQuotaSurfaces, normalizeTuiSessionID, resolveTuiSurfaceRegistration, writeTuiQuotaExportIfEnabled } from "./lib/tui-runtime.js";
+import { QUOTA_DIALOG_COMMANDS, buildQuotaDialogCommandOutput } from "./lib/quota-dialog-commands.js";
 const id = "@slkiser/opencode-quota";
 // Place Quota near the top so variable-height built-in sections
 // (MCP/LSP/Todo/Files) do not push it below the visible fold.
@@ -139,11 +140,13 @@ function acquireSessionQuotaResource(api, sessionID) {
   resources.set(sessionID, next);
   return next;
 }
-function createHomeBottomResource(api) {
+function createHomeBottomResource(api, compactHomeBottomEnabled) {
   const [bottom, setBottom] = createSignal({
     status: "loading",
-    compact: {
+    compact: compactHomeBottomEnabled ? {
       status: "loading"
+    } : {
+      status: "disabled"
     }
   });
   let refCount = 0;
@@ -220,10 +223,10 @@ function createHomeBottomResource(api) {
   reload();
   return resource;
 }
-function acquireHomeBottomResource(api) {
+function acquireHomeBottomResource(api, compactHomeBottomEnabled) {
   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;
 }
@@ -395,46 +398,342 @@ function SessionPromptWithCompactStatus(props) {
   })();
 }
 function HomeBottomView(props) {
-  const resource = acquireHomeBottomResource(props.api);
+  const resource = acquireHomeBottomResource(props.api, props.compactHomeBottomEnabled);
   onCleanup(() => resource.release());
   const announcement = () => getHomeBottomAnnouncementText(resource.bottom());
   const compact = () => resource.bottom().compact;
-  return _$createComponent(Show, {
-    get when() {
-      return shouldRenderHomeBottom(resource.bottom());
-    },
-    get children() {
-      var _el$14 = _$createElement("box"),
-        _el$15 = _$createElement("text");
-      _$insertNode(_el$14, _el$15);
-      _$setProp(_el$14, "gap", 0);
-      _$insertNode(_el$15, _$createTextNode(` `));
-      _$insert(_el$14, _$createComponent(Show, {
-        get when() {
-          return announcement();
-        },
-        get children() {
-          var _el$17 = _$createElement("box"),
-            _el$18 = _$createElement("text");
-          _$insertNode(_el$17, _el$18);
-          _$setProp(_el$17, "flexDirection", "row");
-          _$setProp(_el$17, "justifyContent", "center");
-          _$setProp(_el$18, "wrapMode", "none");
-          _$insert(_el$18, announcement);
-          _$effect(_$p => _$setProp(_el$18, "fg", props.api.theme.current.textMuted, _$p));
-          return _el$17;
-        }
-      }), null);
-      _$insert(_el$14, _$createComponent(CompactStatusLine, {
-        get api() {
-          return props.api;
-        },
-        panel: compact,
-        justifyContent: "center"
-      }), null);
-      return _el$14;
+  const visible = () => shouldRenderHomeBottom(resource.bottom());
+  return (() => {
+    var _el$14 = _$createElement("box");
+    _$setProp(_el$14, "gap", 0);
+    _$insert(_el$14, _$createComponent(Show, {
+      get when() {
+        return visible();
+      },
+      get children() {
+        var _el$15 = _$createElement("text");
+        _$insertNode(_el$15, _$createTextNode(` `));
+        return _el$15;
+      }
+    }), null);
+    _$insert(_el$14, _$createComponent(Show, {
+      get when() {
+        return _$memo(() => !!visible())() && announcement();
+      },
+      get children() {
+        var _el$17 = _$createElement("box"),
+          _el$18 = _$createElement("text");
+        _$insertNode(_el$17, _el$18);
+        _$setProp(_el$17, "flexDirection", "row");
+        _$setProp(_el$17, "justifyContent", "center");
+        _$setProp(_el$18, "wrapMode", "none");
+        _$insert(_el$18, announcement);
+        _$effect(_$p => _$setProp(_el$18, "fg", props.api.theme.current.textMuted, _$p));
+        return _el$17;
+      }
+    }), null);
+    _$insert(_el$14, _$createComponent(Show, {
+      get when() {
+        return visible();
+      },
+      get children() {
+        return _$createComponent(CompactStatusLine, {
+          get api() {
+            return props.api;
+          },
+          panel: compact,
+          justifyContent: "center"
+        });
+      }
+    }), null);
+    return _el$14;
+  })();
+}
+function getActiveTuiSessionID(api) {
+  if (api.route.current.name !== "session") return undefined;
+  return normalizeTuiSessionID(api.route.current.params?.sessionID);
+}
+function getTuiCommandArguments(input) {
+  if (!input || typeof input !== "object") return undefined;
+  const record = input;
+  for (const key of ["arguments", "args", "query"]) {
+    const value = record[key];
+    if (typeof value === "string" && value.trim()) return value.trim();
+  }
+  return undefined;
+}
+function CommandLoadingDialog(props) {
+  return (() => {
+    var _el$19 = _$createElement("box"),
+      _el$20 = _$createElement("text"),
+      _el$21 = _$createElement("b"),
+      _el$22 = _$createElement("text");
+    _$insertNode(_el$19, _el$20);
+    _$insertNode(_el$19, _el$22);
+    _$setProp(_el$19, "gap", 1);
+    _$setProp(_el$19, "paddingLeft", 2);
+    _$setProp(_el$19, "paddingRight", 2);
+    _$setProp(_el$19, "paddingBottom", 1);
+    _$insertNode(_el$20, _el$21);
+    _$insert(_el$21, () => props.title);
+    _$insertNode(_el$22, _$createTextNode(`Loading deterministic local output…`));
+    _$effect(_p$ => {
+      var _v$3 = props.api.theme.current.text,
+        _v$4 = props.api.theme.current.textMuted;
+      _v$3 !== _p$.e && (_p$.e = _$setProp(_el$20, "fg", _v$3, _p$.e));
+      _v$4 !== _p$.t && (_p$.t = _$setProp(_el$22, "fg", _v$4, _p$.t));
+      return _p$;
+    }, {
+      e: undefined,
+      t: undefined
+    });
+    return _el$19;
+  })();
+}
+function CommandOutputDialog(props) {
+  const lines = () => props.output.split("\n");
+  const bodyHeight = () => Math.min(28, Math.max(6, lines().length));
+  return (() => {
+    var _el$24 = _$createElement("box"),
+      _el$25 = _$createElement("text"),
+      _el$26 = _$createElement("b"),
+      _el$27 = _$createElement("scrollbox"),
+      _el$28 = _$createElement("box"),
+      _el$29 = _$createElement("text");
+    _$insertNode(_el$24, _el$25);
+    _$insertNode(_el$24, _el$27);
+    _$insertNode(_el$24, _el$29);
+    _$setProp(_el$24, "gap", 1);
+    _$setProp(_el$24, "width", "100%");
+    _$setProp(_el$24, "flexGrow", 1);
+    _$setProp(_el$24, "paddingLeft", 2);
+    _$setProp(_el$24, "paddingRight", 2);
+    _$setProp(_el$24, "paddingBottom", 1);
+    _$insertNode(_el$25, _el$26);
+    _$insert(_el$26, () => props.title);
+    _$insertNode(_el$27, _el$28);
+    _$setProp(_el$27, "width", "100%");
+    _$setProp(_el$27, "flexGrow", 1);
+    _$setProp(_el$27, "maxHeight", 28);
+    _$setProp(_el$28, "gap", 0);
+    _$setProp(_el$28, "width", "100%");
+    _$setProp(_el$28, "minWidth", 0);
+    _$insert(_el$28, () => lines().map(line => (() => {
+      var _el$31 = _$createElement("text");
+      _$setProp(_el$31, "wrapMode", "word");
+      _$setProp(_el$31, "width", "100%");
+      _$insert(_el$31, line || " ");
+      _$effect(_$p => _$setProp(_el$31, "fg", props.api.theme.current.text, _$p));
+      return _el$31;
+    })()));
+    _$insertNode(_el$29, _$createTextNode(`esc closes`));
+    _$effect(_p$ => {
+      var _v$5 = props.api.theme.current.text,
+        _v$6 = bodyHeight(),
+        _v$7 = props.api.theme.current.textMuted;
+      _v$5 !== _p$.e && (_p$.e = _$setProp(_el$25, "fg", _v$5, _p$.e));
+      _v$6 !== _p$.t && (_p$.t = _$setProp(_el$27, "minHeight", _v$6, _p$.t));
+      _v$7 !== _p$.a && (_p$.a = _$setProp(_el$29, "fg", _v$7, _p$.a));
+      return _p$;
+    }, {
+      e: undefined,
+      t: undefined,
+      a: undefined
+    });
+    return _el$24;
+  })();
+}
+function CommandErrorDialog(props) {
+  const message = props.error instanceof Error ? props.error.message : String(props.error);
+  return (() => {
+    var _el$32 = _$createElement("box"),
+      _el$33 = _$createElement("text"),
+      _el$34 = _$createElement("b"),
+      _el$35 = _$createElement("text"),
+      _el$37 = _$createElement("text"),
+      _el$38 = _$createElement("text");
+    _$insertNode(_el$32, _el$33);
+    _$insertNode(_el$32, _el$35);
+    _$insertNode(_el$32, _el$37);
+    _$insertNode(_el$32, _el$38);
+    _$setProp(_el$32, "gap", 1);
+    _$setProp(_el$32, "paddingLeft", 2);
+    _$setProp(_el$32, "paddingRight", 2);
+    _$setProp(_el$32, "paddingBottom", 1);
+    _$insertNode(_el$33, _el$34);
+    _$insert(_el$34, () => props.title);
+    _$insertNode(_el$35, _$createTextNode(`OpenCode Quota command failed.`));
+    _$setProp(_el$37, "wrapMode", "none");
+    _$insert(_el$37, message || "Unknown error");
+    _$insertNode(_el$38, _$createTextNode(`esc closes`));
+    _$effect(_p$ => {
+      var _v$8 = props.api.theme.current.text,
+        _v$9 = props.api.theme.current.text,
+        _v$0 = props.api.theme.current.textMuted,
+        _v$1 = props.api.theme.current.textMuted;
+      _v$8 !== _p$.e && (_p$.e = _$setProp(_el$33, "fg", _v$8, _p$.e));
+      _v$9 !== _p$.t && (_p$.t = _$setProp(_el$35, "fg", _v$9, _p$.t));
+      _v$0 !== _p$.a && (_p$.a = _$setProp(_el$37, "fg", _v$0, _p$.a));
+      _v$1 !== _p$.o && (_p$.o = _$setProp(_el$38, "fg", _v$1, _p$.o));
+      return _p$;
+    }, {
+      e: undefined,
+      t: undefined,
+      a: undefined,
+      o: undefined
+    });
+    return _el$32;
+  })();
+}
+function getCommandPromptCopy(spec) {
+  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, size, render) {
+  api.ui.dialog.replace(render);
+  // OpenCode dialog.replace() resets size to medium.
+  api.ui.dialog.setSize(size);
+}
+async function runQuotaDialogCommandAsync(api, command, commandDisplay, rawInput, state) {
+  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", () => _$createComponent(api.ui.DialogPrompt, {
+      get title() {
+        return prompt.title;
+      },
+      get placeholder() {
+        return prompt.placeholder;
+      },
+      description: () => (() => {
+        var _el$40 = _$createElement("text");
+        _$setProp(_el$40, "wrapMode", "word");
+        _$insert(_el$40, () => prompt.description);
+        _$effect(_$p => _$setProp(_el$40, "fg", api.theme.current.textMuted, _$p));
+        return _el$40;
+      })(),
+      onCancel: () => api.ui.dialog.clear(),
+      onConfirm: value => {
+        void runQuotaDialogCommandAsync(api, command, commandDisplay, {
+          arguments: value.trim()
+        }, state);
+      }
+    }));
+    return;
+  }
+  const destination = commandDisplay === "inline" && sessionID ? {
+    type: "inline",
+    sessionID
+  } : {
+    type: "dialog"
+  };
+  if (destination.type === "dialog") {
+    replaceDialog(api, spec.dialogSize, () => _$createComponent(CommandLoadingDialog, {
+      api: api,
+      get title() {
+        return 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, () => _$createComponent(CommandOutputDialog, {
+      api: api,
+      get title() {
+        return result.title;
+      },
+      get output() {
+        return result.output;
+      }
+    }));
+  } catch (error) {
+    replaceDialog(api, "large", () => _$createComponent(CommandErrorDialog, {
+      api: api,
+      get title() {
+        return spec.title;
+      },
+      error: error
+    }));
+    api.ui.toast({
+      variant: "error",
+      message: "OpenCode Quota command failed"
+    });
+  }
+}
+function registerQuotaDialogCommands(api, commandDisplay) {
+  const commandState = {};
+  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) {
+        void runQuotaDialogCommandAsync(api, spec.id, commandDisplay, input, commandState);
+      }
+    })),
+    bindings: []
   });
+  api.lifecycle.onDispose(dispose);
 }
 function registerSidebarSlots(api) {
   api.slots.register({
@@ -456,9 +755,11 @@ const tui = async api => {
   try {
     surfaceRegistration = await resolveTuiSurfaceRegistration(api);
   } catch {
+    registerQuotaDialogCommands(api, "inline");
     registerSidebarSlots(api);
     return;
   }
+  registerQuotaDialogCommands(api, surfaceRegistration.commandDisplay);
   if (surfaceRegistration.sidebar.enabled) {
     registerSidebarSlots(api);
   }
@@ -487,7 +788,10 @@ const tui = async api => {
   }
   if (surfaceRegistration.homeBottom) {
     compactSlots.home_bottom = () => _$createComponent(HomeBottomView, {
-      api: api
+      api: api,
+      get compactHomeBottomEnabled() {
+        return surfaceRegistration.compact.homeBottom;
+      }
     });
   }
   if (Object.keys(compactSlots).length > 0) {