1
import express, { Router } from "express";
2
import fs from "fs";
3
import path from "path";
4
import  chalk from "chalk"
5
// 创建路由对象
6
const router: Router = express.Router();
7

8
// 用于存储已存在的路由路径
9
const existingPaths = new Set();
10

11
function loadRoutes(routeFolder: string, parentPath = "") {
12
  const files = fs.readdirSync(routeFolder);
13
  files.forEach(async (file) => {
14
    const filePath = path.join(routeFolder, file);
15
    const isDirectory = fs.statSync(filePath).isDirectory();
16
    if (isDirectory) {
17
      // 如果是文件夹，递归加载子文件夹中的路由文件
18
      let subfolder = path.join(parentPath, file);
19
      subfolder = subfolder.startsWith("/") ? subfolder : `/${subfolder}`;
20
      loadRoutes(filePath, subfolder);
21
    } else if ((file.endsWith(".js") || file.endsWith(".ts")) && !file.endsWith(".d.ts")) {
22
      // 如果是路由文件，导入并注册路由
23
      const routePath = path.parse(file).name === 'index' ? `${parentPath}` : `${parentPath}/${path.parse(file).name}`;
24
      // 检查是否存在重复路由路径
25
      if (existingPaths.has(routePath)) {
26
        console.warn(chalk.yellow("[Warning]"),`重复的路由路径: ${routePath} (${filePath})`);
27
        // 可以采取适当的措施来处理重复路径，如抛出异常或其他处理
28
      } else {
29
        existingPaths.add(routePath);
30
        const route = await import(filePath);
31
        router.use(routePath, route.router);
32
      }
33
    }
34
  });
35
}
36

37
// 调用函数开始递归加载路由文件
38
const routesFolderPath = path.join(__dirname, "routes");
39
loadRoutes(routesFolderPath);
40

41
router.use(apiLimiter);
42

43
export default router;