编辑代码

const express = require('express');
const cors = require('cors');
const bodyParser = require('body-parser');
const helmet = require('helmet');
const rateLimit = require('express-rate-limit');
const path = require('path');
require('dotenv').config();

const { initDatabase } = require('./config/database');
const apiRoutes = require('./routes/api');
const webhookRoutes = require('./routes/webhook');
const logger = require('./utils/logger');
const SchedulerService = require('./services/SchedulerService');

const app = express();
const PORT = process.env.PORT || 3000;

// 安全中间件
app.use(helmet({
    contentSecurityPolicy: false // 为了前端正常工作
}));

// 速率限制
const limiter = rateLimit({
    windowMs: 15 * 60 * 1000, // 15分钟
    max: 100, // 限制每个IP每15分钟最多100个请求
    message: '请求过于频繁，请稍后再试'
});
app.use('/api', limiter);

// 中间件
app.use(cors());
app.use(bodyParser.json());
app.use(bodyParser.urlencoded({ extended: true }));

// 静态文件服务
app.use(express.static(path.join(__dirname, 'public')));

// API路由
app.use('/api', apiRoutes);
app.use('/webhook', webhookRoutes);

// 首页路由
app.get('/', (req, res) => {
    res.sendFile(path.join(__dirname, 'public', 'index.html'));
});

// 错误处理中间件
app.use((err, req, res, next) => {
    logger.error('服务器错误:', err);
    res.status(500).json({
        success: false,
        message: '服务器内部错误',
        error: process.env.NODE_ENV === 'development' ? err.message : undefined
    });
});

// 404处理
app.use((req, res) => {
    res.status(404).json({
        success: false,
        message: '接口不存在'
    });
});

// 初始化数据库并启动服务器
async function startServer() {
    try {
        await initDatabase();
        logger.info('数据库初始化完成');
        
        app.listen(PORT, () => {
            logger.info(`服务器运行在端口 ${PORT}`);
            console.log(`�� 微信AccessToken管理系统已启动`);
            console.log(`�� 访问地址: http://localhost:${PORT}`);
            console.log(`�� API文档: http://localhost:${PORT}/api/docs`);
        });
        
        // 启动调度服务
        SchedulerService.init();
        
    } catch (error) {
        logger.error('服务启动失败:', error);
        process.exit(1);
    }
}

// 优雅关闭
process.on('SIGTERM', () => {
    logger.info('收到SIGTERM信号，正在关闭服务器...');
    SchedulerService.stop();
    process.exit(0);
});

process.on('SIGINT', () => {
    logger.info('收到SIGINT信号，正在关闭服务器...');
    SchedulerService.stop();
    process.exit(0);
});

startServer();