77 lines
2.2 KiB
JavaScript
77 lines
2.2 KiB
JavaScript
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 verifyThirdPartyToken = require("./auth/verifyThirdPartyToken");
|
||
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 = `/api/${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";
|
||
});
|
||
|
||
// 自定义中间件：解析并验证第三方Token
|
||
app.use(async (ctx, next) => {
|
||
const authHeader = ctx.headers.authorization;
|
||
if (authHeader && authHeader.startsWith('Bearer ')) {
|
||
const thirdPartyToken = authHeader.split(' ')[1];
|
||
try {
|
||
// 这里假设第三方Token可以通过某种方式验证并转换为JWT Token
|
||
const decoded = verifyThirdPartyToken(thirdPartyToken); // 假设有一个验证函数
|
||
const jwtToken = jwt.sign(decoded, process.env.JWT_SECRET, { expiresIn: "1h" });
|
||
ctx.state.user = decoded; // 将用户信息存储在ctx.state中
|
||
ctx.headers.authorization = `Bearer ${jwtToken}`; // 替换为JWT Token
|
||
} catch (err) {
|
||
ctx.throw(401, 'Invalid third-party token');
|
||
}
|
||
}
|
||
await next();
|
||
});
|
||
|
||
// 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}`);
|
||
});
|