69 lines
1.9 KiB
TypeScript
69 lines
1.9 KiB
TypeScript
import KoaRouter from '@koa/router';
|
|
import type Koa from 'koa';
|
|
import type { Middleware } from './types.ts';
|
|
import { RouteScanner } from '../libs/route-scanner.ts';
|
|
import { ProjectController } from '../controllers/project.ts';
|
|
import { UserController } from '../controllers/user.ts';
|
|
import { AuthController } from '../controllers/auth.ts';
|
|
import { log } from '../libs/logger.ts';
|
|
|
|
export class Router implements Middleware {
|
|
private router: KoaRouter;
|
|
private routeScanner: RouteScanner;
|
|
private readonly TAG = 'Router';
|
|
|
|
constructor() {
|
|
this.router = new KoaRouter({
|
|
prefix: '/api',
|
|
});
|
|
|
|
// 初始化路由扫描器
|
|
this.routeScanner = new RouteScanner('/api');
|
|
|
|
// 注册装饰器路由
|
|
this.registerDecoratorRoutes();
|
|
|
|
// 注册传统路由（向后兼容）
|
|
this.registerTraditionalRoutes();
|
|
}
|
|
|
|
/**
|
|
* 注册装饰器路由
|
|
*/
|
|
private registerDecoratorRoutes(): void {
|
|
// 注册所有使用装饰器的控制器
|
|
this.routeScanner.registerControllers([
|
|
ProjectController,
|
|
UserController,
|
|
AuthController,
|
|
]);
|
|
|
|
// 输出注册的路由信息
|
|
const routes = this.routeScanner.getRegisteredRoutes();
|
|
log.debug(this.TAG, '装饰器路由注册完成:');
|
|
routes.forEach((route) => {
|
|
log.debug(
|
|
this.TAG,
|
|
` ${route.method} ${route.path} -> ${route.controller}.${route.action}`,
|
|
);
|
|
});
|
|
}
|
|
|
|
/**
|
|
* 注册传统路由（向后兼容）
|
|
*/
|
|
private registerTraditionalRoutes(): void {
|
|
// 保持对老版本的兼容，如果需要可以在这里注册非装饰器路由
|
|
// this.router.get('/application/list-legacy', wrapController(application.list));
|
|
}
|
|
|
|
apply(app: Koa) {
|
|
// 应用装饰器路由
|
|
this.routeScanner.applyToApp(app);
|
|
|
|
// 应用传统路由
|
|
app.use(this.router.routes());
|
|
app.use(this.router.allowedMethods());
|
|
}
|
|
}
|