@ia-ccun/code-agent-cli 0.0.2 → 0.0.4-8.beta.1

package/config/agent/extensions/aicode-tips.ts

@@ -0,0 +1,191 @@
+/**
+ * @author: xujianjiang
+ * Aicode Tips Extension — 在边框下方、状态栏上方随机显示使用技巧
+ *
+ * 显示 10 秒后消失，3 秒后显示下一条
+ * 技巧列表包含快捷键、命令使用及联系方式提示
+ */
+
+import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+
+const AICODE_TIPS = [
+  "@ 快速搜索项目文件",
+  "Ctrl+L 快速切换模型",
+  "Ctrl+P/Shift+Ctrl+P 切换范围模型",
+  "Shift+Tab 切换思考级别",
+  "Ctrl+O 展开/折叠工具输出",
+  "Ctrl+T 展开/折叠思考块",
+  "Alt+Enter 队列跟进消息",
+  "Ctrl+V 粘贴图片",
+  "Shift+Enter 输入多行",
+  "/ 查看所有命令",
+  "/tree 查看会话历史",
+  "/session 查看统计",
+  "/settings 自定义设置",
+  "/export 导出会话为HTML",
+  "/copy 复制最后消息",
+  "/name 设置会话名称",
+  "/hotkeys 显示所有快捷键",
+  "/reload 重新加载配置文件",
+  "/new 开始新会话",
+  "!cmd 执行 bash",
+  "!!cmd 执行bash但不发送给AI",
+  "/compact 压缩上下文",
+  "/skill:name 调用技能",
+  "/resume 恢复会话",
+  "/fork 创建新分支",
+  "Ctrl+C 清空编辑器",
+  "双击Esc 打开树视图",
+  "拖拽文件到终端附加",
+  // 扩展命令提示
+  "/version 查看当前版本",
+  "/update 更新 aicode-cli",
+  "/status 显示会话状态",
+  "/clear 清理当前会话",
+  "/quit 退出当前会话",
+  // 常用Prompt提示
+  "/code-simplifier 自动简化刚修改的代码，提升清晰度和一致性",
+  "/feat 新功能开发，支持需求规划→讨论迭代→执行实施完整流程（推荐使用）",
+  "/git-commit 分析Git改动并生成Conventional Commits规范提交信息（推荐使用）",
+  "/init-project 初始化项目AI上下文，生成根级和模块级CLAUDE.md索引",
+  "/workflow 专业开发者6阶段结构化工作流（研究→构思→计划→执行→优化→评审）",
+  // 常用技能提示
+  "/skill:brainstorming 探索用户意图、需求分析和方案设计",
+  "/skill:code-review 执行全面的代码质量/安全/性能审查（推荐使用）",
+  "/skill:commit 分析代码变更并生成规范的Git提交信息",
+  "/skill:frontend-design 创建高质量生产级前端界面（推荐使用）",
+  "/skill:owasp-security 遵循OWASP Top 10的安全编码实践（推荐使用）",
+  "/skill:receiving-code-review 技术性处理代码评审反馈并验证（推荐使用）",
+  "/skill:requesting-code-review 请求代码评审确保满足合并要求",
+  "/skill:unit-test 为选定的代码文件生成单元测试",
+  "/skill:verification-before-completion 完成前验证实际工作是否完整",
+  "/skill:writing-plans 为多步骤任务编写详细的实施计划（推荐使用）",
+  "/skill:writing-skills 创建、修改和优化Agent技能并测量性能", 
+  "/skill:session-learning 总结会话经验并沉淀为可复用的技能",
+  "/skill:test-driven-development 先写测试再写实现的TDD开发模式",
+  "/skill:subagent-driven-development 通过派发子Agent执行复杂实施计划",
+  "/skill:dispatching-parallel-agents 并行分发多个独立任务无共享状态",
+  "/skill:csv-data-summarizer 分析CSV文件并生成统计和可视化图表",
+  "/skill:executing-plans 执行带有检查点的书面实施计划",
+  "/skill:frontend-slides 使用HTML/CSS创建精美的演示幻灯片",
+  "/skill:markdown-to-html 将Markdown文档转换为精美的HTML页面",
+  "/skill:refactor 重构代码提升可读性和可维护性保持功能不变",
+  "/skill:skill-creator 创建、修改和提升技能性能的工具",
+  "/skill:ui-ux-pro-max 提供50种设计风格和21种配色方案的设计助手（推荐使用）",
+  // 常用技能提示
+  "/btw 并行侧边对话，可在不中断主流程的情况下进行临时讨论或查询 ",
+  "/rewind 回退到任意工具执行前的状态，支持快照恢复和撤销重做 ",
+  "/mcp 连接现有的MCP服务器(如xx平台、企业微信等等)，扩展AI能力支持多种外部工具和服务 ",
+  // 消息传递模式提示
+  "Enter 插话消息 - 在工具执行后立即发送，中断AI思考流程",
+  "Alt+Enter 排队消息 - 等待AI完全结束后发送，不中断AI工作",
+  "Escape 中断当前AI并恢复编辑器中的排队消息",
+  "Alt+Up 查看并编辑所有排队的消息",
+  "插话消息会打断AI正在执行的工具，等待响应",
+  "排队消息静默等待，AI空闲时自动发送",
+  "可在 /settings 中调整消息发送模式和行为偏好",
+  // 联系方式提示
+  "⚠️  任何疑问或需求可联系 xujianjiang，支持需求对接开发、系统集成与扩展",
+  "⚠️  内网手册参考 UniDoc《2026.03 AICode Agent CLI编码智能体使用手册》",
+  "⚠️  LLM API 4xx/5xx 错误时请确认模型配置是否正确，或在企微群咨询 jiangchi、maliang",
+];
+
+export default function (pi: ExtensionAPI) {
+  let currentTipIndex = -1;
+  let autoHideTimeoutId: NodeJS.Timeout | undefined;
+  let autoShowTimeoutId: NodeJS.Timeout | undefined;
+  let currentCtx: ExtensionContext | undefined;
+  let isAutoPlaying = false;
+
+  // 基于 #49bccf (73, 188, 207) 的随机颜色生成
+  const getRandomTipColor = (): string => {
+    const baseR = 73, baseG = 188, baseB = 207;
+    const offset = 20; // 增大偏移范围，透明度降低
+    const r = Math.max(0, Math.min(255, baseR + Math.floor(Math.random() * offset * 2 - offset)));
+    const g = Math.max(0, Math.min(255, baseG + Math.floor(Math.random() * offset * 2 - offset)));
+    const b = Math.max(0, Math.min(255, baseB + Math.floor(Math.random() * offset * 2 - offset)));
+    return `\x1b[38;2;${r};${g};${b}m`;
+  };
+
+  const hideTip = () => {
+    if (!currentCtx?.hasUI) return;
+    currentCtx.ui.setWidget("aicode-tips", [], { placement: "aboveEditor" });
+  };
+
+  const clearAllTimers = () => {
+    if (autoHideTimeoutId) {
+      clearTimeout(autoHideTimeoutId);
+      autoHideTimeoutId = undefined;
+    }
+    if (autoShowTimeoutId) {
+      clearTimeout(autoShowTimeoutId);
+      autoShowTimeoutId = undefined;
+    }
+  };
+
+  const padRight = (str: string, width: number): string => {
+    const chinese = (str.match(/[\u4e00-\u9fff]/g) || []).length;
+    const ascii = str.length - chinese;
+    const fullWidth = chinese * 2 + ascii;
+    const padding = Math.max(0, width - fullWidth);
+    return str + " ".repeat(padding);
+  };
+
+  const showRandomTip = () => {
+    if (!currentCtx?.hasUI) return;
+
+    // 清除之前的定时器
+    clearAllTimers();
+
+    let randomIndex = Math.floor(Math.random() * AICODE_TIPS.length);
+    if (randomIndex === currentTipIndex && AICODE_TIPS.length > 1) {
+      randomIndex = (randomIndex + 1) % AICODE_TIPS.length;
+    }
+    currentTipIndex = randomIndex;
+    const tip = AICODE_TIPS[randomIndex];
+    // 使用随机颜色
+    const color = getRandomTipColor();
+    const coloredTip = `${color}tips: ${tip}\x1b[0m`;
+    currentCtx.ui.setWidget("aicode-tips", [coloredTip], { placement: "aboveEditor" });
+
+    // 自动轮播：10秒后隐藏，隐藏后3秒显示下一条
+    if (isAutoPlaying) {
+      autoHideTimeoutId = setTimeout(() => {
+        hideTip();
+        autoShowTimeoutId = setTimeout(showRandomTip, 3000);
+      }, 10000);
+    }
+  };
+
+  const startAutoPlay = () => {
+    isAutoPlaying = true;
+    showRandomTip();
+  };
+
+  const stopAutoPlay = () => {
+    isAutoPlaying = false;
+    clearAllTimers();
+  };
+
+  // 会话开始时启动自动轮播
+  pi.on("session_start", (_event, ctx) => {
+    currentCtx = ctx;
+    startAutoPlay();
+  });
+
+  // 会话结束时清理
+  pi.on("session_shutdown", () => {
+    stopAutoPlay();
+    currentCtx?.ui.setWidget("aicode-tips", undefined);
+    currentCtx = undefined;
+  });
+
+  // 手动切换技巧
+  pi.registerCommand("tip", {
+    description: "显示一个新的 aicode 使用技巧",
+    handler: async (_args, ctx) => {
+      currentCtx = ctx;
+      showRandomTip();
+    },
+  });
+}

package/config/agent/extensions/bash-check.ts

@@ -0,0 +1,227 @@
+/**
+ * @author: xujianjiang
+ * Bash 命令危险检测扩展
+ *
+ * 放置位置: config/agent/extensions/
+ * 安装后会复制到 ~/.aicode-cli/agent/extensions/
+ *
+ * 拦截危险 bash 命令，执行前需要用户回答数学题验证
+ * 
+ * 危险模式配置: config/agent/dangerous-patterns.json
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { readFileSync, existsSync } from "fs";
+import { join } from "path";
+import { homedir } from "os";
+
+// === 加载 JSON 配置文件 ===
+interface DangerPattern {
+  pattern: string;
+  commandPrefix: string;
+  name: string;
+  risk: string;
+  description: string;
+  impact: string;
+  explanation: string;
+}
+
+interface DangerPatternsConfig {
+  patterns: DangerPattern[];
+}
+
+function loadDangerPatterns(): DangerPattern[] {
+  const configPath = join(homedir(), ".aicode-cli", "agent", "dangerous-patterns.json");
+  
+  if (existsSync(configPath)) {
+    try {
+      const content = readFileSync(configPath, "utf-8");
+      const config: DangerPatternsConfig = JSON.parse(content);
+      return config.patterns || [];
+    } catch (e) {
+      console.error(`[bash-check] Failed to load config from ${configPath}:`, e);
+    }
+  }
+  
+  console.warn("[bash-check] No dangerous-patterns.json found, using empty patterns");
+  return [];
+}
+
+// === 编译危险模式（同步加载）===
+const patterns = loadDangerPatterns();
+const compiledPatterns = patterns.map(p => ({
+  ...p,
+  compiled: new RegExp(p.pattern, ""),
+}));
+
+// === 生成数学验证题目 ===
+function generateMathProblem(): { question: string, answer: number } {
+  const operators = ['+', '-'];
+  const operator = operators[Math.floor(Math.random() * operators.length)];
+  let a = Math.floor(Math.random() * 10) + 1;
+  let b = Math.floor(Math.random() * 10) + 1;
+  
+  // 确保减法结果非负
+  if (operator === '-' && a < b) {
+    [a, b] = [b, a];
+  }
+  
+  const question = `${a} ${operator} ${b} = ?`;
+  const answer = operator === '+' ? a + b : a - b;
+  
+  return { question, answer };
+}
+
+export default function (pi: ExtensionAPI) {
+  // === 拦截 bash 工具 - 检查危险操作 ===
+  pi.on("tool_call", async (event, ctx) => {
+    if (event.toolName !== "bash") return;
+
+    let command = event.input.command || "";
+
+    // 过滤掉 git commit -m 后面的 message 内容，只检查实际命令部分
+    const commitMsgMatch = command.match(/^(git\s+commit\s+.*?-m\s+)(.*)$/s);
+    if (commitMsgMatch) {
+      // 只检查 -m 之前的命令部分，忽略 message 内容
+      command = commitMsgMatch[1];
+    }
+
+    // 检查是否匹配危险模式
+    for (const danger of compiledPatterns) {
+      // 1. 首先检查命令是否以指定前缀开头
+      const trimmedCmd = command.trim();
+      if (!trimmedCmd.startsWith(danger.commandPrefix)) continue;
+      
+      // 2. 然后检查是否匹配危险模式（使用编译后的正则）
+      if (!danger.compiled.test(command)) continue;
+
+      // 解析命令中的路径参数
+      const { readdirSync, statSync, existsSync } = await import("fs");
+      const details: string[] = [];
+
+      // 提取相关路径
+      const pathMatch = command.match(/(['"])([^\1]+)\1|\S+(?=\s|$)/g);
+      if (pathMatch) {
+        for (const p of pathMatch) {
+          const cleanPath = p.replace(/^['"]|['"]$/g, '').trim();
+          if (cleanPath && cleanPath.length > 1 && cleanPath.length < 200) {
+            try {
+              if (existsSync(cleanPath)) {
+                const stat = statSync(cleanPath);
+                if (stat.isDirectory()) {
+                  const files = readdirSync(cleanPath);
+                  details.push(`📁 ${cleanPath}/ (${files.length} 个文件)`);
+                } else {
+                  details.push(`📄 ${cleanPath}`);
+                }
+              }
+            } catch (e) {
+              // 忽略读取错误
+            }
+          }
+        }
+      }
+
+      // 构建确认消息
+      const truncatedCmd = command.length > 100 ? command.slice(0, 100) + '...' : command;
+      const fileInfo = details.length > 0
+        ? `\n\n🗂️ 将操作的文件:\n${details.slice(0, 5).join('\n')}${details.length > 5 ? `\n... 还有 ${details.length - 5} 个` : ''}`
+        : '';
+
+      // 让用户选择操作
+      const choice = await ctx.ui.select(
+        `${danger.risk} 危险操作检测: ${danger.name}`,
+        [
+          `⚠️ 强制执行 (${danger.name})`,
+          "🔍 让 AI 解读此命令",
+          "❌ 取消操作",
+        ]
+      );
+
+      if (!choice) {
+        ctx.ui.notify("⛔ 操作已取消", "warning");
+        return { block: true, reason: "用户取消" };
+      }
+
+      if (choice.includes("取消")) {
+        ctx.ui.notify("⛔ 操作已取消", "warning");
+        return { block: true, reason: "用户取消" };
+      }
+
+      if (choice.includes("AI 解读")) {
+        // 显示详细解释
+        const explanation = `
+📋 命令: ${danger.name}
+
+📝 描述: ${danger.description}
+
+⚠️ 风险等级: ${danger.risk}
+
+💥 影响: ${danger.impact}
+
+📖 详细解释:
+${danger.explanation}
+
+🗂️ 涉及路径:${fileInfo || '\n   (无)'}`.trim();
+
+        ctx.ui.notify(explanation, "info");
+
+        // 解释后再让用户选择
+        const confirmChoice = await ctx.ui.select(
+          "请确认操作",
+          [
+            `⚠️ 我已了解风险，继续执行 (${danger.name})`,
+            "❌ 取消操作",
+          ]
+        );
+
+        if (!confirmChoice || confirmChoice.includes("取消")) {
+          ctx.ui.notify("⛔ 操作已取消", "warning");
+          return { block: true, reason: "用户取消" };
+        }
+
+        // 执行数学验证
+        const mathProblem = generateMathProblem();
+        const mathAnswer = await ctx.ui.input(
+          `🔴 最终验证 - 请回答数学题\n\n⚠️ 命令: ${danger.name}\n\n${truncatedCmd}${fileInfo}\n\n📝 请计算: ${mathProblem.question}`,
+          "输入答案后回车"
+        );
+
+        if (!mathAnswer) {
+          ctx.ui.notify("⛔ 操作已取消", "warning");
+          return { block: true, reason: "用户未输入答案" };
+        }
+
+        const userAnswer = parseInt(mathAnswer.trim(), 10);
+        if (isNaN(userAnswer) || userAnswer !== mathProblem.answer) {
+          ctx.ui.notify(`⛔ 答案错误！正确答案是: ${mathProblem.answer}`, "error");
+          return { block: true, reason: "数学验证失败" };
+        }
+
+        ctx.ui.notify(`✅ 验证通过！执行: ${danger.name}`, "success");
+        return;
+      }
+
+      // 直接执行 - 数学验证
+      const mathProblem = generateMathProblem();
+      const mathAnswer = await ctx.ui.input(
+        `🔴 最终验证 - 请回答数学题\n\n⚠️ 命令: ${danger.name}\n\n${truncatedCmd}${fileInfo}\n\n📝 请计算: ${mathProblem.question}`,
+        "输入答案后回车"
+      );
+
+      if (!mathAnswer) {
+        ctx.ui.notify("⛔ 操作已取消", "warning");
+        return { block: true, reason: "用户未输入答案" };
+      }
+
+      const userAnswer = parseInt(mathAnswer.trim(), 10);
+      if (isNaN(userAnswer) || userAnswer !== mathProblem.answer) {
+        ctx.ui.notify(`⛔ 答案错误！正确答案是: ${mathProblem.answer}`, "error");
+        return { block: true, reason: "数学验证失败" };
+      }
+
+      ctx.ui.notify(`✅ 验证通过！执行: ${danger.name}`, "success");
+      break;
+    }
+  });
+}

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

@@ -0,0 +1,275 @@
+/**
+ * @author: xujianjiang
+ * Context Monitor Extension — 自动检测上下文大小并在接近阈值时警告
+ *
+ * 功能：
+ * 1. 自动检测每轮对话后的上下文使用情况
+ * 2. 在超过阈值时自动发出不同级别的警告
+ * 3. 提供清理建议和上下文统计
+ *
+ * 阈值级别：
+ * - 70%   : info 级别提示
+ * - 80%   : warning 级别警告
+ * - 90%   : error 级别严重警告
+ * - 100%  : 阻止继续并强制提示压缩
+ */
+
+import type { ExtensionAPI, ExtensionContext, ToolResultEvent } from "@mariozechner/pi-coding-agent";
+
+interface ContextThresholds {
+	info: number;    // 70%
+	warning: number; // 80%
+	error: number;   // 90%
+	critical: number; // 100%
+}
+
+const DEFAULT_THRESHOLDS: ContextThresholds = {
+	info: 70,
+	warning: 80,
+	error: 90,
+	critical: 100,
+};
+
+interface ContextStats {
+	contextWindow: number;
+	usedTokens: number;
+	remainingTokens: number;
+	percent: number;
+	systemTokens: number;
+	toolsTokens: number;
+	conversationTokens: number;
+}
+
+function estimateTokens(text: string): number {
+	return Math.max(0, Math.ceil(text.length / 4));
+}
+
+function formatTokens(n: number): string {
+	if (n < 1000) return n.toString();
+	if (n < 1000000) return `${(n / 1000).toFixed(1)}k`;
+	return `${(n / 1000000).toFixed(1)}M`;
+}
+
+function getContextLevel(percent: number): "normal" | "info" | "warning" | "error" | "critical" {
+	if (percent >= DEFAULT_THRESHOLDS.critical) return "critical";
+	if (percent >= DEFAULT_THRESHOLDS.error) return "error";
+	if (percent >= DEFAULT_THRESHOLDS.warning) return "warning";
+	if (percent >= DEFAULT_THRESHOLDS.info) return "info";
+	return "normal";
+}
+
+function getContextMessage(stats: ContextStats): { level: "info" | "warning" | "error"; title: string; body: string } {
+	const level = getContextLevel(stats.percent);
+	const contextBar = renderContextBar(stats.percent);
+
+	let title: string;
+	let body: string;
+
+	switch (level) {
+		case "critical":
+			title = "🚨 上下文已满!";
+			body = `上下文窗口已完全使用! (${contextBar})\n剩余: ${formatTokens(stats.remainingTokens)} tokens\n\n⚠️ 请立即使用 /clear 清理会话或开启新会话!`;
+			break;
+		case "error":
+			title = "⚠️ 上下文即将耗尽";
+			body = `上下文使用: ${contextBar}\n已用: ${formatTokens(stats.usedTokens)} / ${formatTokens(stats.contextWindow)}\n剩余: ${formatTokens(stats.remainingTokens)} tokens\n\n💡 建议: 使用 /clear 清理会话`;
+			break;
+		case "warning":
+			title = "📊 上下文使用较高";
+			body = `上下文使用: ${contextBar}\n已用: ${formatTokens(stats.usedTokens)} / ${formatTokens(stats.contextWindow)}\n剩余: ${formatTokens(stats.remainingTokens)} tokens\n\n💡 提示: 可以考虑压缩会话`;
+			break;
+		case "info":
+			title = "ℹ️ 上下文使用提醒";
+			body = `上下文使用: ${contextBar}\n已用: ${formatTokens(stats.usedTokens)} / ${formatTokens(stats.contextWindow)}\n\n💡 提示: 上下文已使用过半，注意控制对话长度`;
+			break;
+		default:
+			title = "✅ 上下文状态正常";
+			body = `上下文使用: ${contextBar}\n已用: ${formatTokens(stats.usedTokens)} / ${formatTokens(stats.contextWindow)}\n剩余: ${formatTokens(stats.remainingTokens)} tokens`;
+	}
+
+	return { level, title, body };
+}
+
+function renderContextBar(percent: number, width: number = 20): string {
+	const filled = Math.min(width, Math.round((percent / 100) * width));
+	const empty = width - filled;
+
+	const filledChar = "█";
+	const emptyChar = "░";
+
+	return filledChar.repeat(filled) + emptyChar.repeat(empty) + ` ${percent.toFixed(1)}%`;
+}
+
+export default function contextMonitorExtension(pi: ExtensionAPI) {
+	// 配置：是否启用自动警告
+	let autoWarnEnabled = true;
+	let lastWarnedLevel: string = "normal";
+	let warnCooldown = 0; // 防止连续警告
+
+	/**
+	 * 获取当前上下文统计
+	 */
+	function getContextStats(ctx: ExtensionContext): ContextStats | null {
+		const usage = ctx.getContextUsage();
+		if (!usage) return null;
+
+		const contextWindow = usage.contextWindow || ctx.model?.contextWindow || 0;
+		if (contextWindow === 0) return null;
+
+		const usedTokens = usage.tokens || 0;
+		const remainingTokens = Math.max(0, contextWindow - usedTokens);
+		const percent = (usedTokens / contextWindow) * 100;
+
+		// 估算各部分 token
+		const systemPrompt = ctx.getSystemPrompt();
+		const systemTokens = systemPrompt ? estimateTokens(systemPrompt) : 0;
+		const toolsTokens = Math.round(
+			pi.getActiveTools().reduce((sum, name) => {
+				const info = pi.getAllTools().find((t) => t.name === name);
+				return sum + estimateTokens(`${name}\n${info?.description || ""}`);
+			}, 0) * 1.5
+		);
+		const conversationTokens = Math.max(0, usedTokens - systemTokens - toolsTokens);
+
+		return {
+			contextWindow,
+			usedTokens,
+			remainingTokens,
+			percent,
+			systemTokens,
+			toolsTokens,
+			conversationTokens,
+		};
+	}
+
+	/**
+	 * 处理上下文警告
+	 */
+	async function handleContextWarning(stats: ContextStats, ctx: ExtensionContext) {
+		if (!autoWarnEnabled) return;
+
+		const level = getContextLevel(stats.percent);
+		const message = getContextMessage(stats);
+
+		// 避免重复警告（同级别）
+		if (level === lastWarnedLevel && warnCooldown > 0) {
+			warnCooldown--;
+			return;
+		}
+
+		// 只在有新级别时警告
+		const levels = ["normal", "info", "warning", "error", "critical"];
+		const currentIdx = levels.indexOf(level);
+		const lastIdx = levels.indexOf(lastWarnedLevel);
+
+		if (currentIdx <= lastIdx && level !== "critical") return;
+
+		lastWarnedLevel = level;
+		warnCooldown = 3; // 设置冷却
+
+		// 发送通知
+		ctx.ui.notify(message.title, message.level as any);
+	}
+
+	// 监听轮次结束事件，检查上下文使用
+	pi.on("turn_end", async (_event, ctx: ExtensionContext) => {
+		const stats = getContextStats(ctx);
+		if (!stats) return;
+
+		await handleContextWarning(stats, ctx);
+	});
+
+	// 监听会话开始，初始化
+	pi.on("session_start", async (_event, ctx: ExtensionContext) => {
+		lastWarnedLevel = "normal";
+		warnCooldown = 0;
+	});
+
+	/**
+	 * /context-stats 命令 - 显示详细上下文统计
+	 */
+	pi.registerCommand("context-stats", {
+		description: "显示当前上下文详细统计",
+		usage: "/context-stats",
+		handler: async (_args, ctx: ExtensionContext) => {
+			const stats = getContextStats(ctx);
+
+			if (!stats) {
+				ctx.ui.notify("无法获取上下文统计", "warning");
+				return;
+			}
+
+			const message = getContextMessage(stats);
+			const lines = message.body.split("\n");
+
+			// 构建详细统计文本
+			const modelName = ctx.model?.name || "unknown";
+			const modelMaxTokens = ctx.model?.maxTokens || 0;
+			const contextWindow = stats.contextWindow;
+
+			// 根据使用情况给出建议
+			let suggestion = "";
+			if (stats.percent >= 90) {
+				suggestion = `⚠️ 上下文使用过高，建议清理会话\n💡 模型 ${modelName} 最大窗口 ${formatTokens(contextWindow)}`;
+			} else if (stats.percent >= 70) {
+				suggestion = `💡 上下文已过半，注意控制对话长度`;
+			}
+
+			const detailLines = [
+				`${"─".repeat(40)}`,
+				`📊 上下文详细统计`,
+				`${"─".repeat(40)}`,
+				``,
+				`模型:     ${modelName}`,
+				`总窗口:   ${formatTokens(contextWindow)} tokens`,
+				`最大输出: ${formatTokens(modelMaxTokens)} tokens`,
+				`已使用:   ${formatTokens(stats.usedTokens)} tokens (${stats.percent.toFixed(1)}%)`,
+				`剩余:     ${formatTokens(stats.remainingTokens)} tokens`,
+				``,
+				`📝 分布估算:`,
+				`  系统:   ${formatTokens(stats.systemTokens)} tokens`,
+				`  工具:   ${formatTokens(stats.toolsTokens)} tokens`,
+				`  对话:   ${formatTokens(stats.conversationTokens)} tokens`,
+				``,
+				suggestion,
+				``,
+				`${"─".repeat(40)}`,
+			];
+
+			// 显示给用户
+			const fullText = detailLines.join("\n");
+			pi.sendMessage(
+				{
+					customType: "context-stats",
+					content: fullText,
+					display: true,
+				},
+				{ triggerTurn: false }
+			);
+
+			// 如果超过阈值也显示通知
+			if (stats.percent >= 80) {
+				ctx.ui.notify(message.title, message.level as any);
+			}
+		},
+	});
+
+	/**
+	 * /context-warn 命令 - 切换自动警告
+	 */
+	pi.registerCommand("context-warn", {
+		description: "开启/关闭上下文警告",
+		usage: "/context-warn [on|off]",
+		handler: async (args, ctx: ExtensionContext) => {
+			if (!args || args === "on") {
+				autoWarnEnabled = true;
+				ctx.ui.notify("✅ 上下文警告已开启", "info");
+			} else if (args === "off") {
+				autoWarnEnabled = false;
+				ctx.ui.notify("🔕 上下文警告已关闭", "info");
+			} else {
+				ctx.ui.notify(`当前状态: ${autoWarnEnabled ? "开启" : "关闭"}`, "info");
+			}
+		},
+	});
+}

package/config/agent/extensions/context.ts

@@ -1,4 +1,5 @@
 /**
+ * @author: xujianjiang
  * /context
  *
  * Small TUI view showing what's loaded/available: