import puppeteer from "@cloudflare/puppeteer";
import Router from "./library/router/index.js";
import talent96 from "./data/talent96.js";

export default {
  /*
    Fetch event handler, this function will be called whenever a request is made to the worker.
    The function will parse the request and return a response based on the request path.

    @param {Request} request - the incoming request object
    @param {Environment} environment - the environment object
    @param {Context} context - the context object

    @returns {Response} a new Response object
  */
  async fetch(request, environment, context) {
    const router = new Router(request, environment, context);

    router.add("/", () => {
      return router.respond({
        message: "Hello, World! :3",
      });
    });

    router.add("/status", (request) => {
      const url = new URL(request.url);
      const service = url.searchParams.get("service");

      const acceptedServices = [
        "dns",
        "load-balancer",
        "cdn",
        "functions",
        "mysql-cluster",
        "mongodb-cluster",
        "redis-cluster",
        "elasticsearch-cluster",
        "git-connector",
        "job-runners",
        "container-registry",
        "kubernetes-cluster",
        "bare-metal-servers",
        "game-server-api",
        "anti-ddos-protection",
        "anti-cheat-api",
      ];

      if (acceptedServices.includes(service))
        return router.respond({
          status: "ok",
        });

      return router.respond({
        error: `service not found`,
      });
    });

    router.add("/96/twitch/streaming", async (request) => {
      /*
        Checks if a twitch channel is live by fetching the "live" preview image of the channel,
        if the image is fetched successfully, then the channel is live, otherwise it's offline.

        @param {string} channel - the twitch channel name
        @returns {boolean} true if the channel is live, false otherwise
      */
      async function isTwitchChannelLive(channel) {
        // construct preview image url with channel name
        const livePreviewUrl = `https://static-cdn.jtvnw.net/previews-ttv/live_user_${channel}-320x180.jpg`;

        // fetch the preview image, don't follow redirects
        const response = await fetch(livePreviewUrl, {
          redirect: "manual",
        });

        // check if the image was fetched successfully
        return response.ok;
      }

      const url = new URL(request.url);
      const channel = url.searchParams.get("channel");

      if (!channel)
        return router.respond({
          error: `channel not provided`,
        });

      const whitelist = ["zuedev", ...talent96];

      if (!whitelist.includes(channel))
        return router.respond({
          error: `channel not whitelisted`,
        });

      const channelLive = await isTwitchChannelLive(channel);

      if (channelLive)
        return router.respond({
          status: "live",
        });

      return router.respond({
        status: "offline",
      });
    });

    router.add("/browser-rendering/screenshot", async (request) => {
      const { searchParams } = new URL(request.url);
      const url = searchParams.get("url");
      const type = searchParams.get("type") || "webp";
      const fullPage = searchParams.get("fullPage") !== null;
      const width = parseInt(searchParams.get("width"), 10) || 1920;
      const height = parseInt(searchParams.get("height"), 10) || 1080;

      if (!url)
        return router.respond({
          error: "URL parameter is required",
        });

      if (!/^https?:\/\//.test(url))
        return router.respond({
          error: "Invalid URL format",
        });

      const browser = await puppeteer.launch(environment.MYBROWSER);
      const page = await browser.newPage();
      await page.setViewport({ width, height });
      await page.goto(url);
      const screenshot = await page.screenshot({
        type,
        fullPage,
      });
      await browser.close();

      return new Response(screenshot, {
        headers: {
          "content-type": `image/${type}`,
        },
      });
    });

    return router.route();
  },

  /*
    Email event handler, this function will be called whenever an email is sent to the worker.
    The function will parse the email message and forward it to a specified email address.

    @param {Message} message - the incoming email message object
    @param {Environment} environment - the environment object
    @param {Context} context - the context object

    @returns {void}
  */
  async email(message, environment, context) {
    message.forward("alex@zue.dev");
  },

  /*
    Scheduled event handler, this function will be called whenever a scheduled event is triggered.
    The function will perform a task and return a response based on the task outcome.

    @param {Event} event - the incoming event object
    @param {Environment} environment - the environment object
    @param {Context} context - the context object

    @returns {void}
  */
  async scheduled(event, environment, context) {
    console.log("Scheduled event triggered!");
  },
};