ppai/server/main.js

import 'dotenv/config';

import mongoose from 'mongoose';
import express from 'express';
import expressWs from 'express-ws';
import cookieParser from 'cookie-parser';
import Logs from './libs/logs.js';
import authRoutes from './routes/auth.js';
import gamesRoutes from './routes/games.js';
import localesRoutes from './routes/locales.js';
import platformsRoutes from './routes/platforms.js';
import requireAuth from './middleware/requireAuth.js';
import requireInternalToken from './middleware/requireInternalToken.js';

const app = express();
const wsInstance = expressWs(app);

// 添加 CORS 支持.env
app.use((req, res, next) => {
  const origin = req.headers.origin;
  res.header('Access-Control-Allow-Origin', origin || '*');
  res.header('Access-Control-Allow-Credentials', 'true');
  res.header('Access-Control-Allow-Headers', 'Origin, X-Requested-With, Content-Type, Accept, Authorization, X-Internal-Token');
  res.header('Access-Control-Allow-Methods', 'GET, POST, PUT, DELETE, OPTIONS');
  res.header('Vary', 'Origin');

  if (req.method === 'OPTIONS') {
    return res.sendStatus(200);
  }
  next();
});

// 中间件
app.use(express.json({ limit: '10mb' }));
app.use(cookieParser());

app.use((req, res, next) => {
  res.badRequest = (data, msg) => {
    if (!msg && typeof data === 'string') {
      msg = data;
      data = undefined;
    }
    return res.status(400).json({ statusCode: 400, code: -1, message: msg ?? 'Bad Request', data });
  }
  res.unauthorized = (data, msg) => {
    if (!msg && typeof data === 'string') {
      msg = data;
      data = undefined;
    }
    return res.status(401).json({ statusCode: 401, code: -1,  message: msg ?? 'Unauthorized', data });
  }
  res.notFound = (data, msg) => {
    if (!msg && typeof data === 'string') {
      msg = data;
      data = undefined;
    }
    return res.status(404).json({ statusCode: 404, code: -1,  message: msg ?? 'Not Found', data });
  }
  res.serverError = (data, msg) => {
    if (!msg && typeof data === 'string') {
      msg = data;
      data = undefined;
    }
    return res.status(500).json({ statusCode: 500, code: -1,  message: msg ?? 'Internal Server Error', data });
  }
  res.sendSuccess = (data, msg) => {
    const response = { statusCode: 200, code: 0,  message: msg ?? 'OK' }
    if (data) {
      response.data = data;
    }
    return res.status(200).json(response);
  }
  res.sendError = (err) => {
    if (err.cause == 400) {
      return res.badRequest(err.data, err.message);
    }
    else if (err.cause == 401) {
      return res.unauthorized(err.data, err.message);
    }
    else if (err.cause == 404) {
      return res.notFound(err.data, err.message);
    }
    else if (err.response) {
      return res.serverError(err.response.data, err.response.statusText);
    }
    return res.serverError(err.data, err.message);
  }
  next();
});

app.use('/api/auth', authRoutes);
app.use('/api/games', requireAuth, gamesRoutes);
app.use('/api/locales', requireAuth, localesRoutes);
app.use('/api/platforms', requireInternalToken, platformsRoutes);

// 启动服务
const PORT = process.env.PORT || 9020;

mongoose.connect(process.env.MONGO_URI)
.then(() => {
  Logs.out('MongoDB connected');
  app.listen(PORT, () => console.log(`Server running on port ${PORT}`));
})
.catch(Logs.err);