app.js

const Koa = require("koa");
const app = new Koa();
const cors = require("@koa/cors");
const views = require("koa-views");
const json = require("koa-json");
const onerror = require("koa-onerror");
const bodyparser = require("koa-bodyparser");
const logger = require("koa-logger");

const passport = require("./middleware/passport");
const pay = require("./middleware/pay");
const index = require("./routes/index");
const users = require("./routes/users");
const dailyProblem = require("./routes/dailyProblem");
const lectures = require("./routes/lectures");
const github = require("./routes/github");
const fallback = require("./routes/redirect");
const my = require("./routes/my");
const lc = require("./routes/lc");
const mockUserInfo = require("./middleware/mockUserInfo");

// error handler
onerror(app);

// middlewares
app.use(
  bodyparser({
    enableTypes: ["json", "form", "text"],
  })
);
app.use(cors({ credentials: true }));
app.use(json());
app.use(logger());

const loginWhiteList = [
  "/api/v1/github/content",
  "/api/v1/user/logout",
  "/api/v1/github/webhook",
  "/api/v1/daily-problem/ranking",
];

app.use(fallback.routes(), fallback.allowedMethods());
if (process.env.NODE_ENV === "development") {
  app.use(
    mockUserInfo({
      whitelist: loginWhiteList,
    })
  );
} else {
  app.use(
    passport({
      whitelist: loginWhiteList,
    })
  );
}
app.use(
  pay({
    whitelist: ["/api/v1/user", "/api/v1/logout"].concat(loginWhiteList),
    nextSeasonWhiteList: ["/api/v1/lectures/basic", "/api/v1/lectures/intro", "/api/v1/lectures/2001", "/api/v1/lectures/2002", "/api/v1/lectures/2003", "/api/v1/lectures/2004", "/api/v1/lectures/2005", "/api/v1/lectures/2006", "/api/v1/lectures/2007", "/api/v1/lectures/1000", "/api/v1/lectures/1001", "/api/v1/lectures/1002", "/api/v1/lectures/1003", "/api/v1/lectures/1004"],
  })
);
app.use(require("koa-static")(__dirname + "/public"));

app.use(
  views(__dirname + "/views", {
    extension: "pug",
  })
);

// logger
app.use(async (ctx, next) => {
  const start = new Date();
  await next();
  const ms = new Date() - start;
  console.log(`${ctx.method} ${ctx.url} - ${ms}ms`);
});

// routes
app.use(index.routes(), index.allowedMethods());
app.use(users.routes(), users.allowedMethods());
app.use(lectures.routes(), lectures.allowedMethods());
app.use(dailyProblem.routes(), dailyProblem.allowedMethods());
app.use(my.routes(), my.allowedMethods());
app.use(github.routes(), github.allowedMethods());
app.use(lc.routes(), lc.allowedMethods());
// error-handling
app.on("error", (err, ctx) => {
  console.error("server error", err, ctx);
});

module.exports = app;