danmaku/app.js

const createError = require('http-errors');
const express = require('express');
const path = require('path');
const cookieParser = require('cookie-parser');
const logger = require('morgan');
const rateLimit = require('express-rate-limit');

// 引入环境变量
require('dotenv')
  .config();

// 引入一个个路由模块
const danmakuRouter = require('./routes/danmaku');
const ipinfoRouter = require('./routes/ipinfo');
const airportsubRouter = require('./routes/airportsub');
const schedule = require('./schedule/schedule');
const DEBUG = process.env.DEBUG === 'true' || false;

const app = express();

// view engine setup
app.set('views', path.join(__dirname, 'views'));
app.set('view engine', 'ejs');
app.set('trust proxy', true);

app.use(logger('dev'));
app.use(express.json());
app.use(express.urlencoded({ extended: false }));
app.use(cookieParser());
// 加载静态资源
app.use(express.static(path.join(__dirname, 'public')));
app.use('/assets', [
  express.static(__dirname + '/node_modules/jquery/dist/'),
  express.static(__dirname + '/node_modules/bootstrap/dist/'),
]);

// Rate Limit
const allowlist = ['::1', '::ffff:127.0.0.1'];
const apiLimiter = rateLimit({
  windowMs: 60 * 1000, // 1 minute
  max: 6, // limit each IP to 6 requests per windowMs
  message: 'Too many requests from this IP, please try again after an minute',
  standardHeaders: true, // Return rate limit info in the `RateLimit-*` headers
  skipFailedRequests: true, // Don't count failed requests (status >= 400)
  skip: (request, response) => allowlist.includes(request.ip),
});
app.use(apiLimiter);

// 加载路由
app.use('/', danmakuRouter);
app.use('/ipinfo', ipinfoRouter);
app.use('/sub', airportsubRouter);

// catch 404 and forward to error handler
app.use(function (req, res, next) {
  next(createError(404));
});

// error handler
app.use(function (err, req, res) {
  // set locals, only providing error in development
  res.locals.message = err.message;
  res.locals.error = req.app.get('env') === 'development' ? err : {};

  // render the error page
  res.status(err.status || 500);
  res.render('error');
});

if (!DEBUG) {
  console.log('PRODUCTION MODE!该模式下TG机器人与定时任务正常运行');
  // 引入TG机器人
  require('./tgbot/tgbot');
  // 引入定时任务模块
  const schedule = require('./schedule/schedule');
  schedule(app);
} else {
  console.log('DEBUG MODE!该模式下将关闭TG机器人与定时任务');
}

module.exports = app;