@ia-ccun/code-agent-cli 0.0.52 → 0.0.54-beta.1

package/config/agent/extensions/custom.ts

@@ -2,7 +2,7 @@
  * @author: xujianjiang
  * 自定义扩展示例
  *
- * 放置位置: config/agent/extensions/
+ * 放置位置：config/agent/extensions/
  * 安装后会复制到 ~/.pi/agent/extensions/
  *
  * 可以拦截内置工具、注册自定义工具和命令
@@ -10,82 +10,66 @@
 
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import type { AssistantMessage } from "@earendil-works/pi-ai";
-import { truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
 
 export default function (pi: ExtensionAPI) {
   // === 记录开始时间 ===
   let agentStartTime = 0;
 
-  pi.on("agent_start", async () => {
-    agentStartTime = Date.now();
-  });
-
-  // === 消息完成通知（只在最后一条显示）===
+  // === 拦截工具调用 ===
 
-  pi.on("agent_end", async (event, ctx) => {
+  pi.on("tool_call", async (event, ctx) => {
     try {
-      // 计算耗时
-      const elapsed = Date.now() - agentStartTime;
-      let timeStr: string;
-      if (elapsed < 1000) {
-        timeStr = `${elapsed}ms`;
-      } else if (elapsed < 60000) {
-        timeStr = `${(elapsed / 1000).toFixed(1)}s`;
-      } else {
-        const mins = Math.floor(elapsed / 60000);
-        const secs = Math.floor((elapsed % 60000) / 1000);
-        timeStr = `${mins}m ${secs}s`;
+      // 拦截 login 工具 - 阻止执行
+      if (event.toolName === "login") {
+        await ctx.ui.notify("⚠️ login 功能已禁用", "warning");
+        return { block: true, reason: "login 功能已禁用" };
       }
 
-      // 格式化完整时间
-      const now = new Date();
-      const timeStr2 = `${now.getFullYear()}-${String(now.getMonth() + 1).padStart(2, '0')}-${String(now.getDate()).padStart(2, '0')} ${String(now.getHours()).padStart(2, '0')}:${String(now.getMinutes()).padStart(2, '0')}:${String(now.getSeconds()).padStart(2, '0')}`;
-
-      // agent_end 表示一轮对话结束，显示完成提示
-      ctx.ui.notify(`${timeStr2} | ✨ 消息处理完成 (${timeStr})`, "info");
+      // 拦截 logout 工具 - 阻止执行
+      if (event.toolName === "logout") {
+        await ctx.ui.notify("⚠️ logout 功能已禁用", "warning");
+        return { block: true, reason: "logout 功能已禁用" };
+      }
     } catch (e) {
       // 忽略错误（ctx 可能已过期）
     }
   });
 
-  // === 拦截工具调用 ===
-
-  pi.on("tool_call", async (event, ctx) => {
-    // 拦截 login 工具 - 阻止执行
-    if (event.toolName === "login") {
-      await ctx.ui.notify("⚠️ login 功能已禁用", "warning");
-      return { block: true, reason: "login 功能已禁用" };
-    }
-
-    // 拦截 logout 工具 - 阻止执行
-    if (event.toolName === "logout") {
-      await ctx.ui.notify("⚠️ logout 功能已禁用", "warning");
-      return { block: true, reason: "logout 功能已禁用" };
-    }
-  });
-
   // === 注册自定义命令 ===
 
   pi.registerCommand("hello", {
     description: "打招呼命令",
     usage: "/hello [名字]",
     handler: async (args, ctx) => {
-      const name = args || "大佬，您吃了吗";
-      ctx.ui.notify(`你好，${name}！🤝`, "info");
+      try {
+        const name = args || "大佬，您吃了吗";
+        ctx.ui.notify(`你好，${name}！🤝`, "info");
+      } catch (e) {
+        // 忽略错误
+      }
     },
   });
 
   pi.registerCommand("status", {
     description: "显示当前会话状态",
     handler: async (_args, ctx) => {
-      const info = `
+      try {
+        // 保存需要的数据，避免多次触发 getter
+        const model = ctx.session.model?.name || "未知";
+        const workspace = ctx.session.workspace || "未知";
+        const messages = ctx.session.messages?.length || 0;
+        
+        const info = `
 📊 会话状态
 
-- 当前模型: ${ctx.session.model?.name || "未知"}
-- 项目目录: ${ctx.session.workspace || "未知"}
-- 消息数量: ${ctx.session.messages?.length || 0}
+- 当前模型：${model}
+- 项目目录：${workspace}
+- 消息数量：${messages}
       `.trim();
-      ctx.ui.notify(info, "info");
+        ctx.ui.notify(info, "info");
+      } catch (e) {
+        // 忽略错误（ctx 可能已过期）
+      }
     },
   });
 
@@ -93,35 +77,33 @@ export default function (pi: ExtensionAPI) {
     description: "清理当前会话上下文",
     usage: "/clear",
     handler: async (_args, ctx) => {
-      // 获取当前会话条目数
-      const entries = ctx.sessionManager.getBranch();
-      const msgCount = entries.filter(e => e.type === "message").length;
-
-      if (msgCount === 0) {
-        ctx.ui.notify("会话已经是空的", "info");
-        return;
-      }
-
-      // 确认清理
-      const ok = await ctx.ui.confirm("清理会话", `确定要清理当前会话的所有 ${msgCount} 条消息吗？`);
-      if (!ok) {
-        ctx.ui.notify("已取消", "info");
-        return;
-      }
+      try {
+        // 获取当前会话条目数（在 await 前完成所有 ctx 访问）
+        const entries = ctx.sessionManager.getBranch();
+        const msgCount = entries.filter(e => e.type === "message").length;
+        const uiContext = ctx.ui;  // ← 保存 UI context
+
+        if (msgCount === 0) {
+          uiContext.notify("会话已经是空的", "info");
+          return;
+        }
 
-      // 创建新会话来清空上下文
-      const result = await ctx.newSession({
-        setup: async (_sm) => {
-          // 新会话默认为空
-        },
-      });
+        // 确认清理
+        const ok = await uiContext.confirm("清理会话", `确定要清理当前会话的所有 ${msgCount} 条消息吗？`);
+        if (!ok) {
+          uiContext.notify("已取消", "info");
+          return;
+        }
 
-      if (result.cancelled) {
-        ctx.ui.notify("已取消", "info");
-        return;
+        // 创建新会话来清空上下文
+        await ctx.newSession({
+          withSession: async (newCtx) => {
+            newCtx.ui.notify(`✅ 已清理 ${msgCount} 条消息，进入新会话`, "success");
+          },
+        });
+      } catch (e) {
+        // 忽略错误（ctx 可能已过期）
       }
-
-      ctx.ui.notify(`✅ 已清理 ${msgCount} 条消息，进入新会话`, "success");
     },
   });
 
@@ -131,8 +113,12 @@ export default function (pi: ExtensionAPI) {
     description: "退出当前会话",
     usage: "/exit",
     handler: async (_args, ctx) => {
-      ctx.ui.notify("👋 再见！", "info");
-      ctx.shutdown();
+      try {
+        ctx.ui.notify("👋 再见！", "info");
+        ctx.shutdown();
+      } catch (e) {
+        // 忽略错误
+      }
     },
   });
 
@@ -142,28 +128,40 @@ export default function (pi: ExtensionAPI) {
     description: "更新 aicode-cli 工具",
     usage: "/update",
     handler: async (_args, ctx) => {
-      const ok = await ctx.ui.confirm("更新确认", "确定要更新 aicode-cli 吗？");
-      if (!ok) {
-        ctx.ui.notify("已取消", "info");
-        return;
-      }
-
-      ctx.ui.notify("🔄 正在更新 aicode-cli...", "info");
-      const { exec } = await import("child_process");
-      exec("npm install -g @ia-ccun/code-agent-cli", (error, stdout, stderr) => {
-        if (error) {
-          ctx.ui.notify(`❌ 更新失败: ${error.message}，请重试`, "error");
-        } else {
-          ctx.ui.notify("✅ aicode-cli 更新成功！请重启 CLI", "success");
+      try {
+        // ← 在 await 前保存 UI context
+        const uiContext = ctx.ui;
+        
+        const ok = await uiContext.confirm("更新确认", "确定要更新 aicode-cli 吗？");
+        if (!ok) {
+          uiContext.notify("已取消", "info");
+          return;
         }
-      });
+
+        uiContext.notify("🔄 正在更新 aicode-cli...", "info");
+        
+        // 使用 Promise 包装 exec
+        const { exec } = await import("child_process");
+        await new Promise<void>((resolve) => {
+          exec("npm install -g @ia-ccun/code-agent-cli", (error, stdout, stderr) => {
+            if (error) {
+              uiContext.notify(`❌ 更新失败：${error.message}，请重试`, "error");
+            } else {
+              uiContext.notify("✅ aicode-cli 更新成功！请重启 CLI", "success");
+            }
+            resolve();
+          });
+        });
+      } catch (e) {
+        // 忽略错误（ctx 可能已过期）
+      }
     },
   });
 
   // === 注册自定义工具 ===
 
-  // 注意: 注册自定义工具需要更复杂的参数定义
-  // 详情请参考: https://github.com/mariozechner/pi-coding-agent/docs/extensions.md
+  // 注意：注册自定义工具需要更复杂的参数定义
+  // 详情请参考：https://github.com/mariozechner/pi-coding-agent/docs/extensions.md
 
   // === 注册 /version 命令 ===
 
@@ -204,7 +202,7 @@ export default function (pi: ExtensionAPI) {
                           (latestParts[0] === localParts[0] && latestParts[1] === localParts[1] && latestParts[2] > localParts[2]);
 
         if (needsUpdate) {
-          updateMsg = `\n\n⚠️  发现新版本 v${localVersion} → v${latestVersion}\n   运行: npm install -g @ia-ccun/code-agent-cli`;
+          updateMsg = `\n\n⚠️  发现新版本 v${localVersion} → v${latestVersion}\n   运行：npm install -g @ia-ccun/code-agent-cli`;
         } else {
           updateMsg = "\n\n✅ 已是最新版本";
         }
@@ -215,8 +213,8 @@ export default function (pi: ExtensionAPI) {
       const info = `
 📦 版本信息
 
-   当前版本: v${localVersion}
-   最新版本: v${latestVersion}
+   当前版本：v${localVersion}
+   最新版本：v${latestVersion}
 ${updateMsg}
       `.trim();
 

package/config/agent/extensions/error-monitor.ts

@@ -122,22 +122,5 @@ export default function (pi: ExtensionAPI) {
 		}
 	});
 
-	// Summary on session shutdown
-	pi.on("session_shutdown", () => {
-		const uptime = Math.round((Date.now() - startTime) / 1000);
-		const totalErrors = toolErrorCount + bashErrorCount;
-
-		if (totalErrors > 0) {
-			console.log(`
-━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
-📊 Error Monitor 会话统计
-━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
-总错误数: ${totalErrors}
-工具错误数: ${toolErrorCount}
-Bash 错误数: ${bashErrorCount}
-运行时长: ${uptime}秒
-━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
-`);
-		}
-	});
+	// Session shutdown - silent (errors already logged to file)
 }