Update index.js

index.js

@@ -11,13 +11,14 @@ import { fileURLToPath } from 'url';
 const __dirname = path.dirname(fileURLToPath(import.meta.url));
 import puppeteer from "puppeteer"
 import sharp from 'sharp';
-
+import { fileTypeFromBuffer } from 'file-type';
 const require = createRequire(import.meta.url);
 const PORT = process.env.PORT || 7860;
 const app = express();
 const writeFileAsync = promisify(fs.writeFile);
 const fss = fs.promises;
-
+const sizeOf = promisify(require('image-size'));
+const { exec } = require('child_process');
 
 app.use('/static', express.static(os.tmpdir()));
 
@@ -220,18 +221,25 @@ async function komiku_download(url) {
 }
 
 async function downloadImage(image, tempDir, instanceID) {
-  const response = await axios.get(image.path, {
-    responseType: 'arraybuffer',
-    headers: {
-      'User-Agent': 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;]',
-      'Referer': image.path,
-      'X-Forwarded-For': generateRandomIP()
-    }
-  });
-  const imagePath = path.join(tempDir, `image_${instanceID}_${Date.now()}_${Math.floor(Math.random() * 1000)}.jpeg`);
-  await writeFileAsync(imagePath, response.data);
-
-  return imagePath;
+  try {
+    // Fetch the image data
+    const response = await axios.get(image.path, {
+      responseType: 'arraybuffer',
+      headers: {
+        'User-Agent': '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;]',
+        'Referer': image.path,
+      }
+    });
+    const buffer = Buffer.from(response.data, 'binary');
+    const mimesd = await fileTypeFromBuffer(buffer);
+    const ext = mimesd ? mimesd.ext : 'jpeg';
+    const imagePath = `${tempDir}/image_${instanceID}_${Date.now()}_${Math.floor(Math.random() * 1000)}.${ext}`;
+    await writeFileAsync(imagePath, buffer);
+    return imagePath;
+  } catch (error) {
+    console.error('Error downloading image:', error);
+    throw error;
+  }
 }
 
 async function downloadImages(imgList, tempDir, instanceID) {
@@ -244,15 +252,33 @@ async function downloadImages(imgList, tempDir, instanceID) {
 }
 
 async function createPDF(imagePaths, instanceID, tempDir) {
-  const pdfPath = path.join(os.tmpdir(), `${instanceID}.pdf`);
+  const pdfPath = path.join(tempDir, `${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 });
+    try {
+      // Check if image needs conversion
+      const imageExt = path.extname(imagePath).toLowerCase();
+      const supportedExtensions = ['.jpg', '.jpeg', '.png'];
+      let convertedImagePath = imagePath;
+
+      if (!supportedExtensions.includes(imageExt)) {
+        convertedImagePath = await convertImage(imagePath);
+      }
+
+      const { width, height } = await getImageDimensions(convertedImagePath);
+      doc.addPage({ size: [width, height] });
+      doc.image(convertedImagePath, 0, 0, { width: width, height: height });
+
+      // Remove converted images if they were created
+      if (convertedImagePath !== imagePath) {
+        fs.unlinkSync(convertedImagePath);
+      }
+    } catch (error) {
+      console.error(`Error processing image ${imagePath}:`, error.message);
+    }
   }
 
   doc.end();
@@ -261,11 +287,22 @@ async function createPDF(imagePaths, instanceID, tempDir) {
 }
 
 async function getImageDimensions(imagePath) {
-  const sizeOf = promisify(require('image-size'));
   const dimensions = await sizeOf(imagePath);
   return dimensions;
 }
 
+async function convertImage(imagePath) {
+  return new Promise((resolve, reject) => {
+    const outputImagePath = imagePath.replace(/\.\w+$/, '.jpg');
+    exec(`convert "${imagePath}" "${outputImagePath}"`, (error, stdout, stderr) => {
+      if (error) {
+        console.error('ImageMagick Error:', stderr);
+        return reject(error);
+      }
+      resolve(outputImagePath);
+    });
+  });
+
 app.get('/download', async (req, res) => {
   const { url } = req.query;
   if (!url) {