123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334
import { serve, Server, ServerRequest, decode } from "./deps.ts";
export interface Router {  method: string;  url: string;  handler(req: Request): void;}export interface ListenOptions {  port: number;  hostname?: string;}export interface Plugin {  (req: Request, callback: Function): any;}export interface Handler {  (req: Request): void;}export interface Parameter {  [key: string]: string;}
interface Hook {  name: string;  handler: Plugin;}
export class Request extends ServerRequest {  /** URL parameter */  parameter!: Parameter;  /** Payload */  payload!: string;  /**   * Send payload   *         *      // send basic message, default http status 200   *      send('ok')   *    *      // send json object   *      send({ message: 'Hello' })   *    *      // send message with custom http status   *      send('not found', 404)   *    *      // send message with custom status & headers   *      const headers = new Headers()   *      headers.set('Authorization', `Bearer ${your_token}`)   *      send({ login: true }, 200, headers)   *    * @param payload    * @param status   * @param headers    *        */  send!: {    <T>(payload: string | T, status?: number, headers?: Headers): boolean;  };  [key: string]: any}
export function FastroError(title: string, error: Error) {  error.name = title;  return error;}
function checkUrl(incoming: string, registered: string): boolean {  try {    const incomingSplit = incoming.substr(1, incoming.length).split("/");    const registeredSplit = registered.substr(1, registered.length).split("/");    const filtered = registeredSplit      .filter((value, idx) => {        if (value.startsWith(":")) return incomingSplit[idx];        return value === incomingSplit[idx];      });    return incomingSplit.length === filtered.length;  } catch (error) {    throw FastroError("CHECK_URL_ERROR", error);  }}
/** * Fastro class *  *  *      const server = new Fastro() */export class Fastro {  private getParameter(incoming: string, registered: string) {    try {      const incomingSplit = incoming.substr(1, incoming.length).split("/");      const registeredSplit = registered.substr(1, registered.length).split(        "/",      );      const param: Parameter = {};      registeredSplit        .map((path, idx) => {          return { path, idx };        })        .filter((value) => value.path.startsWith(":"))        .map((value) => {          const name = value.path.substr(1, value.path.length);          param[name] = incomingSplit[value.idx];        });      return param;    } catch (error) {      throw FastroError("GET_URL_PARAMETER_ERROR", error);    }  }
  /**   * Add plugin   *    *      server.use((req) => {   *        console.log(req.headers.get("token"));   *      });   * @param plugin   */  use(plugin: Plugin) {    this.#plugins.push(plugin);    return this;  }
  private mutateRequest(req: Request, route: Router) {    let mutate = false;    this.#plugins.forEach((plugin) => {      const pluginResult = plugin(req, () => {});      if (pluginResult) mutate = true;    });    this.routeHandler(req, route, mutate);  }
  private requestHandler = async (req: ServerRequest) => {    const request = req as Request;    const [route] = this.#router.filter(function (value) {      return checkUrl(req.url, value.url) && (req.method == value.method);    });
    if (route) request.parameter = this.getParameter(req.url, route.url);    request.payload = decode(await Deno.readAll(req.body));    request.send = (payload, status, headers): boolean => {      return this.send(payload, status, headers, req);    };    if (this.#plugins.length > 0) this.mutateRequest(request, route);    else this.routeHandler(request, route);  };
  private routeHandler = async (    req: Request,    route?: Router,    mutate?: boolean,  ) => {    try {      if (mutate) return;      if (!route) {        return req.respond({ body: `${req.url} not found`, status: 404 });      }      return route.handler(req);    } catch (error) {      throw FastroError("SERVER_REQUEST_HANDLER_ERROR", error);    }  };
  private send<T>(    payload: string | number | boolean | T,    status: number | undefined = 200,    headers: Headers | undefined = new Headers(),    req: ServerRequest,  ) {    try {      let body: any;      headers.set("X-Powered-By", "fastro");      if (        typeof payload === "string" ||        typeof payload === "number" ||        typeof payload === "boolean"      ) {        body = payload;      } else body = JSON.stringify(payload);      req.respond({ status, headers, body });      return true; // this is used for request mutation flag    } catch (error) {      throw FastroError("SERVER_SEND_ERROR", error);    }  }
  /**    * Listen    *         *      server.listen({ port: 8000 })   **/  listen = async (    options?: ListenOptions,    callback?: (error: Error | undefined, address: string | undefined) => void,  ): Promise<void> => {    try {      let opt = options ? options : { port: 8000 };      this.#server = serve(opt);      if (!callback) console.info(options);      else callback(undefined, opt as any);      // creates a loop iterating over async iterable objects      // https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Statements/for-await...of      for await (const req of this.#server) {        await this.requestHandler(req);      }    } catch (error) {      const errStr = "SERVER_LISTEN_ERROR";      if (callback) callback(FastroError(errStr, error), undefined);      else throw FastroError(errStr, error);    }  };
  /**   * Add route   *    *      server.route({   *        url: "/hello",   *        method: "GET",   *        handler: (req) => {   *          req.send("hello");   *        },   *       });   *    * @param options   *    **/  route(options: Router) {    try {      const filteredRoutes = this.#router.filter(function (value) {        return checkUrl(options.url, value.url) &&          options.method === value.method;      });      if (filteredRoutes.length > 0) {        const [route] = filteredRoutes;        const errStr = `Duplicate route: ${          JSON.stringify(route)        } already added.`;        throw FastroError("SERVER_ROUTE_ERROR", new Error(errStr));      }      this.#router.push(options);      return this;    } catch (error) {      throw FastroError("SERVER_ROUTE_ERROR", error);    }  }
  /**   * GET route shorthand declaration   *    *       server.get('/', (req) => req.send('hello'))   *    * @param url    * @param handler    */  get(url: string, handler: Handler) {    return this.route({ method: "GET", url, handler });  }
  /**   * POST route shorthand declaration   *    *       server.post('/', (req) => {   *          const payload = req.payload;   *          req.send(payload);   *       })   *    * @param url    * @param handler    */  post(url: string, handler: Handler) {    return this.route({ method: "POST", url, handler });  }
  /**   * HEAD route shorthand declaration   * @param url    * @param handler    */  head(url: string, handler: Handler) {    return this.route({ method: "HEAD", url, handler });  }
  /**   * PATCH route shorthand declaration   * @param url    * @param handler    */  patch(url: string, handler: Handler) {    return this.route({ method: "PATCH", url, handler });  }
  /**   * PUT route shorthand declaration   * @param url    * @param handler    */  put(url: string, handler: Handler) {    return this.route({ method: "PUT", url, handler });  }
  /**   * OPTIONS route shorthand declaration   * @param url    * @param handler    */  options(url: string, handler: Handler) {    return this.route({ method: "OPTIONS", url, handler });  }
  /**   * DELETE shorthand declaration   * @param url    * @param handler    */  delete(url: string, handler: Handler) {    return this.route({ method: "DELETE", url, handler });  }
  // hook(name: string, handler: Plugin) {  //   this.#hooks.push({ name, handler });  //   return this;  // }
  /** Close server */  async close(): Promise<void> {    if (this.#server) {      this.#server.close();    }  }  // definite assignment assertion  // https://www.typescriptlang.org/docs/handbook/release-notes/typescript-2-7.html  #server!: Server;  #router: Router[] = [];  #plugins: Plugin[] = [];}