Update main.js

main.js

@@ -0,0 +1,42 @@
+const express = require('express');
+const puppeteer = require('puppeteer');
+const app = express();
+
+// 定义一个异步函数来处理截图并返回给客户端
+async function takeScreenshotAndReturn(req, res) {
+    try {
+        // 启动Puppeteer浏览器实例
+        const browser = await puppeteer.launch({
+            headless: true,
+            args: ["--no-sandbox"]
+        });
+        const page = await browser.newPage();
+
+        // 从请求参数中获取要截图的网址，默认为百度
+        const url = req.query.url || 'https://www.baidu.com';
+        await page.goto(url);
+
+        // 进行截图
+        const screenshot = await page.screenshot({ fullPage: true });
+
+        // 关闭浏览器实例
+        await browser.close();
+
+        // 设置响应头，告诉客户端返回的是图片数据
+        res.set('Content-Type', 'image/png');
+        // 将截图数据作为响应返回给客户端
+        res.send(screenshot);
+    } catch (error) {
+        console.error('截图操作出现错误：', error);
+        res.status(500).send('截图失败：' + error.message);
+    }
+}
+
+// 定义一个路由，当客户端访问 /screenshot 路径时，调用 takeScreenshotAndReturn 函数
+app.get('/screenshot', takeScreenshotAndReturn);
+
+// 启动服务器，监听3000端口
+const port = 3000;
+app.listen(port, () => {
+    console.log(`服务器已启动，监听端口 ${port}`);
+});