Update index.js

index.js

@@ -1,220 +1,129 @@
-const express = require('express');
-const puppeteer = require('puppeteer-extra');
-const StealthPlugin = require('puppeteer-extra-plugin-stealth');
-const AnonymizeUA = require('puppeteer-extra-plugin-anonymize-ua');
+ const express = require('express');
+const { chromium } = require('playwright');
 const QRCode = require('qrcode');
 const Jimp = require('jimp');
 const QrCodeReader = require('qrcode-reader');
-
-puppeteer.use(StealthPlugin());
-puppeteer.use(AnonymizeUA());
+const fetch = require('node-fetch');
 
 const app = express();
 const port = 7860;
 
 app.use(express.json());
-app.get('/', async (req, res) => {
-	console.log('request to /')
-	res.send('halo kontol')
-})
+
+app.get('/', (req, res) => {
+	console.log('request to /');
+	res.send('halo kontol');
+});
+
 app.post('/screenshot', async (req, res) => {
-	console.log('request to /screenshot')
-	const {
-		ua,
-		url,
-		type,
-		width,
-		height,
-		language,
-		fullpage
-	} = req.body;
-	if (!url) return res.send('please input url')
-	await processScreenshot(req, res, {
-		ua,
-		url,
-		type,
-		width,
-		height,
-		language,
-		fullpage
-	});
+	console.log('request to /screenshot');
+	const { ua, url, type, width, height, language, fullpage } = req.body;
+	if (!url) return res.send('please input url');
+	await processScreenshot(req, res, { ua, url, type, width, height, language, fullpage });
 });
 
 app.get('/screenshot', async (req, res) => {
-	console.log('request to /screenshot (get)')
-	const {
-		ua,
-		url,
-		type,
-		width,
-		height,
-		language,
-		fullpage
-	} = req.query;
-	if (!url) return res.send('please input url')
-	await processScreenshot(req, res, {
-		ua,
-		url,
-		type,
-		width,
-		height,
-		language,
-		fullpage
-	});
+	console.log('request to /screenshot (get)');
+	const { ua, url, type, width, height, language, fullpage } = req.query;
+	if (!url) return res.send('please input url');
+	await processScreenshot(req, res, { ua, url, type, width, height, language, fullpage });
 });
+
 app.get('/qrcode', async (req, res) => {
 	try {
-		const {
-			mode,
-			url
-		} = req.query;
+		const { mode, url } = req.query;
 
 		if (!mode || !url) {
 			return res.status(400).send('what?');
 		}
 
 		if (mode === 'create') {
-			// Generate a QR code
 			QRCode.toDataURL(url, (err, src) => {
-				if (err) return res.send({
-					error: 'Error generating QR code'
-				});
+				if (err) return res.send({ error: 'Error generating QR code' });
 				res.type('png').send(Buffer.from(src.split(",")[1], "base64"));
 			});
 		} else if (['read', 'scan'].includes(mode)) {
-			// Fetch the image and read the QR code
 			const rts = await fetch(url);
-			if (!rts.ok || rts.status !== 200) return res.send({
-				status: 400,
-				message: 'Can\'t buffer url'
-			})
-			if (!/image/.test(rts.headers.get('content-type'))) return res.send({
-				status: 400,
-				message: 'Only image type'
-			})
+			if (!rts.ok || rts.status !== 200) return res.send({ status: 400, message: 'Can\'t buffer url' });
+			if (!/image/.test(rts.headers.get('content-type'))) return res.send({ status: 400, message: 'Only image type' });
 			const buffer = Buffer.from(await rts.arrayBuffer());
 			const image = await Jimp.read(buffer);
 
-			// Preprocess the image to enhance QR code readability
-			image
-				.grayscale() // Convert to grayscale
-				.contrast(1) // Increase contrast
-				.normalize(); // Normalize the image
+			image.grayscale().contrast(1).normalize();
 
 			const qrCodeReader = new QrCodeReader();
 
 			qrCodeReader.callback = (err, value) => {
-				if (err) return res.send({
-					error: 'Error reading QR code'
-				});
-				res.json({
-					result: value.result
-				});
+				if (err) return res.send({ error: 'Error reading QR code' });
+				res.json({ result: value.result });
 			};
 
 			const bitmap = image.bitmap;
-			qrCodeReader.decode({
-				width: bitmap.width,
-				height: bitmap.height,
-				data: bitmap.data
-			});
+			qrCodeReader.decode({ width: bitmap.width, height: bitmap.height, data: bitmap.data });
 		} else {
 			QRCode.toDataURL(url, (err, src) => {
-				if (err) return res.send({
-					error: 'Error generating QR code'
-				});
+				if (err) return res.send({ error: 'Error generating QR code' });
 				res.type('png').send(Buffer.from(src.split(",")[1], "base64"));
 			});
 		}
 	} catch (error) {
 		console.error(error);
-		res.send({
-			error: 'Internal Server Error'
-		});
+		res.send({ error: 'Internal Server Error' });
 	}
 });
 
-async function processScreenshot(req, res, {
-	ua,
-	url,
-	type,
-	width,
-	height,
-	language,
-	fullpage
-}) {
+async function processScreenshot(req, res, { ua, url, type, width, height, language, fullpage }) {
 	try {
-		console.log('process screenshot')
-		const browser = await puppeteer.launch({
-			headless: "new",
-			args: ["--aggressive-tab-discard",
-				"--disable-accelerated-2d-canvas",
-				"--disable-application-cache",
-				"--disable-cache",
-				"--disable-dev-shm-usage",
-				"--disable-gpu",
-				"--disable-offline-load-stale-cache",
-				"--disable-setuid-sandbox",
-				"--disable-setuid-sandbox",
-				"--disk-cache-size=0",
-				"--ignore-certificate-errors",
-				"--no-first-run",
-				"--no-sandbox",
-				"--no-zygote"
-			]
+		console.log('process screenshot');
+		const browser = await chromium.launch({
+			headless: true,
+			args: ["--no-sandbox", "--disable-setuid-sandbox"]
 		});
-		const page = await browser.newPage();
-
-		// Set language and user agent
-		await page.setExtraHTTPHeaders({
-			'Accept-Language': language || 'id-ID',
-			'User-Agent': ua || 'Mozilla/5.0 (Linux; Android 13; Pixel Build/RPB1.210523.001) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.120 Mobile Safari/537.36',
+		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();
+
+		if (language) {
+			await page.emulateMedia({ colorScheme: language });
+		}
+
+		await page.goto(url, { waitUntil: 'networkidle' });
+
+		// 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
+		// });
 
-		await page.goto(url, {
-			waitUntil: 'networkidle2',
-		});
-		let full = fullpage == 1 ? true : false
 		let screenshotOptions = {
-			fullPage: true
+			fullPage: fullpage == 1 ? true : false,
+			type: 'png',
+			omitBackground: false
 		};
+
 		if (type === 'desktop') {
-			screenshotOptions = {
-				fullPage: full,
-				width: 1280,
-				height: 800
-			};
-			await page.setViewport({
-				width: 1280,
-				height: 800,
+			await page.setViewportSize({
+				width: 1920,
+				height: 1080,
 			});
 		} else if (type === 'mobile') {
-			screenshotOptions = {
-				fullPage: full,
-				width: 375,
-				height: 667
-			};
-			await page.setViewport({
+			await page.setViewportSize({
 				width: 375,
 				height: 667,
 			});
 		} else if (type === 'tablet') {
-			screenshotOptions = {
-				fullPage: full,
-				width: 768,
-				height: 1024
-			};
-			await page.setViewport({
+			await page.setViewportSize({
 				width: 768,
 				height: 1024,
 			});
 		} else if (type === 'iphone') {
-			screenshotOptions = {
-				fullPage: full,
-				width: 375,
-				height: 667
-			};
-			await page.setViewport({
+			await page.setViewportSize({
 				width: 375,
 				height: 667,
 			});
@@ -233,19 +142,13 @@ async function processScreenshot(req, res, {
 				});
 				await browser.close();
 			}
-			screenshotOptions = {
-				fullPage: full,
-				width: width || 375,
-				height: height || 667
-			};
-			await page.setViewport({
+			await page.setViewportSize({
 				width: Number(width) || 375,
 				height: Number(height) || 667,
 			});
 		}
 
-		// Wait for a few seconds to mimic more natural behavior
-		await page.waitForTimeout(3000);
+		await page.waitForTimeout(5000);
 
 		const screenshot = await page.screenshot(screenshotOptions);
 
@@ -257,14 +160,14 @@ async function processScreenshot(req, res, {
 		console.log('process done');
 		await browser.close();
 	} catch (e) {
-		log(e)
+		console.error(e);
 		return res.json({
 			status: 400,
 			message: 'error when take a screenshot'
-		})
+		});
 	}
 }
 
 app.listen(port, () => {
 	console.log(`Server is running on port ${port}`);
-});
+});