deepfish-ai 1.0.8

package/src/core/extension/ExtensionManager.js

@@ -0,0 +1,172 @@
+const { descriptions, functions } = require('./DefaultExtension')
+const path = require('path')
+const fs = require('fs-extra')
+const axios = require('axios')
+const dayjs = require('dayjs')
+const lodash = require('lodash')
+const shelljs = require('shelljs')
+const iconv = require('iconv-lite') // 用于编码转换
+const os = require('os') // 用于判断系统类型
+const { logError } = require('../utils')
+
+class ExtensionManager {
+  constructor(aiCli) {
+    this.aiCli = aiCli
+    this.extensions = {
+      descriptions,
+      functions,
+    }
+    this.parseExtends(this.aiCli.config.extensions || [])
+  }
+
+  parseExtends(configExtends) {
+    // 自动扫描扩展模块
+    const autoScannedExtends = this.autoScanExtensions()
+    configExtends = configExtends.concat(autoScannedExtends)
+    for (const extensionPath of configExtends) {
+      try {
+        // 解析扩展路径
+        const resolvedPath = path.isAbsolute(extensionPath)
+          ? extensionPath
+          : path.resolve(process.cwd(), extensionPath)
+
+        if (!fs.existsSync(resolvedPath)) {
+          logError(`Extension file not found: ${resolvedPath}`)
+          continue
+        }
+
+        // 动态加载扩展模块
+        let { descriptions, functions } = require(resolvedPath)
+        descriptions = descriptions.map((item) => {
+          if (!item.type) {
+            return {
+              type: 'function',
+              function: item,
+            }
+          } else {
+            return item
+          }
+        })
+        this.extensions.descriptions =
+          this.extensions.descriptions.concat(descriptions)
+        this.extensions.functions = Object.assign(
+          this.extensions.functions,
+          functions,
+        )
+      } catch (error) {
+        logError(`Error loading extension ${extensionPath}: ${error.message}`)
+      }
+    }
+    const functions = this.extensions.functions
+    for (const fnName of Object.keys(functions)) {
+      functions[fnName] = functions[fnName].bind(this.aiCli)
+      if (fnName === 'test') {
+        functions[fnName]()
+      }
+    }
+    functions['fs'] = fs
+    functions['axios'] = axios
+    functions['dayjs'] = dayjs
+    functions['lodash'] = lodash
+  }
+
+  // 自动扫描node_modules和命令执行目录下的扩展模块
+  autoScanExtensions() {
+    const result = []
+    // 扫描本程序所在目录下node_modules目录
+    const nodeModulesPath1 = path.resolve(__dirname, '../../../node_modules')
+    // 扫描根node_modules目录
+    const nodeModulesPath2 = this._executeCommand('npm root -g')
+    // 扫描命令执行目录下node_modules目录
+    const nodeModulesPath3 = path.resolve(process.cwd(), 'node_modules')
+    // 扫描命令执行目录
+    const nodeModulesPath4 = process.cwd()
+    for (const dirPath of [
+      nodeModulesPath1,
+      nodeModulesPath2,
+      nodeModulesPath3,
+      nodeModulesPath4,
+    ]) {
+      if (!fs.existsSync(dirPath)) {
+        continue
+      }
+      const fileNames = fs.readdirSync(dirPath)
+      for (const fileName of fileNames) {
+        // 如果是目录且目录名称前缀是"deepfish-"，则认为是扩展模块
+        const extensionDir = path.resolve(dirPath, fileName)
+        if (
+          fileName.startsWith('deepfish-') && fileName !== 'deepfish-ai' && 
+          fs.statSync(extensionDir).isDirectory()
+        ) {
+          const subDirNames = fs.readdirSync(extensionDir)
+          for (const subDirName of subDirNames) {
+            const subDirPath = path.resolve(extensionDir, subDirName)
+            if (fs.statSync(subDirPath).isDirectory()) {
+              const extNames = fs.readdirSync(subDirPath)
+              const jsFiles = extNames.filter(
+                (file) => file.endsWith('.js') || file.endsWith('.cjs'),
+              )
+              jsFiles.forEach((jsFile) => {
+                const jsFilePath = path.resolve(subDirPath, jsFile)
+                // 读取文件，查询文件内是否存在‘descriptions’和‘functions’
+                const fileContent = fs.readFileSync(jsFilePath, 'utf-8')
+                if (
+                  fileContent.includes('descriptions') &&
+                  fileContent.includes('functions')
+                ) {
+                  result.push(jsFilePath)
+                }
+              })
+            }
+          }
+        }
+      }
+    }
+    return result
+  }
+
+  _executeCommand(command) {
+    return new Promise((resolve, reject) => {
+      const platform = os.platform()
+      const targetEncoding = platform === 'win32' ? 'gbk' : 'utf-8' // Windows(含PowerShell)用gbk，Linux/macOS用utf-8
+      shelljs.exec(
+        command,
+        {
+          async: true,
+          cwd: process.cwd(),
+          encoding: 'binary',
+          silent: true,
+        },
+        (code, stdout, stderr) => {
+          try {
+            const stdoutUtf8 = iconv.decode(
+              Buffer.from(stdout, 'binary'),
+              targetEncoding,
+            )
+            const stderrUtf8 = iconv.decode(
+              Buffer.from(stderr, 'binary'),
+              targetEncoding,
+            )
+            if (stderrUtf8 && !stderrUtf8.trim().startsWith('WARNING')) {
+              // 过滤无关警告
+              const error = new Error(
+                `Command failed (code ${code}): ${stderrUtf8}`,
+              )
+              reject(error)
+              return
+            }
+            resolve(stdoutUtf8)
+          } catch (decodeError) {
+            reject(
+              new Error(
+                `Failed to parse command output: ${decodeError.message}`,
+              ),
+            )
+          }
+        },
+      )
+    })
+  }
+}
+
+module.exports = ExtensionManager

package/src/core/utils.js

@@ -0,0 +1,261 @@
+const chalk = require("chalk");
+const path = require("path");
+const os = require("os");
+const fs = require("fs-extra");
+
+// 日志相关工具函数
+function logInfo(message) {
+  log(message, "#6dd2ea");
+}
+
+function logSuccess(message) {
+  log(message, "#9bed7f");
+}
+
+function logError(message) {
+  log(message, "#ed7f7f");
+}
+
+function writeLine(msg1, msg2 = "", color = "blue") {
+  if (color === "blue") {
+    process.stdout.write("\r" + chalk.hex("#6dd2ea")(msg1) + " " + msg2);
+  } else if (color === "green") {
+    process.stdout.write("\r" + chalk.hex("#9bed7f")(msg1) + " " + msg2);
+  } else if (color === "red") {
+    process.stdout.write("\r" + chalk.hex("#ed7f7f")(msg1) + " " + msg2);
+  } else {
+    process.stdout.write("\r" + chalk.hex(color)(msg1) + " " + msg2);
+  }
+}
+
+// 流式输出
+async function streamOutput(text, color = "#9bed7f") {
+  process.stdout.write(chalk.hex(color)(text));
+}
+
+// 流式换行
+async function streamLineBreak() {
+  process.stdout.write("\n");
+}
+
+function objStrToObj(str) {
+  try {
+    if (typeof str === 'string') {
+      return eval(`(${str})`);
+    } else {
+      return str
+    }
+  } catch (error) {
+    throw new Error(`对象转换失败：${error.message}`);
+  }
+}
+
+
+
+function delay(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+
+function loading(label = "Thinking...") {
+  let animationInterval;
+  const spinners = ["|", "/", "-", "\\"];
+  let spinnerIndex = 0;
+  process.stdout.write("\r");
+  animationInterval = setInterval(() => {
+    writeLine(spinners[spinnerIndex], label);
+    spinnerIndex = (spinnerIndex + 1) % spinners.length;
+  }, 200);
+  return (endLabel, isError = false) => {
+    clearInterval(animationInterval);
+    if (endLabel) {
+      writeLine(endLabel, "", isError ? "red" : "green");
+    }
+    process.stdout.write("\r\n");
+  };
+}
+
+function log(msg, color) {
+  if (!color) {
+    console.log(msg);
+  } else {
+    console.log(chalk.hex(color)(msg));
+  }
+}
+
+// 添加扩展
+function addExtensionToConfig(fileName) {
+  // 检查 fileName 是否为空
+  if (!fileName) {
+    logError("Extension file name is required.");
+    return;
+  }
+  const filePath = path.resolve(process.cwd(), fileName);
+  // 判断是否路径是文件还是目录
+  if (fs.statSync(filePath).isDirectory()) {
+    // 扫描目录和子目录下所有js、cjs文件
+    const files = traverseFiles()
+    const jsFiles = files.filter((file) => file.endsWith(".js") || file.endsWith(".cjs"));
+    jsFiles.forEach((jsFile) => {
+      // 读取文件，查询文件内是否存在‘descriptions’和‘functions’
+      const fileContent = fs.readFileSync(jsFile, "utf-8");
+      if (fileContent.includes("descriptions") && fileContent.includes("functions")) {
+        addExtensionToConfig(jsFile);
+      }
+    });
+    return;
+  }
+  // 判断文件是否存在
+  if (!fs.existsSync(filePath)) {
+    logError(`File not found: ${filePath}`);
+    return;
+  }
+  const userConfigPath = path.join(os.homedir(), ".ai-cmd.config.js");
+  if (!fs.existsSync(userConfigPath)) {
+    logError(
+      `User config file not found: ${userConfigPath}. Please run 'ai config reset' first.`,
+    );
+    return;
+  }
+  const userConfig = require(userConfigPath);
+  if (userConfig.extensions && Array.isArray(userConfig.extensions)) {
+    userConfig.extensions.push(filePath);
+  } else {
+    userConfig.extensions = [filePath];
+  }
+  // 数组去重
+  userConfig.extensions = [...new Set(userConfig.extensions)];
+  fs.writeFileSync(
+    userConfigPath,
+    `module.exports = ${JSON.stringify(userConfig, null, 2)}`,
+  );
+  logSuccess(
+    `Extension added to config: ${filePath}.`,
+  );
+}
+
+// 移除扩展
+function removeExtensionFromConfig(fileName) {
+  const userConfigPath = path.join(os.homedir(), ".ai-cmd.config.js");
+  if (!fs.existsSync(userConfigPath)) {
+    logError(
+      `User config file not found: ${userConfigPath}. Please run 'ai config reset' first.`,
+    );
+    return;
+  }
+  const userConfig = require(userConfigPath);
+  // 增加对数字索引的支持
+  if (!isNaN(Number(fileName))) {
+    const extIndex = Number(fileName);
+    if (extIndex < 0 || extIndex >= userConfig.extensions.length) {
+      logError(`Invalid extension index: ${extIndex}`);
+      return;
+    }
+    const filePath = userConfig.extensions.splice(extIndex, 1);
+    fs.writeFileSync(
+      userConfigPath,
+      `module.exports = ${JSON.stringify(userConfig, null, 2)}`,
+    );
+    logSuccess(
+      `Extension removed from config: ${filePath}.You can run 'ai ext ls' to view the changes.`,
+    );
+    return;
+  }
+  const filePath = path.resolve(process.cwd(), fileName);
+  // 判断文件是否存在
+  if (!fs.existsSync(filePath)) {
+    logError(`File not found: ${filePath}`);
+    return;
+  }
+  if (userConfig.extensions && Array.isArray(userConfig.extensions)) {
+    userConfig.extensions = userConfig.extensions.filter(
+      (ext) => ext !== filePath,
+    );
+  }
+  fs.writeFileSync(
+    userConfigPath,
+    `module.exports = ${JSON.stringify(userConfig, null, 2)}`,
+  );
+  logSuccess(
+    `Extension removed from config: ${filePath}.You can run 'ai ext ls' to view the changes.`,
+  );
+}
+
+// 查看扩展列表
+function viewExtensionsFromConfig() {
+  const userConfigPath = path.join(os.homedir(), ".ai-cmd.config.js");
+  if (!fs.existsSync(userConfigPath)) {
+    logError(
+      `User config file not found: ${userConfigPath}. Please run 'ai config reset' first.`,
+    );
+    return;
+  }
+  const userConfig = require(userConfigPath);
+  if (userConfig.extensions && Array.isArray(userConfig.extensions)) {
+    console.log("=".repeat(50));
+    // 打印扩展列表，并加上索引
+    if (userConfig.extensions.length === 0) {
+      console.log(`No extensions in config.`);
+    } else {
+      console.log("Extensions in config:");
+      userConfig.extensions.forEach((ext, index) => {
+        console.log(`[${index}] ${ext}`);
+      });
+    }
+    console.log("=".repeat(50));
+  } else {
+    logSuccess(`No extensions in config.`);
+  }
+}
+
+
+function traverseFiles() {
+  try {
+    const currentDir = process.cwd();
+    const allFiles = [];
+    const currentItems = fs.readdirSync(currentDir, { withFileTypes: true });
+    for (const item of currentItems) {
+      const itemPath = path.join(currentDir, item.name);
+      if (item.isFile()) {
+        allFiles.push(itemPath);
+        continue;
+      }
+      if (item.isDirectory()) {
+        try {
+          const subItems = fs.readdirSync(itemPath, { withFileTypes: true });
+          for (const subItem of subItems) {
+            if (subItem.isFile()) {
+              allFiles.push(path.join(itemPath, subItem.name));
+            }
+          }
+        } catch (subErr) {
+          console.warn(`读取子目录失败 ${itemPath}：${subErr.message}`);
+        }
+      }
+    }
+    return allFiles;
+  } catch (err) {
+    console.error(`遍历目录失败：${err.message}`);
+    return [];
+  }
+}
+
+// 获取配置文件所在目录
+function getConfigPath() {
+  return path.join(os.homedir(), ".ai-cmd.config.js");
+}
+
+module.exports = {
+  logInfo,
+  logSuccess,
+  logError,
+  loading,
+  writeLine,
+  streamOutput,
+  streamLineBreak,
+  addExtensionToConfig,
+  removeExtensionFromConfig,
+  viewExtensionsFromConfig,
+  objStrToObj,
+  delay,
+  getConfigPath
+};

package/src/index.js

@@ -0,0 +1,7 @@
+const AICLI = require("./core/AICLI");
+const BaseExtension = require("./core/extension/BaseExtension");
+
+module.exports = {
+  AICLI,
+  BaseExtension,
+};