codemate-ai 1.0.2 → 1.0.3

package/dist/cli.js

@@ -3,6 +3,7 @@ import * as path14 from 'path';
 import path14__default from 'path';
 import { fileURLToPath } from 'url';
 import * as fs15 from 'fs';
+import fs15__default from 'fs';
 import 'dotenv/config';
 import { Command } from 'commander';
 import * as fs16 from 'fs/promises';
@@ -14,6 +15,7 @@ import { exec, spawn } from 'child_process';
 import { promisify } from 'util';
 import * as readline from 'readline/promises';
 import * as os from 'os';
+import os__default from 'os';
 import express from 'express';
 import { createServer } from 'http';
 import { WebSocketServer as WebSocketServer$1, WebSocket } from 'ws';
@@ -25,6 +27,7 @@ import React2, { createContext, useState, useMemo, useEffect, useContext } from
 import { render, useApp, Box, Text, Static } from 'ink';
 import TextInput from 'ink-text-input';
 import Spinner from 'ink-spinner';
+import readline3 from 'readline';
 
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
@@ -1326,14 +1329,14 @@ var EditFileTool = class extends Tool {
     regex: z.boolean().optional().describe("Use regex for matching")
   });
   async execute(input) {
-    const { path: path17, oldContent, newContent, regex = false } = input;
-    console.log(`\u270F\uFE0F  Editing file: ${path17}`);
+    const { path: path18, oldContent, newContent, regex = false } = input;
+    console.log(`\u270F\uFE0F  Editing file: ${path18}`);
     try {
       if (this.container) {
         const fileHistory = this.container.getFileHistory();
         await fileHistory.trackFile(input.path);
       }
-      const content = await fs16.readFile(path17, "utf-8");
+      const content = await fs16.readFile(path18, "utf-8");
       let newFileContent;
       if (regex) {
         const pattern = new RegExp(oldContent, "g");
@@ -1347,15 +1350,15 @@ var EditFileTool = class extends Tool {
           message: "No changes made (old content not found)"
         };
       }
-      await fs16.writeFile(path17, newFileContent, "utf-8");
+      await fs16.writeFile(path18, newFileContent, "utf-8");
       console.log(`\u2705 File edited successfully`);
       return {
         success: true,
-        message: `File edited: ${path17}`,
+        message: `File edited: ${path18}`,
         changes: newFileContent.length - content.length
       };
     } catch (error) {
-      console.error(`\u274C Failed to edit file: ${path17}`, error);
+      console.error(`\u274C Failed to edit file: ${path18}`, error);
       if (error instanceof Error) {
         throw new Error(`Failed to edit file: ${error.message}`);
       }
@@ -1374,30 +1377,30 @@ var DeleteFileTool = class extends Tool {
     recursive: z.boolean().optional().describe("Delete directory recursively")
   });
   async execute(input) {
-    const { path: path17, recursive = false } = input;
-    console.log(`\u{1F5D1}\uFE0F  Deleting: ${path17}`);
+    const { path: path18, recursive = false } = input;
+    console.log(`\u{1F5D1}\uFE0F  Deleting: ${path18}`);
     try {
       if (this.container) {
         const fileHistory = this.container.getFileHistory();
         await fileHistory.trackFile(input.path);
       }
-      const stats = await fs16.stat(path17);
+      const stats = await fs16.stat(path18);
       if (stats.isDirectory()) {
         if (!recursive) {
           throw new Error("Cannot delete directory without recursive flag");
         }
-        await fs16.rm(path17, { recursive: true, force: true });
+        await fs16.rm(path18, { recursive: true, force: true });
       } else {
-        await fs16.unlink(path17);
+        await fs16.unlink(path18);
       }
       console.log(`\u2705 Deleted successfully`);
       return {
         success: true,
-        message: `Deleted: ${path17}`,
+        message: `Deleted: ${path18}`,
         type: stats.isDirectory() ? "directory" : "file"
       };
     } catch (error) {
-      console.error(`\u274C Failed to delete: ${path17}`, error);
+      console.error(`\u274C Failed to delete: ${path18}`, error);
       if (error instanceof Error) {
         throw new Error(`Failed to delete: ${error.message}`);
       }
@@ -2023,14 +2026,14 @@ var AskUserTool = class extends Tool {
     defaultAnswer: z.string().optional().describe("Default answer if user presses Enter")
   });
   async execute(input) {
-    const { question, defaultAnswer } = input;
-    console.log(`\u2753 Asking user: ${question}`);
+    const { question: question2, defaultAnswer } = input;
+    console.log(`\u2753 Asking user: ${question2}`);
     try {
       const rl = readline.createInterface({
         input: process.stdin,
         output: process.stdout
       });
-      let prompt = question;
+      let prompt = question2;
       if (defaultAnswer) {
         prompt += ` (default: ${defaultAnswer})`;
       }
@@ -2041,7 +2044,7 @@ var AskUserTool = class extends Tool {
       console.log(`\u2705 User answered: ${finalAnswer}`);
       return {
         success: true,
-        question,
+        question: question2,
         answer: finalAnswer
       };
     } catch (error) {
@@ -3563,65 +3566,65 @@ function normalizeWindowsPath(input = "") {
 var _UNC_REGEX = /^[/\\]{2}/;
 var _IS_ABSOLUTE_RE = /^[/\\](?![/\\])|^[/\\]{2}(?!\.)|^[A-Za-z]:[/\\]/;
 var _DRIVE_LETTER_RE = /^[A-Za-z]:$/;
-var normalize = function(path17) {
-  if (path17.length === 0) {
+var normalize = function(path18) {
+  if (path18.length === 0) {
     return ".";
   }
-  path17 = normalizeWindowsPath(path17);
-  const isUNCPath = path17.match(_UNC_REGEX);
-  const isPathAbsolute = isAbsolute3(path17);
-  const trailingSeparator = path17[path17.length - 1] === "/";
-  path17 = normalizeString(path17, !isPathAbsolute);
-  if (path17.length === 0) {
+  path18 = normalizeWindowsPath(path18);
+  const isUNCPath = path18.match(_UNC_REGEX);
+  const isPathAbsolute = isAbsolute3(path18);
+  const trailingSeparator = path18[path18.length - 1] === "/";
+  path18 = normalizeString(path18, !isPathAbsolute);
+  if (path18.length === 0) {
     if (isPathAbsolute) {
       return "/";
     }
     return trailingSeparator ? "./" : ".";
   }
   if (trailingSeparator) {
-    path17 += "/";
+    path18 += "/";
   }
-  if (_DRIVE_LETTER_RE.test(path17)) {
-    path17 += "/";
+  if (_DRIVE_LETTER_RE.test(path18)) {
+    path18 += "/";
   }
   if (isUNCPath) {
     if (!isPathAbsolute) {
-      return `//./${path17}`;
+      return `//./${path18}`;
     }
-    return `//${path17}`;
+    return `//${path18}`;
   }
-  return isPathAbsolute && !isAbsolute3(path17) ? `/${path17}` : path17;
+  return isPathAbsolute && !isAbsolute3(path18) ? `/${path18}` : path18;
 };
 var join10 = function(...segments) {
-  let path17 = "";
+  let path18 = "";
   for (const seg of segments) {
     if (!seg) {
       continue;
     }
-    if (path17.length > 0) {
-      const pathTrailing = path17[path17.length - 1] === "/";
+    if (path18.length > 0) {
+      const pathTrailing = path18[path18.length - 1] === "/";
       const segLeading = seg[0] === "/";
       const both = pathTrailing && segLeading;
       if (both) {
-        path17 += seg.slice(1);
+        path18 += seg.slice(1);
       } else {
-        path17 += pathTrailing || segLeading ? seg : `/${seg}`;
+        path18 += pathTrailing || segLeading ? seg : `/${seg}`;
       }
     } else {
-      path17 += seg;
+      path18 += seg;
     }
   }
-  return normalize(path17);
+  return normalize(path18);
 };
-function normalizeString(path17, allowAboveRoot) {
+function normalizeString(path18, allowAboveRoot) {
   let res = "";
   let lastSegmentLength = 0;
   let lastSlash = -1;
   let dots = 0;
   let char = null;
-  for (let index = 0; index <= path17.length; ++index) {
-    if (index < path17.length) {
-      char = path17[index];
+  for (let index = 0; index <= path18.length; ++index) {
+    if (index < path18.length) {
+      char = path18[index];
     } else if (char === "/") {
       break;
     } else {
@@ -3657,9 +3660,9 @@ function normalizeString(path17, allowAboveRoot) {
         }
       } else {
         if (res.length > 0) {
-          res += `/${path17.slice(lastSlash + 1, index)}`;
+          res += `/${path18.slice(lastSlash + 1, index)}`;
         } else {
-          res = path17.slice(lastSlash + 1, index);
+          res = path18.slice(lastSlash + 1, index);
         }
         lastSegmentLength = index - lastSlash - 1;
       }
@@ -4152,60 +4155,60 @@ var CodeLocator = class {
     let found = null;
     traverseFunction(ast, {
       // 查找函数
-      FunctionDeclaration(path17) {
-        if (type === "function" && path17.node.id?.name === name) {
-          const parentPath = path17.parentPath;
+      FunctionDeclaration(path18) {
+        if (type === "function" && path18.node.id?.name === name) {
+          const parentPath = path18.parentPath;
           const isExported = parentPath && t.isExportNamedDeclaration(parentPath.node);
           found = {
             type: "function",
             name,
-            location: isExported ? getLocation(parentPath.node) : getLocation(path17.node),
-            code: isExported ? parentPath.toString() : path17.toString()
+            location: isExported ? getLocation(parentPath.node) : getLocation(path18.node),
+            code: isExported ? parentPath.toString() : path18.toString()
           };
-          path17.stop();
+          path18.stop();
         }
       },
       // 查找类
-      ClassDeclaration(path17) {
-        if (type === "class" && path17.node.id?.name === name) {
-          const parentPath = path17.parentPath;
+      ClassDeclaration(path18) {
+        if (type === "class" && path18.node.id?.name === name) {
+          const parentPath = path18.parentPath;
           const isExported = parentPath && t.isExportNamedDeclaration(parentPath.node);
           found = {
             type: "class",
             name,
-            location: isExported ? getLocation(parentPath.node) : getLocation(path17.node),
-            code: isExported ? parentPath.toString() : path17.toString()
+            location: isExported ? getLocation(parentPath.node) : getLocation(path18.node),
+            code: isExported ? parentPath.toString() : path18.toString()
           };
-          path17.stop();
+          path18.stop();
         }
       },
       // 查找方法
-      ClassMethod(path17) {
+      ClassMethod(path18) {
         if (type === "method") {
-          const classPath = path17.findParent((p) => p.isClassDeclaration());
+          const classPath = path18.findParent((p) => p.isClassDeclaration());
           const className = classPath && t.isClassDeclaration(classPath.node) ? classPath.node.id?.name : void 0;
-          if (className === parent && t.isIdentifier(path17.node.key) && path17.node.key.name === name) {
+          if (className === parent && t.isIdentifier(path18.node.key) && path18.node.key.name === name) {
             found = {
               type: "method",
               name,
-              location: getLocation(path17.node),
-              code: path17.toString(),
+              location: getLocation(path18.node),
+              code: path18.toString(),
               parent: className
             };
-            path17.stop();
+            path18.stop();
           }
         }
       },
       // 查找变量
-      VariableDeclarator(path17) {
-        if (type === "variable" && t.isIdentifier(path17.node.id) && path17.node.id.name === name) {
+      VariableDeclarator(path18) {
+        if (type === "variable" && t.isIdentifier(path18.node.id) && path18.node.id.name === name) {
           found = {
             type: "variable",
             name,
-            location: getLocation(path17.node),
-            code: path17.toString()
+            location: getLocation(path18.node),
+            code: path18.toString()
           };
-          path17.stop();
+          path18.stop();
         }
       }
     });
@@ -4220,15 +4223,15 @@ var CodeLocator = class {
   listFunctions(ast) {
     const functions = [];
     traverseFunction(ast, {
-      FunctionDeclaration(path17) {
-        if (path17.node.id) {
-          const parentPath = path17.parentPath;
+      FunctionDeclaration(path18) {
+        if (path18.node.id) {
+          const parentPath = path18.parentPath;
           const isExported = parentPath && t.isExportNamedDeclaration(parentPath.node);
           functions.push({
             type: "function",
-            name: path17.node.id.name,
-            location: isExported ? getLocation(parentPath.node) : getLocation(path17.node),
-            code: isExported ? parentPath.toString() : path17.toString()
+            name: path18.node.id.name,
+            location: isExported ? getLocation(parentPath.node) : getLocation(path18.node),
+            code: isExported ? parentPath.toString() : path18.toString()
           });
         }
       }
@@ -4244,15 +4247,15 @@ var CodeLocator = class {
   listClasses(ast) {
     const classes = [];
     traverseFunction(ast, {
-      ClassDeclaration(path17) {
-        if (path17.node.id) {
-          const parentPath = path17.parentPath;
+      ClassDeclaration(path18) {
+        if (path18.node.id) {
+          const parentPath = path18.parentPath;
           const isExported = parentPath && t.isExportNamedDeclaration(parentPath.node);
           classes.push({
             type: "class",
-            name: path17.node.id.name,
-            location: isExported ? getLocation(parentPath.node) : getLocation(path17.node),
-            code: isExported ? parentPath.toString() : path17.toString()
+            name: path18.node.id.name,
+            location: isExported ? getLocation(parentPath.node) : getLocation(path18.node),
+            code: isExported ? parentPath.toString() : path18.toString()
           });
         }
       }
@@ -5305,13 +5308,140 @@ function App({ app }) {
   return /* @__PURE__ */ React2.createElement(AppContextProvider, { app }, /* @__PURE__ */ React2.createElement(AppContent, null));
 }
 
+// src/utils/firstRun.ts
+init_esm_shims();
+function isFirstRun() {
+  const configPath = getConfigPath();
+  return !fs15__default.existsSync(configPath);
+}
+function getConfigPath() {
+  const homeDir = os__default.homedir();
+  return path14__default.join(homeDir, ".aiclirc.json");
+}
+function createDefaultConfig(apiKey, provider, model, baseURL) {
+  const defaultModels = {
+    openai: "gpt-4",
+    anthropic: "claude-3-5-sonnet-20241022",
+    deepseek: "deepseek-chat",
+    openrouter: "anthropic/claude-3.5-sonnet",
+    custom: "gpt-4"
+  };
+  const config = {
+    provider,
+    apiKey,
+    model: model || defaultModels[provider] || "gpt-4",
+    createdAt: (/* @__PURE__ */ new Date()).toISOString()
+  };
+  if (baseURL) {
+    config.baseURL = baseURL;
+  }
+  const configPath = getConfigPath();
+  fs15__default.writeFileSync(configPath, JSON.stringify(config, null, 2));
+}
+
+// src/utils/setup.ts
+init_esm_shims();
+var PROVIDERS = {
+  openai: {
+    name: "OpenAI",
+    defaultModel: "gpt-4",
+    baseURL: void 0
+  },
+  anthropic: {
+    name: "Anthropic",
+    defaultModel: "claude-3-5-sonnet-20241022",
+    baseURL: void 0
+  },
+  deepseek: {
+    name: "DeepSeek",
+    defaultModel: "deepseek-chat",
+    baseURL: "https://api.deepseek.com"
+  },
+  openrouter: {
+    name: "OpenRouter",
+    defaultModel: "anthropic/claude-3.5-sonnet",
+    baseURL: "https://openrouter.ai/api/v1"
+  },
+  custom: {
+    name: "\u81EA\u5B9A\u4E49 (OpenAI \u517C\u5BB9)",
+    defaultModel: "gpt-4",
+    baseURL: void 0
+  }
+};
+async function runFirstTimeSetup() {
+  const rl = readline3.createInterface({
+    input: process.stdin,
+    output: process.stdout
+  });
+  console.log("\u8BF7\u9009\u62E9 AI \u63D0\u4F9B\u5546:");
+  console.log("  1. OpenAI (GPT-4, GPT-3.5)");
+  console.log("  2. Anthropic (Claude)");
+  console.log("  3. DeepSeek (\u56FD\u5185\u53EF\u7528\uFF0C\u6027\u4EF7\u6BD4\u9AD8)");
+  console.log("  4. OpenRouter (\u591A\u6A21\u578B\u805A\u5408)");
+  console.log("  5. \u81EA\u5B9A\u4E49 (OpenAI \u517C\u5BB9 API)");
+  console.log("");
+  const choice = await question(rl, "\u8BF7\u8F93\u5165\u9009\u9879 (1-5): ");
+  const providerKey = getProviderKey(choice.trim());
+  console.log("");
+  const apiKey = await question(rl, "\u8BF7\u8F93\u5165 API Key: ");
+  console.log("");
+  let baseURL;
+  if (providerKey === "custom") {
+    baseURL = await question(rl, "\u8BF7\u8F93\u5165 API Base URL (\u4F8B\u5982: https://api.example.com/v1): ");
+    console.log("");
+  }
+  const defaultModel = PROVIDERS[providerKey].defaultModel;
+  const modelInput = await question(rl, `\u8BF7\u8F93\u5165\u6A21\u578B\u540D\u79F0 (\u76F4\u63A5\u56DE\u8F66\u4F7F\u7528\u9ED8\u8BA4: ${defaultModel}): `);
+  const model = modelInput.trim() || defaultModel;
+  rl.close();
+  createDefaultConfig(apiKey.trim(), providerKey, model, baseURL || PROVIDERS[providerKey].baseURL);
+  console.log("");
+  console.log("\u2705 \u914D\u7F6E\u5DF2\u4FDD\u5B58\u5230: ~/.aiclirc.json");
+  console.log("");
+  console.log(`\u63D0\u4F9B\u5546: ${PROVIDERS[providerKey].name}`);
+  console.log(`\u6A21\u578B: ${model}`);
+  if (baseURL || PROVIDERS[providerKey].baseURL) {
+    console.log(`Base URL: ${baseURL || PROVIDERS[providerKey].baseURL}`);
+  }
+}
+function getProviderKey(choice) {
+  switch (choice) {
+    case "1":
+      return "openai";
+    case "2":
+      return "anthropic";
+    case "3":
+      return "deepseek";
+    case "4":
+      return "openrouter";
+    case "5":
+      return "custom";
+    default:
+      return "openai";
+  }
+}
+function question(rl, query) {
+  return new Promise((resolve8) => {
+    rl.question(query, resolve8);
+  });
+}
+
 // src/cli.ts
 var program = new Command();
-program.name("codemate").description("AI-powered CLI assistant").version("1.0.2");
+program.name("codemate").description("AI-powered CLI assistant").version("1.0.3");
 program.command("chat", { isDefault: true }).description("Start interactive chat mode").action(async () => {
   try {
+    if (isFirstRun()) {
+      console.log("\n\u6B22\u8FCE\u4F7F\u7528 CodeMate AI! \u{1F389}\n");
+      console.log("\u68C0\u6D4B\u5230\u8FD9\u662F\u9996\u6B21\u8FD0\u884C\uFF0C\u9700\u8981\u8FDB\u884C\u521D\u59CB\u914D\u7F6E\u3002\n");
+      await runFirstTimeSetup();
+      console.log("\n\u914D\u7F6E\u5B8C\u6210\uFF01\u73B0\u5728\u53EF\u4EE5\u5F00\u59CB\u4F7F\u7528\u4E86\u3002\n");
+      console.log('\u8BD5\u8BD5\uFF1Acodemate "\u521B\u5EFA\u4E00\u4E2A Hello World \u7A0B\u5E8F"\n');
+      return;
+    }
     const configService = new ConfigService();
-    await configService.load();
+    const configPath = getConfigPath();
+    await configService.load(configPath);
     const app = new Application(void 0, configService);
     await app.start();
     const sessionService = app.getContainer().get("session");
@@ -5324,12 +5454,34 @@ program.command("chat", { isDefault: true }).description("Start interactive chat
     process.exit(1);
   }
 });
+program.command("config").description("Configure or reconfigure CodeMate AI").action(async () => {
+  try {
+    console.log("\n\u{1F527} CodeMate AI \u914D\u7F6E\u5411\u5BFC\n");
+    if (!isFirstRun()) {
+      console.log("\u5F53\u524D\u914D\u7F6E\u6587\u4EF6: ~/.aiclirc.json");
+      console.log("\u6B64\u64CD\u4F5C\u5C06\u8986\u76D6\u73B0\u6709\u914D\u7F6E\u3002\n");
+    }
+    await runFirstTimeSetup();
+    console.log("\n\u2705 \u914D\u7F6E\u5DF2\u66F4\u65B0\uFF01\n");
+    console.log("\u73B0\u5728\u53EF\u4EE5\u4F7F\u7528 codemate \u5F00\u59CB\u5DE5\u4F5C\u4E86\u3002\n");
+  } catch (error) {
+    console.error("\u274C \u914D\u7F6E\u5931\u8D25:", error);
+    process.exit(1);
+  }
+});
 program.command("server").description("Start HTTP server mode").option("-p, --port <port>", "Port to listen on", "3000").action(async (options) => {
   try {
+    if (isFirstRun()) {
+      console.log("\n\u274C \u9519\u8BEF\uFF1A\u5C1A\u672A\u914D\u7F6E CodeMate AI\n");
+      console.log("\u8BF7\u5148\u8FD0\u884C\u4EE5\u4E0B\u547D\u4EE4\u8FDB\u884C\u914D\u7F6E\uFF1A\n");
+      console.log("  codemate config\n");
+      process.exit(1);
+    }
     const configService = new ConfigService();
-    await configService.load();
+    const configPath = getConfigPath();
+    await configService.load(configPath);
     const app = new Application(void 0, configService);
-    const port = parseInt(options.port, 10);
+    const port = Number.parseInt(options.port, 10);
     await app.startServer(port);
     process.on("SIGINT", async () => {
       console.log("\n\n\u{1F44B} Shutting down server...");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "codemate-ai",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "type": "module",
   "description": "Enterprise-grade AI coding assistant CLI",
   "main": "dist/cli.js",