@chongyan/autospec 1.0.2 → 1.0.4

package/plugins/hooks/lib/environment-detector.js

@@ -1,553 +0,0 @@
-/**
- * 环境知识检测器
- *
- * 检测项目使用了哪些环境知识，统一使用 PatternLoader
- */
-
-import fs from 'fs';
-import path from 'path';
-import { ENV_DIRS } from '../../constants.js';
-import { PatternLoader } from './detection-pattern-loader.js';
-import { EnvironmentRegistry } from './environment-registry.js';
-import { DirectoryDiscovery } from './directory-discovery.js';
-
-/**
- * 检测环境知识
- * @param {string} projectDir - 项目根目录
- * @param {Object} config - 项目配置
- * @returns {Object} 检测结果
- */
-export function detectEnvironmentKnowledge(projectDir, config = {}) {
-  const autospecDir = path.join(projectDir, '.autospec');
-  const envDir = path.join(autospecDir, ENV_DIRS.ROOT);
-
-  // 使用 PatternLoader 加载检测模式
-  const patternLoader = new PatternLoader(autospecDir);
-  patternLoader.load();
-
-  // 使用 DirectoryDiscovery 发现源码目录
-  const discovery = new DirectoryDiscovery(projectDir, config);
-  const sourceDirs = discovery.discoverSourceDirectories();
-
-  // 1. 从代码中检测中间件使用
-  const detectedMiddleware = detectMiddlewareFromCode(projectDir, sourceDirs, patternLoader);
-
-  // 2. 从代码中检测组件使用
-  const detectedComponents = detectComponentsFromCode(projectDir, sourceDirs, patternLoader);
-
-  // 3. 与已注册的知识对比
-  const registry = new EnvironmentRegistry(autospecDir);
-  registry.load();
-
-  const registeredMiddleware = registry.list('middleware').map(m => m.name);
-  const registeredComponents = registry.list('components').map(c => c.name);
-
-  // 4. 发现缺失
-  const missingMiddleware = detectedMiddleware.filter(
-    (m) => !registeredMiddleware.includes(m.name)
-  );
-
-  const missingComponents = detectedComponents.filter(
-    (c) => !registeredComponents.includes(c.name)
-  );
-
-  // 5. 返回建议
-  return {
-    detected: {
-      middleware: detectedMiddleware,
-      components: detectedComponents,
-    },
-    registered: {
-      middleware: registeredMiddleware,
-      components: registeredComponents,
-    },
-    missing: {
-      middleware: missingMiddleware,
-      components: missingComponents,
-    },
-    suggestions: {
-      actionItems: generateActionItems(missingMiddleware, missingComponents),
-    },
-    sourceDirs,
-  };
-}
-
-/**
- * 从代码中检测中间件使用
- */
-function detectMiddlewareFromCode(projectDir, sourceDirs, patternLoader) {
-  const detected = [];
-  const middlewarePatterns = patternLoader.getMiddlewarePatterns();
-
-  // 如果没有源码目录，尝试常见目录
-  if (sourceDirs.length === 0) {
-    sourceDirs = [{ path: 'src' }];
-  }
-
-  for (const dir of sourceDirs) {
-    const dirPath = path.join(projectDir, dir.path);
-    if (!fs.existsSync(dirPath)) continue;
-
-    for (const [name, pattern] of Object.entries(middlewarePatterns)) {
-      const evidence = scanForMiddleware(dirPath, name, pattern, projectDir);
-
-      if (evidence.length > 0) {
-        // 检查是否已检测到
-        const existing = detected.find(d => d.name === name);
-        if (existing) {
-          existing.evidence.push(...evidence);
-          existing.confidence = Math.max(existing.confidence, ...evidence.map(e => e.confidence || 0.8));
-        } else {
-          detected.push({
-            name,
-            type: pattern.type || inferMiddlewareType(name),
-            evidence,
-            confidence: Math.max(...evidence.map(e => e.confidence || 0.8)),
-          });
-        }
-      }
-    }
-  }
-
-  // 从依赖文件检测
-  const depEvidence = detectFromDependencies(projectDir, middlewarePatterns);
-  for (const dep of depEvidence) {
-    const existing = detected.find(d => d.name === dep.name);
-    if (existing) {
-      existing.evidence.push(dep.evidence);
-      existing.confidence = Math.max(existing.confidence, 0.9);
-    } else {
-      detected.push({
-        name: dep.name,
-        type: inferMiddlewareType(dep.name),
-        evidence: [dep.evidence],
-        confidence: 0.9,
-      });
-    }
-  }
-
-  // 去重证据
-  for (const item of detected) {
-    item.evidence = item.evidence.slice(0, 5);
-  }
-
-  return detected;
-}
-
-/**
- * 扫描目录检测中间件
- */
-function scanForMiddleware(dirPath, middlewareName, pattern, projectDir) {
-  const evidence = [];
-  const extensions = ['.py', '.js', '.ts', '.jsx', '.tsx', '.java', '.go', '.rs'];
-
-  const scanDir = (dir, depth = 0) => {
-    if (depth > 5) return;
-
-    try {
-      const entries = fs.readdirSync(dir, { withFileTypes: true });
-
-      for (const entry of entries) {
-        // 跳过忽略目录
-        if (entry.isDirectory()) {
-          if (['node_modules', '.git', 'dist', 'build', '__pycache__', 'venv', '.venv', 'target'].includes(entry.name)) {
-            continue;
-          }
-          scanDir(path.join(dir, entry.name), depth + 1);
-        } else if (entry.isFile()) {
-          const ext = path.extname(entry.name);
-
-          // 检查代码模式
-          if (extensions.includes(ext) && pattern.codePatterns) {
-            const filePath = path.join(dir, entry.name);
-            try {
-              const content = fs.readFileSync(filePath, 'utf-8');
-
-              for (const cp of pattern.codePatterns) {
-                try {
-                  const regex = new RegExp(cp.pattern, 'i');
-                  if (regex.test(content)) {
-                    evidence.push({
-                      type: 'code',
-                      file: path.relative(projectDir, filePath),
-                      pattern: cp.pattern,
-                      confidence: cp.confidence || 0.8,
-                    });
-                    break;
-                  }
-                } catch (e) {
-                  // 忽略无效正则
-                }
-              }
-            } catch (e) {
-              // 忽略读取错误
-            }
-          }
-
-          // 检查环境变量模式
-          if (pattern.envVarPatterns) {
-            const filePath = path.join(dir, entry.name);
-            try {
-              const content = fs.readFileSync(filePath, 'utf-8');
-
-              for (const evp of pattern.envVarPatterns) {
-                try {
-                  const regex = new RegExp(evp.pattern, 'g');
-                  const matches = content.match(regex);
-                  if (matches) {
-                    evidence.push({
-                      type: 'env-var',
-                      file: path.relative(projectDir, filePath),
-                      envVars: [...new Set(matches)],
-                      confidence: 0.85,
-                    });
-                  }
-                } catch (e) {
-                  // 忽略无效正则
-                }
-              }
-            } catch (e) {
-              // 忽略读取错误
-            }
-          }
-        }
-      }
-    } catch (e) {
-      // 忽略目录读取错误
-    }
-  };
-
-  scanDir(dirPath);
-  return evidence;
-}
-
-/**
- * 从依赖文件检测中间件
- */
-function detectFromDependencies(projectDir, middlewarePatterns) {
-  const detected = [];
-
-  // 检测 package.json
-  const pkgPath = path.join(projectDir, 'package.json');
-  if (fs.existsSync(pkgPath)) {
-    try {
-      const pkg = JSON.parse(fs.readFileSync(pkgPath, 'utf-8'));
-      const deps = { ...pkg.dependencies, ...pkg.devDependencies };
-
-      for (const [depName] of Object.entries(deps)) {
-        for (const [mwName, pattern] of Object.entries(middlewarePatterns)) {
-          if (pattern.dependencies) {
-            for (const dep of pattern.dependencies) {
-              if (dep.packageManager === 'npm' && matchDependency(dep.pattern, depName)) {
-                detected.push({
-                  name: mwName,
-                  evidence: {
-                    type: 'dependency',
-                    file: 'package.json',
-                    dependency: depName,
-                    packageManager: 'npm',
-                    confidence: 0.9,
-                  },
-                });
-              }
-            }
-          }
-        }
-      }
-    } catch (e) {
-      // 忽略解析错误
-    }
-  }
-
-  // 检测 requirements.txt
-  const reqPath = path.join(projectDir, 'requirements.txt');
-  if (fs.existsSync(reqPath)) {
-    try {
-      const content = fs.readFileSync(reqPath, 'utf-8');
-      const lines = content.split('\n');
-
-      for (const line of lines) {
-        const depName = line.split(/[=<>#\[]/)[0].trim();
-        if (!depName) continue;
-
-        for (const [mwName, pattern] of Object.entries(middlewarePatterns)) {
-          if (pattern.dependencies) {
-            for (const dep of pattern.dependencies) {
-              if (dep.packageManager === 'pip' && matchDependency(dep.pattern, depName)) {
-                detected.push({
-                  name: mwName,
-                  evidence: {
-                    type: 'dependency',
-                    file: 'requirements.txt',
-                    dependency: depName,
-                    packageManager: 'pip',
-                    confidence: 0.9,
-                  },
-                });
-              }
-            }
-          }
-        }
-      }
-    } catch (e) {
-      // 忽略解析错误
-    }
-  }
-
-  // 检测 pom.xml
-  const pomPath = path.join(projectDir, 'pom.xml');
-  if (fs.existsSync(pomPath)) {
-    try {
-      const content = fs.readFileSync(pomPath, 'utf-8');
-
-      for (const [mwName, pattern] of Object.entries(middlewarePatterns)) {
-        if (pattern.dependencies) {
-          for (const dep of pattern.dependencies) {
-            if (dep.packageManager === 'maven' && content.includes(dep.pattern)) {
-              detected.push({
-                name: mwName,
-                evidence: {
-                  type: 'dependency',
-                  file: 'pom.xml',
-                  dependency: dep.pattern,
-                  packageManager: 'maven',
-                  confidence: 0.9,
-                },
-              });
-            }
-          }
-        }
-      }
-    } catch (e) {
-      // 忽略解析错误
-    }
-  }
-
-  return detected;
-}
-
-/**
- * 匹配依赖名称
- */
-function matchDependency(pattern, depName) {
-  if (pattern.startsWith('^') || pattern.includes('*')) {
-    const regex = new RegExp('^' + pattern.replace(/\*/g, '.*') + '$');
-    return regex.test(depName);
-  }
-  return depName === pattern || depName.includes(pattern);
-}
-
-/**
- * 从代码中检测组件使用
- */
-function detectComponentsFromCode(projectDir, sourceDirs, patternLoader) {
-  const detected = [];
-  const componentPatterns = patternLoader.getComponentPatterns();
-
-  // 如果没有组件模式定义，返回空
-  if (Object.keys(componentPatterns).length === 0) {
-    return detected;
-  }
-
-  if (sourceDirs.length === 0) {
-    sourceDirs = [{ path: 'src' }];
-  }
-
-  for (const dir of sourceDirs) {
-    const dirPath = path.join(projectDir, dir.path);
-    if (!fs.existsSync(dirPath)) continue;
-
-    for (const [name, pattern] of Object.entries(componentPatterns)) {
-      const evidence = scanForComponent(dirPath, pattern, projectDir);
-
-      if (evidence.length > 0) {
-        detected.push({
-          name,
-          type: pattern.type || 'service-component',
-          evidence,
-        });
-      }
-    }
-  }
-
-  return detected;
-}
-
-/**
- * 扫描目录检测组件
- */
-function scanForComponent(dirPath, pattern, projectDir) {
-  const evidence = [];
-  const extensions = ['.py', '.js', '.ts', '.jsx', '.tsx', '.java'];
-
-  if (!pattern.codePatterns) return evidence;
-
-  const scanDir = (dir, depth = 0) => {
-    if (depth > 5) return;
-
-    try {
-      const entries = fs.readdirSync(dir, { withFileTypes: true });
-
-      for (const entry of entries) {
-        if (entry.isDirectory()) {
-          if (['node_modules', '.git', 'dist', 'build', '__pycache__'].includes(entry.name)) {
-            continue;
-          }
-          scanDir(path.join(dir, entry.name), depth + 1);
-        } else if (entry.isFile()) {
-          const ext = path.extname(entry.name);
-          if (extensions.includes(ext)) {
-            const filePath = path.join(dir, entry.name);
-            try {
-              const content = fs.readFileSync(filePath, 'utf-8');
-
-              for (const cp of pattern.codePatterns) {
-                try {
-                  const regex = new RegExp(cp.pattern, 'i');
-                  if (regex.test(content)) {
-                    evidence.push({
-                      type: 'code',
-                      file: path.relative(projectDir, filePath),
-                      pattern: cp.pattern,
-                    });
-                    break;
-                  }
-                } catch (e) {
-                  // 忽略无效正则
-                }
-              }
-            } catch (e) {
-              // 忽略读取错误
-            }
-          }
-        }
-      }
-    } catch (e) {
-      // 忽略目录读取错误
-    }
-  };
-
-  scanDir(dirPath);
-  return evidence.slice(0, 3);
-}
-
-/**
- * 推断中间件类型
- */
-function inferMiddlewareType(name) {
-  const typeMap = {
-    'mysql': 'relational-database',
-    'postgresql': 'relational-database',
-    'postgres': 'relational-database',
-    'mongodb': 'document-database',
-    'redis': 'cache',
-    'kafka': 'message-queue',
-    'rabbitmq': 'message-queue',
-    'elasticsearch': 'search-engine',
-    'elastic': 'search-engine',
-  };
-  return typeMap[name.toLowerCase()] || 'unknown';
-}
-
-/**
- * 生成行动项
- */
-function generateActionItems(missingMiddleware, missingComponents) {
-  const items = [];
-
-  for (const middleware of missingMiddleware) {
-    items.push({
-      type: 'middleware',
-      name: middleware.name,
-      action: `将自动注册并激活 ${middleware.name} 中间件知识`,
-      priority: 'high',
-    });
-  }
-
-  for (const component of missingComponents) {
-    items.push({
-      type: 'component',
-      name: component.name,
-      action: `将自动注册并激活 ${component.name} 组件知识`,
-      priority: 'medium',
-    });
-  }
-
-  return items;
-}
-
-/**
- * 检测环境变量使用
- */
-export function detectEnvVariables(projectDir, config = {}) {
-  const envVars = new Map();
-
-  const discovery = new DirectoryDiscovery(projectDir, config);
-  const sourceDirs = discovery.discoverSourceDirectories();
-
-  if (sourceDirs.length === 0) {
-    sourceDirs = [{ path: 'src' }];
-  }
-
-  const envPattern = /process\.env\.([A-Z_][A-Z0-9_]*)|os\.environ\[['"]([A-Z_][A-Z0-9_]*)['"]\]|System\.getenv\(['"]([A-Z_][A-Z0-9_]*)['"]\)/g;
-
-  for (const dir of sourceDirs) {
-    const dirPath = path.join(projectDir, dir.path);
-    if (!fs.existsSync(dirPath)) continue;
-
-    const scanDir = (currentDir, depth = 0) => {
-      if (depth > 5) return;
-
-      try {
-        const entries = fs.readdirSync(currentDir, { withFileTypes: true });
-
-        for (const entry of entries) {
-          if (entry.isDirectory()) {
-            if (['node_modules', '.git', 'dist', 'build', '__pycache__', 'venv', '.venv'].includes(entry.name)) {
-              continue;
-            }
-            scanDir(path.join(currentDir, entry.name), depth + 1);
-          } else if (entry.isFile()) {
-            const ext = path.extname(entry.name);
-            if (['.py', '.js', '.ts', '.java', '.go'].includes(ext)) {
-              const filePath = path.join(currentDir, entry.name);
-              try {
-                const content = fs.readFileSync(filePath, 'utf-8');
-                let match;
-
-                while ((match = envPattern.exec(content)) !== null) {
-                  const varName = match[1] || match[2] || match[3];
-                  if (varName) {
-                    if (!envVars.has(varName)) {
-                      envVars.set(varName, {
-                        name: varName,
-                        files: [],
-                      });
-                    }
-                    const entry = envVars.get(varName);
-                    if (entry.files.length < 3) {
-                      entry.files.push(path.relative(projectDir, filePath));
-                    }
-                  }
-                }
-              } catch (e) {
-                // 忽略读取错误
-              }
-            }
-          }
-        }
-      } catch (e) {
-        // 忽略目录读取错误
-      }
-    };
-
-    scanDir(dirPath);
-  }
-
-  return Array.from(envVars.values());
-}
-
-export default {
-  detectEnvironmentKnowledge,
-  detectEnvVariables,
-};