@chongyan/autospec 1.0.1

package/plugins/hooks/lib/technology-stack-detector.js

@@ -0,0 +1,374 @@
+/**
+ * 技术栈检测模块（配置驱动）
+ *
+ * 通过分析构建文件识别技术栈，规则从配置文件加载。
+ */
+
+import fs from 'fs';
+import path from 'path';
+import { getBuildFileRules, getFrameworkRules, getIgnoreDirectories } from './environment-config-loader.js';
+
+/**
+ * 获取构建文件规则（从配置加载）
+ */
+function getBuildFileRulesWithParsers() {
+  const rules = getBuildFileRules();
+  const rulesWithParsers = {};
+
+  for (const [filename, rule] of Object.entries(rules)) {
+    rulesWithParsers[filename] = {
+      ...rule,
+      parser: getParserForFile(filename)
+    };
+  }
+
+  return rulesWithParsers;
+}
+
+/**
+ * 获取文件对应的解析器
+ */
+function getParserForFile(filename) {
+  const parserMap = {
+    'package.json': parsePackageJson,
+    'pom.xml': parsePomXml,
+    'build.gradle': parseBuildGradle,
+    'build.gradle.kts': parseBuildGradle,
+    'go.mod': parseGoMod,
+    'Cargo.toml': parseCargoToml,
+    'pyproject.toml': parsePyprojectToml,
+    'requirements.txt': parseRequirements,
+    'setup.py': parseSetupPy,
+    'dbt_project.yml': parseDbtProject,
+    'pubspec.yaml': parsePubspec,
+    'composer.json': parseComposer,
+    'Gemfile': parseGemfile
+  };
+
+  return parserMap[filename] || parseGeneric;
+}
+
+/**
+ * 检测技术栈
+ * @param {string} projectDir - 项目根目录
+ * @returns {Array} 技术栈列表
+ */
+export function detectTechStacks(projectDir) {
+  const techStacks = [];
+  const buildFileRules = getBuildFileRulesWithParsers();
+  const frameworkRules = getFrameworkRules();
+
+  // 扫描所有构建文件
+  const buildFiles = scanBuildFiles(projectDir, Object.keys(buildFileRules));
+
+  for (const file of buildFiles) {
+    const rule = buildFileRules[file.name];
+    if (!rule) continue;
+
+    const content = fs.readFileSync(file.absolutePath, 'utf-8');
+    const details = rule.parser(content, file.relativePath, frameworkRules);
+
+    if (details) {
+      techStacks.push({
+        buildFile: file.relativePath,
+        path: path.dirname(file.relativePath) || '.',
+        buildTool: rule.buildTool,
+        lang: rule.lang,
+        framework: details.framework,
+        type: details.type,
+        dependencies: details.dependencies || [],
+        confidence: 'high'  // 基于构建文件的检测是高置信度
+      });
+    }
+  }
+
+  return techStacks;
+}
+
+/**
+ * 扫描构建文件
+ */
+function scanBuildFiles(projectDir, buildFileNames) {
+  const buildFiles = [];
+  const ignoreDirs = getIgnoreDirectories();
+
+  function scan(dir, depth) {
+    if (depth > 3) return;  // 最多扫描 3 层
+
+    try {
+      const entries = fs.readdirSync(dir, { withFileTypes: true });
+
+      for (const entry of entries) {
+        const fullPath = path.join(dir, entry.name);
+
+        if (entry.isDirectory()) {
+          // 跳过忽略目录
+          if (!shouldIgnoreDir(entry.name, ignoreDirs)) {
+            scan(fullPath, depth + 1);
+          }
+        } else if (entry.isFile()) {
+          // 检查是否是构建文件
+          if (buildFileNames.includes(entry.name)) {
+            buildFiles.push({
+              name: entry.name,
+              relativePath: path.relative(projectDir, fullPath),
+              absolutePath: fullPath
+            });
+          }
+        }
+      }
+    } catch (e) {
+      // 忽略无权限目录
+    }
+  }
+
+  scan(projectDir, 0);
+  return buildFiles;
+}
+
+/**
+ * 判断是否应该忽略的目录
+ */
+function shouldIgnoreDir(name, ignoreDirs) {
+  return ignoreDirs.includes(name) || name.startsWith('.');
+}
+
+// ============================================================
+// 构建文件解析器
+// ============================================================
+
+/**
+ * 解析 package.json
+ */
+function parsePackageJson(content, filePath, frameworkRules) {
+  try {
+    const pkg = JSON.parse(content);
+    const deps = { ...pkg.dependencies, ...pkg.devDependencies };
+    const depNames = Object.keys(deps);
+
+    // 匹配框架（优先使用传入的规则）
+    const rules = frameworkRules?.javascript || [];
+    for (const rule of rules) {
+      if (rule.deps.some(d => depNames.includes(d))) {
+        return {
+          framework: rule.framework,
+          type: rule.type,
+          dependencies: depNames
+        };
+      }
+    }
+
+    // 默认
+    return {
+      framework: pkg.name || 'Node.js',
+      type: inferTypeFromDeps(depNames, 'javascript'),
+      dependencies: depNames
+    };
+  } catch (e) {
+    return { framework: 'Node.js', type: 'unknown' };
+  }
+}
+
+/**
+ * 解析 pom.xml
+ */
+function parsePomXml(content, filePath, frameworkRules) {
+  const rules = frameworkRules?.java || [];
+
+  for (const rule of rules) {
+    if (rule.deps.some(d => content.includes(d))) {
+      return { framework: rule.framework, type: rule.type };
+    }
+  }
+
+  return { framework: 'Java', type: 'backend' };
+}
+
+/**
+ * 解析 build.gradle
+ */
+function parseBuildGradle(content, filePath, frameworkRules) {
+  return parsePomXml(content, filePath, frameworkRules);
+}
+
+/**
+ * 解析 go.mod
+ */
+function parseGoMod(content, filePath, frameworkRules) {
+  const rules = frameworkRules?.go || [];
+
+  for (const rule of rules) {
+    if (rule.deps.some(d => content.includes(d))) {
+      return { framework: rule.framework, type: rule.type };
+    }
+  }
+
+  // 提取模块名
+  const moduleMatch = content.match(/module\s+(\S+)/);
+  const moduleName = moduleMatch ? moduleMatch[1] : 'Go';
+
+  return { framework: moduleName, type: 'backend' };
+}
+
+/**
+ * 解析 Cargo.toml
+ */
+function parseCargoToml(content, filePath, frameworkRules) {
+  const rules = frameworkRules?.rust || [];
+
+  for (const rule of rules) {
+    if (rule.deps.some(d => content.includes(d))) {
+      return { framework: rule.framework, type: rule.type };
+    }
+  }
+
+  return { framework: 'Rust', type: 'backend' };
+}
+
+/**
+ * 解析 pyproject.toml
+ */
+function parsePyprojectToml(content, filePath, frameworkRules) {
+  return parsePythonDeps(content, filePath, frameworkRules);
+}
+
+/**
+ * 解析 requirements.txt
+ */
+function parseRequirements(content, filePath, frameworkRules) {
+  return parsePythonDeps(content, filePath, frameworkRules);
+}
+
+/**
+ * 解析 setup.py
+ */
+function parseSetupPy(content, filePath, frameworkRules) {
+  return parsePythonDeps(content, filePath, frameworkRules);
+}
+
+/**
+ * 解析 Python 依赖
+ */
+function parsePythonDeps(content, filePath, frameworkRules) {
+  const rules = frameworkRules?.python || [];
+
+  for (const rule of rules) {
+    if (rule.deps.some(d => content.toLowerCase().includes(d.toLowerCase()))) {
+      return {
+        framework: rule.framework,
+        type: rule.type,
+        dependencies: extractDependencies(content)
+      };
+    }
+  }
+
+  return {
+    framework: 'Python',
+    type: inferTypeFromDeps(extractDependencies(content), 'python'),
+    dependencies: extractDependencies(content)
+  };
+}
+
+/**
+ * 解析 dbt_project.yml
+ */
+function parseDbtProject(content, filePath, frameworkRules) {
+  return {
+    framework: 'DBT',
+    type: 'data'
+  };
+}
+
+/**
+ * 解析 pubspec.yaml
+ */
+function parsePubspec(content, filePath, frameworkRules) {
+  const hasFlutter = content.includes('flutter');
+
+  return {
+    framework: hasFlutter ? 'Flutter' : 'Dart',
+    type: hasFlutter ? 'mobile' : 'unknown'
+  };
+}
+
+/**
+ * 解析 composer.json
+ */
+function parseComposer(content, filePath, frameworkRules) {
+  try {
+    const pkg = JSON.parse(content);
+    const deps = { ...pkg.require, ...pkg['require-dev'] };
+    const depNames = Object.keys(deps);
+
+    // 检查常见 PHP 框架
+    if (depNames.some(d => d.includes('laravel'))) {
+      return { framework: 'Laravel', type: 'backend', dependencies: depNames };
+    }
+    if (depNames.some(d => d.includes('symfony'))) {
+      return { framework: 'Symfony', type: 'backend', dependencies: depNames };
+    }
+
+    return { framework: 'PHP', type: 'backend', dependencies: depNames };
+  } catch (e) {
+    return { framework: 'PHP', type: 'unknown' };
+  }
+}
+
+/**
+ * 解析 Gemfile
+ */
+function parseGemfile(content, filePath, frameworkRules) {
+  if (content.includes('rails')) {
+    return { framework: 'Rails', type: 'backend' };
+  }
+  if (content.includes('sinatra')) {
+    return { framework: 'Sinatra', type: 'backend' };
+  }
+  return { framework: 'Ruby', type: 'backend' };
+}
+
+/**
+ * 通用解析器（fallback）
+ */
+function parseGeneric(content, filePath, frameworkRules) {
+  return { framework: 'Unknown', type: 'unknown' };
+}
+
+// ============================================================
+// 辅助函数
+// ============================================================
+
+/**
+ * 从内容中提取依赖名
+ */
+function extractDependencies(content) {
+  const deps = [];
+
+  // 匹配 pip 格式：package==version 或 package>=version
+  const pipMatches = content.match(/^[a-zA-Z0-9_-]+/gm);
+  if (pipMatches) {
+    deps.push(...pipMatches.filter(d => d && !d.startsWith('#')));
+  }
+
+  return [...new Set(deps)];
+}
+
+/**
+ * 根据依赖推断类型
+ */
+function inferTypeFromDeps(deps, lang) {
+  const backendIndicators = ['api', 'server', 'web', 'http', 'rest', 'grpc'];
+  const frontendIndicators = ['react', 'vue', 'angular', 'svelte', 'ui', 'css'];
+
+  const depStr = deps.join(' ').toLowerCase();
+
+  if (frontendIndicators.some(i => depStr.includes(i))) {
+    return 'frontend';
+  }
+
+  if (backendIndicators.some(i => depStr.includes(i))) {
+    return 'backend';
+  }
+
+  return 'unknown';
+}