Remove puppeteer based scraping, use yt-dlp

1 files changed, 40 insertions, 4 deletions

diff --git a/src/tasks.ts b/src/tasks.ts
index 8fb1b2f..0646da5 100644
--- a/src/tasks.ts
+++ b/src/tasks.ts
@@ -1,8 +1,44 @@
-import { InstagramTask } from "./instagram_task";
-import { TikTokTask } from "./tiktok_task";
+import { asyncExec } from "./utils";
+import fs from "fs";
+export const checkIfYtdlpIsInstalled = async () => {
+  try {
+    await asyncExec("yt-dlp --version");
+    return true;
+  } catch (e) {
+    console.error("yt-dlp is not installed or available in PATH", e);
+    return false;
+  }
+};
+
+export const runDownloadUrlTask = async (task: Task) => {
+  try {
+    const filename = "veeti.mp4";
+    // remove the file if it exists
+    try {
+      fs.rmSync(filename);
+    } catch (e: any) {
+      if (e.code !== "ENOENT") {
+        console.error("Error deleting file", e);
+      }
+    }
 
-// union of possible task types
-export type Task = InstagramTask | TikTokTask;
+    const { stdout } = await asyncExec(
+      `yt-dlp -o "downloads/${filename}" "${task.url}"`,
+    );
+    console.log(stdout);
+    task.respondWithFile(filename);
+  } catch (e) {
+    console.error("Error downloading URL", e);
+    throw e;
+  }
+};
+
+export type Task = {
+  url: string;
+  type: "download_url";
+  respondWithFile: (file: string) => void;
+  respondTaskFailed: (err: unknown) => void;
+};
 
 const taskQueue: Task[] = [];