diff --git a/src/expresskit.ts b/src/expresskit.ts index 37cd9a8..fbaa3f8 100644 --- a/src/expresskit.ts +++ b/src/expresskit.ts @@ -38,10 +38,14 @@ export class ExpressKit { if (setup) { setup({ express: this.express, + nodekit: this.nodekit, + routes, setupBaseMiddleware: () => setupBaseMiddleware(this.nodekit.ctx, this.express), setupLangMiddleware: () => setupLangMiddleware(this.nodekit.ctx, this.express), setupParsers: () => setupParsers(this.nodekit.ctx, this.express), - setupRoutes: () => setupRoutes(this.nodekit.ctx, this.express, routes), + setupRoutes: (routesToSetup?: AppRoutes) => { + setupRoutes(this.nodekit.ctx, this.express, routesToSetup ?? routes); + }, setupErrorHandlers: () => setupErrorHandlers(this.nodekit.ctx, this.express), }); } else { diff --git a/src/types.ts b/src/types.ts index 63abc46..fa44cc8 100644 --- a/src/types.ts +++ b/src/types.ts @@ -1,4 +1,4 @@ -import type {AppContext} from '@gravity-ui/nodekit'; +import type {AppContext, NodeKit} from '@gravity-ui/nodekit'; import type bodyParser from 'body-parser'; import type { ErrorRequestHandler, @@ -169,10 +169,12 @@ export interface AppErrorHandler { export interface SetupParams { express: Express; + nodekit: NodeKit; + routes: AppRoutes; setupBaseMiddleware: () => void; setupLangMiddleware: () => void; setupParsers: () => void; - setupRoutes: () => void; + setupRoutes: (routes?: AppRoutes) => void; setupErrorHandlers: () => void; }