require("dotenv").config();
const Koa = require("koa");
const Router = require("koa-router");
const jwt = require("jsonwebtoken");
const koaJwt = require("koa-jwt");
const fs = require("fs");
const path = require("path");
const bodyParser = require("koa-bodyparser");

const app = new Koa();
app.use(bodyParser()); // 添加在路由中间件之前
const router = new Router();

// 自动加载API路由函数
const loadAPIRoutes = () => {
  const apiDir = path.join(__dirname, "api");
  const files = fs.readdirSync(apiDir);

  files.forEach((file) => {
    if (file.endsWith(".js") && file !== "index.js") {
      const routePath = `/${file.replace(".js", "")}`;
      const handler = require(path.join(apiDir, file));

      router.post(routePath, async (ctx) => {
        await handler(ctx);
      });
    }
  });
};

// 公开路由
router.get("/public", (ctx) => {
  ctx.body = "Public content";
});

// 登录路由
router.post("/login", (ctx) => {
  const user = { id: 1, username: "admin" };
  const token = jwt.sign(user, process.env.JWT_SECRET, { expiresIn: "1h" });
  ctx.body = { token };
});

// JWT中间件（保护下方所有路由）
app.use(
  koaJwt({
    secret: process.env.JWT_SECRET,
  }).unless({
    path: [/^\/public/, /^\/login/],
  })
);

// 加载自动生成的路由
loadAPIRoutes();

// 受保护路由
router.get("/protected", (ctx) => {
  ctx.body = `Protected content for ${ctx.state.user.username}`;
});

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

app.listen(process.env.PORT || 3000, () => {
  console.log(`Server running on http://localhost:${process.env.PORT || 3000}`);
});