79 lines
1.9 KiB
TypeScript

import express from 'express'
import {IRoutes, IMethod} from '@rondo/common'
import {ITypedHandler} from './ITypedHandler'
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: ITypedHandler<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: ITypedHandler<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: ITypedHandler<R, P, 'get'>,
) {
this.addRoute('get', path, handler)
}
post<P extends keyof R & string>(
path: P,
handler: ITypedHandler<R, P, 'post'>,
) {
this.addRoute('post', path, handler)
}
put<P extends keyof R & string>(
path: P, handler: ITypedHandler<R, P, 'put'>,
) {
this.addRoute('put', path, handler)
}
delete<P extends keyof R & string>(
path: P, handler: ITypedHandler<R, P, 'delete'>) {
this.addRoute('delete', path, handler)
}
head<P extends keyof R & string>(
path: P,
handler: ITypedHandler<R, P, 'head'>,
) {
this.addRoute('head', path, handler)
}
options<P extends keyof R & string>(
path: P, handler: ITypedHandler<R, P, 'options'>) {
this.addRoute('options', path, handler)
}
patch<P extends keyof R & string>(
path: P, handler: ITypedHandler<R, P, 'patch'>) {
this.addRoute('patch', path, handler)
}
}