@ia-ccun/code-agent-cli 0.0.1

package/config/agent/extensions/custom-footer.ts

@@ -0,0 +1,170 @@
+/**
+ * Custom Footer Extension — Single-line compact powerline style
+ *
+ * ~/path (branch) │ ↑in ↓out $cost │ 42%/200k │ ⚡ model • thinking
+ * [gold]  [dim]       [dim]       [color ctx]  [accent]
+ */
+
+import type { AssistantMessage } from "@mariozechner/pi-ai";
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { truncateToWidth, visibleWidth } from "@mariozechner/pi-tui";
+
+function fmtTokens(n: number): string {
+	if (n < 1000) return n.toString();
+	if (n < 10000) return `${(n / 1000).toFixed(1)}k`;
+	if (n < 1000000) return `${Math.round(n / 1000)}k`;
+	return `${(n / 1000000).toFixed(1)}M`;
+}
+
+// Catppuccin Mocha Yellow — #f9e2af
+const gold = (s: string) => `\x1b[38;2;166;227;161m${s}\x1b[0m`;
+
+export default function (pi: ExtensionAPI) {
+	pi.on("session_start", async (_event, ctx) => {
+		ctx.ui.setFooter((tui, theme, footerData) => {
+			const unsub = footerData.onBranchChange(() => tui.requestRender());
+
+			return {
+				dispose: unsub,
+				invalidate() {},
+				render(width: number): string[] {
+					const sep = theme.fg("dim", " │ ");
+					const sepWidth = 3; // " │ "
+
+					// ── Segment 1: project name (branch) ─────────────────────
+					const pwd = process.cwd();
+					const projectName = pwd.split('/').pop() || "project";
+					const branch = footerData.getGitBranch();
+					const pathRaw = projectName + (branch ? ` (${branch})` : "");
+
+					// ── Segment 2: token stats ──────────────────────────────
+					let totalIn = 0,
+						totalOut = 0,
+						totalCost = 0;
+					for (const e of ctx.sessionManager.getBranch()) {
+						if (e.type === "message" && e.message.role === "assistant") {
+							const m = e.message as AssistantMessage;
+							totalIn += m.usage.input;
+							totalOut += m.usage.output;
+							totalCost += m.usage.cost.total;
+						}
+					}
+					const statsParts: string[] = [];
+					if (totalIn || totalOut) {
+						statsParts.push(`↑${fmtTokens(totalIn)}`);
+						statsParts.push(`↓${fmtTokens(totalOut)}`);
+					}
+					if (totalCost) statsParts.push(`$${totalCost.toFixed(2)}`);
+					const statsRaw = statsParts.join(" ");
+					const totalTokens = totalIn + totalOut;
+
+					// ── Segment 3: context % ────────────────────────────────
+					const usage = ctx.getContextUsage();
+					const pct = usage?.percent ?? 0;
+					const win = usage?.contextWindow ?? ctx.model?.contextWindow ?? 0;
+					const ctxRaw = `${pct.toFixed(0)}%/${fmtTokens(win)}`;
+
+					// ── Segment 4: model + thinking ─────────────────────────────
+					const modelName = ctx.model?.id || "no-model";
+					const thinking = pi.getThinkingLevel();
+					const thinkSuffix = thinking !== "off" ? ` • ${thinking}` : "";
+					const modelRaw = `⚡ ${modelName}${thinkSuffix}`;
+
+					// ── Assemble: figure out what fits ──────────────────────
+					// Always show: path | ctx | model. Stats only if room.
+					const fixedRight = ctxRaw + modelRaw;
+					const fixedRightWidth = visibleWidth(fixedRight);
+					const sepsNeeded = statsRaw ? 3 : 2; // number of separators
+					const sepsWidth = sepsNeeded * sepWidth;
+
+					// Available width for path after reserving right segments
+					const rightBlockWidth = (statsRaw ? visibleWidth(statsRaw) + sepWidth : 0)
+						+ visibleWidth(ctxRaw) + sepWidth + visibleWidth(modelRaw);
+					const pathBudget = width - rightBlockWidth - sepWidth; // -sepWidth for first separator
+
+					let pathDisplay: string;
+					if (pathBudget >= 10) {
+						// Enough room for path
+						if (visibleWidth(pathRaw) <= pathBudget) {
+							pathDisplay = gold(pathRaw);
+						} else {
+							// Truncate from the left: …ail/path (branch)
+							const truncated = "…" + pathRaw.slice(-(pathBudget - 1));
+							pathDisplay = gold(truncated);
+						}
+					} else {
+						// Very narrow — skip path entirely
+						pathDisplay = "";
+					}
+
+					// Context coloring
+					let ctxColored: string;
+					if (pct > 70) ctxColored = theme.fg("error", ctxRaw);
+					else if (pct > 60) ctxColored = theme.fg("warning", ctxRaw);
+					else ctxColored = theme.fg("success", ctxRaw);
+
+					// Thinking level coloring — semaphore, hardcoded to decouple from border
+					const thinkingColors: Record<string, string> = {
+						off:     "\x1b[38;2;51;45;62m",   // surface1 — invisible
+						minimal: "\x1b[38;2;67;61;78m",   // surface2 — barely there
+						low:     "\x1b[38;2;166;227;161m", // green
+						medium:  "\x1b[38;2;137;180;250m", // blue
+						high:    "\x1b[38;2;250;179;135m", // peach
+						xhigh:   "\x1b[38;2;243;139;168m", // red
+					};
+					const thinkAnsi = thinkingColors[thinking] ?? thinkingColors.off;
+					let modelColored = theme.fg("accent", `⚡ ${modelName}`);
+					if (thinking !== "off") {
+						modelColored += theme.fg("dim", " • ") + `${thinkAnsi}${thinking}\x1b[0m`;
+					}
+
+					// Blue (#89B4FA) for token stats
+					const blue = (s: string) => `\x1b[38;2;137;180;250m${s}\x1b[0m`;
+					const statsColored = statsRaw ? blue(statsRaw) : "";
+
+					// quots
+					const quotes = [
+                        "代码是有灵魂的",
+                        "敬畏生产，双人复核，稳妥变更",
+                        "让代码更有温度",
+                        "民生所系，支付所向",
+                        "智慧风控，安全同行",
+                        "绿色支付，低碳生活",
+						"保持热爱，奔赴山海",
+						"每一行代码都是故事",
+						"技术只是手段，创意才是灵魂",
+						"安全为基，稳健致远",
+						"科技赋能，智慧支付",
+						"普惠金融，温暖万家",
+						"客户为本，体验至上",
+						"责任在肩，守护每笔",
+						"创新驱动，价值共生",
+						"诚信立本，信誉致远",
+						"高效清算，畅通无忧",
+						"支付协同，生态共赢",
+						"支付有温度，服务有深度",
+						"匠心运维，精益求精",
+						"数字纽带，连接美好",
+						"信任所系，使命必达",
+						"稳中求进，变中求效",
+						"跨境无界，体验无痕",
+						"数据守护，隐私无忧",
+						"同心筑梦，支付同行",
+					];
+					const quote = quotes[Math.floor(totalTokens / 10000) % quotes.length];
+
+					// Build the line
+					const segments: string[] = [];
+					if (pathDisplay) segments.push(pathDisplay);
+					if (statsColored) segments.push(statsColored);
+					segments.push(ctxColored);
+					segments.push(modelColored);
+					segments.push(theme.fg("muted", quote));
+
+					const joined = segments.join(sep);
+					return [truncateToWidth(joined, width)];
+				},
+			};
+		});
+	});
+}

package/config/agent/extensions/custom.ts

@@ -0,0 +1,289 @@
+/**
+ * 自定义扩展示例
+ *
+ * 放置位置: config/agent/extensions/
+ * 安装后会复制到 ~/.pi/agent/extensions/
+ *
+ * 可以拦截内置工具、注册自定义工具和命令
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import type { AssistantMessage } from "@mariozechner/pi-ai";
+import { truncateToWidth, visibleWidth } from "@mariozechner/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) => {
+    // 计算耗时
+    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`;
+    }
+
+    // 格式化时间
+    const now = new Date();
+    const timeStr2 = `${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");
+  });
+
+  // === 拦截工具调用 ===
+
+  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 功能已禁用" };
+    }
+
+    // 拦截其他工具
+    if (event.toolName === "bash") {
+      const command = event.input.command;
+      if (command.includes("rm -rf")) {
+        // 解析要删除的路径
+        const match = command.match(/rm\s+-rf\s+('[^']+'|"[^"]+"|\S+)/g);
+        if (match) {
+          const paths = match.map(m => m.replace(/^rm\s+-rf\s+/, '').replace(/^['"]|['"]$/g, ''));
+
+          // 尝试列出文件
+          const { readdirSync, statSync, existsSync } = await import("fs");
+          let fileList: string[] = [];
+          for (const p of paths) {
+            try {
+              if (existsSync(p)) {
+                const stat = statSync(p);
+                if (stat.isDirectory()) {
+                  const files = readdirSync(p);
+                  fileList.push(`📁 ${p}/ (${files.length} 个文件)`);
+                } else {
+                  fileList.push(`📄 ${p}`);
+                }
+              }
+            } catch (e) { /* 忽略读取错误 */ }
+          }
+
+          const msg = fileList.length > 0
+            ? `将删除以下文件/目录:\n${fileList.join('\n')}`
+            : '确定要执行此删除命令吗？';
+
+          // 第一次确认
+          const firstConfirm = await ctx.ui.confirm("🔴 危险操作 - rm -rf", msg);
+          if (!firstConfirm) {
+            return { block: true, reason: "被用户拒绝" };
+          }
+
+          // 二次确认
+          const secondConfirm = await ctx.ui.confirm("🔴 二次确认 - 不可恢复", "确定要继续删除吗？此操作不可恢复！");
+          if (!secondConfirm) {
+            return { block: true, reason: "被用户二次拒绝" };
+          }
+        }
+      }
+    }
+  });
+
+  // === 注册自定义命令 ===
+
+  pi.registerCommand("hello", {
+    description: "打招呼命令",
+    usage: "/hello [名字]",
+    handler: async (args, ctx) => {
+      const name = args || "大佬，您吃了吗";
+      ctx.ui.notify(`你好，${name}！🤝`, "info");
+    },
+  });
+
+  pi.registerCommand("status", {
+    description: "显示当前会话状态",
+    handler: async (_args, ctx) => {
+      const info = `
+📊 会话状态
+
+- 当前模型: ${ctx.session.model?.name || "未知"}
+- 项目目录: ${ctx.session.workspace || "未知"}
+- 消息数量: ${ctx.session.messages?.length || 0}
+      `.trim();
+      ctx.ui.notify(info, "info");
+    },
+  });
+
+  pi.registerCommand("clear", {
+    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;
+      }
+
+      // 创建新会话来清空上下文
+      const result = await ctx.newSession({
+        setup: async (_sm) => {
+          // 新会话默认为空
+        },
+      });
+
+      if (result.cancelled) {
+        ctx.ui.notify("已取消", "info");
+        return;
+      }
+
+      ctx.ui.notify(`✅ 已清理 ${msgCount} 条消息，进入新会话`, "success");
+    },
+  });
+
+  // === 注册 /exit 命令 ===
+
+  pi.registerCommand("exit", {
+    description: "退出当前会话",
+    usage: "/exit",
+    handler: async (_args, ctx) => {
+      ctx.ui.notify("👋 再见！", "info");
+      ctx.shutdown();
+    },
+  });
+
+  // === 注册 /update 命令 ===
+
+  pi.registerCommand("update", {
+    description: "更新 CLI 工具",
+    usage: "/update",
+    handler: async (_args, ctx) => {
+      const options = [
+        "更新自己 (aicode-cli)",
+        "更新依赖 agent",
+      ];
+
+      // 让用户选择
+      const selected = await ctx.ui.select("选择更新内容", options);
+      if (!selected) {
+        ctx.ui.notify("已取消", "info");
+        return;
+      }
+
+      if (selected.includes("aicode-cli")) {
+        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/aicode-cli", (error, stdout, stderr) => {
+          if (error) {
+            ctx.ui.notify(`❌ 更新失败: ${error.message}，请重试`, "error");
+          } else {
+            ctx.ui.notify("✅ aicode-cli 更新成功！", "success");
+          }
+        });
+      } else if (selected.includes("agent")) {
+        const ok = await ctx.ui.confirm("更新确认", "确定要更新依赖的agent 吗？");
+        if (!ok) {
+          ctx.ui.notify("已取消", "info");
+          return;
+        }
+        ctx.ui.notify("🔄 正在更新 ...", "info");
+        const { exec } = await import("child_process");
+        exec("npm install -g @mariozechner/pi-coding-agent", (error, stdout, stderr) => {
+          if (error) {
+            ctx.ui.notify(`❌ 更新失败: ${error.message}，请重试`, "error");
+          } else {
+            ctx.ui.notify("✅ 依赖的 agent 更新成功！", "success");
+          }
+        });
+      }
+    },
+  });
+
+  // === 注册自定义工具 ===
+
+  // 注意: 注册自定义工具需要更复杂的参数定义
+  // 详情请参考: https://github.com/mariozechner/pi-coding-agent/docs/extensions.md
+
+  // === 注册 /version 命令 ===
+
+  pi.registerCommand("version", {
+    description: "查看当前版本和最新版本",
+    usage: "/version",
+    handler: async (_args, ctx) => {
+      const { execSync } = await import("child_process");
+      const pkgPath = "./package.json";
+      const { readFileSync } = await import("fs");
+
+      // 读取本地版本
+      let localVersion = "未知";
+      try {
+        const pkg = JSON.parse(readFileSync(pkgPath, "utf-8"));
+        localVersion = pkg.version;
+      } catch (e) {
+        // 忽略
+      }
+
+      // 获取最新版本
+      let latestVersion = "未知";
+      let updateMsg = "";
+      try {
+        const registry = execSync("npm config get registry", { encoding: "utf8" }).trim();
+        latestVersion = execSync(`npm view @ia-ccun/code-agent-cli version --registry=${registry}`, { encoding: "utf8" }).trim();
+
+        // 比较版本
+        const localParts = localVersion.split(".").map(Number);
+        const latestParts = latestVersion.split(".").map(Number);
+        const needsUpdate = latestParts[0] > localParts[0] ||
+                          (latestParts[0] === localParts[0] && latestParts[1] > localParts[1]) ||
+                          (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`;
+        } else {
+          updateMsg = "\n\n✅ 已是最新版本";
+        }
+      } catch (e) {
+        updateMsg = "\n\n❌ 最新版本检查异常，请稍后重试";
+      }
+
+      const info = `
+📦 版本信息
+
+   当前版本: v${localVersion}
+   最新版本: v${latestVersion}
+${updateMsg}
+      `.trim();
+
+      ctx.ui.notify(info, "info");
+    },
+  });
+}