audio_codebuddy/server/src/app.ts

import Koa from 'koa';
import cors from '@koa/cors';
import Router from '@koa/router';
import bodyParser from '@koa/bodyparser';
import koaBody from 'koa-body';
import serve from 'koa-static';
import mount from 'koa-mount';
import path from 'path';
import http from 'http';
import { config } from './config';
import { connectDatabase } from './models';
import { errorHandler } from './middleware/errorHandler';
import { requestLogger } from './services/requestLogger';
import { httpLogger } from './services/logger.service';
import { redisService } from './services/redis.service';
import { ossService } from './services/oss.service';
import { storageService } from './services/storage.service';
import { queueService } from './services/queue.service';
import { initBookGenerationQueue, resumeInterruptedTasks } from './modules/book-generator/book-queue.processor';
import { websocketService } from './services/websocket.service';
import { initSentry, sentryErrorHandler } from './services/sentry.service';
import { xssProtection, sqlInjectionProtection } from './middleware/security';
import { performanceMonitor, getMetrics } from './middleware/performance';
import { apiRateLimiter } from './middleware/rate-limiter';
import logRoutes from './services/log.controller';
import authRoutes from './modules/auth/auth.controller';
import ttsRoutes from './modules/tts/tts.controller';
import memberRoutes from './modules/member/member.controller';
import shareRoutes from './modules/share/share.controller';
import playerRoutes from './modules/player/player.controller';
import favoritesRoutes from './modules/favorites/favorites.controller';
import preferencesRoutes from './modules/preferences/preferences.controller';
import searchRoutes from './modules/search/search.controller';
import categoriesRoutes from './modules/categories/categories.controller';
import commentsRoutes from './modules/comments/comments.controller';
import notificationsRoutes from './modules/notifications/notifications.controller';
import bgmRoutes from './modules/bgm/bgm.controller';
import audioEditRoutes from './modules/audioedit/audioedit.controller';
import langGraphRoutes from './modules/book-generator/langgraph-controller';
import aiGenerateRoutes from './modules/book-generator/ai-generate-controller';
import albumRoutes from './modules/book-generator/album-controller';
import albumManagementRoutes from './modules/book-generator/album-management.controller';
import playlistRoutes from './modules/player/playlist.controller';
import draftsRoutes from './modules/drafts/drafts.controller';
import videoGeneratorRoutes from './modules/video-generator/video-generator.controller';
import publishRoutes from './modules/publish/publish.controller';
import signRoutes from './modules/sign/sign.controller';
import subscriptionRoutes from './modules/subscription/subscription.controller';
import paymentRoutes from './modules/payment/payment.controller';
import historyRoutes from './modules/history/history.controller';
import historyBatchRoutes from './modules/history/history-batch.controller';
import { initializePlans } from './modules/subscription/subscription.service';

const app = new Koa();
const router = new Router();

// 创建 HTTP 服务器
const server = http.createServer(app.callback());

// 中间件
app.use(errorHandler);
app.use(sentryErrorHandler()); // Sentry 错误监控
app.use(performanceMonitor()); // 性能监控
app.use(httpLogger); // Winston 日志
app.use(xssProtection()); // XSS 防护
app.use(sqlInjectionProtection()); // SQL 注入防护
app.use(cors({
  origin: '*',
  allowMethods: ['GET', 'POST', 'PUT', 'DELETE', 'OPTIONS'],
  allowHeaders: ['Content-Type', 'Authorization'],
}));
// app.use(apiRateLimiter); // 全局限流 - 已关闭
app.use(koaBody({
  multipart: true,
  formidable: {
    uploadDir: path.join(process.cwd(), 'uploads', 'materials'),
    keepExtensions: true,
    maxFileSize: 100 * 1024 * 1024, // 100MB
  },
}));

// 静态文件服务 - 音频文件
app.use(mount('/uploads', serve(config.upload.dir)));

// 静态文件服务 - 视频文件
app.use(mount('/videos', serve(path.join(process.cwd(), 'public', 'videos'))));

// 健康检查
router.get('/health', (ctx) => {
  ctx.body = { code: 0, message: 'ok', data: { status: 'healthy' } };
});

// 性能指标
router.get('/api/metrics', getMetrics);

// 注册路由
router.use('/api/auth', authRoutes.routes());
router.use('/api/tts', ttsRoutes.routes());
router.use('/api/member', memberRoutes.routes());
router.use('/api/share', shareRoutes.routes());
router.use('/api/player', playerRoutes.routes());
router.use('/api/favorites', favoritesRoutes.routes());
router.use('/api/user/preferences', preferencesRoutes.routes());
router.use('/api/search', searchRoutes.routes());
router.use('/api/categories', categoriesRoutes.routes());
router.use('/api/comments', commentsRoutes.routes());
router.use('/api/notifications', notificationsRoutes.routes());
router.use('/api/bgm', bgmRoutes.routes());
router.use('/api/audio', audioEditRoutes.routes());
router.use('/api/book-generator/langgraph', langGraphRoutes.routes());
router.use('/api/book-generator', aiGenerateRoutes.routes());
router.use('/api/book-generator', albumRoutes.routes());
router.use('/api/book-generator/album', albumManagementRoutes.routes());
router.use('/api/playlists', playlistRoutes.routes());
router.use('/api/drafts', draftsRoutes.routes());
router.use('/api/history', historyRoutes.routes());
router.use('/api/history', historyBatchRoutes.routes());
router.use('/api/video', videoGeneratorRoutes.routes());
router.use('/api/sign', signRoutes.routes());
router.use('/api/subscription', subscriptionRoutes.routes());
router.use('/api/payment', paymentRoutes.routes());
router.use('/api/publish', publishRoutes.routes());
router.use('/api', logRoutes.routes());

app.use(router.routes()).use(router.allowedMethods());

// 启动服务
async function start() {
  try {
    // 0. 初始化 Sentry
    initSentry();
    
    // 1. 连接数据库
    await connectDatabase();
    console.log('✅ MySQL 连接成功');
    
    // 2. 测试 Redis 连接
    const redisOk = await redisService.testConnection();
    console.log(redisOk ? '✅ Redis 连接成功' : '⚠️  Redis 连接失败（缓存功能将不可用）');
    
    // 3. 测试存储连接
    const storageOk = await storageService.testConnection();
    const storageType = storageService.getStorageType();
    console.log(storageOk 
      ? `✅ 存储连接成功 (${storageType === 'oss' ? '阿里云 OSS' : '本地存储'})` 
      : '⚠️  存储连接失败');
    
    // 4. 初始化订阅套餐数据
    await initializePlans();
    
    // 5. 初始化 WebSocket
    websocketService.initialize(server);
    
    // 6. 启动服务
    server.listen(config.port, () => {
      console.log(`🚀 服务启动成功: http://localhost:${config.port}`);
      console.log(`📁 上传目录: ${config.upload.dir}`);
      console.log(`💾 存储模式: ${storageType === 'oss' ? '阿里云 OSS' : '本地存储'}`);
      console.log(`⚡ 缓存模式: ${redisOk ? 'Redis' : '内存'}`);
      console.log(`🔌 WebSocket: ws://localhost:${config.port}/ws`);
      console.log(`\n💡 提示: 修改 .env 中的 STORAGE_TYPE 可切换存储模式`);
    });
    
    // 7. 初始化书籍生成队列处理器
    initBookGenerationQueue();
    
    // 自动恢复中断的任务
    resumeInterruptedTasks();
    
    // 7. 优雅关闭
    process.on('SIGTERM', async () => {
      console.log('收到 SIGTERM 信号，正在关闭服务...');
      websocketService.close();
      await queueService.closeAll();
      await redisService.disconnect();
      process.exit(0);
    });
    
    process.on('SIGINT', async () => {
      console.log('收到 SIGINT 信号，正在关闭服务...');
      websocketService.close();
      await queueService.closeAll();
      await redisService.disconnect();
      process.exit(0);
    });
  } catch (error) {
    console.error('❌ 服务启动失败:', error);
    process.exit(1);
  }
}

start();