77 lines
2.1 KiB
TypeScript
77 lines
2.1 KiB
TypeScript
import {IRoutes, IRoute, IMethod} from '../../common/REST'
|
|
import express, {NextFunction} from 'express'
|
|
|
|
export interface IRequest<T extends IRoute> extends express.Request {
|
|
body: T['body']
|
|
params: T['params']
|
|
query: T['query']
|
|
}
|
|
|
|
export type IHandler<R extends IRoutes, P extends keyof R, M extends IMethod> =
|
|
(req: IRequest<R[P][M]>, res: express.Response, next: NextFunction) =>
|
|
Promise<R[P][M]['response']>
|
|
|
|
export class AsyncRouter<R extends IRoutes> {
|
|
readonly router: express.Router
|
|
readonly use: express.IRouterHandler<void> & express.IRouterMatcher<void>
|
|
|
|
constructor(router?: express.Application | express.Router) {
|
|
this.router = router ? router : express.Router()
|
|
this.use = this.router.use.bind(this.router) as any
|
|
}
|
|
|
|
protected addRoute<M extends IMethod, P extends keyof R & string>(
|
|
method: M,
|
|
path: P,
|
|
handler: IHandler<R, P, M>,
|
|
) {
|
|
const addRoute = this.router[method].bind(this.router)
|
|
|
|
addRoute(path, this.wrapHandler(handler))
|
|
}
|
|
|
|
protected wrapHandler<M extends IMethod, P extends keyof R & string>(
|
|
handler: IHandler<R, P, M>,
|
|
): express.RequestHandler {
|
|
return (req, res, next) => {
|
|
handler(req, res, next)
|
|
.then(response => {
|
|
res.json(response)
|
|
})
|
|
.catch(next)
|
|
}
|
|
}
|
|
|
|
get<P extends keyof R & string>(path: P, handler: IHandler<R, P, 'get'>) {
|
|
this.addRoute('get', path, handler)
|
|
}
|
|
|
|
post<P extends keyof R & string>(path: P, handler: IHandler<R, P, 'post'>) {
|
|
this.addRoute('post', path, handler)
|
|
}
|
|
|
|
put<P extends keyof R & string>(path: P, handler: IHandler<R, P, 'put'>) {
|
|
this.addRoute('put', path, handler)
|
|
}
|
|
|
|
delete<P extends keyof R & string>(
|
|
path: P, handler: IHandler<R, P, 'delete'>) {
|
|
this.addRoute('delete', path, handler)
|
|
}
|
|
|
|
head<P extends keyof R & string>(path: P, handler: IHandler<R, P, 'head'>) {
|
|
this.addRoute('head', path, handler)
|
|
}
|
|
|
|
options<P extends keyof R & string>(
|
|
path: P, handler: IHandler<R, P, 'options'>) {
|
|
this.addRoute('options', path, handler)
|
|
}
|
|
|
|
patch<P extends keyof R & string>(
|
|
path: P, handler: IHandler<R, P, 'patch'>) {
|
|
this.addRoute('patch', path, handler)
|
|
}
|
|
|
|
}
|