Upload index.js

index.js

@@ -279,6 +279,68 @@ async function downloadFromUrl(url, filePath) {
     });
 }
 
+async function Rules34info(url_r34) {
+    try {
+        // Mengambil halaman HTML dengan Axios
+        const response = await axios.get(url_r34, {
+            headers: {
+                '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
+            }
+        });
+        
+        const htmlContent = response.data;
+        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: [],
+            category: []
+        };
+
+        if (infoElements.length > 0) {
+            const tags = infoElements.eq(infoElements.length - 2);
+            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
+            }
+
+            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') });
+                    });
+                });
+            }
+        }
+
+        return result;
+    } catch (error) {
+        console.error('Error fetching data:', error);
+        throw error; // Rethrow error to handle it upstream
+    }
+}
+
 async function Rules34(url_r34) {
     try {
         const browser = await puppeteer.launch();
@@ -386,6 +448,92 @@ app.get('/r34', async (req, res) => {
   }
 });
 
+
+app.get('/r34/download', async (req, res) => {
+  const { url, type = 'download' } = req.query;
+  if (!url) {
+    return res.status(400).send('URL is required');
+  }
+
+  try {
+    const result = await Rules34(url);  // Mendapatkan jalur unduhan dari modul Rules34
+    const filePath = result.downloadPath;
+
+    // Cek apakah file benar-benar ada sebelum mengirim
+    if (!fs.existsSync(filePath)) {
+      return res.status(404).send('File not found');
+    }
+
+    const fileName = path.basename(filePath);
+    const mimeType = 'video/mp4';
+
+    res.setHeader('Content-Type', mimeType);
+
+    if (type === 'stream') {
+      const range = req.headers.range;
+
+      if (range) {
+        const stat = fs.statSync(filePath);
+        const fileSize = stat.size;
+        
+        const parts = range.replace(/bytes=/, "").split("-");
+        const start = parseInt(parts[0], 10);
+        const end = parts[1] ? parseInt(parts[1], 10) : fileSize - 1;
+
+        const chunksize = (end - start) + 1;
+        const file = fs.createReadStream(filePath, { start, end });
+        const head = {
+          'Content-Range': `bytes ${start}-${end}/${fileSize}`,
+          'Accept-Ranges': 'bytes',
+          'Content-Length': chunksize,
+          'Content-Type': mimeType,
+        };
+
+        res.writeHead(206, head);
+        file.pipe(res);
+      } else {
+        res.setHeader('Content-Length', fileSize);
+        res.sendFile(filePath); // Menggunakan res.sendFile untuk mengirim file
+      }
+    } else if (type === 'buffer') {
+      fs.readFile(filePath, (err, data) => {
+        if (err) {
+          console.error('Gagal membaca file:', err);
+          res.status(500).send('Gagal membaca file: \n' + err);
+        } else {
+          res.setHeader('Content-Length', data.length);
+          res.send(data);
+        }
+      });
+    } else { // Default to 'download'
+      res.download(filePath, fileName, (err) => {
+        if (err) {
+          console.error('Gagal mengirim file:', err);
+          res.status(500).send('Gagal mengirim file: \n' + err);
+        }
+      });
+    }
+  } catch (error) {
+    console.error('Error processing request:', error);
+    res.status(500).send('Error processing request');
+  }
+});
+
+
+app.get('/r34/info', async (req, res) => {
+  const { url } = req.query;
+  if (!url) {
+    return res.status(400).send('URL is required');
+  }
+  try {
+    const result = await Rules34info(url);
+    res.json(result);
+  } catch (error) {
+    res.status(500).send('Error processing request');
+  }
+});
+
+
 async function r34Cookie(url) {
   const browser = await puppeteer.launch({
     headless: true, // Non-headless untuk debugging