Update index.js

index.js

@@ -258,7 +258,7 @@ async function downloadVideoR34(url) {
         args: ['--no-sandbox', '--disable-setuid-sandbox']
     });
     const page = await browser.newPage();
-    const downloadPath = path.resolve(os.tmpdir()); // Set your download directory here
+    const downloadPath = path.resolve(os.tmpdir());
 
     await page.setExtraHTTPHeaders({
         'User-Agent': getRandomUserAgent(),
@@ -267,44 +267,31 @@ async function downloadVideoR34(url) {
     });
 
     await page.goto(url, { waitUntil: 'networkidle0' });
-
-    // Get download link and other information
-    const result = await page.evaluate(() => {
+    
+    const downloadLink = await page.evaluate(() => {
         const infoElements = document.querySelectorAll("#tab_video_info > div");
-        const links = Array.from(document.querySelectorAll('a')).map(a => ({
-            href: a.href,
-            text: a.textContent.trim()
-        }));
-        
         if (infoElements.length > 0) {
             const download = infoElements[infoElements.length - 1];
             if (download) {
                 const wrapDivsDownload = download.querySelectorAll("div.wrap > a");
-                return { downloadLink: wrapDivsDownload[0].href, links };
+                return wrapDivsDownload[0].href;
             }
         }
-        return { downloadLink: null, links };
+        return null;
     });
-
-    const { downloadLink, links } = result;
-
+    
     let filePath = null;
-    let filename = null;
+    let title;
+
     if (downloadLink) {
         const response = await axios({
             method: 'GET',
             url: downloadLink,
             responseType: 'stream'
         });
+        title = url.match(/video\/(\d+)\/([a-zA-Z0-9-]+)\//)[2];
 
-        // Get filename from content-disposition header or generate from URL
-        const contentDisposition = response.headers['content-disposition'];
-        filename = contentDisposition ? contentDisposition.split('filename=')[1] : null;
-        if (!filename) {
-            filename = url.match(/video\/(\d+)\/([a-zA-Z0-9-]+)\//)[2] + '.mp4'; // Adjust as needed
-        }
-
-        filePath = path.join(downloadPath, filename);
+        filePath = path.join(downloadPath, `/${title}.mp4`);
         const writer = fs.createWriteStream(filePath);
 
         response.data.pipe(writer);
@@ -313,6 +300,7 @@ async function downloadVideoR34(url) {
             writer.on('finish', resolve);
             writer.on('error', reject);
         });
+
     } else {
         console.error('Download link not found.');
     }
@@ -320,10 +308,8 @@ async function downloadVideoR34(url) {
     await browser.close();
 
     return {
-        download: "https://arashicode-komik.hf.space/static/" + filename,
         path: filePath,
-        url: downloadLink,
-        links
+        download: `https://arashicode-komik.hf.space/static/${title}.mp4`,
     };
 }