@ikuma.cloud/pix-mcp 0.0.11 → 0.0.12

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ikuma.cloud/pix-mcp",
-  "version": "0.0.11",
+  "version": "0.0.12",
   "publishConfig": {
     "access": "public"
   },

package/src/index.ts

@@ -25,6 +25,7 @@ import {
   formatResult,
   removeEditablePromptGuard,
 } from "./output.ts";
+import { inputPromptArgument } from "./prompt-argument-input.ts";
 import { pickPrompt } from "./prompt-picker.ts";
 import { pickResource } from "./resource-picker.ts";
 import {
@@ -457,19 +458,14 @@ export default function mcp(pi: ExtensionAPI) {
     };
     ensureCurrent();
 
-    const dialogOptions = ctx.signal ? { signal: ctx.signal } : undefined;
     const argumentTokens: PromptArgumentToken[] = [];
     for (const argument of item.prompt.arguments ?? []) {
       const description = compact(argument.description ?? "", 160);
-      const label = `${argument.name} (${argument.required ? "required" : "optional"})`;
-      const guidance = argument.required
-        ? "Enter a value"
-        : "Leave empty to omit";
-      const value = await ctx.ui.input(
-        [label, description, guidance].filter(Boolean).join("\n"),
-        undefined,
-        dialogOptions,
-      );
+      const value = await inputPromptArgument(ctx, {
+        label: `${argument.name} (${argument.required ? "required" : "optional"})`,
+        ...(description ? { description } : {}),
+        guidance: argument.required ? "Enter a value" : "Leave empty to omit",
+      });
       if (value === undefined) return;
       ctx.signal?.throwIfAborted();
       ensureCurrent();

package/src/prompt-argument-input.ts

@@ -0,0 +1,83 @@
+import {
+  ExtensionInputComponent,
+  type ExtensionCommandContext,
+  type Theme,
+} from "@earendil-works/pi-coding-agent";
+import type { TUI } from "@earendil-works/pi-tui";
+
+export interface PromptArgumentInputOptions {
+  label: string;
+  description?: string;
+  guidance: string;
+  signal?: AbortSignal;
+}
+
+class PromptArgumentInputComponent extends ExtensionInputComponent {
+  readonly #tui: TUI;
+  readonly #signal: AbortSignal | undefined;
+  readonly #onAbort: () => void;
+
+  constructor(
+    tui: TUI,
+    theme: Theme,
+    done: (value: string | undefined) => void,
+    options: PromptArgumentInputOptions,
+  ) {
+    let finished = false;
+    const finish = (value: string | undefined) => {
+      if (finished) return;
+      finished = true;
+      done(value);
+    };
+    const details = [
+      ...(options.description ? [theme.fg("dim", options.description)] : []),
+      theme.fg("muted", options.guidance),
+    ].join("\n");
+    super(
+      options.label,
+      undefined,
+      (value) => finish(value),
+      () => finish(undefined),
+      { tui, description: details },
+    );
+    this.#tui = tui;
+    this.#signal = options.signal;
+    this.#onAbort = () => finish(undefined);
+    options.signal?.addEventListener("abort", this.#onAbort, { once: true });
+    if (options.signal?.aborted) this.#onAbort();
+  }
+
+  override handleInput(data: string): void {
+    super.handleInput(data);
+    this.#tui.requestRender();
+  }
+
+  override dispose(): void {
+    this.#signal?.removeEventListener("abort", this.#onAbort);
+    super.dispose();
+  }
+}
+
+export function createPromptArgumentInput(
+  tui: TUI,
+  theme: Theme,
+  done: (value: string | undefined) => void,
+  options: PromptArgumentInputOptions,
+): ExtensionInputComponent {
+  return new PromptArgumentInputComponent(tui, theme, done, options);
+}
+
+// Pi 0.87.1's ctx.ui.input renders every title line as accent and ignores its
+// placeholder. Reuse its native component through custom UI to style details.
+export async function inputPromptArgument(
+  ctx: ExtensionCommandContext,
+  options: Omit<PromptArgumentInputOptions, "signal">,
+): Promise<string | undefined> {
+  if (ctx.signal?.aborted) return;
+  return ctx.ui.custom<string | undefined>((tui, theme, _keybindings, done) =>
+    createPromptArgumentInput(tui, theme, done, {
+      ...options,
+      ...(ctx.signal ? { signal: ctx.signal } : {}),
+    }),
+  );
+}