123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
// Copyright 2018-2022 the oak authors. All rights reserved. MIT license.
import { type Application } from "./application.ts";import { deferred, Status, STATUS_TEXT } from "./deps.ts";import { HttpRequest } from "./http_request.ts";import type { Listener, Server } from "./types.d.ts";import { assert } from "./util.ts";
type ServeHandler = (  request: Request,) => Response | Promise<Response> | void | Promise<void>;
interface ServeOptions {  port?: number;  hostname?: string;  signal?: AbortSignal;  onError?: (error: unknown) => Response | Promise<Response>;  onListen?: (params: { hostname: string; port: number }) => void;}
interface ServeTlsOptions extends ServeOptions {  cert: string;  key: string;}
type FlashServerOptions = Omit<Partial<ServeTlsOptions>, "onListen" | "signal">;
const serve: (  handler: ServeHandler,  options?: ServeOptions | ServeTlsOptions,) => Promise<void> = "serve" in Deno  // deno-lint-ignore no-explicit-any  ? (Deno as any).serve.bind(Deno)  : undefined;
function isServeTlsOptions(  value: ServeOptions | ServeTlsOptions,): value is ServeTlsOptions {  return "cert" in value && "key" in value;}
/** A function that determines if the current environment supports Deno flash.*/export function hasFlash(): boolean {  return Boolean(serve);}
/** A server abstraction which manages requests from Deno's flash server. * * You can pass the class as the `server` property when constructing a new * application to force the application to use Deno's flash server. */export class FlashServer implements Server<HttpRequest> {  // deno-lint-ignore no-explicit-any  #app: Application<any>;  #closed = false;  #controller?: ReadableStreamDefaultController<HttpRequest>;  #abortController = new AbortController();  #options: FlashServerOptions;  #servePromise?: Promise<void>;  #stream?: ReadableStream<HttpRequest>;
  // deno-lint-ignore no-explicit-any  constructor(app: Application<any>, options: FlashServerOptions) {    if (!serve) {      throw new Error("The flash bindings for serving HTTP are not available.");    }    this.#app = app;    this.#options = options;  }
  async close(): Promise<void> {    if (this.#closed) {      return;    }    this.#closed = true;    try {      this.#controller?.close();      this.#controller = undefined;      this.#stream = undefined;      this.#abortController.abort();      if (this.#servePromise) {        await this.#servePromise;        this.#servePromise = undefined;      }    } catch {      // just swallow here    }  }
  listen(): Promise<Listener> {    const p = deferred<Listener>();    const start: ReadableStreamDefaultControllerCallback<HttpRequest> = (      controller,    ) => {      this.#controller = controller;      const options: ServeOptions | ServeTlsOptions = {        ...this.#options,        signal: this.#abortController.signal,        onListen: (addr) => p.resolve({ addr }),        onError: (error) => {          this.#app.dispatchEvent(new ErrorEvent("error", { error }));          return new Response("Internal server error", {            status: Status.InternalServerError,            statusText: STATUS_TEXT[Status.InternalServerError],          });        },      };      const handler: ServeHandler = (request) => {        const resolve = deferred<Response>();        const flashRequest = new HttpRequest(request, resolve);        controller.enqueue(flashRequest);        return resolve;      };      this.#servePromise = serve(handler, options);    };    this.#stream = new ReadableStream({ start });    return p;  }
  [Symbol.asyncIterator](): AsyncIterableIterator<HttpRequest> {    assert(      this.#stream,      ".listen() was not called before iterating or server is closed.",    );    return this.#stream[Symbol.asyncIterator]();  }}