Update index.js

index.js

@@ -2,24 +2,23 @@ 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 QRCode = require('qrcode');
+const Jimp = require('jimp');
+const QrCodeReader = require('qrcode-reader');
 
 puppeteer.use(StealthPlugin());
 puppeteer.use(AnonymizeUA());
 
 const app = express();
-const port = 7860;
+const port = 5000;
 
 app.use(express.json());
 app.get('/', async (req, res) => {
 	console.log('request to /')
-	console.log('IP => ' + req.headers['x-real-ip'])
-	console.log('User Agent => ' + req.headers['user-agent'])
 	res.send('halo kontol')
 })
 app.post('/screenshot', async (req, res) => {
 	console.log('request to /screenshot')
-	console.log('IP => ' + req.headers['x-real-ip'])
-	console.log('User Agent => ' + req.headers['user-agent'])
 	const {
 		ua,
 		url,
@@ -43,8 +42,6 @@ app.post('/screenshot', async (req, res) => {
 
 app.get('/screenshot', async (req, res) => {
 	console.log('request to /screenshot (get)')
-	console.log('IP => ' + req.headers['x-real-ip'])
-	console.log('User Agent => ' + req.headers['user-agent'])
 	const {
 		ua,
 		url,
@@ -65,6 +62,69 @@ app.get('/screenshot', async (req, res) => {
 		fullpage
 	});
 });
+app.get('/qrcode', async (req, res) => {
+	try {
+		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'
+				});
+				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 response = await fetch(url);
+			const buffer = await response.buffer();
+			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
+
+			const qrCodeReader = new QrCodeReader();
+
+			qrCodeReader.callback = (err, value) => {
+				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
+			});
+		} else {
+			QRCode.toDataURL(url, (err, src) => {
+				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'
+		});
+	}
+});
 
 async function processScreenshot(req, res, {
 	ua,
@@ -75,7 +135,7 @@ async function processScreenshot(req, res, {
 	language,
 	fullpage
 }) {
-	
+	try {
 		console.log('process screenshot')
 		const browser = await puppeteer.launch({
 			headless: "new",
@@ -102,7 +162,7 @@ async function processScreenshot(req, res, {
 			'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',
 		});
-		try {
+
 		await page.goto(url, {
 			waitUntil: 'networkidle2',
 		});
@@ -152,18 +212,18 @@ async function processScreenshot(req, res, {
 			});
 		} else if (type === 'custom') {
 			if (width && height > 4096) {
-				await browser.close();
 				return res.json({
 					"status": 400,
 					"message": "Width and height values should not exceed 4096 pixels."
 				});
-			}
-			if (width && height < 90) {
 				await browser.close();
+			}
+			if (width && height < 400) {
 				return res.json({
 					"status": 400,
-					"message": "Width and height values should not be less than 90 pixels."
+					"message": "Width and height values should not be less than 400 pixels."
 				});
+				await browser.close();
 			}
 			screenshotOptions = {
 				fullPage: full,
@@ -186,11 +246,10 @@ async function processScreenshot(req, res, {
 			'Content-Length': screenshot.length,
 		});
 		res.end(screenshot);
-
+        console.log('process done')
 		await browser.close();
 	} catch (e) {
-		console.log(e)
-		await browser.close();
+		log(e)
 		return res.json({
 			status: 400,
 			message: 'error when take a screenshot'
@@ -200,4 +259,4 @@ async function processScreenshot(req, res, {
 
 app.listen(port, () => {
 	console.log(`Server is running on port ${port}`);
-});
+});