Update brat generator (using playwright)

index.js

@@ -6,14 +6,24 @@ import fs from 'fs'
 import morgan from 'morgan'
 import os from 'os'
 import path from 'path'
+import playwright from 'playwright'
 import PDFDocument from 'pdfkit'
 import sharp from 'sharp'
 import util from 'util'
 import yts from 'yt-search'
 
 const utils = {
+	getBrowser: (...opts) => playwright.chromium.launch({
+		args: [
+			'--incognito', '--single-process',
+			'--no-sandbox', '--no-zygote', '--no-cache'
+		],
+		executablePath: process.env.CHROME_BIN,
+		headless: true,
+		...opts
+	}),
 	// from https://github.com/petersolopov/carbonara
-	fetchCarbonaraAPI: async (code, output, opts = {}) => {
+	fetchCarbonaraAPI: async (code, opts = {}) => {
 		let resp = await utils.fetchPOST('https://carbonara.solopov.dev/api/cook', JSON.stringify({ code, ...opts }), {
 			headers: { 'Content-Type': 'application/json' }
 		})
@@ -22,12 +32,12 @@ const utils = {
 			if (/json$/.test(content)) {
 				let json = await resp.json()
 				throw json.error || 'An error occurred'
-			} else {
-				throw resp.statusText
 			}
+			throw resp.statusText
 		}
 		
 		let img = await resp.arrayBuffer()
+		const output = `${tmpDir}/${utils.randomName('.png')}`
 		await fs.promises.writeFile(output, Buffer.from(img))
 		return output
 	},
@@ -38,6 +48,23 @@ const utils = {
 	).json(),
 	fetchPOST: (url, body, opts = {}) => fetch(url, { method: 'POST', body, ...opts }),
 	formatSize: (n) => bytes(+n, { unitSeparator: ' ' }),
+	generateBrat: async (text) => {
+		const browser = await utils.getBrowser()
+		try {
+			const page = await browser.newPage()
+			await page.goto('https://www.bratgenerator.com/')
+			await page.click('#toggleButtonWhite')
+			await page.locator('#textInput').fill(text)
+			const output = `${tmpDir}/${utils.randomName('.jpg')}`
+			const ss = await page.locator('#textOverlay')
+				.screenshot({ path: output })
+			return output
+		} catch (e) {
+			throw e
+		} finally {
+			await browser.close()
+		}
+	},
 	getError: (e) => String(e).startsWith('[object ') ? 'Internal Server Error' : String(e),
 	isBase64: (str) => {
 		try {
@@ -121,25 +148,31 @@ app.all('/', (_, res) => {
 	status['memoryUsage'] = `${utils.formatSize(totalmem - freemem)} / ${utils.formatSize(totalmem)}`
 	
 	res.json({
-		creator: `@${process.env['SPACE_AUTHOR_NAME'] || 'rippanteq7'}`,
 		message: 'Hello World!',
 		uptime: new Date(process.uptime() * 1000).toUTCString().split(' ')[4],
 		status
 	})
 })
 
-app.all('/carbon', async (req, res) => {
+app.all(/^\/(brat|carbon)/, async (req, res) => {
 	if (!['GET', 'POST'].includes(req.method)) return res.status(405).json({ success: false, message: 'Method Not Allowed' })
 	
 	try {
 		const obj = req.allParams
-		if (!obj.code) return res.status(400).json({ success: false, message: 'Required parameter \'code\'' })
+		const isBrat = req.params[0] === 'brat'
+		if (isBrat && !obj.text) {
+			return res.status(400).json({ success: false, message: 'Required parameter \'text\'' })
+		} else if (!isBrat && !(obj.code || obj.text)) {
+			return res.status(400).json({ success: false, message: 'Required parameter \'code\'' })
+		}
 		
-		const filePath = `${tmpDir}/${utils.randomName('.png')}`
-		// nembak API carbonara njir
-		const image = await utils.fetchCarbonaraAPI(obj.code, filePath, obj)
+		const image = isBrat ?
+			await utils.generateBrat(obj.text) :
+			utils.fetchCarbonaraAPI(obj.code || obj.text, obj)
 		const resultUrl = `https://${req.hostname}/${image.replace(tmpDir, 'file')}`
-		utils.isTrue(obj.json) ? res.json({ success: true, result: resultUrl }) : res[utils.isTrue(obj.raw) ? 'send' : 'redirect'](resultUrl)
+		utils.isTrue(obj.json) ?
+			res.json({ success: true, result: resultUrl }) :
+		res[utils.isTrue(obj.raw) ? 'send' : 'redirect'](resultUrl)
 	} catch (e) {
 		console.log(e)
 		res.status(500).json({ error: true, message: utils.getError(e) })
@@ -161,7 +194,9 @@ app.all('/topdf', async (req, res) => {
 		await fs.promises.writeFile(`${tmpDir}/${fileName}`, bufferPDF)
 		
 		const resultUrl = `https://${req.hostname}/file/${fileName}`
-		utils.isTrue(json) ? res.json({ success: true, result: resultUrl }) : res[utils.isTrue(raw) ? 'send' : 'redirect'](resultUrl)
+		utils.isTrue(json) ?
+			res.json({ success: true, result: resultUrl }) :
+		res[utils.isTrue(raw) ? 'send' : 'redirect'](resultUrl)
 	} catch (e) {
 		console.log(e)
 		res.status(500).json({ error: true, message: utils.getError(e) })
@@ -183,24 +218,29 @@ app.all(/^\/webp2(gif|mp4|png)/, async (req, res) => {
 			await fs.promises.writeFile(`${tmpDir}/${fileName}`, fileBuffer)
 			
 			const resultUrl = `https://${req.hostname}/file/${fileName}`
-			utils.isTrue(json) ? res.json({ success: true, result: resultUrl }) : res[utils.isTrue(raw) ? 'send' : 'redirect'](resultUrl)
-		} else {
-			const fileName = utils.randomName('.webp')
-			const filePath = `${tmpDir}/${fileName}`
-			await fs.promises.writeFile(filePath, Buffer.from(file, 'base64'))
-			
-			const exec = util.promisify(cp.exec).bind(cp)
-			await exec(`convert ${filePath} ${filePath.replace('webp', 'gif')}`)
-			
-			let resultUrl
-			if (type === 'gif') resultUrl = `https://${req.hostname}/file/${fileName.replace('webp', 'gif')}`
-			else {
-				await exec(`ffmpeg -i ${filePath.replace('webp', 'gif')} -movflags faststart -pix_fmt yuv420p -vf "scale=trunc(iw/2)*2:trunc(ih/2)*2" ${filePath.replace('webp', 'mp4')}`)
-				resultUrl = `https://${req.hostname}/file/${fileName.replace('webp', 'mp4')}`
-			}
-			
-			utils.isTrue(json) ? res.json({ success: true, result: resultUrl }) : res[utils.isTrue(raw) ? 'send' : 'redirect'](resultUrl)
+			utils.isTrue(json) ?
+				res.json({ success: true, result: resultUrl }) :
+			res[utils.isTrue(raw) ? 'send' : 'redirect'](resultUrl)
+			return
 		}
+		
+		const fileName = utils.randomName('.webp')
+		const filePath = `${tmpDir}/${fileName}`
+		await fs.promises.writeFile(filePath, Buffer.from(file, 'base64'))
+		
+		const exec = util.promisify(cp.exec).bind(cp)
+		await exec(`convert ${filePath} ${filePath.replace('webp', 'gif')}`)
+		
+		let resultUrl
+		if (type === 'gif') resultUrl = `https://${req.hostname}/file/${fileName.replace('webp', 'gif')}`
+		else {
+			await exec(`ffmpeg -i ${filePath.replace('webp', 'gif')} -movflags faststart -pix_fmt yuv420p -vf "scale=trunc(iw/2)*2:trunc(ih/2)*2" ${filePath.replace('webp', 'mp4')}`)
+			resultUrl = `https://${req.hostname}/file/${fileName.replace('webp', 'mp4')}`
+		}
+		
+		utils.isTrue(json) ?
+			res.json({ success: true, result: resultUrl }) :
+		res[utils.isTrue(raw) ? 'send' : 'redirect'](resultUrl)
 	} catch (e) {
 		console.log(e)
 		res.status(500).json({ error: true, message: utils.getError(e) })
@@ -238,20 +278,20 @@ app.all(/^\/y(outube|t)(\/(d(ownload|l)|search)?)?/, async (req, res) => {
 			if (result.status === 'error') return res.status(400).json({ success: false, message: 'An error occurred' })
 			res.redirect(result.url)
 			return
-		} else {
-			if (!obj.query) return res.status(400).json({ success: false, message: 'Required parameter \'query\'' })
-			
-			let result = await yts(utils.ytIdRegex.test(obj.query) ? { videoId: utils.ytIdRegex.exec(obj.query)[1] } : obj.query)
-			result = result.videos ? result.videos[0] : result
-			if (!result?.url) return res.status(400).json({ success: false, message: 'Video unavailable' })
-			
-			const dlUrl = `https://${req.hostname}/yt/dl?url=${result.url}`
-			const download = { audio: `${dlUrl}&type=audio`, video: `${dlUrl}&type=video` }
-			res.json({
-				success: true,
-				result: { ...result, download }
-			})
 		}
+	
+		if (!obj.query) return res.status(400).json({ success: false, message: 'Required parameter \'query\'' })
+		
+		let result = await yts(utils.ytIdRegex.test(obj.query) ? { videoId: utils.ytIdRegex.exec(obj.query)[1] } : obj.query)
+		result = result.videos ? result.videos[0] : result
+		if (!result?.url) return res.status(400).json({ success: false, message: 'Video unavailable' })
+		
+		const dlUrl = `https://${req.hostname}/yt/dl?url=${result.url}`
+		const download = { audio: `${dlUrl}&type=audio`, video: `${dlUrl}&type=video` }
+		res.json({
+			success: true,
+			result: { ...result, download }
+		})
 	} catch (e) {
 		console.log(e)
 		res.status(500).json({ error: true, message: utils.getError(e) })

package.json

@@ -4,6 +4,7 @@
     "express": "*",
     "morgan": "*",
     "pdfkit": "*",
+    "playwright": "*",
     "serve-favicon": "*",
     "sharp": "*",
     "yt-search": "*"