123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208
import type {  Response,  ServerRequest,} from "./vendor/https/deno.land/std/http/server.ts";import type { Cookie } from "./vendor/https/deno.land/std/http/cookie.ts";import type { Application } from "./app.ts";import type { ContextOptions } from "./types.ts";
import { Status } from "./vendor/https/deno.land/std/http/http_status.ts";import { join } from "./vendor/https/deno.land/std/path/mod.ts";import {  getCookies,  setCookie,} from "./vendor/https/deno.land/std/http/cookie.ts";import { MultipartReader } from "./vendor/https/deno.land/std/mime/multipart.ts";import { Header, MIME } from "./constants.ts";import { contentType, NotFoundHandler } from "./util.ts";
const { cwd, readFile, readAll } = Deno;
const encoder = new TextEncoder();const decoder = new TextDecoder();
export class Context {  app!: Application;  request!: ServerRequest;  url!: URL;
  response: Response & { headers: Headers } = { headers: new Headers() };  params: Record<string, string> = {};  customContext: any;
  #store?: Map<string | symbol, unknown>;
  #body: Promise<unknown> | undefined;
  get cookies(): Record<string, string> {    return getCookies(this.request);  }
  get path(): string {    return this.url.pathname;  }
  get method(): string {    return this.request.method;  }
  get queryParams(): Record<string, string> {    const params: Record<string, string> = {};    for (const [k, v] of this.url.searchParams) {      params[k] = v;    }    return params;  }
  get body(): Promise<unknown> {    return this.#body ?? (this.#body = this.#readBody());  }
  get(key: string | symbol): unknown {    return this.#store?.get(key);  }
  set(key: string | symbol, val: unknown): void {    if (this.#store === undefined) {      this.#store = new Map();    }
    this.#store.set(key, val);  }
  constructor(opts: ContextOptions);  constructor(c: Context);  constructor(optionsOrContext: ContextOptions | Context) {    if (optionsOrContext instanceof Context) {      Object.assign(this, optionsOrContext);      this.customContext = this;      return;    }
    const opts = optionsOrContext;    this.app = opts.app;    this.request = opts.r;
    this.url = new URL(this.request.url, `http://0.0.0.0`);  }
  #writeContentType = (v: string): void => {    if (!this.response.headers.has(Header.ContentType)) {      this.response.headers.set(Header.ContentType, v);    }  };
  #readBody = async (): Promise<unknown> => {    const contentType = this.request.headers.get(Header.ContentType);    walk: {      let data: Record<string, unknown> = {};      if (contentType) {        if (contentType.includes(MIME.ApplicationJSON)) {          data = JSON.parse(decoder.decode(await readAll(this.request.body)));        } else if (contentType.includes(MIME.ApplicationForm)) {          for (            const [k, v] of new URLSearchParams(              decoder.decode(await readAll(this.request.body)),            )          ) {            data[k] = v;          }        } else if (contentType.includes(MIME.MultipartForm)) {          const match = contentType.match(/boundary=([^\s]+)/);          const boundary = match ? match[1] : undefined;          if (boundary) {            const mr = new MultipartReader(this.request.body, boundary);            const form = await mr.readForm();            for (const [k, v] of form.entries()) {              data[k] = v;            }          }        } else {          break walk;        }      } else {        break walk;      }
      return data;    }
    return decoder.decode(await readAll(this.request.body));  };
  string(v: string, code: Status = Status.OK): void {    this.#writeContentType(MIME.TextPlainCharsetUTF8);    this.response.status = code;    this.response.body = encoder.encode(v);  }
  json(v: Record<string, any> | string, code: Status = Status.OK): void {    this.#writeContentType(MIME.ApplicationJSONCharsetUTF8);    this.response.status = code;    this.response.body = encoder.encode(      typeof v === "object" ? JSON.stringify(v) : v,    );  }
  /** Sends an HTTP response with status code. */  html(v: string, code: Status = Status.OK): void {    this.#writeContentType(MIME.TextHTMLCharsetUTF8);    this.response.status = code;    this.response.body = encoder.encode(v);  }
  /** Sends an HTTP blob response with status code. */  htmlBlob(b: Uint8Array | Deno.Reader, code: Status = Status.OK): void {    this.blob(b, MIME.TextHTMLCharsetUTF8, code);  }
  /**   * Renders a template with data and sends a text/html response with status code.   * renderer must be registered first.   */  async render<T>(    name: string,    data: T = {} as T,    code: Status = Status.OK,  ): Promise<void> {    if (!this.app.renderer) {      throw new Error();    }    const r = await this.app.renderer.render(name, data);    this.htmlBlob(r, code);  }
  /** Sends a blob response with content type and status code. */  blob(    b: Uint8Array | Deno.Reader,    contentType?: string,    code: Status = Status.OK,  ): void {    if (contentType) {      this.#writeContentType(contentType);    }    this.response.status = code;    this.response.body = b;  }
  async file(filepath: string): Promise<void> {    filepath = join(cwd(), filepath);    try {      this.blob(await readFile(filepath), contentType(filepath));    } catch {      NotFoundHandler();    }  }
  /** append a `Set-Cookie` header to the response */  setCookie(c: Cookie): void {    setCookie(this.response, c);  }
  /** Redirects a response to a specific URL. the `code` defaults to `302` if omitted */  redirect(url: string, code = Status.Found): void {    this.response.headers.set(Header.Location, url);    this.response.status = code;  }}