v3.2.0-beta.1

Author: DestroyCom

app/api/video-converter/route.ts

@@ -1,11 +1,15 @@
 "use server";
 
 import ffmpeg from "fluent-ffmpeg";
-import ytdl from "@distube/ytdl-core";
+import * as oldYtdl from "@distube/ytdl-core";
 import { PassThrough, Readable } from "stream";
 import path from "path";
 import * as fs from "fs";
-import { initializeConf, sanitizeFilename } from "@/lib/serverUtils";
+import {
+  initializeConf,
+  sanitizeFilename,
+  selectYtDlpPath,
+} from "@/lib/serverUtils";
 import { FormatData, VideoData } from "@/lib/types";
 import { NextResponse } from "next/server";
 import { PrismaClient } from "@prisma/client";
@@ -54,18 +58,32 @@ const downloadStreams = async (
   url: string,
   audio_path: string,
   video_path: string,
-  quality?: string
+  formatSelector?: string
 ) => {
-  quality = quality || "highestvideo";
-
-  const audioStream = ytdl(url, {
-    quality: "highestaudio",
-  });
-
-  const videoStream = ytdl(url, {
-    quality: quality,
-    filter: "videoonly",
-  });
+  const ytdl = await selectYtDlpPath();
+  formatSelector = formatSelector || "bv";
+
+  const audioStream = ytdl.exec(url, {
+    noCheckCertificates: true,
+    noWarnings: true,
+    preferFreeFormats: true,
+    addHeader: ["referer:youtube.com", "user-agent:googlebot"],
+    format: "ba",
+    output: "-",
+  }).stdout;
+
+  const videoStream = ytdl.exec(url, {
+    noCheckCertificates: true,
+    noWarnings: true,
+    preferFreeFormats: true,
+    addHeader: ["referer:youtube.com", "user-agent:googlebot"],
+    format: formatSelector,
+    output: "-",
+  }).stdout;
+  if (!audioStream || !videoStream) {
+    throw new Error("Failed to download audio or video stream");
+  }
+  console.log("Downloading audio and video streams...");
 
   const audioWriteStream = fs.createWriteStream(audio_path);
   const videoWriteStream = fs.createWriteStream(video_path);
@@ -181,13 +199,17 @@ export async function GET(request: Request) {
     ffmpeg.setFfmpegPath(ffmpegPath);
   }
 
-  const video = await ytdl.getBasicInfo(url);
+  const video = await oldYtdl.getBasicInfo(url);
   const formatMap = new Map();
   (video.player_response.streamingData.adaptiveFormats as FormatData[]).forEach(
     (format: FormatData) => {
-      if (!formatMap.has(format.qualityLabel)) {
+      if (format.qualityLabel && !formatMap.has(format.qualityLabel)) {
         formatMap.set(format.qualityLabel, format);
       }
+      const itagKey = String(format.itag);
+      if (!formatMap.has(itagKey)) {
+        formatMap.set(itagKey, format);
+      }
     }
   );
 
@@ -225,9 +247,18 @@ export async function GET(request: Request) {
   };
 
   if (quality === "audio") {
-    const audioStream = ytdl(url, {
-      quality: "highestaudio",
-    });
+    const ytdl = await selectYtDlpPath();
+    const audioStream = ytdl.exec(url, {
+      noCheckCertificates: true,
+      noWarnings: true,
+      preferFreeFormats: true,
+      addHeader: ["referer:youtube.com", "user-agent:googlebot"],
+      format: "ba",
+      output: "-",
+    }).stdout;
+    if (!audioStream) {
+      throw new Error("Failed to download audio stream");
+    }
 
     //Convert audio stream to mp3
     const audioPassThrough = new PassThrough();
@@ -302,7 +333,29 @@ export async function GET(request: Request) {
     });
 
     if (!requestedFile) {
-      await downloadStreams(url, AUDIO_FILE_PATH, VIDEO_FILE_PATH, quality);
+      const selectedFormat = formatMap.get(quality);
+      if (!selectedFormat) {
+        console.log("formatMap", formatMap);
+        console.log("quality", quality);
+        console.error(
+          `Cannot find the requested format: ${quality}. Available formats are: ${Array.from(
+            formatMap.keys()
+          ).join(", ")}`
+        );
+
+        return new Response(
+          `Cannot find the requested format. Available formats are: ${Array.from(
+            formatMap.keys()
+          ).join(", ")}`,
+          { status: 400 }
+        );
+      }
+      await downloadStreams(
+        url,
+        AUDIO_FILE_PATH,
+        VIDEO_FILE_PATH,
+        String(selectedFormat.itag)
+      );
 
       await mergeAudioVideo(
         VIDEO_FILE_PATH,

components/custom/GetterInput.tsx

@@ -59,7 +59,7 @@ export const GetterInput = () => {
       <div className="relative my-4 w-full">
         <input
           type="text"
-          placeholder="Please enter a instagram, tiktok, youtube... video URL"
+          placeholder="Please enter a youtube video URL or a search query"
           id="video-url"
           name="video-url"
           className="block w-full rounded-md border border-[#081721] bg-[#081721] p-2.5 text-white focus:border-blue-500 focus:ring-blue-500"

copy-binaries.js

@@ -0,0 +1,10 @@
+import path from "path";
+import fs from "fs";
+
+const sourcePath = path.resolve("./node_modules/youtube-dl-exec/bin/yt-dlp");
+const destPath = path.resolve("./.next/server/bin/yt-dlp");
+
+fs.mkdirSync(path.dirname(destPath), { recursive: true });
+
+fs.copyFileSync(sourcePath, destPath);
+console.log(`Copied ${sourcePath} to ${destPath}`);

lib/serverUtils.ts

@@ -3,6 +3,7 @@ import { execSync } from "child_process";
 import * as os from "os";
 import * as fs from "fs";
 import path from "path";
+import { create as createYoutubeDl } from "youtube-dl-exec";
 import { initializeCleanup } from "@/scripts/cleanup";
 
 type Conf = {
@@ -114,3 +115,23 @@ export async function yt_validate(url: string): Promise<"video" | false> {
   }
   return false;
 }
+
+export async function selectYtDlpPath() {
+  let youtubedl;
+
+  if (process.env.NODE_ENV === "development") {
+    const devBinaryPath = path.join(
+      process.cwd(),
+      "node_modules/youtube-dl-exec/bin/yt-dlp"
+    );
+    youtubedl = createYoutubeDl(devBinaryPath);
+  } else {
+    const prodBinaryPath = path.join(
+      process.cwd(),
+      "node_modules/youtube-dl-exec/bin/yt-dlp"
+    );
+    youtubedl = createYoutubeDl(prodBinaryPath);
+  }
+
+  return youtubedl;
+}

next.config.mjs

@@ -4,6 +4,8 @@ import path from "path";
 const packageJsonPath = path.resolve("./package.json");
 const packageJson = JSON.parse(fs.readFileSync(packageJsonPath, "utf-8"));
 
+const __dirname = path.resolve();
+
 /** @type {import('next').NextConfig} */
 const nextConfig = {
   output: "standalone",