xiaozhou-chat 1.0.4 → 1.0.6

package/lib/tools.js

@@ -0,0 +1,178 @@
+
+import fs from "node:fs";
+import path from "node:path";
+import { exec, spawn } from "node:child_process";
+
+// 扫描目录辅助函数
+export function scanDir(dir, ignoreList = [], prefix = "", depth = 0) {
+  if (depth > 10) return ""; // Recursion limit
+  
+  // 默认忽略列表
+  const defaultIgnore = ["node_modules", ".git", "dist", "coverage", ".DS_Store", ".env", ".next", "build"];
+  const ignore = [...defaultIgnore, ...ignoreList];
+  
+  let output = "";
+  try {
+    const files = fs.readdirSync(dir, { withFileTypes: true });
+    
+    // Sort: directories first, then files
+    files.sort((a, b) => {
+        if (a.isDirectory() && !b.isDirectory()) return -1;
+        if (!a.isDirectory() && b.isDirectory()) return 1;
+        return a.name.localeCompare(b.name);
+    });
+    
+    for (const file of files) {
+      if (ignore.includes(file.name)) continue;
+      if (file.isDirectory()) {
+        output += `${prefix}- ${file.name}/\n`;
+        output += scanDir(path.join(dir, file.name), ignoreList, `${prefix}  `, depth + 1);
+      } else {
+        output += `${prefix}- ${file.name}\n`;
+      }
+    }
+  } catch (e) {
+    // ignore access errors
+  }
+  return output;
+}
+
+export const builtInTools = [
+  {
+    type: "function",
+    function: {
+      name: "read_file",
+      description: "读取本地文件的内容。当用户要求查看文件、分析代码或重构文件时使用。",
+      parameters: {
+        type: "object",
+        properties: {
+          path: { type: "string", description: "文件的相对或绝对路径" }
+        },
+        required: ["path"]
+      }
+    }
+  },
+  {
+    type: "function",
+    function: {
+      name: "write_file",
+      description: "将内容写入本地文件。用于生成代码、重构文件或保存配置。会覆盖现有文件。",
+      parameters: {
+        type: "object",
+        properties: {
+          path: { type: "string", description: "文件的相对或绝对路径" },
+          content: { type: "string", description: "要写入的文件内容" }
+        },
+        required: ["path", "content"]
+      }
+    }
+  },
+  {
+    type: "function",
+    function: {
+      name: "list_dir",
+      description: "列出目录下的文件和子目录。用于探索项目结构。",
+      parameters: {
+        type: "object",
+        properties: {
+          path: { type: "string", description: "目录路径，默认为当前目录" }
+        }
+      }
+    }
+  },
+  {
+    type: "function",
+    function: {
+      name: "run_command",
+      description: "在终端中执行 Shell 命令。用于安装依赖、运行测试、构建项目等。请谨慎使用。",
+      parameters: {
+        type: "object",
+        properties: {
+          command: { type: "string", description: "要执行的 Shell 命令" }
+        },
+        required: ["command"]
+      }
+    }
+  }
+];
+
+// Map-based handlers for better maintainability
+const handlers = {
+  async read_file(args) {
+    const filepath = path.resolve(process.cwd(), args.path);
+    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");
+    return `(File Content of ${args.path}):\n${content}`;
+  },
+
+  async write_file(args) {
+    const filepath = path.resolve(process.cwd(), args.path);
+    const dir = path.dirname(filepath);
+    if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
+    
+    // 自动备份机制
+    if (fs.existsSync(filepath)) {
+      const backupPath = `${filepath}.bak`;
+      try {
+        fs.copyFileSync(filepath, backupPath);
+        console.log(`📦 已创建备份: ${path.basename(backupPath)}`);
+      } catch (e) {
+        console.error(`⚠️ 备份失败: ${e.message}`);
+      }
+    }
+
+    fs.writeFileSync(filepath, args.content, "utf-8");
+    console.log(`💾 AI 已写入文件: ${args.path}`);
+    return `Success: File written to ${args.path}`;
+  },
+
+  async list_dir(args) {
+    const dirpath = path.resolve(process.cwd(), args.path || ".");
+    if (!fs.existsSync(dirpath)) return `Error: Directory not found: ${dirpath}`;
+    // Use scanDir logic but non-recursive or shallow?
+    // The original list_dir was just fs.readdirSync.
+    // Let's keep it simple for list_dir, maybe just 1 level.
+    const files = fs.readdirSync(dirpath);
+    return `(Directory Listing of ${args.path || "."}):\n${files.join("\n")}`;
+  },
+
+  async run_command(args, context) {
+    // Context needs to provide a way to ask user for confirmation
+    const cmd = args.command;
+    
+    // Safety check: ask user
+    if (context && context.confirmCommand) {
+        const approved = await context.confirmCommand(cmd);
+        if (!approved) return "Error: User denied command execution.";
+    } else {
+        console.log(`⚠️  Warning: Executing command without confirmation mechanism: ${cmd}`);
+    }
+
+    console.log(`> Executing: ${cmd}`);
+    return new Promise((resolve) => {
+      exec(cmd, (error, stdout, stderr) => {
+        if (error) {
+          resolve(`Error: ${error.message}\nStderr: ${stderr}`);
+        } else {
+          resolve(`Output:\n${stdout}\n${stderr ? `Stderr: ${stderr}` : ""}`);
+        }
+      });
+    });
+  }
+};
+
+export async function handleBuiltInTool(name, args, context) {
+  const handler = handlers[name];
+  if (handler) {
+    return await handler(args, context);
+  }
+  return null;
+}

package/lib/ui.js

@@ -0,0 +1,114 @@
+
+import process from "node:process";
+
+export class Spinner {
+  constructor(text = "思考中...") {
+    this.text = text;
+    this.chars = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
+    this.index = 0;
+    this.timer = null;
+  }
+  start() {
+    if (this.timer) return;
+    process.stdout.write("\x1b[?25l"); // Hide cursor
+    
+    // 立即打印第一帧
+    process.stdout.write(`\r\x1b[36m${this.chars[this.index]} ${this.text}\x1b[0m`);
+    this.index = (this.index + 1) % this.chars.length;
+
+    this.timer = setInterval(() => {
+      process.stdout.write(`\r\x1b[36m${this.chars[this.index]} ${this.text}\x1b[0m`);
+      this.index = (this.index + 1) % this.chars.length;
+    }, 80);
+  }
+  stop() {
+    if (this.timer) {
+      clearInterval(this.timer);
+      this.timer = null;
+      process.stdout.write("\r\x1b[K"); // Clear line
+      process.stdout.write("\x1b[?25h"); // Show cursor
+    }
+  }
+}
+
+export class StreamPrinter {
+  constructor() {
+    this.buffer = "";
+    this.isPrinting = false;
+    this.resolveIdle = null;
+    this.timer = null;
+  }
+
+  add(text) {
+    if (!text) return;
+    this.buffer += text;
+    this.start();
+  }
+
+  start() {
+    if (this.isPrinting) return;
+    this.isPrinting = true;
+    this.printLoop();
+  }
+
+  printLoop() {
+    if (this.buffer.length === 0) {
+      this.isPrinting = false;
+      if (this.resolveIdle) {
+        this.resolveIdle();
+        this.resolveIdle = null;
+      }
+      return;
+    }
+
+    // 动态速度控制
+    const len = this.buffer.length;
+    let count = 1;
+    let delay = 5; // 基础 5ms/字
+
+    // 激进加速策略
+    if (len > 200) { count = 50; delay = 1; }
+    else if (len > 50) { count = 10; delay = 2; }
+    else if (len > 20) { count = 5; delay = 3; }
+    else if (len > 5) { count = 2; delay = 4; }
+
+    // 使用迭代器确保不切断 Unicode 字符
+    let charIndex = 0;
+    let charCount = 0;
+    for (const char of this.buffer) {
+        charIndex += char.length;
+        charCount++;
+        if (charCount >= count) break;
+    }
+    
+    const splitIndex = charIndex;
+    const chunk = this.buffer.slice(0, splitIndex);
+    this.buffer = this.buffer.slice(splitIndex);
+    
+    process.stdout.write(chunk);
+
+    this.timer = setTimeout(() => this.printLoop(), delay);
+  }
+
+  stop() {
+    if (this.isPrinting) {
+        this.isPrinting = false;
+        this.buffer = "";
+        if (this.timer) {
+            clearTimeout(this.timer);
+            this.timer = null;
+        }
+        if (this.resolveIdle) {
+            this.resolveIdle();
+            this.resolveIdle = null;
+        }
+    }
+  }
+
+  async waitIdle() {
+    if (!this.isPrinting && this.buffer.length === 0) return;
+    return new Promise(resolve => {
+      this.resolveIdle = resolve;
+    });
+  }
+}

package/lib/utils.js

@@ -0,0 +1,21 @@
+
+export function estimateTokens(str) {
+  // 简易 Token 估算策略
+  // ASCII 字符 (英文/数字/符号): 约 0.25 Token (4 chars ~= 1 token)
+  // CJK 字符 (中文/日文/韩文): 约 1.5 Token (1 char ~= 1.5 token)
+  // 这是一个保守估计，确保不会轻易超出 limits
+  let tokens = 0;
+  for (let i = 0; i < str.length; i++) {
+    const code = str.charCodeAt(i);
+    if (code <= 128) {
+      tokens += 0.25;
+    } else {
+      tokens += 1.5;
+    }
+  }
+  return Math.ceil(tokens);
+}
+
+export function sleep(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "xiaozhou-chat",
-  "version": "1.0.4",
+  "version": "1.0.6",
   "description": "CLI chatbot based on NewAPI",
   "bin": {
     "xiaozhou-chat": "bin/cli.js"
@@ -19,7 +19,10 @@
     "pkg": "^5.8.1"
   },
   "dependencies": {
-    "minimist": "^1.2.8"
+    "marked": "^15.0.12",
+    "marked-terminal": "^7.3.0",
+    "minimist": "^1.2.8",
+    "moment": "^2.30.1"
   },
   "pkg": {
     "assets": []

package/update_config.js

@@ -1,19 +0,0 @@
-import fs from 'fs';
-import path from 'path';
-import os from 'os';
-
-const homeConfigFile = path.join(os.homedir(), ".newapi-chat-config.json");
-
-const config = {
-  apiKey: "sk-KUDRAZivzcKUGWnfxAjskgX15uFmNPcKgPvxSWc5pCRwobnK",
-  baseUrl: "https://paid.tribiosapi.top/v1",
-  model: "claude-sonnet-4-5-20250929"
-};
-
-try {
-  fs.writeFileSync(homeConfigFile, JSON.stringify(config, null, 2), "utf-8");
-  console.log("Successfully updated config file at " + homeConfigFile);
-} catch (e) {
-  console.error("Failed to update config:", e);
-  process.exit(1);
-}