import fs from "fs";
import puppeteer, { TimeoutError } from "puppeteer";
import { wait } from "./utils";

export interface TikTokTask {
  type: "download_tiktok";
  url: string;
  respondWithFile: (filePath: string) => Promise<void>;
  respondTaskFailed: (err: unknown) => Promise<void>;
}

export const runDownloadTiktokTask = async (task: TikTokTask) => {
  const browser = await puppeteer.launch({
    headless: false,
    args: ["--shm-size=1gb"],
  });
  const page = await browser.newPage();
  const client = await page.createCDPSession();
  await client.send("Page.setDownloadBehavior", {
    behavior: "allow",
    downloadPath: "./downloads",
  });

  try {
    console.log("Starting task:", task.url);

    const downloadDirFiles = fs.readdirSync("./downloads");
    console.log("Emptying downloads folder");
    downloadDirFiles.forEach((file) => {
      fs.unlinkSync(`./downloads/${file}`);
    });

    await page.goto("https://snaptik.app/");

    try {
      const consent = await page.waitForSelector("button.fc-cta-consent", {
        timeout: 3000,
      });
      if (consent) {
        console.log("Looks like a cookie consent dialog, clicking agree");
        await consent.click();
      }
    } catch (e) {
      if (e instanceof TimeoutError) {
        console.log("No cookie consent dialog found");
      } else {
        throw e;
      }
    }

    await page.type("input#url", task.url);
    await page.click("button[type=submit]");

    console.log("Waiting for a bit");
    await wait(1000);
    console.log("Done waiting");

    // try {
    //   const adModalClose = await page.waitForSelector("#adOverlay button", {
    //     timeout: 3000,
    //   });
    //   console.log("Looks like an ad modal, clicking close");
    //   await adModalClose?.click();
    // } catch (e) {
    //   if (e instanceof TimeoutError) {
    //     console.log("No ad modal");
    //   } else {
    //     throw e;
    //   }
    // }

    // try {
    //   const secondAdModalClose = await page.waitForSelector("#dismiss-button", {
    //     timeout: 3000,
    //   });
    //   console.log("Looks like a second ad modal, clicking close");
    //   await secondAdModalClose?.click();
    // } catch (e) {
    //   if (e instanceof TimeoutError) {
    //     console.log("No second ad modal");
    //   } else {
    //     throw e;
    //   }
    // }

    const downloadButton = await page.waitForSelector(
      "a[data-event=server01_file]",
      { timeout: 10_000 }
    );
    if (!downloadButton) {
      throw new Error("Could not find download button");
    }

    console.log("Found download button, pressing enter");
    await downloadButton?.scrollIntoView();
    await downloadButton?.focus();
    await downloadButton?.press("Enter");

    console.log("Waiting to see if one more ad pops up");
    await wait(1000);

    // Click outside shadow-dommed ad modal to close it and start the download
    // console.log(
    //   "Clicking outside the possible ad modal to close it and start the download"
    // );
    // await page.mouse.click(10, 10);

    console.log("Waiting for download to start");
    await wait(3000);

    let downloadedFiles = fs.readdirSync("./downloads");
    while (downloadedFiles.some((file) => file.endsWith(".crdownload"))) {
      console.log("Download still in progress, waiting");
      await wait(1000);
      downloadedFiles = fs.readdirSync("./downloads");
    }

    console.log("Downloaded files", downloadedFiles);
    if (downloadedFiles.length === 0) {
      throw new Error("No files downloaded");
    } else if (downloadedFiles.length > 1) {
      throw new Error("More than one file downloaded, not sure what to do");
    }

    const downloadedFile = downloadedFiles[0];

    await task.respondWithFile(`./downloads/${downloadedFile}`);
  } finally {
    await page.close();
    await browser.close();
  }
};