import { compose } from './compose.ts' import { Context } from './context.ts' import type { ExecutionContext } from './context.ts' import type { Router } from './router.ts' import { METHOD_NAME_ALL, METHOD_NAME_ALL_LOWERCASE, METHODS } from './router.ts' import { RegExpRouter } from './router/reg-exp-router/index.ts' import { SmartRouter } from './router/smart-router/index.ts' import { StaticRouter } from './router/static-router/index.ts' import { TrieRouter } from './router/trie-router/index.ts' import type { TypeResponse, HandlerInterface, ToAppType, Handler, ErrorHandler, NotFoundHandler, Environment, Route, } from './types.ts' import { getPathFromURL, mergePath } from './utils/url.ts' type Methods = typeof METHODS[number] | typeof METHOD_NAME_ALL_LOWERCASE interface RouterRoute { path: string method: string handler: Handler } function defineDynamicClass(): { new < E extends Partial = {}, // eslint-disable-next-line @typescript-eslint/no-unused-vars _M extends string = string, P extends string = string, I = {}, O = {} >(): { [M in Methods]: HandlerInterface } } { return class {} as never } export class Hono< E extends Partial = {}, R extends Route = Route, I = {}, O = {} > extends defineDynamicClass() { readonly router: Router = new SmartRouter({ routers: [new StaticRouter(), new RegExpRouter(), new TrieRouter()], }) readonly strict: boolean = true // strict routing - default is true private _tempPath: string = '' private path: string = '/' routes: RouterRoute[] = [] constructor(init: Partial> = {}) { super() const allMethods = [...METHODS, METHOD_NAME_ALL_LOWERCASE] allMethods.map((method) => { // eslint-disable-next-line @typescript-eslint/ban-ts-comment // @ts-ignore this[method] = (args1: string | Handler, ...args: Handler[]) => { if (typeof args1 === 'string') { this.path = args1 } else { this.addRoute(method, this.path, args1) } args.map((handler) => { if (typeof handler !== 'string') { this.addRoute(method, this.path, handler) } }) return this } }) Object.assign(this, init) } private notFoundHandler: NotFoundHandler = (c: Context) => { return c.text('404 Not Found', 404) } private errorHandler: ErrorHandler = (err: Error, c: Context) => { console.trace(err) const message = 'Internal Server Error' return c.text(message, 500) } // eslint-disable-next-line @typescript-eslint/no-explicit-any route(path: string, app?: Hono) { this._tempPath = path if (app) { app.routes.map((r) => { this.addRoute(r.method, r.path, r.handler) }) this._tempPath = '' } return this } use(...middleware: Handler[]): Hono use = E>( arg1: Path, ...middleware: Handler[] ): Hono use(arg1: string | Handler, ...handlers: Handler[]) { if (typeof arg1 === 'string') { this.path = arg1 } else { handlers.unshift(arg1) } handlers.map((handler) => { this.addRoute(METHOD_NAME_ALL, this.path, handler) }) return this } on( method: Method, path: Path, ...handlers: Handler[] ): Hono on(method: string, path: string, ...handlers: Handler[]) { if (!method) return this this.path = path handlers.map((handler) => { this.addRoute(method.toUpperCase(), this.path, handler) }) return this } build = (): ToAppType => { const app = {} as typeof this type AppType = ToAppType return {} as AppType } onError(handler: ErrorHandler) { this.errorHandler = handler return this } notFound(handler: NotFoundHandler) { this.notFoundHandler = handler return this } showRoutes() { const length = 8 this.routes.map((route) => { console.log( `\x1b[32m${route.method}\x1b[0m ${' '.repeat(length - route.method.length)} ${route.path}` ) }) } // eslint-disable-next-line @typescript-eslint/no-explicit-any private addRoute(method: string, path: string, handler: Handler) { method = method.toUpperCase() if (this._tempPath) { path = mergePath(this._tempPath, path) } this.router.add(method, path, handler) const r: RouterRoute = { path: path, method: method, handler: handler } this.routes.push(r) } private matchRoute(method: string, path: string) { return this.router.match(method, path) } private handleError(err: unknown, c: Context) { if (err instanceof Error) { return this.errorHandler(err, c) } throw err } private dispatch( request: Request, eventOrExecutionCtx?: ExecutionContext | FetchEvent, env?: E['Bindings'] ): Response | Promise { const [path, queryIndex] = getPathFromURL(request.url, this.strict) const method = request.method const result = this.matchRoute(method, path) const paramData = result?.params const c = new Context(request, { env, executionCtx: eventOrExecutionCtx, notFoundHandler: this.notFoundHandler, paramData, queryIndex, }) // Do not `compose` if it has only one handler if (result && result.handlers.length === 1) { const handler = result.handlers[0] as unknown as Handler let res: ReturnType try { res = handler(c, async () => {}) if (!res) { return this.notFoundHandler(c) } } catch (err) { return this.handleError(err, c) } if ('response' in res) { res = res.response } if (res instanceof Response) { return res } return (async () => { let awaited: Response | TypeResponse | void try { awaited = await res if (awaited !== undefined && 'response' in awaited) { awaited = awaited['response'] as Response } if (!awaited) { return this.notFoundHandler(c) } } catch (err) { return this.handleError(err, c) } return awaited })() } const handlers = result ? result.handlers : [this.notFoundHandler] const composed = compose, E>(handlers, this.notFoundHandler, this.errorHandler) return (async () => { try { const tmp = composed(c) const context = tmp instanceof Promise ? await tmp : tmp if (!context.finalized) { throw new Error( 'Context is not finalized. You may forget returning Response object or `await next()`' ) } return context.res } catch (err) { return this.handleError(err, c) } })() } handleEvent = (event: FetchEvent) => { return this.dispatch(event.request, event) } fetch = (request: Request, Environment?: E['Bindings'], executionCtx?: ExecutionContext) => { return this.dispatch(request, executionCtx, Environment) } request = async (input: Request | string, requestInit?: RequestInit) => { const req = input instanceof Request ? input : new Request(input, requestInit) return await this.fetch(req) } }