Hugging Face's logo

Spaces:
aseli
/
eses
Running

App Files Community
eses / index.js
aseli's picture
aseli
Update index.js
5d933f1 verified
raw history blame contribute delete
No virus
6.6 kB
const express = require('express');
const puppeteer = require('puppeteer-extra');
const StealthPlugin = require('puppeteer-extra-plugin-stealth');
const AnonymizeUA = require('puppeteer-extra-plugin-anonymize-ua');
var jimp = require("jimp");
var qrCode = require('qrcode-reader');
puppeteer.use(StealthPlugin());
puppeteer.use(AnonymizeUA());
const app = express();
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({message: '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,
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)')
console.log('IP => ' + req.headers['x-real-ip'])
console.log('User Agent => ' + req.headers['user-agent'])
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
});
});
async function processScreenshot(req, res, {
ua,
url,
type,
width,
height,
language,
fullpage
}) {
console.log('process screenshot')
const browser = await puppeteer.launch({
headless: "new",
executablePath:"/usr/bin/chromium",
protocolTimeout: 0,
args: [
"--fast-start",
"--disable-extensions",
"--no-sandbox",
"--disable-setuid-sandbox",
"--no-gpu",
"--disable-background-timer-throttling",
"--disable-renderer-backgrounding",
"--override-plugin-power-saver-for-testing=never",
"--disable-extensions-http-throttling",
"--user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.3"
]
});
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',
});
try {
await page.goto(url);
let full = fullpage == 1 ? true : false
let screenshotOptions = {
fullPage: true
};
if (type === 'desktop') {
screenshotOptions = {
fullPage: full,
width: 1280,
height: 800
};
await page.setViewport({
width: 1280,
height: 800,deviceScaleFactor: 2
});
} else if (type === 'mobile') {
screenshotOptions = {
fullPage: full,
width: 375,
height: 667
};
await page.setViewport({
width: 375,
height: 667,deviceScaleFactor: 2
});
} else if (type === 'tablet') {
screenshotOptions = {
fullPage: full,
width: 768,
height: 1024
};
await page.setViewport({
width: 768,
height: 1024,deviceScaleFactor: 2
});
} else if (type === 'iphone') {
screenshotOptions = {
fullPage: full,
width: 375,
height: 667
};
await page.setViewport({
width: 375,
height: 667,deviceScaleFactor: 2
});
} 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();
return res.json({
"status": 400,
"message": "Width and height values should not be less than 90 pixels."
});
}
screenshotOptions = {
fullPage: full,
width: width || 375,
height: height || 667
};
await page.setViewport({
width: Number(width) || 375,
height: Number(height) || 667,deviceScaleFactor: 2
});
}
// Wait for a few seconds to mimic more natural behavior
await page.waitForTimeout(3000);
const screenshot = await page.screenshot(screenshotOptions);
res.writeHead(200, {
'Content-Type': 'image/png',
'Content-Length': screenshot.length,
});
res.end(screenshot);
console.log('screenshot successfully')
await browser.close();
} catch (e) {
console.log(e)
await browser.close();
return res.json({
status: 400,
message: 'error when take a screenshot'
})
}
}
const PORT = process.env.PORT || 7860
function restartSpace() {
const { SPACE_ID, HF_TOKEN } = process.env
return fetch(`https://huggingface.co/api/spaces/${SPACE_ID}/restart`, {
method: 'POST',
headers: { Authorization: `Bearer ${HF_TOKEN}` }
})
}
function keepAlive() {
var url = `https://${process.env.SPACE_HOST}`
if (/(\/\/|\.)undefined\./.test(url)) return
setInterval(() => {
fetch(url, { headers: { Authorization: "Bearer " + process.env.HF_TOKEN}})
.then(response => {
if (!response.ok) {
restartSpace().catch(console.error)
}
})
.catch(console.error)
}, 5 * 1000 * 60)
}
app.get("/qrcode", async (req, res) => {
if (!req.query.url) return res.send({status: 400})
if (!req.query.mode) return res.send({status: 400})
const readQRCode = async (buffer) => {
try {
const img = await jimp.read(buffer);
const qr = new qrCode();
const value = await new Promise((resolve, reject) => {
qr.callback = (err, v) => err != null ? reject(err) : resolve(v);
qr.decode(img.bitmap);
});
return value.result;
} catch (error) {
return error.message
}
}
if (req.query.mode == "scan") {
let rts = await fetch(req.query.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'
})
return res.send({ result: await readQRCode(await rts.arrayBuffer()) || null })
} else {
var { toDataURL } = require("qrcode");
let me = await toDataURL(req.query.url);
res.type('png').send(Buffer.from(me.split("base64,")[1], "base64"));
}
})
app.listen(PORT, async () => {
console.log('App running on port', PORT)
console.log(fetch)
console.log(process.env['HF_TOKEN'])
keepAlive()
})