71 lines
1.5 KiB
JavaScript
71 lines
1.5 KiB
JavaScript
const fs = require("fs");
|
|
const path = require("path");
|
|
const Router = require("koa-router");
|
|
|
|
const router = new Router();
|
|
|
|
// 全部路由数据
|
|
const allRouterInfo = {
|
|
name: "全部接口",
|
|
subtitle: "除了特殊接口外的全部接口列表",
|
|
total: 0,
|
|
data: [],
|
|
};
|
|
|
|
// 根目录
|
|
router.get("/", async (ctx) => {
|
|
await ctx.render("index");
|
|
});
|
|
|
|
// 遍历所有路由模块
|
|
fs.readdirSync(__dirname)
|
|
.filter((filename) => filename.endsWith(".js") && filename !== "index.js")
|
|
.forEach((filename) => {
|
|
const routerPath = path.join(__dirname, filename);
|
|
const routerModule = require(routerPath);
|
|
// 自动注册路由
|
|
if (routerModule instanceof Router) {
|
|
// 写入路由数据
|
|
if (routerModule?.info) {
|
|
allRouterInfo.total++;
|
|
allRouterInfo.data.push({
|
|
...routerModule.info,
|
|
stack: routerModule.stack,
|
|
});
|
|
}
|
|
// 引用路由
|
|
router.use(routerModule.routes());
|
|
}
|
|
});
|
|
|
|
// 全部接口路由
|
|
router.get("/all", async (ctx) => {
|
|
console.log("获取全部接口路由");
|
|
if (allRouterInfo.total > 0) {
|
|
ctx.body = {
|
|
code: 200,
|
|
message: "获取成功",
|
|
...allRouterInfo,
|
|
};
|
|
} else if (allRouterInfo.total === 0) {
|
|
ctx.body = {
|
|
code: 200,
|
|
message: "暂无接口，请添加",
|
|
...allRouterInfo,
|
|
};
|
|
} else {
|
|
ctx.body = {
|
|
code: 500,
|
|
message: "获取失败",
|
|
...allRouterInfo,
|
|
};
|
|
}
|
|
});
|
|
|
|
// 404 路由
|
|
router.use(async (ctx) => {
|
|
await ctx.render("404");
|
|
});
|
|
|
|
module.exports = router;
|