@cydm/pie 1.0.1 → 1.0.3

package/dist/builtin/extensions/plan-mode/index.js

@@ -0,0 +1,209 @@
+import { createRequire as __createRequire } from "node:module"; const require = __createRequire(import.meta.url);
+
+// builtin/extensions/plan-mode/index.ts
+function extractTodoItems(text) {
+  const items = [];
+  const planMatch = text.match(/Plan:\s*\n?([\s\S]*?)(?=\n\n|\n[A-Z]|$)/i);
+  if (!planMatch) return items;
+  const planSection = planMatch[1];
+  const lines = planSection.split("\n");
+  for (const line of lines) {
+    const trimmed = line.trim();
+    const match = trimmed.match(/^(?:\d+[.):\s]+|[\-*]\s+)(.+)$/);
+    if (match) {
+      items.push({
+        step: items.length + 1,
+        text: match[1].trim(),
+        completed: false
+      });
+    }
+  }
+  return items;
+}
+function isSafeCommand(command) {
+  const trimmed = command.trim();
+  const safePrefixes = [
+    "cat ",
+    "ls ",
+    "find ",
+    "grep ",
+    "head ",
+    "tail ",
+    "echo ",
+    "pwd",
+    "cd ",
+    "which ",
+    "git status",
+    "git log",
+    "git diff",
+    "git show",
+    "git branch -a",
+    "git remote -v",
+    "git config --list",
+    "npm list",
+    "npm view"
+  ];
+  const dangerousPatterns = [
+    /\brm\s+-[rf]*/i,
+    />[\s]*\/dev\//,
+    /\|.*\brm\b/,
+    /git\s+push/i,
+    /git\s+reset/i,
+    /git\s+checkout\s+-/,
+    /git\s+revert/i,
+    /npm\s+publish/i,
+    /npm\s+uninstall/i,
+    /npm\s+rm/i
+  ];
+  for (const pattern of dangerousPatterns) {
+    if (pattern.test(trimmed)) {
+      return false;
+    }
+  }
+  for (const prefix of safePrefixes) {
+    if (trimmed.startsWith(prefix)) {
+      return true;
+    }
+  }
+  if (/^[a-z]+$/.test(trimmed)) {
+    return ["pwd", "ls", "clear", "whoami"].includes(trimmed);
+  }
+  return false;
+}
+function markCompletedSteps(text, todoItems2) {
+  let markedCount = 0;
+  const doneMatches = text.matchAll(/\[DONE:\s*(\d+)\]/gi);
+  for (const match of doneMatches) {
+    const stepNum = parseInt(match[1], 10);
+    const item = todoItems2.find((t) => t.step === stepNum);
+    if (item && !item.completed) {
+      item.completed = true;
+      markedCount++;
+    }
+  }
+  return markedCount;
+}
+var PLAN_MODE_TOOLS = ["read", "bash", "grep", "find", "ls", "questionnaire"];
+var NORMAL_MODE_TOOLS = ["read", "bash", "edit", "write", "grep", "find", "ls", "questionnaire"];
+var planModeEnabled = false;
+var executionMode = false;
+var todoItems = [];
+function planModeExtension(ctx) {
+  ctx.log("Plan mode extension loaded");
+  function updateStatus() {
+    if (executionMode && todoItems.length > 0) {
+      const completed = todoItems.filter((t) => t.completed).length;
+      ctx.ui.notify(`\u{1F4CB} Plan: ${completed}/${todoItems.length} completed`, "info");
+    } else if (planModeEnabled) {
+      ctx.ui.notify("\u23F8\uFE0F Plan mode active (read-only)", "info");
+    }
+  }
+  function togglePlanMode() {
+    planModeEnabled = !planModeEnabled;
+    executionMode = false;
+    todoItems = [];
+    ctx.setPlanMode(planModeEnabled);
+    if (planModeEnabled) {
+      ctx.setActiveTools(PLAN_MODE_TOOLS);
+      ctx.ui.notify(`Plan mode enabled. Available tools: ${PLAN_MODE_TOOLS.join(", ")}`);
+    } else {
+      ctx.setActiveTools(NORMAL_MODE_TOOLS);
+      ctx.ui.notify("Plan mode disabled. Full access restored.");
+    }
+    updateStatus();
+  }
+  ctx.registerCommand({
+    path: ["tools", "plan"],
+    description: "Toggle plan mode (read-only exploration)",
+    handler: async (_ctx) => {
+      togglePlanMode();
+    }
+  });
+  ctx.registerShortcut("ctrl+alt+p", {
+    description: "Toggle plan mode",
+    handler: async () => {
+      togglePlanMode();
+    }
+  });
+  ctx.on("tool:call", async (event) => {
+    if (!planModeEnabled || event.toolName !== "bash") return;
+    const command = event.args?.command;
+    if (command && !isSafeCommand(command)) {
+      ctx.ui.notify(
+        `Plan mode: Command blocked (not allowlisted). Use /plan to disable plan mode first.
+Command: ${command}`,
+        "error"
+      );
+      return { block: true, reason: `Plan mode: command "${command}" is not in the allowlist. Disable plan mode with /plan to run this command.` };
+    }
+  });
+  ctx.on("agent:start", async () => {
+    if (planModeEnabled) {
+      ctx.log("[plan-mode] Injecting plan mode context");
+    }
+  });
+  ctx.on("turn:end", async (event) => {
+    if (!executionMode || todoItems.length === 0) return;
+    const message = event.message;
+    if (message?.role === "assistant" && Array.isArray(message.content)) {
+      const text = message.content.filter((c) => c.type === "text").map((c) => c.text).join("\n");
+      if (markCompletedSteps(text, todoItems) > 0) {
+        updateStatus();
+      }
+    }
+  });
+  ctx.on("agent:end", async (event) => {
+    if (executionMode && todoItems.length > 0) {
+      if (todoItems.every((t) => t.completed)) {
+        const completedList = todoItems.map((t) => `\u2713 ${t.text}`).join("\n");
+        ctx.ui.notify(`**Plan Complete!** \u2713
+
+${completedList}`, "info");
+        executionMode = false;
+        todoItems = [];
+        ctx.setActiveTools(NORMAL_MODE_TOOLS);
+      }
+      return;
+    }
+    if (!planModeEnabled || !ctx.hasUI) return;
+    const messages = event.messages || [];
+    const lastAssistant = [...messages].reverse().find(
+      (m) => m.role === "assistant" && Array.isArray(m.content)
+    );
+    if (lastAssistant) {
+      const text = lastAssistant.content.filter((c) => c.type === "text").map((c) => c.text).join("\n");
+      const extracted = extractTodoItems(text);
+      if (extracted.length > 0) {
+        todoItems = extracted;
+      }
+    }
+    if (todoItems.length > 0) {
+      const todoListText = todoItems.map((t) => `${t.step}. \u2610 ${t.text}`).join("\n");
+      ctx.ui.notify(`**Plan Steps (${todoItems.length}):**
+${todoListText}`, "info");
+      const choice = await ctx.ui.select("Plan mode - what next?", [
+        todoItems.length > 0 ? "Execute the plan (track progress)" : "Execute the plan",
+        "Stay in plan mode",
+        "Refine the plan"
+      ]);
+      if (choice?.startsWith("Execute")) {
+        planModeEnabled = false;
+        executionMode = todoItems.length > 0;
+        ctx.setPlanMode(false);
+        ctx.setActiveTools(NORMAL_MODE_TOOLS);
+        updateStatus();
+        const execMessage = todoItems.length > 0 ? `Execute the plan. Start with: ${todoItems[0].text}. Mark completed steps with [DONE:n] in your response.` : "Execute the plan you just created.";
+        ctx.sendUserMessage(execMessage);
+      } else if (choice === "Refine the plan") {
+        const refinement = await ctx.ui.editor("Refine the plan:", "");
+        if (refinement?.trim()) {
+          ctx.sendUserMessage(refinement.trim());
+        }
+      }
+    }
+  });
+  ctx.ui.notify("\u23F8\uFE0F Plan mode extension loaded", "info");
+}
+export {
+  planModeExtension as default
+};