Update index.js

index.js

@@ -6,12 +6,60 @@ const bodyParser = require('body-parser');
 const path = require('path');
 const loadImg = require('./loader');
 const app = express();
-const puppeteer = require('puppeteer');
+const { chromium } = require('playwright');
 const port = 7860;
 app.set('trust proxy', 1);
 
 let totalReq = 0; 
 
+//Brat
+const config = {
+    maxTextLength: 100,
+    viewport: { width: 1920, height: 1080 },
+    userAgent: 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36'
+};
+
+let browser, page;
+
+const utils = {
+    async initialize() {
+        if (!browser) {
+            browser = await chromium.launch({ headless: true });
+            const context = await browser.newContext({
+                viewport: config.viewport,
+                userAgent: config.userAgent
+            });
+
+            await context.route('**/*', (route) => {
+                const url = route.request().url();
+                if (url.endsWith('.png') || url.endsWith('.jpg') || url.includes('google-analytics')) {
+                    return route.abort();
+                }
+                route.continue();
+            });
+
+            page = await context.newPage();
+            await page.goto('https://www.bratgenerator.com/', { waitUntil: 'domcontentloaded', timeout: 10000 });
+
+            try {
+                await page.click('#onetrust-accept-btn-handler', { timeout: 2000 });
+            } catch { }
+
+            await page.evaluate(() => setupTheme('white'));
+        }
+    },
+
+    async generateBrat(text) {
+        await page.fill('#textInput', text);
+        const overlay = page.locator('#textOverlay');
+        return overlay.screenshot({ timeout: 3000 });
+    },
+
+  async close() {
+        if (browser) await browser.close();
+    }
+};
+
 //Security
 app.use(helmet());
 const limiter = rateLimit({
@@ -322,41 +370,21 @@ app.post('/generate2', async (req, res) => {
   }
 });
 
-app.post("/brat", async (req, res) => {
-  const { text } = req.body;
-
-  if (!text) {
-    return res.status(400).json({ error: "Teks tidak boleh kosong." });
-  }
-
-  let browser;
-
-  try {
-    // Buka browser baru
-    browser = await puppeteer.launch({
-      headless: true,
-      args: ["--no-sandbox", "--disable-setuid-sandbox"],
-    });
-
-    const page = await browser.newPage();
-    await page.goto("https://www.bratgenerator.com/");
-    await page.click("#toggleButtonWhite"); // Aktifkan tema putih
-    await page.type("#textInput", text); // Isi teks
-
-    // Ambil screenshot elemen hasil
-    const screenshotBuffer = await page.$eval("#textOverlay", async (element) => {
-      return await element.screenshot({ encoding: "binary" });
-    });
-
-    // Kirim hasil sebagai image
-    res.set("Content-Type", "image/jpeg");
-    res.send(screenshotBuffer);
-  } catch (error) {
-    console.error("Error generating Brat:", error);
-    res.status(500).json({ error: "Terjadi kesalahan saat membuat gambar." });
-  } finally {
-    if (browser) await browser.close();
-  }
+app.get('/brat', async (req, res) => {
+    try {
+        const { text } = req.query;
+        if (!text) {
+            return res.json('Input text brat!');
+        }
+        const imageBuffer = await utils.generateBrat(text);
+        res.set('Content-Type', 'image/png');
+        res.send(imageBuffer);
+    } catch (error) {
+        console.error(error);
+        res.status(500).json({
+          message: 'error generating brat'
+        });
+    }
 });
 
 // Mulai server Express