xiaozhou-chat 1.0.30 → 1.0.32

package/bin/cli.js

@@ -46,7 +46,9 @@ import {
   loadConfig, 
   updateConfig, 
   getActiveConfig, 
-  setProfileValue 
+  setProfileValue,
+  setRole,
+  getRoles
 } from "../lib/config.js";
 import { 
   loadHistory, 
@@ -342,6 +344,9 @@ rl.on("line", async (line) => {
 - \`/commit\`: 生成 Git Commit Message
 - \`/compress\`: 压缩对话历史
 - \`/token\`: 估算 Token 消耗
+- `/copy`: 复制上一次 AI 回复的代码或内容
+- `/editor`: 调用系统编辑器输入多行内容
+- `/role`: 切换或管理角色预设 (别名: `/角色`)
 `);
         return rl.prompt();
     }

package/lib/config.js

@@ -37,7 +37,12 @@ export function initConfigFile() {
         model: "claude-sonnet-4-5-20250929"
       }
     },
-    currentProfile: "default"
+    currentProfile: "default",
+    roles: {
+      coder: "你是一个资深的全栈工程师，精通 Node.js, React, TypeScript。请编写高质量、可维护的代码。",
+      writer: "你是一个专业的文案创作者，擅长清晰、有说服力的表达。",
+      translator: "你是一个精通中英文的翻译专家，请提供信达雅的翻译结果。"
+    }
   };
 
   fs.writeFileSync(homeConfigFile, JSON.stringify(defaultConfig, null, 2), "utf-8");
@@ -65,7 +70,12 @@ export function initProjectConfigFile() {
         model: "claude-sonnet-4-5-20250929"
       }
     },
-    currentProfile: "default"
+    currentProfile: "default",
+    roles: {
+      coder: "你是一个资深的全栈工程师，精通 Node.js, React, TypeScript。请编写高质量、可维护的代码。",
+      writer: "你是一个专业的文案创作者，擅长清晰、有说服力的表达。",
+      translator: "你是一个精通中英文的翻译专家，请提供信达雅的翻译结果。"
+    }
   };
 
   fs.writeFileSync(targetFile, JSON.stringify(defaultConfig, null, 2), "utf-8");
@@ -184,6 +194,21 @@ export function setProfileValue(profileName, key, value) {
   return current;
 }
 
+export function setRole(name, prompt) {
+  const target = getWriteConfigFile();
+  let current = loadConfigFrom(target);
+  
+  if (!current.roles) current.roles = {};
+  current.roles[name] = prompt;
+  
+  fs.writeFileSync(target, JSON.stringify(current, null, 2), "utf-8");
+  return current;
+}
+
+export function getRoles(config) {
+    return config.roles || {};
+}
+
 export function getActiveConfig(config) {
     const profileName = config.currentProfile || "default";
     const profile = config.profiles?.[profileName] || {};

package/lib/tools.js

@@ -55,12 +55,29 @@ export const builtInTools = [
       parameters: {
         type: "object",
         properties: {
-          path: { type: "string", description: "文件的相对或绝对路径" }
+          path: { type: "string", description: "文件的相对或绝对路径" },
+          start_line: { type: "integer", description: "起始行号 (从1开始，可选)" },
+          end_line: { type: "integer", description: "结束行号 (可选)" }
         },
         required: ["path"]
       }
     }
   },
+  {
+    type: "function",
+    function: {
+      name: "search_files",
+      description: "在项目中搜索包含特定关键词的文件。返回匹配的文件路径和行号。",
+      parameters: {
+        type: "object",
+        properties: {
+          query: { type: "string", description: "搜索关键词 (不区分大小写)" },
+          path: { type: "string", description: "搜索目录，默认为当前目录" }
+        },
+        required: ["query"]
+      }
+    }
+  },
   {
     type: "function",
     function: {
@@ -112,16 +129,73 @@ const handlers = {
     if (!fs.existsSync(filepath)) return `Error: File not found: ${filepath}`;
     
     // 安全检查：防止读取敏感文件 (如 /etc/passwd 或 .env)
-    // 简单实现：检查文件名
     if (path.basename(filepath) === ".env" || path.basename(filepath).includes("key")) {
        // Warning but allow if user explicitly asked? Better strict for now.
-       // return `Error: Access to sensitive file ${args.path} is restricted.`;
     }
 
-    const content = fs.readFileSync(filepath, "utf-8");
+    let content = fs.readFileSync(filepath, "utf-8");
+    
+    if (args.start_line || args.end_line) {
+        const lines = content.split('\n');
+        const start = (args.start_line || 1) - 1;
+        const end = args.end_line || lines.length;
+        content = lines.slice(start, end).join('\n');
+        return `(File Content of ${args.path} lines ${start+1}-${end}):\n${content}`;
+    }
+
     return `(File Content of ${args.path}):\n${content}`;
   },
 
+  async search_files(args) {
+    const query = args.query.toLowerCase();
+    const dir = path.resolve(process.cwd(), args.path || ".");
+    
+    // 使用 scanDir 的逻辑来遍历文件，但这里我们需要读取内容
+    // 为避免性能问题，我们限制搜索深度和文件大小
+    
+    function searchRecursive(currentDir, depth = 0) {
+        if (depth > 5) return [];
+        let results = [];
+        try {
+            const files = fs.readdirSync(currentDir, { withFileTypes: true });
+            for (const file of files) {
+                const fullPath = path.join(currentDir, file.name);
+                
+                // Ignore common binary/system dirs
+                if (file.name.startsWith(".") || file.name === "node_modules" || file.name === "dist" || file.name === "build") continue;
+                
+                if (file.isDirectory()) {
+                    results = results.concat(searchRecursive(fullPath, depth + 1));
+                } else {
+                    // Check extension (skip images, binaries)
+                    const ext = path.extname(file.name).toLowerCase();
+                    if ([".jpg", ".png", ".exe", ".bin", ".lock", ".pdf"].includes(ext)) continue;
+
+                    try {
+                        const content = fs.readFileSync(fullPath, "utf-8");
+                        const lines = content.split('\n');
+                        lines.forEach((line, index) => {
+                            if (line.toLowerCase().includes(query)) {
+                                // Return relative path
+                                const relPath = path.relative(process.cwd(), fullPath);
+                                results.push(`${relPath}:${index + 1}: ${line.trim().slice(0, 100)}`);
+                            }
+                        });
+                    } catch (e) {
+                        // ignore read errors
+                    }
+                }
+            }
+        } catch (e) {}
+        return results;
+    }
+
+    const matches = searchRecursive(dir);
+    if (matches.length === 0) return "No matches found.";
+    // Limit results
+    return `Found ${matches.length} matches:\n${matches.slice(0, 50).join('\n')}${matches.length > 50 ? `\n...and ${matches.length - 50} more` : ""}`;
+  },
+
   async write_file(args) {
     const filepath = path.resolve(process.cwd(), args.path);
     const dir = path.dirname(filepath);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "xiaozhou-chat",
-  "version": "1.0.30",
+  "version": "1.0.32",
   "description": "CLI chatbot based on NewAPI",
   "bin": {
     "xiaozhou-chat": "bin/cli.js"