import { Route, Router, RouterType } from "itty-router";
import { handleAll, handleOne } from "./handlers";
import { Env, Flags } from "./types";
import { Cache } from "./cache";
import { index } from "./landing";

interface BasicRouter extends RouterType {
  all: Route;
  get: Route;
}

const router = <BasicRouter>Router();

router
  .get<BasicRouter>(
    "/",
    () =>
      new Response(null, { status: 303, headers: { location: "/population/" } })
  )
  .get<BasicRouter>("/population/", index)
  .get<BasicRouter>("/population/all", handleAll)
  .get<BasicRouter>("/population/:id", handleOne)
  .get<BasicRouter>(
    "/population~/flags",
    (_1, _2, _3, cache: Cache, flags: Flags) => {
      return new Response(
        JSON.stringify({
          ...flags,
          disableCache: cache.disableCache,
        }),
        {
          headers: { "content-type": "application/json" },
        }
      );
    }
  )
  .get<BasicRouter>("/population~/health", async () => {
    const [saerro, voidwell, honu, fisu] = await Promise.all([
      fetch("https://saerro.ps2.live/health").then((r) => r.status === 200),
      fetch("https://voidwell.com/").then((r) => r.status === 200),
      fetch("https://wt.honu.pw/api/health").then((r) => r.status === 200),
      fetch("https://ps2.fisu.pw").then((r) => r.status === 200),
    ]);

    return new Response(
      JSON.stringify({
        saerro,
        voidwell,
        honu,
        fisu,
      }),
      {
        headers: { "content-type": "application/json" },
        status: saerro || voidwell || honu || fisu ? 200 : 502,
      }
    );
  })
  .all<BasicRouter>("*", () => {
    return new Response("Not found", {
      headers: { "content-type": "text/plain" },
    });
  });

export default {
  fetch: async (request: Request, env: Env, ctx: ExecutionContext) => {
    const worldCache = new Cache(env.CACHE, env.DISABLE_CACHE === "1");

    const flags: Flags = {
      disableFisu: env.DISABLE_FISU === "1",
      disableHonu: env.DISABLE_HONU === "1",
      disableSaerro: env.DISABLE_SAERRO === "1",
      disableVoidwell: env.DISABLE_VOIDWELL === "1",
      voidwellUsePS4: env.VOIDWELL_USE_PS4 === "1",
      fisuUsePS4EU: env.FISU_USE_PS4EU === "1",
    };

    const start = Date.now();

    return router
      .handle(request as any, env, ctx, worldCache, flags)
      .then((response) => {
        response.headers.set("access-control-allow-origin", "*");
        response.headers.set(
          "access-control-allow-method",
          "GET, HEAD, OPTIONS"
        );
        response.headers.set("x-timing", `${Date.now() - start}ms`);
        return response;
      });
  },
};