Update index.js

index.js

@@ -263,7 +263,6 @@ app.get('/nhentai', async (req, res) => {
 
 
 
-
 // Fungsi untuk mendownload dari URL
 async function downloadFromUrl(url, filePath) {
     const writer = fs.createWriteStream(filePath);
@@ -441,28 +440,107 @@ app.get('/r34/info', async (req, res) => {
 
 // Endpoint untuk download
 app.get('/r34/download', async (req, res) => {
-  const { url } = req.query;
+  const { url, type = 'download' } = req.query;
   if (!url) {
     return res.status(400).send('URL is required');
   }
+
   try {
-    const result = await Rules34(url);
+    const result = await Rules34(url);  // Mendapatkan jalur unduhan dari modul Rules34
     const filePath = result.downloadPath;
-    res.download(filePath, (err) => {
+
+    // 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 stat = fs.statSync(filePath);
+      const fileSize = stat.size;
+      const range = req.headers.range;
+
+      if (range) {
+        const parts = range.replace(/bytes=/, "").split("-");
+        const start = parseInt(parts[0], 10);
+        const end = parts[1] ? parseInt(parts[1], 10) : fileSize - 1;
+
+        if (start >= fileSize) {
+          res.status(416).send('Requested range not satisfiable\n' + start + ' >= ' + fileSize);
+          return;
+        }
+
+        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).on('finish', () => {
+          fs.unlink(filePath, (err) => {
+            if (err) {
+              console.error(`Error deleting file: ${err.message}`);
+            } else {
+              console.log(`File deleted: ${filePath}`);
+            }
+          });
+        });
+      } else {
+        res.setHeader('Content-Length', fileSize);
+        const file = fs.createReadStream(filePath);
+        file.pipe(res).on('finish', () => {
+          fs.unlink(filePath, (err) => {
+            if (err) {
+              console.error(`Error deleting file: ${err.message}`);
+            } else {
+              console.log(`File deleted: ${filePath}`);
+            }
+          });
+        });
+      }
+    } else if (type === 'buffer') {
+      fs.readFile(filePath, (err, data) => {
         if (err) {
-            console.error('Gagal mengirim file:', err);
-            res.status(500).send('Gagal mengirim file: \n' + err);
+          console.error('Gagal membaca file:', err);
+          res.status(500).send('Gagal membaca file: \n' + err);
         } else {
-            fs.unlink(filePath, (err) => {
-                if (err) {
-                    console.error(`Error deleting file: ${err.message}`);
-                } else {
-                    console.log(`File deleted: ${filePath}`);
-                }
-            });
+          res.setHeader('Content-Length', data.length);
+          res.send(data);
+          fs.unlink(filePath, (err) => {
+            if (err) {
+              console.error(`Error deleting file: ${err.message}`);
+            } else {
+              console.log(`File deleted: ${filePath}`);
+            }
+          });
         }
-    });
+      });
+    } else { // Default to 'download'
+      res.setHeader('Content-Type', 'video/mp4');
+      res.download(filePath, fileName, (err) => {
+        if (err) {
+          console.error('Gagal mengirim file:', err);
+          res.status(500).send('Gagal mengirim file: \n' + err);
+        } else {
+          fs.unlink(filePath, (err) => {
+            if (err) {
+              console.error(`Error deleting file: ${err.message}`);
+            } else {
+              console.log(`File deleted: ${filePath}`);
+            }
+          });
+        }
+      });
+    }
   } catch (error) {
+    console.error('Error processing request:', error);
     res.status(500).send('Error processing request');
   }
 });