123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237
import type {  Environment,  NotFoundHandler,  ContextVariableMap,  Bindings,  ValidatedData,} from './hono.ts'import type { CookieOptions } from './utils/cookie.ts'import { serialize } from './utils/cookie.ts'import type { StatusCode } from './utils/http-status.ts'
type HeaderField = [string, string]type Headers = Record<string, string | string[]>export type Data = string | ArrayBuffer | ReadableStream
export interface Context<  RequestParamKeyType extends string = string,  E extends Partial<Environment> = any,  D extends ValidatedData = ValidatedData> {  req: Request<RequestParamKeyType, D>  env: E['Bindings']  event: FetchEvent  executionCtx: ExecutionContext  finalized: boolean
  get res(): Response  set res(_res: Response)  header: (name: string, value: string, options?: { append?: boolean }) => void  status: (status: StatusCode) => void  set: {    <Key extends keyof ContextVariableMap>(key: Key, value: ContextVariableMap[Key]): void    <Key extends keyof E['Variables']>(key: Key, value: E['Variables'][Key]): void    (key: string, value: any): void  }  get: {    <Key extends keyof ContextVariableMap>(key: Key): ContextVariableMap[Key]    <Key extends keyof E['Variables']>(key: Key): E['Variables'][Key]    <T = any>(key: string): T  }  pretty: (prettyJSON: boolean, space?: number) => void  newResponse: (data: Data | null, status: StatusCode, headers: Headers) => Response  body: (data: Data | null, status?: StatusCode, headers?: Headers) => Response  text: (text: string, status?: StatusCode, headers?: Headers) => Response  json: <T>(object: T, status?: StatusCode, headers?: Headers) => Response  html: (html: string, status?: StatusCode, headers?: Headers) => Response  redirect: (location: string, status?: StatusCode) => Response  cookie: (name: string, value: string, options?: CookieOptions) => void  notFound: () => Response | Promise<Response>}
export class HonoContext<  RequestParamKeyType extends string = string,  E extends Partial<Environment> = Environment,  D extends ValidatedData = ValidatedData> implements Context<RequestParamKeyType, E, D>{  req: Request<RequestParamKeyType, D>  env: E['Bindings']  finalized: boolean
  _status: StatusCode = 200  private _executionCtx: FetchEvent | ExecutionContext | undefined  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<E>
  constructor(    req: Request<RequestParamKeyType>,    env: E['Bindings'] | undefined = undefined,    executionCtx: FetchEvent | ExecutionContext | undefined = undefined,    notFoundHandler: NotFoundHandler<E> = () => new Response()  ) {    this._executionCtx = executionCtx    this.req = req as Request<RequestParamKeyType, D>    this.env = env || ({} as Bindings)
    this.notFoundHandler = notFoundHandler    this.finalized = false  }
  get event(): FetchEvent {    if (this._executionCtx instanceof FetchEvent) {      return this._executionCtx    } else {      throw Error('This context has no FetchEvent')    }  }
  get executionCtx(): ExecutionContext {    if (this._executionCtx) {      return this._executionCtx    } else {      throw Error('This context has no ExecutionContext')    }  }
  get res(): Response {    return (this._res ||= new Response('404 Not Found', { status: 404 }))  }
  set res(_res: Response) {    this._res = _res    this.finalized = true  }
  header(name: string, value: string, options?: { append?: boolean }): void {    this._headers ||= {}    const key = name.toLowerCase()
    let shouldAppend = false    if (options && options.append) {      const vAlreadySet = this._headers[key]      if (vAlreadySet && vAlreadySet.length) {        shouldAppend = true      }    }
    if (shouldAppend) {      this._headers[key].push(value)    } else {      this._headers[key] = [value]    }
    if (this.finalized) {      if (shouldAppend) {        this.res.headers.append(name, value)      } else {        this.res.headers.set(name, value)      }    }  }
  status(status: StatusCode): void {    this._status = status  }
  set<Key extends keyof ContextVariableMap>(key: Key, value: ContextVariableMap[Key]): void  set<Key extends keyof E['Variables']>(key: Key, value: E['Variables'][Key]): void  set(key: string, value: any): void  set(key: string, value: any): void {    this._map ||= {}    this._map[key] = value  }
  get<Key extends keyof ContextVariableMap>(key: Key): ContextVariableMap[Key]  get<Key extends keyof E['Variables']>(key: Key): E['Variables'][Key]  get<T = any>(key: string): T  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 {    return new Response(data, {      status: status || this._status || 200,      headers: this._finalizeHeaders(headers),    })  }
  private _finalizeHeaders(incomingHeaders: Headers): HeaderField[] {    const finalizedHeaders: HeaderField[] = []    const headersKv = this._headers || {}    // If Response is already set    if (this._res) {      this._res.headers.forEach((v, k) => {        headersKv[k] = [v]      })    }    for (const key of Object.keys(incomingHeaders)) {      const value = incomingHeaders[key]      if (typeof value === 'string') {        finalizedHeaders.push([key, value])      } else {        for (const v of value) {          finalizedHeaders.push([key, v])        }      }      delete headersKv[key]    }    for (const key of Object.keys(headersKv)) {      for (const value of headersKv[key]) {        const kv: HeaderField = [key, value]        finalizedHeaders.push(kv)      }    }    return finalizedHeaders  }
  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 {    return this.newResponse(null, status, {      Location: location,    })  }
  cookie(name: string, value: string, opt?: CookieOptions): void {    const cookie = serialize(name, value, opt)    this.header('set-cookie', cookie, { append: true })  }
  notFound(): Response | Promise<Response> {    return this.notFoundHandler(this as any)  }}