使用字节的 TRAE 开发一个简单的 node 服务

const { chromium } = require("@playwright/test");
const nodemailer = require("nodemailer");
const cron = require("node-cron");
const winston = require("winston"); // 添加日志库

// 配置日志
const logger = winston.createLogger({
  format: winston.format.combine(winston.format.timestamp(), winston.format.json()),
  transports: [
    new winston.transports.File({ filename: "error.log", level: "error" }),
    new winston.transports.File({ filename: "combined.log" }),
    new winston.transports.Console(),
  ],
});

// 邮件配置
const transporter = nodemailer.createTransport({
  host: "smtp.qq.com",
  port: 465, // 修改为465端口
  secure: true, // 设置为true
  auth: {
    user: "3074994545@qq.com", // 补充完整的QQ邮箱地址
    pass: "xxxx", // QQ邮箱授权码
  },
});

// 要监控的网站列表
const websites = [
  {
    name: "老鱼简历",
    url: "https://laoyujianli.com",
    checkElements: [".laoyu-page-container", ".new-index-page", ".template-item"],
  },
  {
    name: "编程导航",
    url: "https://codefather.cn",
    checkElements: [".ant-layout", ".user-name"],
  },
  {
    name: "面试鸭",
    url: "https://mianshiya.cn", // 移除多余的空格
    checkElements: ["#indexPage", ".ant-list-item", "#ccc"],
    retryCount: 3, // 添加重试次数
    retryDelay: 5000, // 重试延迟时间（毫秒）
  },
];

// 浏览器实例管理
let browserInstance = null;
async function getBrowser() {
  if (!browserInstance) {
    browserInstance = await chromium.launch({
      timeout: 60000,
    });
  }
  return browserInstance;
}

// 检查网站可用性
async function checkWebsite(site) {
  const startTime = Date.now();
  const browser = await getBrowser();
  const context = await browser.newContext({
    userAgent:
      "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
  });

  try {
    const page = await context.newPage();
    logger.info(`正在检查网站: ${site.name}`);

    let lastError;
    // 添加重试机制
    for (let i = 0; i < (site.retryCount || 1); i++) {
      try {
        // 设置页面加载超时时间
        await page.goto(site.url, {
          timeout: 30000,
          waitUntil: "networkidle", // 等待网络空闲
        });

        // 检查关键元素是否存在
        for (const selector of site.checkElements) {
          try {
            await page.waitForSelector(selector, { timeout: 5000 });
          } catch (elementError) {
            console.log(`元素 ${selector} 未找到，继续检查其他元素`);
          }
        }

        console.log(`${site.name} 检查通过`);
        return { success: true };
      } catch (error) {
        lastError = error;
        if (i < (site.retryCount || 1) - 1) {
          console.log(`${site.name} 检查失败，等待 ${site.retryDelay || 5000}ms 后重试...`);
          await new Promise((resolve) => setTimeout(resolve, site.retryDelay || 5000));
        }
      }
    }

    console.error(`${site.name} 检查失败:`, lastError.message);
    return {
      success: false,
      error: lastError.message,
    };
  } finally {
    await context.close(); // 只关闭 context，不关闭浏览器
  }
}

// 添加进程退出时的清理
process.on("SIGTERM", async () => {
  logger.info("服务正在关闭...");
  if (browserInstance) {
    await browserInstance.close();
    browserInstance = null;
  }
  process.exit(0);
});

process.on("SIGINT", async () => {
  logger.info("服务正在关闭...");
  if (browserInstance) {
    await browserInstance.close();
    browserInstance = null;
  }
  process.exit(0);
});

// 发送告警邮件
async function sendAlertEmail(site, error) {
  const mailOptions = {
    from: "3074994545@qq.com", // 使用完整的发件人邮箱
    to: "3074994545@qq.com", // 使用完整的收件人邮箱
    subject: `🚨 网站异常告警: ${site.name}`,
    html: `
      <h2>网站监控告警</h2>
      <p><strong>网站名称:</strong> ${site.name}</p>
      <p><strong>网站地址:</strong> ${site.url}</p>
      <p><strong>错误信息:</strong> ${error}</p>
      <p><strong>发生时间:</strong> ${new Date().toLocaleString()}</p>
    `,
  };

  try {
    await transporter.sendMail(mailOptions);
    console.log(`已发送告警邮件 - ${site.name}`);
  } catch (error) {
    console.error("发送邮件失败:", error);
    // 如果是认证错误，输出更详细的信息
    if (error.code === "EAUTH") {
      console.error("邮箱认证失败，请检查以下内容：");
      console.error("1. 确保QQ邮箱已开启SMTP服务");
      console.error("2. 确保使用的是正确的授权码而不是邮箱密码");
      console.error("3. 访问 https://service.mail.qq.com/cgi-bin/help?subtype=1&&id=28&&no=1001256 获取帮助");
    }
  }
}

// 执行监控任务
async function runMonitoring() {
  // 添加并发限制
  const concurrentLimit = 2;
  const chunks = [];
  for (let i = 0; i < websites.length; i += concurrentLimit) {
    chunks.push(websites.slice(i, i + concurrentLimit));
  }

  for (const chunk of chunks) {
    await Promise.all(
      chunk.map(async (site) => {
        try {
          const result = await checkWebsite(site);
          if (!result.success) {
            await sendAlertEmail(site, result.error);
          }
        } catch (error) {
          logger.error(`监控任务执行失败 (${site.name}):`, {
            error: error.message,
            stack: error.stack,
          });
        }
      })
    );
  }
}

// 添加优雅退出处理
process.on("SIGTERM", async () => {
  logger.info("服务正在关闭...");
  if (browserInstance) {
    await browserInstance.close();
  }
  process.exit(0);
});

// 设置定时任务，每5分钟执行一次监控
cron.schedule("*/5 * * * *", () => {
  console.log("开始执行网站监控...");
  runMonitoring();
});

// 立即执行一次监控
console.log("启动网站监控服务...");
runMonitoring();