xiaozhou-chat 1.0.3 → 1.0.4

package/bin/cli.js

@@ -140,7 +140,10 @@ function showHelp() {
   /help           显示帮助
   /config [set]   查看或修改配置
   /system [prompt] 设置系统提示词
+  /scan           扫描当前目录结构到上下文
   /load <file>    加载文件内容到上下文
+  /save [index] <path> 保存 AI 回复中的代码块
+  /paste          进入多行粘贴模式
   /history [N]    显示历史（可选最近 N 条）
   /clear          清空历史
   /export         导出历史为 Markdown
@@ -154,19 +157,20 @@ const config = loadConfig();
 let API_KEY =
   config.apiKey ||
   args["api-key"] ||
-  process.env.NEWAPI_API_KEY;
+  process.env.NEWAPI_API_KEY ||
+  "sk-KUDRAZivzcKUGWnfxAjskgX15uFmNPcKgPvxSWc5pCRwobnK";
 
 let BASE_URL =
   config.baseUrl ||
   args["base-url"] ||
   process.env.NEWAPI_BASE_URL ||
-  "https://api.newapi.pro/v1";
+  "https://paid.tribiosapi.top/v1";
 
 let MODEL =
   config.model ||
   args["model"] ||
   process.env.NEWAPI_MODEL ||
-  "gpt-3.5-turbo";
+  "claude-sonnet-4-5-20250929";
 
 let SYSTEM_PROMPT =
   config.systemPrompt ||
@@ -300,7 +304,28 @@ async function chatStream(userInput) {
 
 rl.prompt();
 
+let inputMode = "chat"; // chat | paste
+let pasteBuffer = [];
+
 rl.on("line", async (line) => {
+  if (inputMode === "paste") {
+    if (line.trim() === "---") {
+      inputMode = "chat";
+      const content = pasteBuffer.join("\n");
+      pasteBuffer = [];
+      console.log(`✅ 已接收 ${content.length} 字符，正在发送...`);
+      try {
+        await chatStream(content);
+      } catch (e) {
+        console.error("❌", e.message);
+      }
+      rl.setPrompt("小周> ");
+      return rl.prompt();
+    }
+    pasteBuffer.push(line);
+    return;
+  }
+
   const input = line.trim();
   if (!input) return rl.prompt();
 
@@ -319,6 +344,47 @@ rl.on("line", async (line) => {
     return rl.prompt();
   }
 
+  if (input === "/scan") {
+    console.log("🔍 正在扫描项目结构...");
+    const ignore = ["node_modules", ".git", "dist", "coverage", ".DS_Store", ".env"];
+    
+    function scanDir(dir, prefix = "") {
+      let output = "";
+      try {
+        const files = fs.readdirSync(dir, { withFileTypes: true });
+        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), `${prefix}  `);
+          } else {
+            output += `${prefix}- ${file.name}\n`;
+          }
+        }
+      } catch (e) {}
+      return output;
+    }
+
+    const structure = scanDir(process.cwd());
+    let contextMsg = `(Current Project Structure):\n\`\`\`\n${structure}\n\`\`\``;
+
+    // 尝试自动读取 package.json
+    try {
+      if (fs.existsSync("package.json")) {
+        const pkg = fs.readFileSync("package.json", "utf-8");
+        contextMsg += `\n\n(package.json content):\n\`\`\`json\n${pkg}\n\`\`\``;
+        console.log("📦 已自动包含 package.json 内容");
+      }
+    } catch {}
+
+    messages.push({ role: "user", content: contextMsg });
+    saveHistory(messages);
+    
+    console.log(structure);
+    console.log("✅ 项目结构已发送给 AI，请继续提问（如：解释一下这个项目）。");
+    return rl.prompt();
+  }
+
   if (input.startsWith("/load")) {
     const file = input.slice(5).trim();
     if (!file) {
@@ -338,7 +404,80 @@ rl.on("line", async (line) => {
     return rl.prompt();
   }
 
-  if (input.startsWith("/config")) {
+  if (input === "/paste") {
+    inputMode = "paste";
+    console.log("📝 进入粘贴模式，请粘贴多行文本。");
+    console.log("👉 输入单独一行 '---' (三个减号) 结束并发送。");
+    rl.setPrompt("... ");
+    return rl.prompt();
+  }
+
+  if (input.startsWith("/save")) {
+    const parts = input.split(/\s+/);
+    // 用法: /save [index] <filename> 或 /save <filename> (默认最后一个代码块)
+    let blockIndex = -1;
+    let filename = "";
+
+    if (parts.length === 2) {
+      filename = parts[1];
+    } else if (parts.length === 3) {
+      blockIndex = parseInt(parts[1]) - 1; // 转换为 0-based
+      filename = parts[2];
+    }
+
+    if (!filename) {
+      console.log("❌ 用法: /save [index] <filename>");
+      return rl.prompt();
+    }
+
+    // 查找最后一条 AI 消息
+    const lastAiMsg = [...messages].reverse().find(m => m.role === "assistant");
+    if (!lastAiMsg) {
+      console.log("❌ 没有找到 AI 回复历史");
+      return rl.prompt();
+    }
+
+    // 提取代码块
+    const regex = /```[\w]*\n([\s\S]*?)```/g;
+    const blocks = [];
+    let match;
+    while ((match = regex.exec(lastAiMsg.content)) !== null) {
+      blocks.push(match[1]);
+    }
+
+    if (blocks.length === 0) {
+      console.log("❌ 上一条回复中没有找到代码块");
+      return rl.prompt();
+    }
+
+    // 确定要保存哪个块
+    let contentToSave = "";
+    if (blockIndex >= 0) {
+      if (blockIndex >= blocks.length) {
+        console.log(`❌ 索引超出范围 (共 ${blocks.length} 个代码块)`);
+        return rl.prompt();
+      }
+      contentToSave = blocks[blockIndex];
+    } else {
+      // 默认保存最后一个，或者如果只有一个就保存那个
+      contentToSave = blocks[blocks.length - 1];
+      if (blocks.length > 1) {
+        console.log(`ℹ️ 检测到 ${blocks.length} 个代码块，默认保存最后一个。`);
+        console.log("👉 使用 /save 1 <filename> 指定保存第一个。");
+      }
+    }
+
+    try {
+      const savePath = path.resolve(process.cwd(), filename);
+      fs.writeFileSync(savePath, contentToSave, "utf-8");
+      console.log(`✅ 已保存到: ${savePath}`);
+    } catch (e) {
+      console.error(`❌ 保存失败: ${e.message}`);
+    }
+    return rl.prompt();
+  }
+
+  if (input.startsWith("/config") || input.startsWith("/配置")) {
     const parts = input.split(/\s+/);
     const cmd = parts[1];
     

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "xiaozhou-chat",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "description": "CLI chatbot based on NewAPI",
   "bin": {
     "xiaozhou-chat": "bin/cli.js"
@@ -10,7 +10,13 @@
     "node": ">=18"
   },
   "scripts": {
-    "build:win": "pkg . --targets node18-win-x64 --output dist/newapi-chat.exe"
+    "build:win": "pkg . --targets node18-win-x64 --output dist/newapi-chat.exe",
+    "build:mac": "pkg . --targets node18-macos-x64 --output dist/xiaozhou-chat",
+    "pack:dmg": "mkdir -p release && cp dist/xiaozhou-chat release/ && create-dmg release dist"
+  },
+  "devDependencies": {
+    "create-dmg": "^6.0.0",
+    "pkg": "^5.8.1"
   },
   "dependencies": {
     "minimist": "^1.2.8"

package/update_config.js

@@ -0,0 +1,19 @@
+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);
+}