Update main.js

main.js

@@ -1,30 +1,45 @@
 const http = require('http');
 const puppeteer = require('puppeteer');
+const urlModule = require('url');
 
 const server = http.createServer(async (req, res) => {
-  console.log('hello'+req.url)
-    if (req.url === '/screenshot' && req.method === 'GET') {
+    const parsedUrl = urlModule.parse(req.url, true); // 第二个参数为true表示解析查询参数
+    const pathname = parsedUrl.pathname;
+    console.log(pathname)
+    const query = parsedUrl.query;
+    console.log(query)
+    if (pathname === '/screenshot' && req.method.toUpperCase() === 'GET') {
+        const targetUrl = query.url;
+      console.log(targetUrl)
+        if (!targetUrl) {
+            res.statusCode = 400;
+            res.end('Missing url parameter');
+            return;
+        }
+
         try {
-            const url = req.query.url;
-            console.log(url)
-            if (!url) {
-                res.statusCode = 400;
-                res.end('Missing url parameter');
-                return;
-            }
+            // 启动浏览器
             const browser = await puppeteer.launch();
-            const page = await browser.newPage();
-            await page.goto(url);
-            const screenshotBuffer = await page.screenshot();
-            await browser.close();
+            try {
+                const page = await browser.newPage();
+                // 这里确保传入完整协议的网址，比如 http:// 或 https:// 开头的
+                await page.goto(targetUrl);
+                const screenshotBuffer = await page.screenshot();
+                await browser.close();
 
-            res.statusCode = 200;
-            res.setHeader('Content-Type', 'image/png');
-            res.end(screenshotBuffer);
-        } catch (error) {
-            console.error('Error taking screenshot:', error);
+                res.statusCode = 200;
+                res.setHeader('Content-Type', 'image/png');
+                res.end(screenshotBuffer);
+            } catch (pageError) {
+                console.error('Error accessing page:', pageError);
+                await browser.close();
+                res.statusCode = 500;
+                res.end('Error accessing page');
+            }
+        } catch (launchError) {
+            console.error('Error launching browser:', launchError);
             res.statusCode = 500;
-            res.end('Error taking screenshot');
+            res.end('Error launching browser');
         }
     } else {
         res.statusCode = 404;
@@ -32,7 +47,7 @@ const server = http.createServer(async (req, res) => {
     }
 });
 
-const port = process.env.PORT || 7860;
+const port = process.env.PORT || 3000;
 server.listen(port, () => {
     console.log(`Server running on port ${port}`);
 });