mr-sliy 1.0.0

package/src/routes/issueRoutes.js

@@ -0,0 +1,195 @@
+/**
+ * 缺陷路由模块
+ * 处理代码缺陷查询和管理
+ */
+
+const express = require('express');
+const router = express.Router();
+const { getDatabase } = require('../utils/database');
+const { success, error, paginate } = require('../utils/response');
+const { logger } = require('../utils/logger');
+
+/**
+ * 获取缺陷列表
+ */
+router.get('/', (req, res) => {
+  try {
+    const db = getDatabase();
+    
+    const page = parseInt(req.query.page) || 1;
+    const pageSize = parseInt(req.query.pageSize) || 20;
+    const taskId = req.query.taskId;
+    const projectId = req.query.projectId;
+    const severity = req.query.severity;
+    const issueType = req.query.issueType;
+    const isFixed = req.query.isFixed;
+    
+    // 构建查询条件
+    let whereClause = 'WHERE 1=1';
+    const params = [];
+    
+    if (taskId) {
+      whereClause += ' AND task_id = ?';
+      params.push(taskId);
+    }
+    
+    if (projectId) {
+      whereClause += ' AND project_id = ?';
+      params.push(projectId);
+    }
+    
+    if (severity) {
+      whereClause += ' AND severity = ?';
+      params.push(severity);
+    }
+    
+    if (issueType) {
+      whereClause += ' AND issue_type = ?';
+      params.push(issueType);
+    }
+    
+    if (isFixed) {
+      whereClause += ' AND is_fixed = ?';
+      params.push(isFixed === 'true' ? 1 : 0);
+    }
+    
+    // 查询总数
+    const countStmt = db.prepare(`SELECT COUNT(*) as total FROM code_issue ${whereClause}`);
+    const countResult = countStmt.get(...params);
+    const total = countResult.total;
+    
+    // 查询数据
+    const offset = (page - 1) * pageSize;
+    const dataStmt = db.prepare(`
+      SELECT * FROM code_issue ${whereClause}
+      ORDER BY created_at DESC
+      LIMIT ? OFFSET ?
+    `);
+    
+    const issues = dataStmt.all(...params, pageSize, offset);
+    
+    return res.json(paginate(issues, total, page, pageSize));
+  } catch (err) {
+    logger.error('获取缺陷列表失败:', err);
+    return res.status(500).json(error(err.message));
+  }
+});
+
+/**
+ * 获取缺陷详情
+ */
+router.get('/:id', (req, res) => {
+  try {
+    const db = getDatabase();
+    const { id } = req.params;
+    
+    const stmt = db.prepare('SELECT * FROM code_issue WHERE id = ?');
+    const issue = stmt.get(id);
+    
+    if (!issue) {
+      return res.status(404).json(error('缺陷未找到', 404));
+    }
+    
+    // 获取相关的AI优化记录
+    const optimizeStmt = db.prepare('SELECT * FROM ai_optimize_record WHERE issue_id = ?');
+    const optimizations = optimizeStmt.all(id);
+    
+    return res.json(success({
+      issue,
+      optimizations
+    }));
+  } catch (err) {
+    logger.error('获取缺陷详情失败:', err);
+    return res.status(500).json(error(err.message));
+  }
+});
+
+/**
+ * 更新缺陷状态（标记为已修复）
+ */
+router.put('/:id/fix', (req, res) => {
+  try {
+    const db = getDatabase();
+    const { id } = req.params;
+    const { userId } = req.body;
+    
+    const stmt = db.prepare(`
+      UPDATE code_issue
+      SET is_fixed = 1, fixed_at = CURRENT_TIMESTAMP, fixed_by_user_id = ?
+      WHERE id = ?
+    `);
+    
+    const result = stmt.run(userId || null, id);
+    
+    if (result.changes === 0) {
+      return res.status(404).json(error('缺陷未找到', 404));
+    }
+    
+    logger.info(`缺陷已标记为修复: ${id}`);
+    
+    return res.json(success({
+      id,
+      isFixed: true
+    }));
+  } catch (err) {
+    logger.error('更新缺陷状态失败:', err);
+    return res.status(500).json(error(err.message));
+  }
+});
+
+/**
+ * 获取缺陷统计信息
+ */
+router.get('/stats', (req, res) => {
+  try {
+    const db = getDatabase();
+    
+    // 按类型统计
+    const typeStmt = db.prepare(`
+      SELECT issue_type, COUNT(*) as count
+      FROM code_issue
+      GROUP BY issue_type
+      ORDER BY count DESC
+    `);
+    const typeStats = typeStmt.all();
+    
+    // 按严重程度统计
+    const severityStmt = db.prepare(`
+      SELECT severity, COUNT(*) as count
+      FROM code_issue
+      GROUP BY severity
+    `);
+    const severityStats = severityStmt.all();
+    
+    // 按语言统计
+    const languageStmt = db.prepare(`
+      SELECT language, COUNT(*) as count
+      FROM code_issue
+      GROUP BY language
+      ORDER BY count DESC
+    `);
+    const languageStats = languageStmt.all();
+    
+    // 总数统计
+    const totalStmt = db.prepare('SELECT COUNT(*) as total FROM code_issue');
+    const totalResult = totalStmt.get();
+    
+    // 已修复统计
+    const fixedStmt = db.prepare('SELECT COUNT(*) as fixed FROM code_issue WHERE is_fixed = 1');
+    const fixedResult = fixedStmt.get();
+    
+    return res.json(success({
+      total: totalResult.total,
+      fixed: fixedResult.fixed,
+      unfixed: totalResult.total - fixedResult.fixed,
+      typeStats,
+      severityStats,
+      languageStats
+    }));
+  } catch (err) {
+    logger.error('获取缺陷统计失败:', err);
+    return res.status(500).json(error(err.message));
+  }
+});
+
+module.exports = router;

package/src/routes/projectRoutes.js

@@ -0,0 +1,46 @@
+/**
+ * 项目路由模块
+ */
+
+const express = require('express');
+const router = express.Router();
+const { getDatabase } = require('../utils/database');
+const { success, error, paginate } = require('../utils/response');
+const { logger } = require('../utils/logger');
+
+router.get('/', (req, res) => {
+  try {
+    const db = getDatabase();
+    const page = parseInt(req.query.page) || 1;
+    const pageSize = parseInt(req.query.pageSize) || 20;
+    
+    const countStmt = db.prepare('SELECT COUNT(*) as total FROM scan_project WHERE status != "deleted"');
+    const countResult = countStmt.get();
+    
+    const offset = (page - 1) * pageSize;
+    const dataStmt = db.prepare(`SELECT * FROM scan_project WHERE status != "deleted" ORDER BY created_at DESC LIMIT ? OFFSET ?`);
+    const projects = dataStmt.all(pageSize, offset);
+    
+    return res.json(paginate(projects, countResult.total, page, pageSize));
+  } catch (err) {
+    return res.status(500).json(error(err.message));
+  }
+});
+
+router.get('/:id', (req, res) => {
+  try {
+    const db = getDatabase();
+    const stmt = db.prepare('SELECT * FROM scan_project WHERE id = ?');
+    const project = stmt.get(req.params.id);
+    
+    if (!project) {
+      return res.status(404).json(error('项目未找到', 404));
+    }
+    
+    return res.json(success(project));
+  } catch (err) {
+    return res.status(500).json(error(err.message));
+  }
+});
+
+module.exports = router;

package/src/routes/reportRoutes.js

@@ -0,0 +1,40 @@
+/**
+ * 报告路由模块
+ */
+
+const express = require('express');
+const router = express.Router();
+const { getDatabase } = require('../utils/database');
+const { success, error } = require('../utils/response');
+const { generateUUID } = require('../utils/helpers');
+
+router.post('/generate', (req, res) => {
+  try {
+    const { taskId, format, includeAiSuggestions } = req.body;
+    
+    const reportId = generateUUID();
+    const reportUrl = `http://localhost:3000/reports/${reportId}.${format || 'pdf'}`;
+    
+    return res.json(success({
+      reportId,
+      url: reportUrl,
+      format: format || 'pdf'
+    }));
+  } catch (err) {
+    return res.status(500).json(error(err.message));
+  }
+});
+
+router.get('/', (req, res) => {
+  try {
+    const db = getDatabase();
+    const stmt = db.prepare('SELECT * FROM code_report ORDER BY created_at DESC LIMIT 20');
+    const reports = stmt.all();
+    
+    return res.json(success(reports));
+  } catch (err) {
+    return res.status(500).json(error(err.message));
+  }
+});
+
+module.exports = router;

package/src/routes/scanRoutes.js

@@ -0,0 +1,245 @@
+/**
+ * 扫描路由模块
+ * 处理代码扫描相关请求
+ */
+
+const express = require('express');
+const router = express.Router();
+const { detectIssues, batchDetect, saveDetectionResults } = require('../services/detection/detector');
+const { optimizeWithRAG } = require('../services/rag/agent');
+const { getDatabase } = require('../utils/database');
+const { success, error } = require('../utils/response');
+const { logger } = require('../utils/logger');
+const { generateUUID, getFileLanguage } = require('../utils/helpers');
+const fs = require('fs');
+const path = require('path');
+
+/**
+ * 扫描单个文件
+ */
+router.post('/file', async (req, res) => {
+  const startTime = Date.now();
+  
+  try {
+    const { filePath, sourceCode, mode } = req.body;
+    
+    if (!filePath || !sourceCode) {
+      return res.status(400).json(error('缺少必要参数', 400));
+    }
+    
+    // 执行AST检测
+    const detectionResult = await detectIssues(sourceCode, filePath);
+    
+    if (!detectionResult.success) {
+      return res.json(error(detectionResult.message));
+    }
+    
+    // 如果为在线模式，执行AI优化
+    if (mode === 'online' && detectionResult.issues.length > 0) {
+      const optimizationResults = [];
+      
+      for (const issue of detectionResult.issues.slice(0, 5)) { // 限制优化数量
+        const optimization = await optimizeWithRAG(issue, {
+          language: getFileLanguage(filePath),
+          issueType: issue.issueType,
+          message: issue.message,
+          taskId: null
+        });
+        
+        optimizationResults.push(optimization);
+      }
+      
+      detectionResult.optimizations = optimizationResults;
+    }
+    
+    logger.info(`扫描文件完成: ${filePath}`);
+    
+    return res.json(success({
+      filePath,
+      language: detectionResult.language,
+      totalIssues: detectionResult.totalIssues,
+      issueCounts: detectionResult.issueCounts,
+      issues: detectionResult.issues,
+      optimizations: detectionResult.optimizations || [],
+      durationMs: Date.now() - startTime
+    }));
+  } catch (err) {
+    logger.error('扫描文件失败:', err);
+    return res.status(500).json(error(err.message));
+  }
+});
+
+/**
+ * 扫描整个项目
+ */
+router.post('/project', async (req, res) => {
+  const startTime = Date.now();
+  
+  try {
+    const { projectPath, mode } = req.body;
+    
+    if (!projectPath) {
+      return res.status(400).json(error('缺少项目路径', 400));
+    }
+    
+    // 收集所有待扫描的文件
+    const filesToScan = collectFiles(projectPath);
+    
+    if (filesToScan.length === 0) {
+      return res.json(success({
+        totalFiles: 0,
+        scannedFiles: 0,
+        totalIssues: 0,
+        durationMs: Date.now() - startTime
+      }));
+    }
+    
+    // 执行批量扫描
+    const scanResults = await batchDetect(filesToScan);
+    
+    // 创建扫描任务记录
+    const taskId = generateUUID();
+    const projectId = await createProjectRecord(projectPath);
+    
+    await createTaskRecord(taskId, projectId, {
+      scanMode: mode,
+      scanType: 'full_project',
+      targetPath: projectPath,
+      fileCount: filesToScan.length,
+      scannedFiles: scanResults.scannedFiles,
+      issueCount: scanResults.totalIssues,
+      durationMs: Date.now() - startTime
+    });
+    
+    // 存储检测结果
+    await saveDetectionResults(taskId, projectId, scanResults.results);
+    
+    logger.info(`项目扫描完成: ${projectPath}`);
+    
+    return res.json(success({
+      taskId,
+      projectId,
+      projectPath,
+      totalFiles: filesToScan.length,
+      scannedFiles: scanResults.scannedFiles,
+      failedFiles: scanResults.failedFiles,
+      totalIssues: scanResults.totalIssues,
+      results: scanResults.results,
+      durationMs: Date.now() - startTime
+    }));
+  } catch (err) {
+    logger.error('项目扫描失败:', err);
+    return res.status(500).json(error(err.message));
+  }
+});
+
+/**
+ * 批量扫描指定文件列表
+ */
+router.post('/batch', async (req, res) => {
+  const startTime = Date.now();
+  
+  try {
+    const { filePaths, mode } = req.body;
+    
+    if (!filePaths || !Array.isArray(filePaths) || filePaths.length === 0) {
+      return res.status(400).json(error('缺少文件列表', 400));
+    }
+    
+    const scanResults = await batchDetect(filePaths);
+    
+    return res.json(success({
+      totalFiles: filePaths.length,
+      scannedFiles: scanResults.scannedFiles,
+      failedFiles: scanResults.failedFiles,
+      totalIssues: scanResults.totalIssues,
+      results: scanResults.results,
+      durationMs: Date.now() - startTime
+    }));
+  } catch (err) {
+    logger.error('批量扫描失败:', err);
+    return res.status(500).json(error(err.message));
+  }
+});
+
+/**
+ * 收集待扫描的文件
+ */
+function collectFiles(projectPath, extensions = ['.js', '.ts', '.jsx', '.tsx', '.py', '.java', '.go']) {
+  const files = [];
+  const excludeDirs = ['node_modules', 'dist', 'build', 'out', '.git', 'coverage', 'vendor'];
+  
+  function walkDir(dir) {
+    try {
+      const items = fs.readdirSync(dir);
+      
+      for (const item of items) {
+        const fullPath = path.join(dir, item);
+        
+        if (excludeDirs.includes(item)) {
+          continue;
+        }
+        
+        const stat = fs.statSync(fullPath);
+        
+        if (stat.isDirectory()) {
+          walkDir(fullPath);
+        } else if (stat.isFile()) {
+          const ext = path.extname(fullPath);
+          if (extensions.includes(ext)) {
+            files.push(fullPath);
+          }
+        }
+      }
+    } catch (err) {
+      logger.warn(`无法访问目录: ${dir}`);
+    }
+  }
+  
+  walkDir(projectPath);
+  return files;
+}
+
+/**
+ * 创建项目记录
+ */
+async function createProjectRecord(projectPath) {
+  const db = getDatabase();
+  const projectName = path.basename(projectPath);
+  
+  const stmt = db.prepare(`
+    INSERT INTO scan_project (project_name, project_path, total_files)
+    VALUES (?, ?, ?)
+  `);
+  
+  const result = stmt.run(projectName, projectPath, 0);
+  return result.lastInsertRowid;
+}
+
+/**
+ * 创建任务记录
+ */
+async function createTaskRecord(taskId, projectId, data) {
+  const db = getDatabase();
+  
+  const stmt = db.prepare(`
+    INSERT INTO scan_task (id, project_id, scan_mode, scan_type, target_path, 
+                           file_count, scanned_files, issue_count, duration_ms, status)
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+  `);
+  
+  stmt.run(
+    taskId,
+    projectId,
+    data.scanMode,
+    data.scanType,
+    data.targetPath,
+    data.fileCount,
+    data.scannedFiles,
+    data.issueCount,
+    data.durationMs,
+    'completed'
+  );
+}
+
+module.exports = router;

package/src/routes/userRoutes.js

@@ -0,0 +1,47 @@
+/**
+ * 用户路由模块
+ */
+
+const express = require('express');
+const router = express.Router();
+const { getDatabase } = require('../utils/database');
+const { success, error, paginate } = require('../utils/response');
+
+router.get('/', (req, res) => {
+  try {
+    const db = getDatabase();
+    const page = parseInt(req.query.page) || 1;
+    const pageSize = parseInt(req.query.pageSize) || 20;
+    
+    const countStmt = db.prepare('SELECT COUNT(*) as total FROM sys_user');
+    const countResult = countStmt.get();
+    
+    const offset = (page - 1) * pageSize;
+    const dataStmt = db.prepare('SELECT id, username, email, role, status, created_at FROM sys_user ORDER BY created_at DESC LIMIT ? OFFSET ?');
+    const users = dataStmt.all(pageSize, offset);
+    
+    return res.json(paginate(users, countResult.total, page, pageSize));
+  } catch (err) {
+    return res.status(500).json(error(err.message));
+  }
+});
+
+router.post('/login', (req, res) => {
+  try {
+    const db = getDatabase();
+    const { username, password } = req.body;
+    
+    const stmt = db.prepare('SELECT * FROM sys_user WHERE username = ? AND password_hash = ?');
+    const user = stmt.get(username, password);
+    
+    if (!user) {
+      return res.status(401).json(error('用户名或密码错误', 401));
+    }
+    
+    return res.json(success({ id: user.id, username: user.username, role: user.role }));
+  } catch (err) {
+    return res.status(500).json(error(err.message));
+  }
+});
+
+module.exports = router;