mr-sliy 1.0.0

package/src/index.js

@@ -0,0 +1,165 @@
+/**
+ * Express应用主文件
+ * 创建并配置Express服务器
+ */
+
+const express = require('express');
+const path = require('path');
+const { config, validate } = require('./config');
+const { logger, requestLogger } = require('./utils/logger');
+const { closeDatabase } = require('./utils/database');
+const {
+  securityMiddleware,
+  corsMiddleware,
+  rateLimitMiddleware,
+  bodyParserMiddleware,
+  errorHandler,
+  notFoundHandler
+} = require('./middlewares');
+
+// 导入路由
+const routes = require('./routes');
+
+// 创建Express应用
+const app = express();
+
+// 验证配置
+if (!validate()) {
+  process.exit(1);
+}
+
+// ============================================
+// 配置中间件
+// ============================================
+
+// 安全中间件
+app.use(securityMiddleware);
+
+// CORS中间件
+app.use(corsMiddleware);
+
+// 请求体解析中间件
+app.use(bodyParserMiddleware);
+
+// 请求日志中间件
+app.use(requestLogger);
+
+// 静态文件服务（用于报告文件）
+const reportsDir = path.join(__dirname, '..', 'reports');
+app.use('/reports', express.static(reportsDir));
+
+// ============================================
+// API路由
+// ============================================
+
+// API路由前缀
+app.use('/api', rateLimitMiddleware);
+
+// 注册业务路由
+app.use('/api', routes);
+
+// 健康检查路由
+app.get('/health', (req, res) => {
+  res.json({
+    success: true,
+    code: 200,
+    message: '服务运行正常',
+    data: {
+      status: 'healthy',
+      mode: config.defaultMode,
+      timestamp: new Date().toISOString()
+    }
+  });
+});
+
+// 根路由
+app.get('/', (req, res) => {
+  res.json({
+    success: true,
+    code: 200,
+    message: '代码优化智能体API服务',
+    data: {
+      name: 'Code Optimizer Agent',
+      version: '1.0.0',
+      description: '基于Tree-sitter与RAG的多语言代码优化智能体',
+      endpoints: {
+        health: '/health',
+        api: '/api',
+        docs: '/api/docs'
+      }
+    }
+  });
+});
+
+// ============================================
+// 错误处理
+// ============================================
+
+// 404处理
+app.use(notFoundHandler);
+
+// 全局错误处理
+app.use(errorHandler);
+
+// ============================================
+// 启动服务器
+// ============================================
+
+const PORT = config.server.port;
+const HOST = config.server.host;
+
+// 启动HTTP服务器
+const server = app.listen(PORT, HOST, () => {
+  logger.info(`服务器启动成功: http://${HOST}:${PORT}`);
+  logger.info(`运行模式: ${config.defaultMode}`);
+  logger.info(`环境: ${config.server.nodeEnv}`);
+});
+
+// ============================================
+// 进程事件处理
+// ============================================
+
+// 优雅关闭
+process.on('SIGTERM', gracefulShutdown);
+process.on('SIGINT', gracefulShutdown);
+
+// 未捕获异常处理
+process.on('uncaughtException', (err) => {
+  logger.error('未捕获异常:', err);
+  gracefulShutdown();
+});
+
+// 未处理的Promise拒绝
+process.on('unhandledRejection', (reason, promise) => {
+  logger.error('未处理的Promise拒绝:', reason);
+});
+
+/**
+ * 优雅关闭函数
+ */
+function gracefulShutdown() {
+  logger.info('正在关闭服务器...');
+  
+  server.close((err) => {
+    if (err) {
+      logger.error('关闭服务器失败:', err);
+      process.exit(1);
+    }
+    
+    logger.info('服务器已关闭');
+    
+    // 关闭数据库连接
+    closeDatabase();
+    
+    process.exit(0);
+  });
+  
+  // 强制关闭超时
+  setTimeout(() => {
+    logger.error('强制关闭服务器');
+    closeDatabase();
+    process.exit(1);
+  }, 10000);
+}
+
+module.exports = app;

package/src/middlewares/errorHandler.js

@@ -0,0 +1,166 @@
+/**
+ * Express中间件模块
+ */
+
+const helmet = require('helmet');
+const cors = require('cors');
+const rateLimit = require('express-rate-limit');
+const { config } = require('../config');
+const { logger } = require('../utils/logger');
+
+/**
+ * 安全中间件配置
+ */
+const securityMiddleware = [
+  // Helmet安全头
+  helmet({
+    contentSecurityPolicy: {
+      directives: {
+        defaultSrc: ["'self'"],
+        styleSrc: ["'self'", "'unsafe-inline'"],
+        scriptSrc: ["'self'"],
+        imgSrc: ["'self'", 'data:', 'https:'],
+      },
+    },
+    crossOriginEmbedderPolicy: false,
+  }),
+];
+
+/**
+ * CORS中间件配置
+ */
+const corsMiddleware = cors({
+  origin: config.cors.origins,
+  credentials: config.cors.credentials,
+  methods: ['GET', 'POST', 'PUT', 'DELETE', 'OPTIONS'],
+  allowedHeaders: ['Content-Type', 'Authorization', 'X-Requested-With'],
+  exposedHeaders: ['X-Total-Count', 'X-Page-Size'],
+  maxAge: 86400, // 24小时
+});
+
+/**
+ * 速率限制中间件
+ */
+const rateLimitMiddleware = rateLimit({
+  windowMs: config.rateLimit.windowMs,
+  max: config.rateLimit.max,
+  message: {
+    success: false,
+    code: 429,
+    message: '请求过于频繁，请稍后再试',
+    timestamp: new Date().toISOString()
+  },
+  standardHeaders: true,
+  legacyHeaders: false,
+  handler: (req, res, next, options) => {
+    logger.warn(`速率限制触发: ${req.ip} - ${req.method} ${req.url}`);
+    res.status(429).json(options.message);
+  }
+});
+
+/**
+ * 请求体解析中间件
+ */
+const bodyParserMiddleware = [
+  // JSON解析
+  require('express').json({
+    limit: '10mb',
+    strict: true,
+  }),
+  // URL编码解析
+  require('express').urlencoded({
+    extended: true,
+    limit: '10mb',
+  }),
+];
+
+/**
+ * 错误处理中间件
+ */
+function errorHandler(err, req, res, next) {
+  logger.error('错误处理中间件捕获:', err);
+  
+  // JWT错误
+  if (err.name === 'UnauthorizedError') {
+    return res.status(401).json({
+      success: false,
+      code: 401,
+      message: '无效的认证令牌',
+      timestamp: new Date().toISOString()
+    });
+  }
+  
+  // 验证错误
+  if (err.name === 'ValidationError') {
+    return res.status(400).json({
+      success: false,
+      code: 400,
+      message: '参数验证失败',
+      errors: err.errors,
+      timestamp: new Date().toISOString()
+    });
+  }
+  
+  // 语法错误
+  if (err instanceof SyntaxError && err.status === 400 && 'body' in err) {
+    return res.status(400).json({
+      success: false,
+      code: 400,
+      message: 'JSON解析错误',
+      timestamp: new Date().toISOString()
+    });
+  }
+  
+  // 数据库错误
+  if (err.code === 'SQLITE_CONSTRAINT') {
+    return res.status(400).json({
+      success: false,
+      code: 400,
+      message: '数据约束冲突',
+      timestamp: new Date().toISOString()
+    });
+  }
+  
+  // 默认错误
+  const statusCode = err.status || err.statusCode || 500;
+  const message = err.message || '服务器内部错误';
+  
+  res.status(statusCode).json({
+    success: false,
+    code: statusCode,
+    message,
+    errors: config.server.nodeEnv === 'development' ? err.stack : undefined,
+    timestamp: new Date().toISOString()
+  });
+}
+
+/**
+ * 404处理中间件
+ */
+function notFoundHandler(req, res, next) {
+  res.status(404).json({
+    success: false,
+    code: 404,
+    message: `路由未找到: ${req.method} ${req.url}`,
+    timestamp: new Date().toISOString()
+  });
+}
+
+/**
+ * 异步错误捕获包装器
+ */
+function asyncHandler(fn) {
+  return (req, res, next) => {
+    Promise.resolve(fn(req, res, next)).catch(next);
+  };
+}
+
+module.exports = {
+  securityMiddleware,
+  corsMiddleware,
+  rateLimitMiddleware,
+  bodyParserMiddleware,
+  errorHandler,
+  notFoundHandler,
+  asyncHandler
+};

package/src/middlewares/index.js

@@ -0,0 +1,23 @@
+/**
+ * 中间件导出模块
+ */
+
+const {
+  securityMiddleware,
+  corsMiddleware,
+  rateLimitMiddleware,
+  bodyParserMiddleware,
+  errorHandler,
+  notFoundHandler,
+  asyncHandler
+} = require('./errorHandler');
+
+module.exports = {
+  securityMiddleware,
+  corsMiddleware,
+  rateLimitMiddleware,
+  bodyParserMiddleware,
+  errorHandler,
+  notFoundHandler,
+  asyncHandler
+};

package/src/routes/aiRoutes.js

@@ -0,0 +1,117 @@
+/**
+ * AI优化路由模块
+ * 处理AI代码优化相关请求
+ */
+
+const express = require('express');
+const router = express.Router();
+const { optimizeWithRAG } = require('../services/rag/agent');
+const { success, error } = require('../utils/response');
+const { logger } = require('../utils/logger');
+const { getFileLanguage } = require('../utils/helpers');
+const { isOnlineMode } = require('../config');
+
+/**
+ * AI优化代码片段
+ */
+router.post('/optimize', async (req, res) => {
+  const startTime = Date.now();
+  
+  try {
+    const { code, filePath, language, issueType, message } = req.body;
+    
+    if (!code) {
+      return res.status(400).json(error('缺少代码片段', 400));
+    }
+    
+    // 检查是否为在线模式
+    if (!isOnlineMode()) {
+      return res.json(error('当前为离线模式，无法使用AI优化功能'));
+    }
+    
+    // 构建issue对象
+    const issue = {
+      id: null,
+      codeSnippet: code
+    };
+    
+    // 构建上下文
+    const context = {
+      language: language || getFileLanguage(filePath || 'unknown.js'),
+      issueType: issueType || 'general',
+      message: message || '优化建议',
+      taskId: null
+    };
+    
+    // 执行RAG优化
+    const result = await optimizeWithRAG(issue, context);
+    
+    if (!result.success) {
+      return res.json(error(result.message));
+    }
+    
+    logger.info('AI优化完成');
+    
+    return res.json(success({
+      optimizationId: result.optimizationId,
+      optimizedCode: result.optimizedCode,
+      explanation: result.explanation,
+      suggestions: result.suggestions,
+      similarSnippets: result.similarSnippets,
+      tokensUsed: result.tokensUsed,
+      durationMs: Date.now() - startTime
+    }));
+  } catch (err) {
+    logger.error('AI优化失败:', err);
+    return res.status(500).json(error(err.message));
+  }
+});
+
+/**
+ * 获取优化历史
+ */
+router.get('/history', (req, res) => {
+  try {
+    const limit = parseInt(req.query.limit) || 10;
+    const { getOptimizationHistory } = require('../services/rag/agent');
+    const history = getOptimizationHistory(limit);
+    
+    return res.json(success({
+      total: history.length,
+      history
+    }));
+  } catch (err) {
+    logger.error('获取优化历史失败:', err);
+    return res.status(500).json(error(err.message));
+  }
+});
+
+/**
+ * 应用优化建议
+ */
+router.post('/apply', async (req, res) => {
+  try {
+    const { optimizationId, filePath, originalCode, optimizedCode } = req.body;
+    
+    if (!filePath || !optimizedCode) {
+      return res.status(400).json(error('缺少必要参数', 400));
+    }
+    
+    // 这里应该实现实际的代码替换逻辑
+    // 由于安全考虑，实际应用中应该由前端或IDE插件完成替换
+    
+    logger.info(`优化建议已应用: ${optimizationId}`);
+    
+    return res.json(success({
+      optimizationId,
+      filePath,
+      applied: true,
+      message: '优化建议已应用，请在IDE中确认更改'
+    }));
+  } catch (err) {
+    logger.error('应用优化失败:', err);
+    return res.status(500).json(error(err.message));
+  }
+});
+
+module.exports = router;

package/src/routes/configRoutes.js

@@ -0,0 +1,31 @@
+/**
+ * 配置路由模块
+ */
+
+const express = require('express');
+const router = express.Router();
+const { getDatabase } = require('../utils/database');
+const { success, error } = require('../utils/response');
+
+router.get('/', (req, res) => {
+  try {
+    const db = getDatabase();
+    const stmt = db.prepare('SELECT config_key, config_value, config_type, description FROM sys_config WHERE is_public = 1');
+    const configs = stmt.all();
+    
+    const configMap = {};
+    configs.forEach(c => {
+      configMap[c.config_key] = {
+        value: c.config_value,
+        type: c.config_type,
+        description: c.description
+      };
+    });
+    
+    return res.json(success(configMap));
+  } catch (err) {
+    return res.status(500).json(error(err.message));
+  }
+});
+
+module.exports = router;

package/src/routes/index.js

@@ -0,0 +1,75 @@
+/**
+ * 路由注册模块
+ * 统一管理所有API路由
+ */
+
+const express = require('express');
+const router = express.Router();
+
+// 导入各业务模块路由
+const userRoutes = require('./userRoutes');
+const configRoutes = require('./configRoutes');
+const projectRoutes = require('./projectRoutes');
+const scanRoutes = require('./scanRoutes');
+const issueRoutes = require('./issueRoutes');
+const aiRoutes = require('./aiRoutes');
+const reportRoutes = require('./reportRoutes');
+
+// 注册路由
+router.use('/users', userRoutes);
+router.use('/config', configRoutes);
+router.use('/projects', projectRoutes);
+router.use('/scan', scanRoutes);
+router.use('/issues', issueRoutes);
+router.use('/ai', aiRoutes);
+router.use('/reports', reportRoutes);
+
+// API文档路由
+router.get('/docs', (req, res) => {
+  res.json({
+    success: true,
+    code: 200,
+    message: 'API文档',
+    data: {
+      endpoints: {
+        users: {
+          base: '/api/users',
+          methods: ['GET', 'POST', 'PUT', 'DELETE'],
+          description: '用户管理'
+        },
+        config: {
+          base: '/api/config',
+          methods: ['GET', 'PUT'],
+          description: '系统配置'
+        },
+        projects: {
+          base: '/api/projects',
+          methods: ['GET', 'POST', 'PUT', 'DELETE'],
+          description: '项目管理'
+        },
+        scan: {
+          base: '/api/scan',
+          methods: ['POST', 'GET'],
+          description: '代码扫描'
+        },
+        issues: {
+          base: '/api/issues',
+          methods: ['GET', 'PUT'],
+          description: '代码缺陷'
+        },
+        ai: {
+          base: '/api/ai',
+          methods: ['POST', 'GET'],
+          description: 'AI优化'
+        },
+        reports: {
+          base: '/api/reports',
+          methods: ['GET', 'POST'],
+          description: '报告管理'
+        }
+      }
+    }
+  });
+});
+
+module.exports = router;