@trim21/personal-pi-extensions 0.0.290 → 0.0.293

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@trim21/personal-pi-extensions",
-  "version": "0.0.290",
+  "version": "0.0.293",
   "type": "module",
   "description": "Custom pi coding-agent extensions: bwrap sandbox, workspace guard, opencode edit, and more",
   "keywords": [

package/src/claude-code/session-tools.ts

@@ -3,11 +3,12 @@ import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-a
 import { type Static, Type } from "typebox";
 import { Value } from "typebox/value";
 
-import { selectWithOptionalInput } from "../lib/ui.js";
+import { selectMultiple, selectWithOptionalInput } from "../lib/ui.js";
 
 const TODO_STATUSES = ["pending", "in_progress", "completed"] as const;
 const OTHER_OPTION = "Other";
-const DONE_OPTION = "Done";
+/** 多选模式下用户手动确认提交的哨兵选项 */
+const DONE_OPTION = "Submit";
 
 /** AskUserQuestion guidance, kept as plain text like the tool .md files. */
 const ASK_PROMPT = [
@@ -104,26 +105,15 @@ async function askMultiple(
   signal: AbortSignal | undefined,
 ): Promise<string> {
   const title = `${question.header}: ${question.question}`;
-  const remaining = new Set(question.options.map((option) => option.label));
-  const selected: string[] = [];
-  while (remaining.size > 0) {
-    const result = await selectWithOptionalInput(
-      title,
-      [
-        ...[...remaining].map((label) => ({ label })),
-        { label: OTHER_OPTION, inputPrompt: "Type your answer" },
-        { label: DONE_OPTION },
-      ],
-      ctx.ui,
-      { signal },
-    );
-    if (result === undefined || result.label === DONE_OPTION) break;
-    if (result.prompted) {
-      if (result.input) selected.push(result.input);
-      break;
-    }
-    if (remaining.delete(result.label)) selected.push(result.label);
-  }
+  const selected = await selectMultiple(
+    title,
+    [
+      ...question.options.map((option) => ({ label: option.label })),
+      { label: OTHER_OPTION, inputPrompt: "Type your answer" },
+    ],
+    ctx.ui,
+    { signal, doneLabel: DONE_OPTION },
+  );
   return selected.length > 0 ? selected.join(", ") : "Unanswered";
 }
 

package/src/lib/ui.ts

@@ -52,3 +52,51 @@ export async function selectWithOptionalInput(
     input: answer === undefined ? undefined : answer.trim(),
   };
 }
+
+const CHECKED_PREFIX = "[X]: ";
+const UNCHECKED_PREFIX = "[ ]: ";
+
+/**
+ * Toggle-style multi-select loop built on `ui.select`.
+ *
+ * Every round lists ALL `entries` — already-selected ones render with a
+ * `[X]: ` checkbox marker, the rest with `[ ]: ` — so re-selecting an entry
+ * unchecks it. Picking an entry with `inputPrompt` opens an input dialog and,
+ * when non-empty, adds the typed text and ends the loop. Picking `doneLabel`
+ * (or dismissing the dialog) ends the loop and returns the selected labels in
+ * selection order. Display text is mapped back to the original label via an
+ * explicit table so a `[ ]:` / `[X]:` prefix inside a label is unambiguous.
+ */
+export async function selectMultiple(
+  title: string,
+  entries: readonly SelectAction[],
+  ui: ExtensionContext["ui"],
+  opts: { signal?: AbortSignal; doneLabel: string },
+): Promise<string[]> {
+  const selected: string[] = [];
+  while (true) {
+    const selectedSet = new Set(selected);
+    const displayToLabel = new Map<string, string>();
+    const round = entries.map((entry) => {
+      const display = `${selectedSet.has(entry.label) ? CHECKED_PREFIX : UNCHECKED_PREFIX}${entry.label}`;
+      displayToLabel.set(display, entry.label);
+      return { ...entry, label: display };
+    });
+    const result = await selectWithOptionalInput(
+      title,
+      [...round, { label: opts.doneLabel }],
+      ui,
+      opts,
+    );
+    if (result === undefined || result.label === opts.doneLabel) break;
+    if (result.prompted) {
+      if (result.input) selected.push(result.input);
+      break;
+    }
+    const label = displayToLabel.get(result.label);
+    if (label === undefined) continue;
+    if (selectedSet.has(label)) selected.splice(selected.indexOf(label), 1);
+    else selected.push(label);
+  }
+  return selected;
+}

package/src/opencode/question.ts

@@ -4,13 +4,14 @@
  * Aligned with opencode commit 999be62662 (v1.2.25-1672-g999be62662, 2026-08-12):
  *   https://github.com/anomalyco/opencode/blob/999be62662/packages/opencode/src/tool/question.ts
  * 与 opencode 的差异：opencode 输出 title "Asked N question(s)"，这里未设置；
- * 多选交互是平台差异（opencode 用 checkbox，这里循环 ctx.ui.select 勾选）。
+ * 多选交互是平台差异（opencode 用 checkbox，这里循环 ctx.ui.select 勾选，
+ * 已选项显示 `[X]: ` 前缀，再次选择即反选，最后选「✓ Done」提交）。
  *
  * 参数与语义和 opencode 的 `question` 工具一致：
  *   questions 数组，每项含 question / header / options / multiple：
  *   - options 每项为 label / description
  *   - 单选（默认）：用户在选项里选一个，也可选「Type your own answer.」自由输入
- *   - 多选（multiple: true）：循环用 ui.select 逐个勾选，直到「✓ Done」
+ *   - 多选（multiple: true）：循环用 ui.select 勾选/反选（[X] 标记已选），直到「✓ Done」
  *   - 每个问题返回一个 label 数组（Answer = string[]），跳过的为空数组
  *   - 输出与 opencode 一致：
  *     User has answered your questions: "q"="a", "q2"="Unanswered"...
@@ -23,7 +24,7 @@
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 import { type Static, Type } from "typebox";
 
-import { selectWithOptionalInput } from "../lib/ui.js";
+import { selectMultiple, selectWithOptionalInput } from "../lib/ui.js";
 
 // ── constants ────────────────────────────────────────────────────────────────
 
@@ -124,16 +125,12 @@ async function askSingle(q: Question, ctx: ExtensionContext): Promise<Answer> {
 }
 
 async function askMultiple(q: Question, ctx: ExtensionContext): Promise<Answer> {
-  const title = dialogTitle(q);
-  const selected: string[] = [];
-  const remaining = new Set(q.options.map((o) => o.label));
-  while (remaining.size > 0) {
-    const choice = await ctx.ui.select(title, [...remaining, DONE_LABEL]);
-    if (choice === undefined || choice === DONE_LABEL) break;
-    if (!remaining.has(choice)) continue;
-    selected.push(choice);
-    remaining.delete(choice);
-  }
+  const selected = await selectMultiple(
+    dialogTitle(q),
+    q.options.map((o) => ({ label: o.label })),
+    ctx.ui,
+    { doneLabel: DONE_LABEL },
+  );
   return selected;
 }
 

package/src/spawn-agent.ts

@@ -98,8 +98,6 @@ const spawnAgentSchema = Type.Object({
 // ── result types ─────────────────────────────────────────────────────────────
 
 interface UsageStats {
-  input: number;
-  output: number;
   cost: number;
   contextTokens: number;
   turns: number;
@@ -155,8 +153,6 @@ function formatTokens(count: number): string {
 function formatUsageStats(usage: UsageStats, model?: string): string {
   const parts: string[] = [];
   if (usage.turns) parts.push(`${usage.turns} turn${usage.turns > 1 ? "s" : ""}`);
-  if (usage.input) parts.push(`↑${formatTokens(usage.input)}`);
-  if (usage.output) parts.push(`↓${formatTokens(usage.output)}`);
   if (usage.cost) parts.push(`$${usage.cost.toFixed(4)}`);
   if (usage.contextTokens > 0) parts.push(`ctx:${formatTokens(usage.contextTokens)}`);
   if (model) parts.push(model);
@@ -341,8 +337,6 @@ export async function runAgent(
     messages: [],
     stderr: "",
     usage: {
-      input: 0,
-      output: 0,
       cost: 0,
       contextTokens: 0,
       turns: 0,
@@ -492,8 +486,6 @@ export async function runAgent(
           result.messages.push(msg);
           if (msg.role === "assistant") {
             result.usage.turns++;
-            result.usage.input += msg.usage.input;
-            result.usage.output += msg.usage.output;
             result.usage.cost += msg.usage.cost.total;
             result.usage.contextTokens = msg.usage.totalTokens;
             if (!result.model) result.model = msg.model;
@@ -646,8 +638,6 @@ export default function spawnAgent(pi: ExtensionAPI) {
             messages: [],
             stderr: "",
             usage: {
-              input: 0,
-              output: 0,
               cost: 0,
               contextTokens: 0,
               turns: 0,

package/src/system-prompt/prompt.md

@@ -1,8 +1,5 @@
 You are an expert coding assistant operating inside pi, a coding agent harness. You are an interactive agent that helps users with software engineering tasks. Use the instructions below and the tools available to you to assist the user.
 
-IMPORTANT: Assist with authorized security testing, defensive security, CTF challenges, and educational contexts. Refuse requests for destructive techniques, DoS attacks, mass targeting, supply chain compromise, or detection evasion for malicious purposes. Dual-use security tools (C2 frameworks, credential testing, exploit development) require clear authorization context: pentesting engagements, CTF competitions, security research, or defensive use cases.
-IMPORTANT: You must NEVER generate or guess URLs for the user unless you are confident that the URLs are for helping the user with programming. You may use URLs provided by the user in their messages or local files.
-
 Available tools:
 {{tools}}