const express = require('express');
const { chromium } = require('playwright');

const app = express();
const PORT = 7860;

app.get('/scrape', async (req, res) => {
    const url = req.query.url;
    if (!url) return res.status(400).json({ error: 'URL is required' });

    try {
        const browser = await chromium.launch({ headless: true });
        const context = await browser.newContext({
            userAgent: 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
            viewport: { width: 1280, height: 720 },
            javaScriptEnabled: true, 
            bypassCSP: true
        });

        const page = await context.newPage();
        await page.goto(url, { waitUntil: 'load', timeout: 60000 });
        await page.waitForSelector('body');
        
        const html = await page.content();
        await browser.close();

        res.status(200).send(html);
    } catch (error) {
        res.status(500).json({ error: 'Failed to scrape the website', details: error.message });
    }
});

app.get('/bypass-cf-media', async (req, res) => {
    const url = req.query.url;
    if (!url) return res.status(400).json({ error: 'URL is required' });

    try {
        const browser = await chromium.launch({ headless: true });
        const context = await browser.newContext({
            userAgent: 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
            viewport: { width: 1280, height: 720 },
            bypassCSP: true
        });

        const page = await context.newPage();
        await page.goto(url, { waitUntil: 'load', timeout: 60000 });
        await page.waitForSelector('body');
        
        const mediaElement = await page.$('img, video, audio, source');
        if (!mediaElement) {
            await browser.close();
            return res.status(404).json({ error: 'No media found' });
        }
        
        const buffer = await mediaElement.screenshot() || await mediaElement.evaluate(el => el.src);
        await browser.close();
        
        if (typeof buffer === 'string') {
            return res.redirect(buffer);
        }
        
        res.setHeader('Content-Type', 'image/png');
        res.send(buffer);
    } catch (error) {
        res.status(500).json({ error: 'Failed to bypass CF or get media', details: error.message });
    }
});

app.listen(PORT, () => {
    console.log(`Server running on http://localhost:${PORT}`);
});