269 lines
10 KiB
JavaScript
269 lines
10 KiB
JavaScript
import { config } from './config.js';
|
||
|
||
/**
|
||
* 设置临时邮箱并获取验证码
|
||
* @param {Page} page - Playwright 页面对象
|
||
* @returns {Promise<{email: string, code: string}>}
|
||
*/
|
||
export async function setupTempMail(page) {
|
||
console.log('📧 开始设置临时邮箱...');
|
||
|
||
// 访问 tempmail.plus（增加超时时间，因为网络较慢）
|
||
await page.goto(config.tempmail.url, {
|
||
waitUntil: 'domcontentloaded', // 改为 domcontentloaded，不等待所有网络请求
|
||
timeout: 60000 // 60秒超时
|
||
});
|
||
await page.waitForTimeout(3000); // 等待页面稳定
|
||
|
||
// 输入邮箱用户名（固定为配置的 qichen111）
|
||
console.log('输入邮箱用户名:', config.tempmail.username);
|
||
await page.fill('#pre_button', config.tempmail.username);
|
||
|
||
// 点击页面其他位置或按回车来确认输入，触发邮箱地址更新
|
||
console.log('确认邮箱用户名...');
|
||
await page.click('body'); // 点击页面空白处
|
||
await page.waitForTimeout(2000);
|
||
|
||
// 检查 PIN 码弹窗是否已经显示
|
||
console.log('[LOG] 检查 PIN 码弹窗状态...');
|
||
const pinModal = page.locator('#modal-verify');
|
||
await page.waitForTimeout(2000); // 等待页面稳定
|
||
|
||
const initialModalClass = await pinModal.getAttribute('class');
|
||
console.log('[LOG] 弹窗 class:', initialModalClass);
|
||
|
||
// 判断弹窗是否已经显示（包含 "show" class）
|
||
const isModalShown = initialModalClass && initialModalClass.includes('show');
|
||
console.log('[LOG] 弹窗是否已显示:', isModalShown);
|
||
|
||
if (!isModalShown) {
|
||
// 如果弹窗未显示，才需要点击 PIN 保护元素
|
||
console.log('[LOG] 弹窗未显示，点击 PIN 码保护元素...');
|
||
const pinProtectSpan = page.locator('span.pin-text[data-tr="box_protected"]');
|
||
await pinProtectSpan.waitFor({ state: 'visible', timeout: 15000 });
|
||
await pinProtectSpan.click({ force: true });
|
||
console.log('[LOG] ✅ 已点击 PIN 保护元素');
|
||
await page.waitForTimeout(2000);
|
||
} else {
|
||
console.log('[LOG] ✅ 弹窗已自动显示，无需点击');
|
||
}
|
||
|
||
// 查找 PIN 码输入框
|
||
console.log('[LOG] 查找 PIN 码输入框...');
|
||
const pinInput = page.locator('#pin');
|
||
const pinInputCount = await pinInput.count();
|
||
console.log('[LOG] 找到 PIN 输入框数量:', pinInputCount);
|
||
|
||
if (pinInputCount > 0) {
|
||
// 检查输入框是否可见
|
||
const isVisible = await pinInput.isVisible();
|
||
console.log('[LOG] PIN 输入框是否可见:', isVisible);
|
||
|
||
// 输入 PIN 码
|
||
console.log('[LOG] 输入 PIN 码:', config.tempmail.pinCode);
|
||
await pinInput.fill(config.tempmail.pinCode);
|
||
console.log('[LOG] ✅ PIN 码已输入');
|
||
await page.waitForTimeout(500);
|
||
|
||
// 查找提交按钮
|
||
console.log('[LOG] 查找提交按钮 #verify...');
|
||
const verifyButton = page.locator('#verify');
|
||
const verifyButtonCount = await verifyButton.count();
|
||
console.log('[LOG] 找到提交按钮数量:', verifyButtonCount);
|
||
|
||
if (verifyButtonCount > 0) {
|
||
const buttonText = await verifyButton.textContent();
|
||
console.log('[LOG] 按钮文本:', buttonText);
|
||
console.log('[LOG] 点击提交按钮...');
|
||
await verifyButton.click();
|
||
console.log('[LOG] ✅ 已点击提交按钮');
|
||
} else {
|
||
console.log('[LOG] ⚠️ 未找到提交按钮，尝试按回车');
|
||
await pinInput.press('Enter');
|
||
}
|
||
|
||
// 等待验证完成
|
||
await page.waitForTimeout(2000);
|
||
console.log('[LOG] ✅ PIN 码验证流程完成');
|
||
} else {
|
||
console.log('[LOG] ❌ 未找到 PIN 输入框！');
|
||
throw new Error('未找到 PIN 码输入框');
|
||
}
|
||
|
||
// 等待弹窗关闭后获取邮箱地址
|
||
await page.waitForTimeout(2000);
|
||
console.log('[LOG] 获取生成的邮箱地址...');
|
||
|
||
// 尝试多个可能的邮箱地址元素
|
||
let emailAddress = '';
|
||
|
||
// 方法 1: 查找特定的邮箱显示元素
|
||
const emailDisplay = page.locator('#email-address, #mail, .email-address, [id*="email"]').first();
|
||
if (await emailDisplay.count() > 0 && await emailDisplay.isVisible()) {
|
||
const text = await emailDisplay.textContent();
|
||
const match = text.match(/[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/);
|
||
if (match) {
|
||
emailAddress = match[0];
|
||
}
|
||
}
|
||
|
||
// 方法 2: 如果没找到，尝试从页面中提取所有包含 @ 的文本
|
||
if (!emailAddress) {
|
||
const pageContent = await page.content();
|
||
const emailMatches = pageContent.match(/qichen111@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/);
|
||
if (emailMatches && emailMatches.length > 0) {
|
||
emailAddress = emailMatches[0];
|
||
}
|
||
}
|
||
|
||
// 方法 3: 查找复制按钮附近的文本
|
||
if (!emailAddress) {
|
||
const copyButton = page.locator('button:has-text("复制")').first();
|
||
if (await copyButton.count() > 0) {
|
||
const parent = copyButton.locator('..');
|
||
const text = await parent.textContent();
|
||
const match = text.match(/[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/);
|
||
if (match) {
|
||
emailAddress = match[0];
|
||
}
|
||
}
|
||
}
|
||
|
||
console.log('[LOG] 提取到的邮箱地址:', emailAddress);
|
||
|
||
if (!emailAddress || !emailAddress.includes('@')) {
|
||
console.log('[LOG] ⚠️ 未能正确获取邮箱地址，尝试使用默认格式');
|
||
emailAddress = `${config.tempmail.username}@mailto.plus`;
|
||
}
|
||
|
||
console.log('✅ 邮箱地址:', emailAddress);
|
||
|
||
return { inboxReady: true, page };
|
||
}
|
||
|
||
// 全局基线时间，在 setupTempMail 后初始化，每次注册成功后更新
|
||
let globalBaselineTime = null;
|
||
|
||
/**
|
||
* 获取收件箱最新一条邮件的时间作为基线
|
||
*/
|
||
export async function updateBaseline(page) {
|
||
try {
|
||
// 等待收件箱加载
|
||
await page.waitForSelector('.inbox', { timeout: 10000 }).catch(() => {});
|
||
await page.waitForTimeout(2000);
|
||
|
||
// 查找所有含有时间的邮件行
|
||
const timeSpans = page.locator('.inbox span[data-date]');
|
||
const count = await timeSpans.count();
|
||
if (count === 0) {
|
||
globalBaselineTime = null;
|
||
console.log('[基线] 收件箱无邮件，基线为 null');
|
||
return;
|
||
}
|
||
// 第一个就是最新的
|
||
const dateStr = await timeSpans.first().getAttribute('data-date');
|
||
globalBaselineTime = dateStr;
|
||
console.log('[基线] 更新为最新邮件时间:', globalBaselineTime);
|
||
} catch (e) {
|
||
console.log('[基线] 获取失败:', e.message);
|
||
globalBaselineTime = null;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 等待并获取验证码
|
||
* @param {Page} page - Playwright 页面对象
|
||
* @returns {Promise<string>} 验证码
|
||
*/
|
||
export async function waitForVerificationCode(page, sentAfterMs, targetEmail) {
|
||
console.log('⏳ 等待接收验证码邮件...');
|
||
console.log('[LOG] 基线时间:', globalBaselineTime || '未设置');
|
||
|
||
const maxWaitTime = 120000;
|
||
const checkInterval = 1000;
|
||
let elapsed = 0;
|
||
|
||
async function extractCodeFromOpenedMail(expectEmail) {
|
||
// 1) 优先从特定内容区域读取
|
||
const contentLoc = page.locator('.mail-content, .email-body, #mail-body, [class*="mail-content"], .pm-text, .view, .message-body').first();
|
||
if (await contentLoc.count() > 0) {
|
||
const text = await contentLoc.textContent();
|
||
if (expectEmail && text && !text.includes(expectEmail)) {
|
||
console.log('[LOG] 内容不匹配目标邮箱，继续查找');
|
||
} else {
|
||
const code = /\b(\d{6})\b/.exec(text || '');
|
||
if (code) return code[1];
|
||
}
|
||
}
|
||
// 2) 遍历所有 iframe 提取文本
|
||
const frames = page.frames();
|
||
console.log(`[LOG] 当前 frame 数量: ${frames.length}`);
|
||
for (const f of frames) {
|
||
try {
|
||
const body = await f.$('body');
|
||
if (!body) continue;
|
||
const text = await f.evaluate(() => document.body.innerText || '');
|
||
if (expectEmail && text && !text.includes(expectEmail)) continue;
|
||
const m = /\b(\d{6})\b/.exec(text);
|
||
if (m) return m[1];
|
||
} catch {}
|
||
}
|
||
// 3) 退化为整页文本
|
||
const all = await page.textContent('body');
|
||
if (expectEmail && all && !all.includes(expectEmail)) return null;
|
||
const m = /\b(\d{6})\b/.exec(all || '');
|
||
return m ? m[1] : null;
|
||
}
|
||
|
||
while (elapsed < maxWaitTime) {
|
||
await page.waitForTimeout(checkInterval);
|
||
elapsed += checkInterval;
|
||
|
||
console.log(`[LOG] 检查邮件... (已等待 ${elapsed / 1000} 秒)`);
|
||
|
||
try {
|
||
// 每个邮件包含 <span data-date>，找到所有带时间戳的行
|
||
const timeSpans = page.locator('.inbox span[data-date]');
|
||
const count = await timeSpans.count();
|
||
console.log(`[LOG] 当前邮件数量: ${count}`);
|
||
|
||
if (count === 0) continue;
|
||
|
||
// 遍历所有邮件，找第一个满足条件的：发件人 cfbounces && 时间 > 基线
|
||
for (let i = 0; i < count; i++) {
|
||
const timeSpan = timeSpans.nth(i);
|
||
const dateStr = await timeSpan.getAttribute('data-date');
|
||
|
||
// 向上找到包含 from 的父元素获取发件人
|
||
const row = timeSpan.locator('xpath=ancestor::div[contains(@class, "row")]').first();
|
||
const fromSpan = row.locator('.from span').last();
|
||
const sender = await fromSpan.textContent().catch(() => '');
|
||
|
||
const isCfbounces = /cfbounces/i.test(sender || '');
|
||
const isNewer = globalBaselineTime ? (dateStr > globalBaselineTime) : true;
|
||
|
||
console.log(`[LOG] Row ${i+1}: sender=${sender} date=${dateStr} baseline=${globalBaselineTime} newer=${isNewer}`);
|
||
|
||
if (isCfbounces && isNewer) {
|
||
console.log('✅ 发现新验证邮件，点击读取...');
|
||
await row.scrollIntoViewIfNeeded().catch(() => {});
|
||
await row.click({ timeout: 5000 });
|
||
await page.waitForTimeout(2000);
|
||
|
||
const code = await extractCodeFromOpenedMail(targetEmail);
|
||
if (code) {
|
||
console.log('✅ 获取到验证码:', code);
|
||
return code;
|
||
}
|
||
console.log('[LOG] 该邮件未提取到验证码，继续...');
|
||
}
|
||
}
|
||
} catch (e) {
|
||
console.log('[LOG] 检查邮件出错:', e.message);
|
||
}
|
||
}
|
||
|
||
throw new Error('超时：未能在规定时间内收到验证码');
|
||
}
|