@cydm/pie 1.0.3 → 1.0.5

package/dist/cli.js

@@ -65783,8 +65783,8 @@ function createExtensionContext2(extension, cwd, uiContext, abortFn, isIdleFn, h
     sendUserMessage(content) {
       actions?.sendUserMessage(content);
     },
-    setPlanMode(enabled) {
-      actions?.setPlanMode(enabled);
+    setPlanMode(enabled, options) {
+      actions?.setPlanMode(enabled, options);
     },
     // Extended properties for subagent creation
     get apiKey() {
@@ -65799,6 +65799,15 @@ function createExtensionContext2(extension, cwd, uiContext, abortFn, isIdleFn, h
     getAllTools() {
       return actions?.getAllTools ? actions.getAllTools() : actions?.getActiveTools()?.map((name) => ({ name, description: "" })) ?? [];
     },
+    getMessages() {
+      return actions?.getMessages?.() ?? [];
+    },
+    getSessionMetadata() {
+      return actions?.getSessionMetadata?.() ?? {};
+    },
+    setSessionMetadata(key, value) {
+      actions?.setSessionMetadata?.(key, value);
+    },
     get yoloMode() {
       return actions?.yoloMode ?? false;
     }
@@ -66190,7 +66199,7 @@ var ExtensionRunner = class {
       setActiveTools: (toolNames) => this.actions.setActiveTools(toolNames),
       getActiveTools: () => this.actions.getActiveTools(),
       sendUserMessage: (content) => this.actions.sendUserMessage(content),
-      setPlanMode: (enabled) => this.actions.setPlanMode(enabled),
+      setPlanMode: (enabled, options) => this.actions.setPlanMode(enabled, options),
       get apiKey() {
         return runner.actions.apiKey;
       },
@@ -66201,6 +66210,9 @@ var ExtensionRunner = class {
         return runner.actions.skills ?? [];
       },
       getAllTools: () => runner.actions.getAllTools(),
+      getMessages: () => runner.actions.getMessages(),
+      getSessionMetadata: () => runner.actions.getSessionMetadata(),
+      setSessionMetadata: (key, value) => runner.actions.setSessionMetadata(key, value),
       get yoloMode() {
         return runner.actions.yoloMode ?? false;
       }
@@ -69314,6 +69326,75 @@ function findFirstChangedLine(oldText, newText) {
   return void 0;
 }
 
+// src/todo/render.ts
+var TODO_STATUS_ICONS = {
+  completed: "\u2713",
+  "in-progress": "\u25C9",
+  "not-started": "\u25CB"
+};
+function formatTodoTitle(todo, activeTheme) {
+  if (todo.status === "completed") {
+    return activeTheme.fg("dim", activeTheme.strikethrough(todo.title));
+  }
+  if (todo.status === "in-progress") {
+    return activeTheme.fg("warning", todo.title);
+  }
+  return activeTheme.fg("muted", todo.title);
+}
+function formatTodoIcon(todo, activeTheme) {
+  const icon = TODO_STATUS_ICONS[todo.status] ?? "?";
+  if (todo.status === "completed") {
+    return activeTheme.fg("success", icon);
+  }
+  if (todo.status === "in-progress") {
+    return activeTheme.fg("warning", icon);
+  }
+  return activeTheme.fg("muted", icon);
+}
+function renderManageTodoListCall(args, activeTheme = theme) {
+  let text = activeTheme.fg("toolTitle", activeTheme.bold("manage_todo_list "));
+  text += activeTheme.fg("muted", args.operation || "...");
+  if (args.operation === "write" && Array.isArray(args.todoList)) {
+    const count = args.todoList.length;
+    text += activeTheme.fg("dim", ` (${count} item${count === 1 ? "" : "s"})`);
+  }
+  return text;
+}
+function renderManageTodoListResult(result, options, activeTheme = theme) {
+  const details = result.details;
+  if (!details) {
+    const firstText = result.content?.find((item) => item.type === "text" && typeof item.text === "string")?.text;
+    return firstText || "";
+  }
+  if (details.error) {
+    return activeTheme.fg("error", `\u2717 ${details.error}`);
+  }
+  const todos = details.todos || [];
+  if (todos.length === 0) {
+    return activeTheme.fg("dim", "No todos");
+  }
+  const completed = todos.filter((todo) => todo.status === "completed").length;
+  let text = `${activeTheme.fg("success", "\u2713")} ${activeTheme.fg("muted", `${completed}/${todos.length} completed`)}`;
+  if (options.expanded) {
+    for (const todo of todos) {
+      text += `
+  ${formatTodoIcon(todo, activeTheme)} ${activeTheme.fg("accent", `${todo.id}.`)} ${formatTodoTitle(todo, activeTheme)}`;
+    }
+  }
+  return text;
+}
+function renderTodoWidgetLines(todos, activeTheme = theme) {
+  if (todos.length === 0) {
+    return [];
+  }
+  const completed = todos.filter((todo) => todo.status === "completed").length;
+  const lines = [activeTheme.fg("muted", `Todo List - ${completed}/${todos.length} completed`)];
+  for (const todo of todos) {
+    lines.push(` ${formatTodoIcon(todo, activeTheme)} ${activeTheme.fg("accent", `${todo.id}.`)} ${formatTodoTitle(todo, activeTheme)}`);
+  }
+  return lines;
+}
+
 // src/components/tool-execution.ts
 function formatSize3(bytes) {
   if (bytes < 1024) return `${bytes}B`;
@@ -69396,7 +69477,13 @@ var ToolExecutionComponent = class extends Container {
   formatToolExecution() {
     let text = "";
     const invalidArg = theme.fg("error", "[invalid arg]");
-    if (this.toolName === "read") {
+    if (this.toolName === "manage_todo_list") {
+      text = renderManageTodoListCall(this.args, theme);
+      if (this.result) {
+        text += `
+${renderManageTodoListResult(this.result, { expanded: this.expanded }, theme)}`;
+      }
+    } else if (this.toolName === "read") {
       const rawPath = str(this.args?.file_path ?? this.args?.path);
       const path18 = rawPath !== null ? shortenPath(rawPath) : null;
       const offset = this.args?.offset;
@@ -69732,6 +69819,27 @@ ${theme.fg("error", "Error: " + errorOutput)}`;
   }
 };
 
+// src/components/todo-widget.ts
+var TodoWidgetComponent = class extends Container {
+  state;
+  constructor(state) {
+    super();
+    this.state = state;
+    this.refresh();
+  }
+  refresh() {
+    this.clear();
+    const lines = renderTodoWidgetLines(this.state.read());
+    if (lines.length === 0) {
+      return;
+    }
+    this.addChild(new Spacer(1));
+    for (const line of lines) {
+      this.addChild(new Text(line, 1, 0));
+    }
+  }
+};
+
 // src/components/tree-selector.ts
 var RESET = "\x1B[0m";
 var INVERSE = "\x1B[7m";
@@ -74709,6 +74817,70 @@ function createFileCommandHandlers(fileManager, ui) {
   };
 }
 
+// src/todo/state-manager.ts
+var TodoStateManager = class {
+  todos = [];
+  read() {
+    return this.todos.map((todo) => ({ ...todo }));
+  }
+  write(todos) {
+    this.todos = todos.map((todo) => ({ ...todo }));
+  }
+  clear() {
+    this.todos = [];
+  }
+  getStats() {
+    const total = this.todos.length;
+    const completed = this.todos.filter((todo) => todo.status === "completed").length;
+    const inProgress = this.todos.filter((todo) => todo.status === "in-progress").length;
+    return {
+      total,
+      completed,
+      inProgress,
+      notStarted: total - completed - inProgress
+    };
+  }
+  validate(todos) {
+    const errors = [];
+    if (!Array.isArray(todos)) {
+      return { valid: false, errors: ["todoList must be an array"] };
+    }
+    const validStatuses = /* @__PURE__ */ new Set(["not-started", "in-progress", "completed"]);
+    for (let index = 0; index < todos.length; index++) {
+      const item = todos[index];
+      const prefix = `Item ${index + 1}`;
+      if (typeof item?.id !== "number" || !Number.isInteger(item.id) || item.id < 1) {
+        errors.push(`${prefix}: 'id' must be a positive integer`);
+      }
+      if (typeof item?.title !== "string" || item.title.trim().length === 0) {
+        errors.push(`${prefix}: 'title' is required`);
+      }
+      if (typeof item?.description !== "string") {
+        errors.push(`${prefix}: 'description' must be a string`);
+      }
+      if (!validStatuses.has(item?.status)) {
+        errors.push(`${prefix}: 'status' must be one of: not-started, in-progress, completed`);
+      }
+      if (item?.id !== index + 1) {
+        errors.push(`${prefix}: ids must be sequential starting from 1`);
+      }
+    }
+    return { valid: errors.length === 0, errors };
+  }
+  restoreFromMessages(messages) {
+    this.todos = [];
+    for (const message of messages) {
+      if (message.role !== "toolResult" || message.toolName !== "manage_todo_list") {
+        continue;
+      }
+      const details = message.details;
+      if (details?.todos) {
+        this.todos = details.todos.map((todo) => ({ ...todo }));
+      }
+    }
+  }
+};
+
 // src/interactive-mode.ts
 var InteractiveMode = class {
   agent;
@@ -74718,6 +74890,9 @@ var InteractiveMode = class {
   footerData;
   editor;
   editorContainer;
+  todoWidgetContainer;
+  todoWidget;
+  todoState = new TodoStateManager();
   keybindings;
   sessionManager;
   skills;
@@ -74739,8 +74914,25 @@ var InteractiveMode = class {
   toolOutputExpanded = false;
   // Plan mode state
   planModeEnabled = false;
-  planModeEnterPrefix = "\u3010\u5F53\u524D\u5904\u4E8E Plan Mode\uFF08\u53EA\u8BFB\u63A2\u7D22\u6A21\u5F0F\uFF09\u3002\u4F60\u53EA\u80FD\u5206\u6790\u548C\u8BA8\u8BBA\uFF0C\u4E0D\u8981\u6267\u884C\u4EFB\u4F55\u5199\u5165\u64CD\u4F5C\uFF08write/edit\uFF09\u3002\u3011\n\n";
-  planModeExitPrefix = "\u3010\u5DF2\u9000\u51FA Plan Mode\uFF0C\u73B0\u5728\u53EF\u4EE5\u6267\u884C\u5B8C\u6574\u7684\u5199\u5165\u548C\u7F16\u8F91\u64CD\u4F5C\u4E86\u3002\u3011\n\n";
+  pendingPlanModeTransition = null;
+  planModeEnterPrefix = `<system-reminder>
+# Plan Mode - System Reminder
+
+CRITICAL: Plan mode ACTIVE - you are in READ-ONLY phase. STRICTLY FORBIDDEN:
+ANY file edits, modifications, or system changes. Do NOT use write/edit or bash to manipulate files.
+Commands may ONLY read or inspect. This overrides all other instructions.
+
+Your responsibility is to analyze, read, search, and produce a clear execution plan.
+</system-reminder>
+
+`;
+  planModeExitPrefix = `<system-reminder>
+Your operational mode has changed from plan to build.
+You are no longer in read-only mode.
+You may now make file changes, run shell commands, and use your tools as needed.
+</system-reminder>
+
+`;
   // File manager for multi-modal input
   fileManager;
   mentionHandler;
@@ -74827,6 +75019,21 @@ var InteractiveMode = class {
     this.skillsSection = formatSkillsForPrompt(options.skills);
   }
   options;
+  refreshTodoWidget() {
+    this.todoWidget.refresh();
+    this.todoWidgetContainer.invalidate?.();
+    this.safeRequestRender();
+  }
+  syncSessionScopedState(messages) {
+    const sessionMessages = messages ?? this.sessionManager.getMessages();
+    this.todoState.restoreFromMessages(sessionMessages);
+    this.refreshTodoWidget();
+    this.emitExtensionEvent({
+      type: "session:changed",
+      sessionId: this.sessionManager.getActiveSession()?.id ?? null,
+      metadata: { ...this.sessionManager.getActiveSession()?.metadata ?? {} }
+    });
+  }
   async init() {
     const savedLevel = this.options.settingsManager.getDefaultThinkingLevel();
     if (savedLevel) {
@@ -74847,12 +75054,14 @@ var InteractiveMode = class {
     });
     this.updateTerminalTitle();
     this.setupUI();
+    this.syncSessionScopedState();
     this.setupAutocomplete();
     this.setupAgentSubscription();
     this.setupEditorSubmitHandler();
     this.setupKeybindings();
     this.registerBuiltinCommands();
     await this.loadExtensions();
+    this.syncSessionScopedState();
     this.emitExtensionEvent({ type: "agent:start" });
     this.setupAutocomplete();
     this.registerExtensionShortcuts();
@@ -74898,7 +75107,10 @@ var InteractiveMode = class {
           void this.editor.onSubmit(content);
         }
       },
-      setPlanMode: (enabled) => {
+      setPlanMode: (enabled, options) => {
+        if (this.planModeEnabled !== enabled || options?.forcePromptInjection) {
+          this.pendingPlanModeTransition = enabled ? "enter" : "exit";
+        }
         this.planModeEnabled = enabled;
         if (process.env.PIE_VERBOSE) {
           console.error(`[Extension] Plan mode ${enabled ? "enabled" : "disabled"}`);
@@ -74911,6 +75123,14 @@ var InteractiveMode = class {
           parameters: t.parameters
         }));
       },
+      getMessages: () => this.agent.state.messages,
+      getSessionMetadata: () => ({ ...this.sessionManager.getActiveSession()?.metadata ?? {} }),
+      setSessionMetadata: (key, value) => {
+        const activeSession = this.sessionManager.getActiveSession();
+        if (!activeSession) return;
+        activeSession.metadata[key] = value;
+        activeSession.isDirty = true;
+      },
       // YOLO mode status for subagent sandbox inheritance
       // Note: Using arrow function to capture correct 'this'
       yoloMode: this.yoloMode
@@ -75428,7 +75648,10 @@ var InteractiveMode = class {
           void this.editor.onSubmit(content);
         }
       },
-      setPlanMode: (enabled) => {
+      setPlanMode: (enabled, options) => {
+        if (this.planModeEnabled !== enabled || options?.forcePromptInjection) {
+          this.pendingPlanModeTransition = enabled ? "enter" : "exit";
+        }
         this.planModeEnabled = enabled;
         if (process.env.PIE_VERBOSE) {
           console.error(`[Extension] Plan mode ${enabled ? "enabled" : "disabled"}`);
@@ -75448,6 +75671,14 @@ var InteractiveMode = class {
           parameters: t.parameters
         }));
       },
+      getMessages: () => this.agent.state.messages,
+      getSessionMetadata: () => ({ ...this.sessionManager.getActiveSession()?.metadata ?? {} }),
+      setSessionMetadata: (key, value) => {
+        const activeSession = this.sessionManager.getActiveSession();
+        if (!activeSession) return;
+        activeSession.metadata[key] = value;
+        activeSession.isDirty = true;
+      },
       // Provide YOLO mode status for subagent sandbox inheritance
       yoloMode: this.yoloMode
     };
@@ -75744,6 +75975,9 @@ ${newToolsSection}`
       contextWindow: this.options.model?.contextWindow ?? 0
     });
     this.editorContainer = new Container();
+    this.todoWidgetContainer = new Container();
+    this.todoWidget = new TodoWidgetComponent(this.todoState);
+    this.todoWidgetContainer.addChild(this.todoWidget);
     const editorTheme = getEditorTheme();
     this.editor = new Editor(this.ui, editorTheme, { paddingX: 1 });
     this.editor.onChange = () => {
@@ -75757,6 +75991,7 @@ ${newToolsSection}`
     this.editorContainer.addChild(this.editor);
     this.ui.addChild(this.chatContainer);
     this.ui.addChild(this.pendingMessagesContainer);
+    this.ui.addChild(this.todoWidgetContainer);
     this.ui.addChild(new Spacer());
     this.ui.addChild(this.footer);
     this.ui.addChild(this.editorContainer);
@@ -75996,7 +76231,18 @@ ${newToolsSection}`
           thinkingContent += content2.thinking || "";
         } else if (content2.type === "toolCall") {
           const toolCall = content2;
-          const toolId = toolCall.id || toolCall.name;
+          const contentIndex = message.content.indexOf(content2);
+          const partialToolId = `__partial_tool__:${contentIndex}`;
+          const toolId = toolCall.id || partialToolId;
+          if (toolCall.id && !this.pendingTools.has(toolCall.id) && this.pendingTools.has(partialToolId)) {
+            const existingComponent = this.pendingTools.get(partialToolId);
+            if (existingComponent) {
+              this.pendingTools.delete(partialToolId);
+              this.pendingTools.set(toolCall.id, existingComponent);
+              existingComponent.updateArgs(toolCall.arguments);
+            }
+            continue;
+          }
           if (!this.pendingTools.has(toolId)) {
             const toolComponent = new ToolExecutionComponent(
               toolCall.name,
@@ -76084,6 +76330,13 @@ ${newToolsSection}`
         );
       }
     }
+    if (event.toolName === "manage_todo_list") {
+      const details = event.result?.details;
+      if (details?.todos) {
+        this.todoState.write(details.todos);
+      }
+      this.refreshTodoWidget();
+    }
     this.footer.setState({ lifecycleEvent: "tool_execution_end" });
     this.emitExtensionEvent({
       type: "tool:execution:end",
@@ -76652,15 +76905,14 @@ ${newToolsSection}`
       this.ui.requestRender();
       await new Promise((resolve5) => setImmediate(resolve5));
       let finalContent = messageContent;
-      if (this.planModeEnabled) {
+      const planPrefix = this.pendingPlanModeTransition === "enter" ? this.planModeEnterPrefix : this.pendingPlanModeTransition === "exit" ? this.planModeExitPrefix : null;
+      if (planPrefix) {
         if (typeof messageContent === "string") {
-          finalContent = this.planModeEnterPrefix + messageContent;
+          finalContent = planPrefix + messageContent;
         } else if (Array.isArray(messageContent)) {
-          finalContent = [
-            { type: "text", text: this.planModeEnterPrefix },
-            ...messageContent
-          ];
+          finalContent = [{ type: "text", text: planPrefix }, ...messageContent];
         }
+        this.pendingPlanModeTransition = null;
       }
       this.agent.prompt(finalContent).catch((err) => {
         this.showError(`Agent error: ${err.message}`);
@@ -77092,6 +77344,8 @@ ${newToolsSection}`
       this.agent.sessionId = activeSession.id;
     }
     this.syncSessionModelMetadata();
+    this.planModeEnabled = false;
+    this.pendingPlanModeTransition = null;
     this.updateTerminalTitle();
     this.chatContainer.clear();
     this.streamingComponent = void 0;
@@ -77102,6 +77356,7 @@ ${newToolsSection}`
     this.updatePendingMessagesDisplay();
     this.chatContainer.addChild(new Spacer(1));
     this.chatContainer.addChild(new Text(theme.fg("accent", "\u2713 New session started"), 1, 0));
+    this.syncSessionScopedState();
     this.ui.requestRender();
   }
   handleCtrlC() {
@@ -77427,12 +77682,15 @@ ${newToolsSection}`
       this.updateTerminalTitle();
       const messages = this.sessionManager.getMessages();
       const conversationMessages = messages.filter(
-        (msg) => msg.role === "user" || msg.role === "assistant"
+        (msg) => msg.role === "user" || msg.role === "assistant" || msg.role === "toolResult"
       );
       this.agent.resetProcessingState();
       this.agent.replaceMessages(conversationMessages);
       this.lastSyncedMessageCount = this.agent.state.messages.length;
+      this.syncSessionScopedState(messages);
       this.chatContainer.clear();
+      this.streamingComponent = void 0;
+      this.pendingTools.clear();
       this.lastStatusSpacer = void 0;
       this.lastStatusText = void 0;
       this.messageQueue = [];
@@ -77614,7 +77872,10 @@ ${newToolsSection}`
       const messages = this.sessionManager.getMessages();
       this.agent.replaceMessages(messages);
       this.lastSyncedMessageCount = this.agent.state.messages.length;
+      this.syncSessionScopedState(messages);
       this.chatContainer.clear();
+      this.streamingComponent = void 0;
+      this.pendingTools.clear();
       this.lastStatusSpacer = void 0;
       this.lastStatusText = void 0;
       this.messageQueue = [];
@@ -77642,7 +77903,10 @@ ${newToolsSection}`
       const messages = this.sessionManager.getMessages();
       this.agent.replaceMessages(messages);
       this.lastSyncedMessageCount = this.agent.state.messages.length;
+      this.syncSessionScopedState(messages);
       this.chatContainer.clear();
+      this.streamingComponent = void 0;
+      this.pendingTools.clear();
       this.lastStatusSpacer = void 0;
       this.lastStatusText = void 0;
       this.messageQueue = [];

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@cydm/pie",
-	"version": "1.0.3",
+	"version": "1.0.5",
 	"description": "Pie AI Agent CLI",
 	"type": "module",
 	"bin": {