Update index.js

index.js

@@ -251,67 +251,89 @@ app.get('/nhentai', async (req, res) => {
   }
 });
 
-async function r34(url) {
-  const browser = await puppeteer.launch({
-    headless: true, // Non-headless untuk debugging
-    args: ['--no-sandbox', '--disable-setuid-sandbox']
-  });
-
-  const page = await browser.newPage();
-
-  // Setel header User-Agent dan Referer
-  await page.setExtraHTTPHeaders({
-    'User-Agent': getRandomUserAgent(),
-    'Referer': url,
-    'X-Forwarded-For': generateRandomIP()
-  });
-
-  /*
-
-  // Setel direktori unduhan unik untuk setiap permintaan
-  const downloadPath = fs.mkdtempSync(path.join(os.tmpdir(), 'r34')); // Buat subdirektori unik
-
-  await page._client.send('Page.setDownloadBehavior', {
-    behavior: 'allow',
-    downloadPath: downloadPath
-  });
-  */
 
-  await page.goto(url); // Buka URL yang diberikan
+async function downloadVideoR34(url) {
+    const browser = await puppeteer.launch({
+        headless: true,
+        args: ['--no-sandbox', '--disable-setuid-sandbox']
+    });
+    const page = await browser.newPage();
+    const downloadPath = path.resolve(os.tmpdir()); // Set your download directory here
 
-  // Asumsikan ada tombol unduh dengan selector '#downloadButton'
-  await page.click("#tab_video_info > div:nth-child(4) > div > a:nth-child(5)");
+    await page.setExtraHTTPHeaders({
+        'User-Agent': getRandomUserAgent(),
+        'Referer': url,
+        'X-Forwarded-For': generateRandomIP()
+    });
 
-  /*
+    await page.goto(url, { waitUntil: 'networkidle0' });
+
+    // Get download link and other information
+    const result = 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 { downloadLink: null, links };
+    });
 
-  // Tunggu sampai file diunduh dengan memonitor perubahan pada direktori
-  let filename = '';
-  while (filename === '') {
-    const files = fs.readdirSync(downloadPath);
-    if (files.length > 0) {
-      filename = files[0]; // Ambil nama file pertama yang muncul
+    const { downloadLink, links } = result;
+
+    let filePath = null;
+    let filename = null;
+    if (downloadLink) {
+        const response = await axios({
+            method: 'GET',
+            url: downloadLink,
+            responseType: 'stream'
+        });
+
+        // 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);
+        const writer = fs.createWriteStream(filePath);
+
+        response.data.pipe(writer);
+
+        await new Promise((resolve, reject) => {
+            writer.on('finish', resolve);
+            writer.on('error', reject);
+        });
     } else {
-      await new Promise(resolve => setTimeout(resolve, 100)); // Tunggu sebentar sebelum cek lagi
+        console.error('Download link not found.');
     }
-  }
-  */
-
-  const cookies = await page.cookies();
 
-  await browser.close();
+    await browser.close();
 
-  return { cookies } //{ path: path.join(downloadPath, filename), url: url };
+    return {
+        download: "https://arashicode-komik.hf.space/static/" + filename,
+        path: filePath,
+        url: downloadLink,
+        links
+    };
 }
 
-
-
 app.get('/r34', async (req, res) => {
   const { url } = req.query;
   if (!url) {
     return res.status(400).send('URL is required');
   }
   try {
-    const result = await r34(url);
+    const result = await downloadVideoR34(url);
     res.json(result);
   } catch (error) {
     res.status(500).send('Error processing request');