使用 Puppeteer 和 Clash 代理服务器创建一个截图 API ​

/**
 * Required dependencies for the application.
 */
const express = require("express");

/**
 * Create an instance of the Express application.
 */
const app = express();

/**
 * The port number that the application will listen on.
 */
const port = 7860;

/**
 * Define a route for the root URL that sends a "Hello World!" message.
 */
app.get("/", (req, res) => {
  res.send("Hello World!");
});

/**
 * Start the application and listen for incoming requests on the specified port.
 */
app.listen(port, () => {
  console.log(`Example app listening on port ${port}`);
});

/**
 * This route handles GET requests to the '/api' endpoint, which accepts a 'url' query parameter
 * and returns a screenshot of the corresponding webpage in PNG format. The function performs
 * the following steps:
 * 1. Validates the 'url' parameter and checks the request frequency to avoid overloading the server.
 * 2. Launches a headless Chrome browser instance with a proxy server set to 'http://127.0.0.1:7891'.
 * 3. Navigates to the specified URL and takes a screenshot of the page.
 * 4. Sends the PNG image file as a response to the client.
 * 5. Closes the browser instance.
 * @param {Object} req - The HTTP request object.
 * @param {Object} res - The HTTP response object.
 * @returns {Promise<void>} - A Promise that resolves when the response has been sent.
 */
const puppeteer = require("puppeteer");

app.get("/api", async (req, res) => {
  // ...省略校验逻辑

  const browser = await puppeteer.launch({
    headless: true,
    args: [
      "--proxy-server=http://127.0.0.1:7891", // 设置代理
    ],
  });

  const page = await browser.newPage();

  await page.goto(url);
  await page.screenshot({ path: "screenshot.png" });

  res.type("image/png");
  res.sendFile("/screenshot.png");

  await browser.close();
});

const express = require("express");
const puppeteer = require("puppeteer");
const fs = require("fs"); // 引入fs模块
const { exec } = require("child_process");
const app = express();
const port = process.env.PORT || 7860;
const space_host = process.env.SPACE_HOST || `http://localhost:${port}`;

// 创建一个 user-agent 数组
function readUserAgentsFromFile(filename) {
  try {
    // 同步读取文件，将文件内容分割为数组（每行一个元素）
    const data = fs.readFileSync(filename, "utf8");
    return data.split("\n");
  } catch (err) {
    console.error(`Error reading file from disk: ${err}`);
  }
  // 如果读取失败，返回一个空数组
  return [];
}

// 使用上面的函数从文件中读取 user-agents
const userAgents = readUserAgentsFromFile("user-agents.txt");

// 创建一个记录访问 URL 的对象
let visitRecord = {};

app.get("/", (req, res) => {
  const usage = `
    <h1>API usage</h1>
    <p><b>GET ${space_host}/api?url={URL}</b></p>
    <ul>
      <li>{URL} - The URL of the webpage to screenshot.</li>
    </ul>`;
  res.set("Content-Type", "text/html");
  res.send(usage);
});

app.get("/status", (req, res) => {
  exec("ps -ef && ls -a -l", (err, stdout, stderr) => {
    if (err) {
      console.log(err);
      res.status(500).send(`<pre>${stderr}<pre>`);
    }
    res.send(`<pre>${stdout}<pre>`);
  });
});

app.get("/health", (req, res) => {
  console.log("health check");
  res.status(200).json({ message: "ok" });
});

app.get("/api", async (req, res) => {
  const urlParam = req.query.url;

  if (!urlParam) {
    return res.status(400).json({ error: "Missing URL parameter" });
  }

  // 检查是否超过了频率限制 (5分钟内只能访问一次)
  if (
    visitRecord[urlParam] &&
    Date.now() - visitRecord[urlParam] <= 5 * 60 * 1000
  ) {
    console.log("visitRecord: " + visitRecord[urlParam]);
    return res.status(429).json({ error: "Too Many Requests" });
  }

  try {
    const browser = await puppeteer.launch({
      headless: true,
      args: [
        "--no-sandbox",
        "--disable-setuid-sandbox",
        "--proxy-server=socks5://127.0.0.1:7891",
      ],
    });
    const page = await browser.newPage();

    // 设置视窗大小（可选）
    await page.setViewport({
      width: 1366,
      height: 762,
    });

    // 选择一个随机的 user-agent
    const userAgent = userAgents[Math.floor(Math.random() * userAgents.length)];

    // 设置 user-agent
    await page.setUserAgent(userAgent);

    // 模拟浏览器访问目标Url
    await page.goto(urlParam);

    visitRecord[urlParam] = Date.now(); // 更新访问记录

    // 等待加载完成（可根据情况调整等待时间）
    await new Promise((resolve) => setTimeout(resolve, 4000));

    const screenshotBuffer = await page.screenshot({
      fullpage: true,
    });

    res.set("Content-Type", "image/png");
    res.send(screenshotBuffer);

    await browser.close(); // 在这里添加浏览器关闭的代码段
  } catch (err) {
    console.log(`Screenshot failed : ${err}`);
    return res.sendStatus(500);
  }
});

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