Update server.js

server.js

@@ -4,33 +4,43 @@ const { chromium } = require("playwright");
 const app = express();
 
 app.get("/:url", async (req, res) => {
+  let browser;
   try {
-    const { url } = req.params;
-    if (!url.trim()) return res.sendStatus(400);
-
-    const browser = await chromium.launch({
-      args: [
-		'--incognito',
-		'--single-process',
-		'--no-sandbox',
-		'--no-zygote',
-        '--no-cache'
-      ],
+    let { url } = req.params;
+    url = decodeURIComponent(url);
+    if (!url.trim().startsWith("http")) return res.sendStatus(400);
+
+    const { type, width, height } = req.query;
+    const viewports = {
+      desktop: { width: 1280, height: 720 },
+      mobile: { width: 375, height: 812, isMobile: true },
+      tablet: { width: 768, height: 1024, isMobile: true },
+    };
+
+    let viewport;
+    if (width && height) {
+      viewport = { width: parseInt(width), height: parseInt(height) };
+    } else {
+      viewport = viewports[type] || viewports.desktop;
+    }
+
+    browser = await chromium.launch({
+      args: ["--incognito", "--single-process", "--no-sandbox", "--no-zygote", "--no-cache"],
       executablePath: process.env.CHROME_BIN,
-      headless: true
+      headless: true,
     });
+
     const page = await browser.newPage();
-    await page.goto(url);
-    const buffer = await page.screenshot();
+    await page.setViewportSize(viewport);
+    await page.goto(url, { waitUntil: "load", timeout: 10000 });
 
-    res.setHeader("Content-Type", "image/jpeg");
+    const buffer = await page.screenshot();
+    res.setHeader("Content-Type", "image/png");
     res.send(buffer);
-
-  } catch (err) {
-    console.error(err);
+  } catch {
     res.sendStatus(500);
   } finally {
-    if (browser) await browser.close()
+    if (browser) await browser.close();
   }
 });