Building a Simple Node.js Service with ByteDance's TRAE

const { chromium } = require("@playwright/test");
const nodemailer = require("nodemailer");
const cron = require("node-cron");
const winston = require("winston"); // Add logging library

// Configure logging
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(),
  ],
});

// Email configuration
const transporter = nodemailer.createTransport({
  host: "smtp.qq.com",
  port: 465, // Changed to port 465
  secure: true, // Set to true
  auth: {
    user: "3074994545@qq.com", // Complete QQ email address
    pass: "xxxx", // QQ email authorization code
  },
});

// List of websites to monitor
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", // Remove extra space
    checkElements: ["#indexPage", ".ant-list-item", "#ccc"],
    retryCount: 3, // Add retry count
    retryDelay: 5000, // Retry delay (milliseconds)
  },
];

// Browser instance management
let browserInstance = null;
async function getBrowser() {
  if (!browserInstance) {
    browserInstance = await chromium.launch({
      timeout: 60000,
    });
  }
  return browserInstance;
}

// Check website availability
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(`Checking website: ${site.name}`);

    let lastError;
    // Add retry mechanism
    for (let i = 0; i < (site.retryCount || 1); i++) {
      try {
        // Set page load timeout
        await page.goto(site.url, {
          timeout: 30000,
          waitUntil: "networkidle", // Wait until network is idle
        });

        // Check if key elements exist
        for (const selector of site.checkElements) {
          try {
            await page.waitForSelector(selector, { timeout: 5000 });
          } catch (elementError) {
            console.log(`Element ${selector} not found, continuing to check other elements`);
          }
        }

        console.log(`${site.name} check passed`);
        return { success: true };
      } catch (error) {
        lastError = error;
        if (i < (site.retryCount || 1) - 1) {
          console.log(`${site.name} check failed, retrying after ${site.retryDelay || 5000}ms...`);
          await new Promise((resolve) => setTimeout(resolve, site.retryDelay || 5000));
        }
      }
    }

    console.error(`${site.name} check failed:`, lastError.message);
    return {
      success: false,
      error: lastError.message,
    };
  } finally {
    await context.close(); // Only close context, not the browser
  }
}

// Add cleanup on process exit
process.on("SIGTERM", async () => {
  logger.info("Service shutting down...");
  if (browserInstance) {
    await browserInstance.close();
    browserInstance = null;
  }
  process.exit(0);
});

process.on("SIGINT", async () => {
  logger.info("Service shutting down...");
  if (browserInstance) {
    await browserInstance.close();
    browserInstance = null;
  }
  process.exit(0);
});

// Send alert email
async function sendAlertEmail(site, error) {
  const mailOptions = {
    from: "3074994545@qq.com", // Use full sender email
    to: "3074994545@qq.com", // Use full recipient email
    subject: `🚨 Website Alert: ${site.name}`,
    html: `
      <h2>Website Monitoring Alert</h2>
      <p><strong>Website Name:</strong> ${site.name}</p>
      <p><strong>Website URL:</strong> ${site.url}</p>
      <p><strong>Error Message:</strong> ${error}</p>
      <p><strong>Time:</strong> ${new Date().toLocaleString()}</p>
    `,
  };

  try {
    await transporter.sendMail(mailOptions);
    console.log(`Alert email sent - ${site.name}`);
  } catch (error) {
    console.error("Failed to send email:", error);
    // If it's an auth error, output more details
    if (error.code === "EAUTH") {
      console.error("Email authentication failed. Please check the following:");
      console.error("1. Make sure QQ email SMTP service is enabled");
      console.error("2. Make sure you are using the correct authorization code, not your email password");
      console.error("3. Visit https://service.mail.qq.com/cgi-bin/help?subtype=1&&id=28&&no=1001256 for help");
    }
  }
}

// Execute monitoring task
async function runMonitoring() {
  // Add concurrency limit
  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(`Monitoring task failed (${site.name}):`, {
            error: error.message,
            stack: error.stack,
          });
        }
      })
    );
  }
}

// Add graceful shutdown handler
process.on("SIGTERM", async () => {
  logger.info("Service shutting down...");
  if (browserInstance) {
    await browserInstance.close();
  }
  process.exit(0);
});

// Schedule monitoring every 5 minutes
cron.schedule("*/5 * * * *", () => {
  console.log("Starting website monitoring...");
  runMonitoring();
});

// Run monitoring immediately
console.log("Starting website monitoring service...");
runMonitoring();