p

lib/ytdl.js

@@ -1,18 +1,20 @@
 const { spawn } = require("child_process");
 const path = require("path");
+const fs = require("fs");
 
-const ytdlp = "yt-dlp"; // Asumsi yt-dlp sudah terinstall global di Docker
-const cookiesPath = path.join(__dirname, "./cookies/twt.txt");
+const tmp = path.join(__dirname, "../tmp"); // path to your tmp folder
+const ytdlp = "yt-dlp"; // Assume yt-dlp is installed globally inside the Docker container
+const cookiesPath = path.join(__dirname, "./cookies/yt.txt");
 
 function runYtDlp(args) {
   return new Promise((resolve, reject) => {
     const process = spawn(ytdlp, args, { stdio: ["ignore", "pipe", "pipe"] });
     let stdout = "";
     let stderr = "";
-
+    
     process.stdout.on("data", (data) => (stdout += data.toString()));
     process.stderr.on("data", (data) => (stderr += data.toString()));
-
+    
     process.on("close", (code) => {
       if (code === 0) resolve(stdout.trim());
       else reject(new Error(stderr.trim() || `yt-dlp exited with code ${code}`));
@@ -20,42 +22,63 @@ function runYtDlp(args) {
   });
 }
 
-/**
- * Mengambil metadata video Twitter menggunakan yt-dlp.
- * @param {string} url - URL tweet yang berisi video.
- * @returns {Promise<Object>} - Metadata video dalam format JSON.
- */
-async function getTwitterVideoInfo(url) {
+async function getInfo(url) {
   try {
     const result = await runYtDlp(["-j", "--cookies", cookiesPath, url]);
     const info = JSON.parse(result);
-
-    // Pastikan `formats` ada, jika tidak buat array kosong
-    const videoFormats = (info.formats || []).filter(
-      (f) => (f.video_ext !== "none" && f.audio_ext !== "none") || f.url.includes(".mp4")
-    );
-
-    const audioFormats = (info.formats || []).filter((f) => f.video_ext === "none" && f.audio_ext !== "none");
-
     return {
-      status: 200,
-      title: info.title || "No Title",
-      description: info.description || "",
-      thumbnail: info.thumbnail || "",
-      video: videoFormats.map((v) => ({
-        format_id: v.format_id,
-        resolution: v.resolution || `${v.width}x${v.height}`,
-        url: v.url,
-      })),
-      audio: audioFormats.map((a) => ({
-        format_id: a.format_id,
-        bitrate: a.abr,
-        url: a.url,
-      })),
+      id: info.id,
+      title: info.title,
+      thumbnail: `https://i.ytimg.com/vi/${info.id}/maxresdefault.jpg`,
+      description: info.description,
+      duration: info.duration,
+      views: info.view_count,
+      likes: info.like_count,
+      comments: info.comment_count,
+      uploaded: info.timestamp,
+      channel: {
+        id: info.channel_id,
+        handle: info.uploader_id,
+        name: info.uploader,
+        picture: info.channel_thumbnail || "",
+        subscribers: info.channel_follower_count,
+        verified: !!info.channel_is_verified,
+      },
+      videos: [...new Set(info.formats.filter(v => v.video_ext === "mp4").map(v => v.height + "p"))],
     };
   } catch (err) {
     throw new Error(`Failed to get video info: ${err.message}`);
   }
 }
 
-module.exports = getTwitterVideoInfo;
+async function getVideo(url, quality = "480p") {
+  const ts = Date.now();
+  const outputFile = `${tmp}/${ts}.mp4`;
+  try {
+    await runYtDlp(["-f", `ba[ext=m4a]+bv[height=${quality.slice(0, -1)}]`, "--merge-output-format", "mp4", "-o", outputFile, url]);
+    const result = fs.readFileSync(outputFile);
+    fs.unlinkSync(outputFile);
+    return result;
+  } catch (err) {
+    throw new Error(`Failed to download video: ${err.message}`);
+  }
+}
+
+async function getAudio(url) {
+  const ts = Date.now();
+  const outputFile = `${tmp}/${ts}.mp3`;
+  try {
+    await runYtDlp(["-f", "ba", "-o", outputFile, url]);
+    const result = fs.readFileSync(outputFile);
+    fs.unlinkSync(outputFile);
+    return result;
+  } catch (err) {
+    throw new Error(`Failed to download audio: ${err.message}`);
+  }
+}
+
+module.exports = {
+  getInfo,
+  getAudio,
+  getVideo,
+};