Update main.js

main.js

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