Update index.js

index.js

@@ -76,6 +76,111 @@ function generateRandomID(length = 8) {
   }
   return result;
 }
+/**************************************/
+
+async function komiku_downloadV2(url) {
+  const instanceID = generateRandomID();
+  const tempDir = path.join(os.tmpdir(), instanceID);
+  await fss.mkdir(tempDir);
+
+  const title = url.split('/').filter(part => part).pop();
+
+  const browser = await puppeteer.launch({
+        headless: true,
+        args: ['--no-sandbox', '--disable-setuid-sandbox']
+    });
+  try {
+    const page = await browser.newPage();
+    await page.setUserAgent(getRandomUserAgent() || 'Mozilla/5.0 (Linux; Android 6.0.1; SM-N916S Build/MMB29K; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/106.0.5249.126 Mobile Safari/537.36 [FB_IAB/FB4A;FBAV/389.0.0.42.111;]');
+    await page.setExtraHTTPHeaders({
+      'Referer': 'https://komiku.id/devious-son-of-heaven-chapter-04/',
+      'X-Forwarded-For': generateRandomIP()
+    });
+
+    await page.goto(url, { waitUntil: 'networkidle2' });
+    const imgList = await page.evaluate(() => {
+      return Array.from(document.querySelectorAll('#Baca_Komik img')).map(img => ({ path: img.src }));
+    });
+
+    const imagePaths = await downloadImageV2s(imgList, tempDir, instanceID, page);
+    const pdfPath = await createPDFV2(imagePaths, instanceID, tempDir);
+
+    return { url: "https://arashicode-komik.hf.space/static/" + instanceID + ".pdf", path: `/static/${instanceID}.pdf`, title: title, path: pdfPath};
+  } catch (error) {
+    console.log(error);
+    throw error;
+  } finally {
+    await browser.close();
+    await fss.rmdir(tempDir, { recursive: true });
+  }
+}
+
+async function downloadImageV2(image, tempDir, instanceID, page) {
+  await page.setUserAgent(getRandomUserAgent() || 'Mozilla/5.0 (Linux; Android 6.0.1; SM-N916S Build/MMB29K; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/106.0.5249.126 Mobile Safari/537.36 [FB_IAB/FB4A;FBAV/389.0.0.42.111;]');
+  await page.setExtraHTTPHeaders({
+    'Referer': 'https://komiku.id/devious-son-of-heaven-chapter-04/',
+    'X-Forwarded-For': generateRandomIP()
+  });
+
+  const viewSource = await page.goto(image.path);
+  const imagePath = path.join(tempDir, `image_${instanceID}_${Date.now()}_${Math.floor(Math.random() * 1000)}.jpg`);
+  await fs.promises.writeFile(imagePath, await viewSource.buffer());
+
+  return imagePath;
+}
+
+async function downloadImageV2s(imgList, tempDir, instanceID, page) {
+  const imagePaths = [];
+  for (const img of imgList) {
+    const imagePath = await downloadImageV2(img, tempDir, instanceID, page);
+    imagePaths.push(imagePath);
+  }
+  return imagePaths;
+}
+
+async function createPDFV2(imagePaths, instanceID, tempDir) {
+  const pdfPath = path.join(os.tmpdir(), `${instanceID}.pdf`);
+  const doc = new PDFDocument({ autoFirstPage: false });
+
+  doc.pipe(fs.createWriteStream(pdfPath));
+
+  for (const imagePath of imagePaths) {
+    const { width, height } = await getImageDimensions(imagePath);
+    doc.addPage({ size: [width, height] });
+    doc.image(imagePath, 0, 0, { width: width, height: height });
+  }
+
+  doc.end();
+
+  return pdfPath;
+}
+
+
+app.get('/downloadv2', async (req, res) => {
+  const { url } = req.query;
+  if (!url) {
+    return res.status(400).send('URL is required');
+  }
+
+  try {
+    const result = await komiku_downloadV2(url);
+    res.json(result);
+    setTimeout(() => {
+      fs.unlink(result.path, (err) => {
+        if (err) {
+          console.error(`Error deleting file: ${err.message}`);
+        } else {
+          console.log(`File deleted: ${result.path}`);
+        }
+      });
+    }, 280000);
+  } catch (error) {
+    res.status(500).send('Error processing request');
+  }
+});
+
+
+/**************************************/
 
 async function komiku_download(url) {
   const instanceID = generateRandomID();