w

index.mjs

@@ -107,6 +107,30 @@ async function screenshot(url, cookie, js_enabled) {
     return buffer;
 }
 
+async function evaluate(url, code) {
+    let { browser, context } = globalThis.state;
+    if (!browser) {
+        await locks.request('playwright_browser', async lock => {
+            browser = globalThis.state.browser = await chromium.launch();
+            context = globalThis.state.context = await browser.newContext({        
+                    javaScriptEnabled: true
+            }/*devices['iPhone 11']*/);
+        });
+    }
+
+    const page = await context.newPage();
+
+    page.on('console', async (msg) => {console.log(msg);});
+
+    //await context.route("**/*.{png,jpg,jpeg,css,js}", route => route.abort());
+    await page.goto(url);
+
+    let result = await page.evaluate(code);
+
+    await page.close();
+
+    return result;
+}
 
 const server = http.createServer(async function(req, res) {
     const {query, pathname} = url.parse(req.url, true);
@@ -132,6 +156,21 @@ const server = http.createServer(async function(req, res) {
             // res.writeHead(200,{'Content-type':'image/png'});
             // readStream.pipe(res);
             // res.end();
+        } else if (pathname == '/evaluate') {
+            const buffers = [];
+            for await (const chunk of req) {
+              buffers.push(chunk);
+            }
+            const data = Buffer.concat(buffers).toString();
+
+            const {url, code} = JSON.parse(data);
+
+            try {
+                let result = await evaluate(url, code);
+                res.end(''+result);
+            } catch (e) {
+                res.end(e.toString());
+            }
         } else {
             res.end('hello');
         }