Update app.js

app.js

@@ -1,111 +1,86 @@
 import express from 'express';
-import { chromium } from 'playwright';
-import cors from 'cors';
-import dotenv from 'dotenv';
-import os from 'os';
+import { chromium } from 'playwright-extra';
+import stealth from 'playwright-extra/plugins/stealth';
+import cluster from 'cluster';
+import { cpus } from 'os';
 import sharp from 'sharp';
 
-dotenv.config();
+chromium.use(stealth());
 
-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();
-            });
+const BROWSERS_COUNT = cpus().length;
+const PORT = process.env.PORT || 7860;
 
-            page = await context.newPage();
-            await page.goto('https://www.bratgenerator.com/', { waitUntil: 'domcontentloaded', timeout: 10000 });
+class FastBratGenerator {
+    constructor() {
+        this.browsers = [];
+        this.pages = [];
+    }
 
-            try {
-                await page.click('#onetrust-accept-btn-handler', { timeout: 2000 });
-            } catch { }
+    async init() {
+        const promises = Array(BROWSERS_COUNT).fill().map(async () => {
+            const browser = await chromium.launch({ headless: true });
+            const context = await browser.newContext({ viewport: { width: 1920, height: 1080 } });
+            const page = await context.newPage();
+            
+            await page.goto('https://www.bratgenerator.com/', { waitUntil: 'domcontentloaded' });
+            
+            this.browsers.push(browser);
+            this.pages.push(page);
+        });
 
-            await page.evaluate(() => setupTheme('white'));
-        }
-    },
+        await Promise.all(promises);
+    }
 
-    async generateBrat(text) {
+    async generate(text) {
+        const page = this.pages[Math.floor(Math.random() * this.pages.length)];
         await page.fill('#textInput', text);
-        const overlay = page.locator('#textOverlay');
-        // Ambil screenshot dalam format PNG
-        const pngBuffer = await overlay.screenshot({ 
-            timeout: 3000,
-            type: 'png'
+        
+        const screenshot = await page.locator('#textOverlay').screenshot({ 
+            type: 'png', 
+            timeout: 2000 
         });
         
-        // Konversi ke WebP menggunakan sharp
-        return sharp(pngBuffer)
-            .webp({ quality: 80 })
+        return sharp(screenshot)
+            .webp({ quality: 85, speed: 6 })
             .toBuffer();
-    },
+    }
 
     async close() {
-        if (browser) await browser.close();
+        await Promise.all(this.browsers.map(b => b.close()));
     }
-};
+}
+
+const bratGenerator = new FastBratGenerator();
 
-const app = express();
-app.use(express.json());
-app.use(cors());
+async function startServer() {
+    const app = express();
+    app.use(express.json());
 
-app.get('*', async (req, res) => {
-    try {
-        const { q } = req.query;
-        if (!q) {
-            return res.json({
-                name: 'HD Bart Generator API',
-                message: 'Parameter q di perlukan',
-                version: '2.1.0',
-                runtime: {
-                    os: os.type(),
-                    platform: os.platform(),
-                    architecture: os.arch(),
-                    cpuCount: os.cpus().length,
-                    uptime: `${os.uptime()} seconds`,
-                    memoryUsage: `${Math.round((os.totalmem() - os.freemem()) / 1024 / 1024)} MB used of ${Math.round(os.totalmem() / 1024 / 1024)} MB`
-                }
-            });
+    app.get('/', async (req, res) => {
+        try {
+            const { q } = req.query;
+            if (!q) return res.status(400).send('Text required');
+            
+            const imageBuffer = await bratGenerator.generate(q);
+            res.contentType('image/webp');
+            res.send(imageBuffer);
+        } catch (error) {
+            res.status(500).send('Generation failed');
         }
-        const imageBuffer = await utils.generateBrat(q);
-        res.set('Content-Type', 'image/webp');
-        res.send(imageBuffer);
-    } catch (error) {
-        console.error(error);
-        res.status(500).json({
-            status: false,
-            message: 'Error generating image',
-            error: process.env.NODE_ENV === 'development' ? error.message : undefined
-        });
-    }
-});
+    });
 
-const PORT = process.env.PORT || 7860;
+    await bratGenerator.init();
+    app.listen(PORT, () => console.log(`Running on ${PORT}`));
+}
 
-app.listen(PORT, async () => {
-    console.log(`Server running on port ${PORT}`);
-    await utils.initialize();
-});
+if (cluster.isPrimary) {
+    cpus().forEach(() => cluster.fork());
+    cluster.on('exit', (worker) => cluster.fork());
+} else {
+    startServer();
+}
 
 process.on('SIGINT', async () => {
-    await utils.close();
+    await bratGenerator.close();
     process.exit(0);
 });