Update index.js

index.js

@@ -3,6 +3,7 @@ const morgan = require('morgan');
 const express = require('express');
 const ytdl = require('ytdl-core');
 const { Writable, pipeline } = require('stream');
+const ffmpeg = require("fluent-ffmpeg")
 const util = require('util');
 const axios = require('axios');
 
@@ -227,7 +228,85 @@ const youtubedl = async (url) => {
         audio
     };
 };
+async function streamToBuffer(stream) {
+    const chunks = [];
+    const captureChunks = new Writable({
+        write(chunk, encoding, callback) {
+            chunks.push(chunk);
+            callback();
+        }
+    });
+      
+    await util.promisify(pipeline)(stream, captureChunks);
+
+    return Buffer.concat(chunks);
+} 
+async function fileDitch(media){
+  return new Promise(async (resolve, reject) => {
+    let {fileTypeFromBuffer} = await (await import('file-type'))
+     let mime = await fileTypeFromBuffer(media)
+     let form = new FormData()
+
+     form.append("files[]", media, `file-${new Date().getTime()}.${mime.ext}`)
 
+     axios.post("https://up1.fileditch.com/temp/upload.php", form, {
+        headers: {
+           "User-Agent": generateRandomUserAgent(),
+          "X-Forwarded-For": generateRandomIP(),
+           ...form.getHeaders()
+        }
+     }).then(({ data }) => resolve(data?.files[0]?.url)).catch(reject)
+  })
+//https://up1.fileditch.com/upload.php
+}
+async function convertMp4ToAudio(inputBuffer) {
+    return new Promise((resolve, reject) => {
+        const inputStream = new Readable();
+        inputStream.push(inputBuffer);
+        inputStream.push(null);
+
+        const outputBuffer = [];
+        const outputStream = new Writable({
+            write(chunk, encoding, callback) {
+                outputBuffer.push(chunk);
+                callback();
+            }
+        });
+
+        ffmpeg(inputStream)
+            .toFormat('mp3')
+            .on('end', () => {
+                console.log('Conversion finished!');
+                resolve(Buffer.concat(outputBuffer));
+            })
+            .on('error', (err) => {
+                console.error('Error during conversion:', err);
+                reject(err);
+            })
+            .pipe(outputStream);
+    });
+}
+async function ytAPI(url) {
+  try {
+    const ID = ytdl.getVideoID(url)
+          //let videoStream = await ytdl(ID, { filter: 'audioandvideo', quality: 'highestvideo' });
+    let data = await ytdl.getInfo('https://www.youtube.com/watch?v=' + ID)
+      let format = ytdl.chooseFormat(data.formats, { filter: 'videoandaudio', quality: 'highestvideo' });
+          let audioStream = await ytdl(ID, {filter: "audioandvideo", quality:"lowestvideo"})
+          //let buffermp4 = await streamToBuffer(videoStream)
+          let buffermp3 = await streamToBuffer(audioStream)
+          buffermp3 = await convertMp4ToAudio(buffermp3)
+          //buffermp4 = await fileDitch(buffermp4)
+          buffermp3 = await fileDitch(buffermp3)
+      return {
+        mp4_url: format.url,
+        mp3_url: buffermp3,
+          }
+  } catch (err) {
+    console.error('Error occurred:', err);
+    return null;
+  }
+}
 const app = express()
     .set('json spaces', 4)
     .use(morgan('dev'))
@@ -276,19 +355,27 @@ const app = express()
             res.json({ message: e });
         }
     })
-    .get('/ytdl', async (req, res) => {
-        try {
-            let { url } = req.query
-            if (!ytIdRegex.test(url)) return res.json({ message: 'Invalid URL' });
-            let video = await ytmp4(url)
-            let audio = await ytmp3(url)
-            return res.json({
-                video: video,
-                audio: audio
-            })
-        } catch (e) {
-            console.log(e);
-            res.json({ message: e });
-        }
-    })
+  .get('/ytdl', async (req, res) => {
+    try {
+        let { url } = req.query;
+        //if (!ytIdRegex.test(url)) return res.json({ message: 'Invalid URL' });
+        if (!ytdl.validateURL(url)) return res.json({ message: 'Invalid URL' });
+        let data = await ytAPI(url) || {};
+        let videoID = ytdl.getVideoID(url);
+        let dataInfo = await yts({ videoId: videoID }) || {};
+        //let otherAPI =  await (await axios.get("https://line.1010diy.com/web/free-mp3-finder/detail?url="+url))?.data?.data ? await (await axios.get("https://line.1010diy.com/web/free-mp3-finder/detail?url="+url))?.data?.data : null
+
+        let response = {
+            ...dataInfo,
+            mp4: data.mp4_url,
+            mp3: data.mp3_url,
+            //other: otherAPI || null
+        };
+
+        return res.json(response);
+    } catch (e) {
+        console.log(e);
+        return res.status(500).json({ message: e.message });
+    }
+})
     .listen(7860, () => console.log('App running on port 7860'));