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 { config } from './config';
import { connectDatabase } from './models';
import { errorHandler } from './middleware/errorHandler';
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();

// 中间件
app.use(errorHandler);
app.use(cors({
  origin: '*',
  allowMethods: ['GET', 'POST', 'PUT', 'DELETE', 'OPTIONS'],
  allowHeaders: ['Content-Type', 'Authorization'],
}));
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.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());

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

// 启动服务
async function start() {
  try {
    await connectDatabase();
    console.log('✅ MySQL 连接成功');
    
    // 初始化订阅套餐数据
    await initializePlans();
    
    app.listen(config.port, () => {
      console.log(`🚀 服务启动成功: http://localhost:${config.port}`);
      console.log(`📁 上传目录: ${config.upload.dir}`);
    });
  } catch (error) {
    console.error('❌ 服务启动失败:', error);
    process.exit(1);
  }
}

start();