1 files changed, 130 insertions, 0 deletions
diff --git a/src/instagram_task.ts b/src/instagram_task.ts
new file mode 100644
index 0000000..6ab9838
--- /dev/null
+++ b/src/instagram_task.ts
@@ -0,0 +1,130 @@
+import puppeteer, { Browser, Page, TimeoutError } from "puppeteer";
+import { InstagramTask, addTask } from "./tasks";
+import fs from "fs";
+import { wait } from "./utils";
+
+export const runDownloadInstagramTask = async (task: InstagramTask) => {
+  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://snapinsta.app/");
+    await page.type("input#url", task.url);
+    await page.click("button[type=submit]");
+
+    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;
+      }
+    }
+
+    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(
+      ".download-content a[data-event=click_download_btn]",
+      { 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];
+
+    addTask({
+      type: "slack_reply",
+      replyId: task.replyId,
+      message: "Here's the video you requested",
+      filePath: `./downloads/${downloadedFile}`,
+    });
+  } finally {
+    await page.close();
+    await browser.close();
+  }
+};