Update index.js

index.js

@@ -252,65 +252,104 @@ app.get('/nhentai', async (req, res) => {
 });
 
 
-async function downloadVideoR34(url) {
-    const browser = await puppeteer.launch({
-        headless: true,
-        args: ['--no-sandbox', '--disable-setuid-sandbox']
+async function downloadFromUrl(url, filePath) {
+    const writer = fs.createWriteStream(filePath);
+
+    const response = await axios({
+        url,
+        method: 'GET',
+        responseType: 'stream'
     });
-    const page = await browser.newPage();
-    const downloadPath = path.resolve(os.tmpdir());
 
-    await page.setExtraHTTPHeaders({
-        'User-Agent': getRandomUserAgent(),
-        'Referer': url,
-        'X-Forwarded-For': generateRandomIP()
+    response.data.pipe(writer);
+
+    return new Promise((resolve, reject) => {
+        writer.on('finish', resolve);
+        writer.on('error', reject);
     });
+}
+
+async function Rules34(url_r34) {
+    try {
+        const browser = await puppeteer.launch();
+        const page = await browser.newPage();
+
+        await page.setExtraHTTPHeaders({
+			'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
+			'Referer': url_r34
+	    });
+        await page.goto(url_r34, { referer: url_r34 });
+
+        const htmlContent = await page.content();
+        const $ = cheerio.load(htmlContent);
+
+        const title = $("body > div > div.wrapper > div.main > div.container > div > div > div:nth-child(1) > div.heading > h1").text().trim() || url_r34.match(/\/video\/\d+\/(.*)\//)?.[1];
+
+        const infoElements = $("#tab_video_info > div");
+        const result = {
+            title: title,
+            artist: [],
+            uploader: [],
+            tags: [],
+            download: [],
+            category: []
+        };
 
-    await page.goto(url, { waitUntil: 'networkidle0' });
-    
-    const downloadLink = await page.evaluate(() => {
-        const infoElements = document.querySelectorAll("#tab_video_info > div");
         if (infoElements.length > 0) {
-            const download = infoElements[infoElements.length - 1];
-            if (download) {
-                const wrapDivsDownload = download.querySelectorAll("div.wrap > a");
-                return wrapDivsDownload[0].href;
+            const tags = infoElements.eq(infoElements.length - 2);
+            const download = infoElements.eq(infoElements.length - 1);
+            const videoInfo = infoElements.eq(infoElements.length - 3);
+
+            if (tags.length > 0) {
+                tags.find("div.wrap > a").each((index, element) => {
+                    result.tags.push({ title: $(element).text().trim(), url: $(element).attr('href') });
+                });
+                result.tags.pop(); // Menghapus elemen terakhir dari tags
             }
-        }
-        return null;
-    });
-    
-    let filePath = 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];
-
-        filePath = path.join(downloadPath, `/${title}.mp4`);
-        const writer = fs.createWriteStream(filePath);
-
-        response.data.pipe(writer);
-
-        await new Promise((resolve, reject) => {
-            writer.on('finish', resolve);
-            writer.on('error', reject);
-        });
-
-    } else {
-        console.error('Download link not found.');
-    }
 
-    await browser.close();
+            if (download.length > 0) {
+                download.find("div.wrap > a").each((index, element) => {
+                    result.download.push({ quality: $(element).text().trim(), url: $(element).attr('href') });
+                });
+
+                // Mendownload dari URL terakhir di array download
+                const lastDownloadUrl = result.download[result.download.length - 1].url;
+                const downloadPath = path.resolve(os.tmpdir())
+                let filenem = `/${Date.now()}_file.mp4`
+                const filePath = path.join(downloadPath, filenem)
+
+                await downloadFromUrl(lastDownloadUrl, filePath);
 
-    return {
-        path: filePath,
-        download: `https://arashicode-komik.hf.space/static/${title}.mp4`,
-    };
+                result.downloadPath = filePath; // Menambahkan path file yang telah di-download ke dalam objek result
+                result.downloadUrl = "https://arashicode-komik.hf.space/static/" + filenem; // Menambahkan path file yang telah di-download ke dalam objek result
+            }
+
+            if (videoInfo.length > 0) {
+                videoInfo.find("div.cols > div").each((index, col) => {
+                    const categoryLinks = $(col).find("div.col:nth-child(1) > a");
+                    categoryLinks.each((index, link) => {
+                        result.category.push({ title: $(link).text().trim(), url: $(link).attr('href') });
+                    });
+
+                    const artistLinks = $(col).find("div.col:nth-child(2) > a");
+                    artistLinks.each((index, link) => {
+                        result.artist.push({ title: $(link).text().trim(), url: $(link).attr('href') });
+                    });
+
+                    const uploaderLinks = $(col).find("div.col:nth-child(3) > a");
+                    uploaderLinks.each((index, link) => {
+                        result.uploader.push({ title: $(link).text().trim(), url: $(link).attr('href') });
+                    });
+                });
+            }
+        }
+
+        await browser.close();
+        return result;
+    } catch (error) {
+        console.error('Error fetching data:', error);
+        throw error; // Rethrow error to handle it upstream
+    }
 }
 
 app.get('/r34', async (req, res) => {
@@ -319,7 +358,7 @@ app.get('/r34', async (req, res) => {
     return res.status(400).send('URL is required');
   }
   try {
-    const result = await downloadVideoR34(url);
+    const result = await Rules34(url);
     res.json(result);
   } catch (error) {
     res.status(500).send('Error processing request');