Update index.js

index.js

@@ -1,7 +1,12 @@
 const yts = require('yt-search');
 const morgan = require('morgan');
 const express = require('express');
-const ytdl = require('ytdl-core')
+const ytdl = require('ytdl-core');
+const { Writable } = require('stream');
+const util = require('util');
+const FormData = require('form-data');
+const axios = require('axios');
+const { fileTypeFromBuffer } = require('file-type')
 
 // Regex untuk mengidentifikasi URL YouTube dan Mega
 const ytIdRegex = /(?:http(?:s|):\/\/|)(?:(?:www\.|)?youtube(?:\-nocookie|)\.com\/(?:shorts\/)?(?:watch\?.*(?:|\&)v=|embed\/|v\/)?|youtu\.be\/)([-_0-9A-Za-z]{11})/;
@@ -17,6 +22,35 @@ const post = async (url, form, headers = {}) => {
 };
 
 //YTDL-CORE
+async function uploadBuffer(buffer) {
+    //file: (binary)
+    //expirationTime: 1
+    let mime = await fileTypeFromBuffer(buffer)
+    let formData = new FormData()
+    formData.append('file', buffer, `${Date.now()}.${mime.ext}`)
+    formData.append('expirationTime', 1)
+    let { data } = await axios.post(`https://cdn.itzpire.site/upload`, formData, {
+        headers: {
+            ...formData.getHeaders(),
+            'accept': 'application/json',
+      'Content-Type': 'multipart/form-data'
+        }
+    })
+    return data?.fileUrl
+}
+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);
+} 
 function formatViews(viewCount) {
   if (viewCount >= 1000000000) {
     return (viewCount / 1000000000).toFixed(1) + 'B';
@@ -70,8 +104,11 @@ function formatSize(bytes, si = false, dp = 2) {
 async function ytmp4(url, quality = 'highestvideo') {
   try {
     const ID = ytdl.getVideoID(url),
-      data = await ytdl.getInfo('https://www.youtube.com/watch?v=' + ID),
-      format = ytdl.chooseFormat(data.formats, { quality: quality });
+          data = await ytdl.getInfo('https://www.youtube.com/watch?v=' + ID)
+      const videoStream = await ytdl(ID, { quality: 'highestvideo' });
+      let buffer = await streamToBuffer(videoStream)
+      let upload = await uploadBuffer(buffer)
+      let format = ytdl.chooseFormat(data.formats, { quality: quality });
 
     if (format) {
       return {
@@ -85,7 +122,8 @@ async function ytmp4(url, quality = 'highestvideo') {
         quality: format.qualityLabel,
         thumb: data.videoDetails.thumbnails[0].url,
         dl_url: format.url,
-      }
+        url_v2: upload,
+          }
     } else {
       throw new Error('No suitable format found');
     }
@@ -97,8 +135,11 @@ async function ytmp4(url, quality = 'highestvideo') {
 async function ytmp3(url, bitrate = 'lowestaudio') {
   try {
     const ID = ytdl.getVideoID(url),
-      data = await ytdl.getInfo('https://www.youtube.com/watch?v=' + ID),
-      format = ytdl.chooseFormat(data.formats, { filter: 'audioonly', quality: bitrate });
+          data = await ytdl.getInfo('https://www.youtube.com/watch?v=' + ID)
+      let audioStream = await ytdl(ID, { filter: 'audioonly', quality: 'lowestaudio', });
+      let buffer = await streamToBuffer(audioStream)
+      let upload = await uploadBuffer(buffer)
+      let format = ytdl.chooseFormat(data.formats, { filter: 'audioonly', quality: bitrate });
 
     if (format) {
       return {
@@ -111,7 +152,8 @@ async function ytmp3(url, bitrate = 'lowestaudio') {
         size: format.contentLength ? formatSize(format.contentLength) : 0,
         quality: format.audioQuality,
         thumb: data.videoDetails.thumbnails[0].url,
-        dl_url: format.url,
+          dl_url: format.url,
+        url_v2: upload
       }
     } else {
       throw new Error('No suitable format found');