123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132
import type { NotFoundHandler } from './hono.ts'import type { StatusCode } from './utils/http-status.ts'import { isAbsoluteURL } from './utils/url.ts'
type Headers = Record<string, string>export type Data = string | ArrayBuffer | ReadableStream
export type Env = Record<string, any>
export class Context<RequestParamKeyType extends string = string, E = Env> {  req: Request<RequestParamKeyType>  env: E  event: FetchEvent | undefined  executionCtx: ExecutionContext | undefined  finalized: boolean
  private _status: StatusCode = 200  private _pretty: boolean = false  private _prettySpace: number = 2  private _map: Record<string, any> | undefined  private _headers: Record<string, string> | undefined  private _res: Response | undefined  private notFoundHandler: NotFoundHandler
  constructor(    req: Request,    env: E | undefined = undefined,    eventOrExecutionCtx: FetchEvent | ExecutionContext | undefined = undefined,    notFoundHandler: NotFoundHandler = () => new Response()  ) {    this.req = req    this.env = env ? env : ({} as E)
    if (eventOrExecutionCtx && 'respondWith' in eventOrExecutionCtx) {      this.event = eventOrExecutionCtx    } else {      this.executionCtx = eventOrExecutionCtx    }
    this.notFoundHandler = notFoundHandler    this.finalized = false  }
  get res(): Response {    return (this._res ||= new Response())  }
  set res(_res: Response) {    this._res = _res    this.finalized = true  }
  header(name: string, value: string): void {    this._headers ||= {}    this._headers[name] = value    if (this.finalized) {      this.res.headers.set(name, value)    }  }
  status(status: StatusCode): void {    this._status = status  }
  set(key: string, value: any): void {    this._map ||= {}    this._map[key] = value  }
  get(key: string) {    if (!this._map) {      return undefined    }    return this._map[key]  }
  pretty(prettyJSON: boolean, space: number = 2): void {    this._pretty = prettyJSON    this._prettySpace = space  }
  newResponse(data: Data | null, status: StatusCode, headers: Headers = {}): Response {    const _headers = { ...this._headers, ...headers }    if (this._res) {      this._res.headers.forEach((v, k) => {        _headers[k] = v      })    }    return new Response(data, {      status: status || this._status || 200,      headers: _headers,    })  }
  body(data: Data | null, status: StatusCode = this._status, headers: Headers = {}): Response {    return this.newResponse(data, status, headers)  }
  text(text: string, status: StatusCode = this._status, headers: Headers = {}): Response {    headers['Content-Type'] ||= 'text/plain; charset=UTF-8'    return this.body(text, status, headers)  }
  json<T>(object: T, status: StatusCode = this._status, headers: Headers = {}): Response {    const body = this._pretty      ? JSON.stringify(object, null, this._prettySpace)      : JSON.stringify(object)    headers['Content-Type'] ||= 'application/json; charset=UTF-8'    return this.body(body, status, headers)  }
  html(html: string, status: StatusCode = this._status, headers: Headers = {}): Response {    headers['Content-Type'] ||= 'text/html; charset=UTF-8'    return this.body(html, status, headers)  }
  redirect(location: string, status: StatusCode = 302): Response {    if (!isAbsoluteURL(location)) {      const url = new URL(this.req.url)      url.pathname = location      location = url.toString()    }    return this.newResponse(null, status, {      Location: location,    })  }
  notFound(): Response | Promise<Response> {    return this.notFoundHandler(this as any)  }}