Update index.js

index.js

@@ -1,5 +1,15 @@
+import { createRequire } from 'module';
+import { fileURLToPath } from 'url';
+import path from 'path';
+import PDFDocument from 'pdfkit';
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+const require = createRequire(import.meta.url);
+
+const { promisify } = require('util');
 const express = require('express');
 const axios = require('axios');
+const cheerio = require('cheerio');
 const fs = require('fs');
 const { tmpdir } = require('os');
 const { join } = require('path');
@@ -8,6 +18,112 @@ const puppeteer = require('puppeteer');
 const app = express();
 const PORT = 7860;
 const { format } = require("util");
+const os = require("os");
+const writeFileAsync = promisify(fs.writeFile);
+const fss = fs.promises;
+
+function generateRandomID(length = 8) {
+  const characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
+  let result = '';
+  for (let i = 0; i < length; i++) {
+    result += characters.charAt(Math.floor(Math.random() * characters.length));
+  }
+  return result;
+}
+
+async function komiku_download(url) {
+  const instanceID = generateRandomID();
+  const tempDir = path.join(os.tmpdir(), instanceID);
+  await fss.mkdir(tempDir);
+
+  // Extracting the title from the URL
+  const title = url.split('/').filter(part => part).pop();
+
+  try {
+    const response = await axios.get(url);
+    const html = response.data;
+    const $ = cheerio.load(html);
+    const imgList = [];
+
+    $('#Baca_Komik img').each((index, element) => {
+      const src = $(element).attr('src');
+      imgList.push({ path: src });
+    });
+
+    const imagePaths = await downloadImages(imgList, tempDir, instanceID);
+    const pdfPath = await createPDF(imagePaths, instanceID, tempDir);
+
+    console.log(`PDF berhasil dibuat: ${pdfPath}`);
+    return { path: pdfPath, title: title };
+  } catch (error) {
+    console.log(error);
+    throw error;
+  } finally {
+    await fss.rmdir(tempDir, { recursive: true });
+  }
+}
+
+async function downloadImage(image, tempDir, instanceID) {
+  const response = await axios.get(image.path, { responseType: 'arraybuffer' });
+  const imagePath = path.join(tempDir, `image_${instanceID}_${Date.now()}_${Math.floor(Math.random() * 1000)}.jpg`);
+  await writeFileAsync(imagePath, response.data);
+
+  return imagePath;
+}
+
+async function downloadImages(imgList, tempDir, instanceID) {
+  const imagePaths = [];
+  for (const img of imgList) {
+    const imagePath = await downloadImage(img, tempDir, instanceID);
+    imagePaths.push(imagePath);
+  }
+  return imagePaths;
+}
+
+async function createPDF(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;
+}
+
+async function getImageDimensions(imagePath) {
+  const { promisify } = require('util');
+  const sizeOf = promisify(require('image-size'));
+  const dimensions = await sizeOf(imagePath);
+  return dimensions;
+}
+
+app.get('/komikudl', async (req, res) => {
+  const url = req.query.url;
+
+  if (!url) {
+    return res.status(400).send('URL is required');
+  }
+
+  try {
+    const data = await komiku_download(url);
+    res.download(data.path, `${data.title}.pdf`, (err) => {
+      if (err) {
+        console.error(err);
+      }
+
+      fs.unlinkSync(data.path); // Remove the file after sending it
+    });
+  } catch (error) {
+    res.status(500).send('An error occurred while generating the PDF');
+  }
+});
 
 app.get('/add', async (req, res) => {
     try {
@@ -420,6 +536,31 @@ app.get('/r34/v2', async (req, res) => {
     }
 });
 
+
+async function pingWebsite() {
+    const browser = await puppeteer.launch({
+        headless: true,
+        args: ['--no-sandbox', '--disable-setuid-sandbox']
+    });
+    const page = await browser.newPage();
+    await page.setUserAgent("Mozilla/5.0 (Linux; Android 10; SM-G965U Build/QP1A.190711.020; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/114.0.5735.141 Mobile Safari/537.36 [FB_IAB/FB4A;FBAV/420.0.0.32.61;]");
+    await page.goto('https://huggingface.co/spaces/Nexchan/yutub'); 
+    console.log("Ping");
+    await browser.close();
+}
+
+// Ping website setiap 5 jam
+async function pingEvery5Hours() {
+    await pingWebsite();
+    setInterval(async () => {
+        await pingWebsite();
+    }, 5 * 60 * 60 * 1000); // 5 hours in milliseconds
+}
+
+// Mulai ping
+pingEvery5Hours();
+
+
 app.listen(PORT, () => {
     console.log(`Server berjalan di port ${PORT}`);
 });