Update index.js

index.js

@@ -1,4 +1,4 @@
- const express = require('express');
+const express = require('express');
 const { chromium } = require('playwright');
 const QRCode = require('qrcode');
 const Jimp = require('jimp');
@@ -8,6 +8,10 @@ const QrCodeReader = require('qrcode-reader');
 const app = express();
 const port = 7860;
 
+// Global variable to hold the browser instance and context
+let browser;
+let browserContext;
+
 app.use(express.json());
 
 app.get('/', (req, res) => {
@@ -75,16 +79,25 @@ app.get('/qrcode', async (req, res) => {
 async function processScreenshot(req, res, { ua, url, type, width, height, language, fullpage }) {
 	try {
 		console.log('process screenshot');
-		const browser = await chromium.launch({
-			headless: true,
-			args: ["--no-sandbox", "--disable-setuid-sandbox"]
-		});
-		const context = await browser.newContext({
-			userAgent: ua || 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
-			viewport: { width: width || 1280, height: height || 800 },
-			deviceScaleFactor: 2
-		});
-		const page = await context.newPage();
+
+		// Launch browser and create a context if it doesn't exist
+		if (!browser) {
+			browser = await chromium.launch({
+				headless: true,
+				args: ["--no-sandbox", "--disable-setuid-sandbox"]
+			});
+		}
+
+		// Reuse or create a new context
+		if (!browserContext) {
+			browserContext = await browser.newContext({
+				userAgent: ua || 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
+				viewport: { width: width || 1280, height: height || 800 },
+				deviceScaleFactor: 2
+			});
+		}
+
+		const page = await browserContext.newPage();
 
 		if (language) {
 			await page.emulateMedia({ colorScheme: language });
@@ -94,12 +107,6 @@ async function processScreenshot(req, res, { ua, url, type, width, height, langu
 
 		// Handle JavaScript challenges
 		await page.waitForLoadState('networkidle');
-		
-		// Example to handle JavaScript challenge
-		// This is a placeholder, replace with actual challenge handling if needed
-		// await page.evaluate(() => {
-		// 	// Solve any inline JavaScript challenges
-		// });
 
 		let screenshotOptions = {
 			fullPage: fullpage == 1 ? true : false,
@@ -129,18 +136,18 @@ async function processScreenshot(req, res, { ua, url, type, width, height, langu
 			});
 		} else if (type === 'custom') {
 			if (width && height > 4096) {
+				await page.close();
 				return res.json({
 					"status": 400,
 					"message": "Width and height values should not exceed 4096 pixels."
 				});
-				await browser.close();
 			}
 			if (width && height < 400) {
+				await page.close();
 				return res.json({
 					"status": 400,
 					"message": "Width and height values should not be less than 400 pixels."
 				});
-				await browser.close();
 			}
 			await page.setViewportSize({
 				width: Number(width) || 375,
@@ -158,7 +165,18 @@ async function processScreenshot(req, res, { ua, url, type, width, height, langu
 		});
 		res.end(screenshot);
 		console.log('process done');
-		await browser.close();
+		await page.close(); // Close the tab instead of the browser
+
+		// Periodically close and reopen the browser to avoid memory bloat
+		if (browserContext.pages().length > 10) { // Arbitrary threshold
+			await browserContext.close();
+			browserContext = await browser.newContext();
+		}
+
+		// Trigger garbage collection
+		if (global.gc) {
+			global.gc();
+		}
 	} catch (e) {
 		console.error(e);
 		return res.json({