123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129
// Copyright 2019 Yusuke Sakurai. All rights reserved. MIT license.import Writer = Deno.Writer;import { ServerResponse } from "./server.ts";import { cookieSetter, CookieSetter } from "./cookie.ts";import { writeResponse } from "./serveio.ts";import { extname, basename } from "./vendor/https/deno.land/std/path/mod.ts";import { contentTypeByExt } from "./media_types.ts";/** Basic responder for http response */export interface ServerResponder extends CookieSetter {  /** Respond to request */  respond(response: ServerResponse): Promise<void>;
  /**   * Send file as a response. Content-Type will be guessed but may not be found.   * Default Content-Type is application/octet-stream   *  */  sendFile(    path: string,    opts?: {      contentDisposition?: "inline" | "attachment";      headers?: Headers;    },  ): Promise<void>;
  /** Redirect request with 302 (Found ) */  redirect(    url: string,    opts?: {      headers?: Headers;      body?: ServerResponse["body"];    },  ): Promise<void>;
  /** Mark as responded manually */  markAsResponded(status: number): void;
  isResponded(): boolean;
  respondedStatus(): number | undefined;}
/** create ServerResponder object */export function createResponder(  w: Writer,  onResponse: (r: ServerResponse) => Promise<void> = (resp) =>    writeResponse(w, resp),): ServerResponder {  const responseHeaders = new Headers();  const cookie = cookieSetter(responseHeaders);  let responseStatus: number | undefined;  function isResponded() {    return responseStatus !== undefined;  }  async function redirect(    url: string,    {      headers = new Headers(),      body,    }: { headers?: Headers; body?: ServerResponse["body"] } = {},  ) {    headers.set("location", url);    await respond({ status: 302, headers, body });  }  async function respond(response: ServerResponse): Promise<void> {    if (isResponded()) {      throw new Error("Request already responded");    }    const { status, headers, body } = response;    responseStatus = status;    if (headers) {      for (const [k, v] of headers.entries()) {        responseHeaders.append(k, v);      }    }    await onResponse({      status,      headers: responseHeaders,      body,    });  }  async function sendFile(    path: string,    opts?: {      contentDisposition?: "inline" | "attachment";      headers?: Headers;    },  ): Promise<void> {    const body = await Deno.open(path);    const headers = opts?.headers ?? new Headers();    try {      const contentType =        contentTypeByExt(extname(path)) ?? "application/octet-stream";      headers.set("content-type", contentType);      const contentDisposition = opts?.contentDisposition;      if (contentDisposition === "inline") {        headers.set("content-disposition", contentDisposition);      } else if (contentDisposition === "attachment") {        const filename = basename(path);        headers.set(          "content-disposition",          `attachment; filename="${filename}"`,        );      }      await onResponse({        status: 200,        headers,        body,      });    } finally {      body.close();    }  }  function markAsResponded(status: number) {    responseStatus = status;  }  function respondedStatus() {    return responseStatus;  }  return {    respond,    redirect,    sendFile,    isResponded,    respondedStatus,    markAsResponded,    ...cookie,  };}