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 };
}

/**
 * 等待并获取验证码
 * @param {Page} page - Playwright 页面对象
 * @returns {Promise<string>} 验证码
 */
export async function waitForVerificationCode(page, sentAfterMs) {
  console.log('⏳ 等待接收验证码邮件...');
  console.log('[LOG] 不会刷新页面，只等待新邮件出现...');

  const maxWaitTime = 120000; // 最多等待 120 秒
  const checkInterval = 1000; // 每 1 秒检查一次
  let elapsed = 0;

  async function extractCodeFromOpenedMail() {
    // 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();
      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 || '');
        const m = /\b(\d{6})\b/.exec(text);
        if (m) return m[1];
      } catch {}
    }
    // 3) 退化为整页文本
    const all = await page.textContent('body');
    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} 秒)`);

    // 邮件列表行（兼容多种结构）
    const emailRows = page.locator('.inbox-dataList .mail-item, .inbox-dataList > div, div.row.no-gutters');
    const emailCount = await emailRows.count();
    console.log(`[LOG] 当前邮件数量: ${emailCount}`);

    if (emailCount > 0) {
      console.log('📬 发现邮件，尝试读取...');

      // 遍历所有邮件，优先读取发送之后的邮件（按顺序尝试）
      for (let i = 0; i < emailCount; i++) {
        const emailRow = emailRows.nth(i);

        // 过滤发送时间之前的旧邮件（根据 data-date）
        try {
          const timeSpan = emailRow.locator('span[data-date]').first();
          if (await timeSpan.count() > 0) {
            const dateStr = await timeSpan.getAttribute('data-date');
            if (dateStr) {
              const parsed = Date.parse(dateStr.replace(' ', 'T')) || new Date(dateStr).getTime();
              if (sentAfterMs && parsed && parsed < sentAfterMs) {
                console.log(`[LOG] 跳过旧邮件 ${dateStr}`);
                continue;
              }
            }
          }
        } catch {}

        try {
          // 点击打开邮件（有些需要双击/再次点击切换到详情）
          await emailRow.click({ timeout: 5000 });
          await page.waitForTimeout(500);
          try { await emailRow.dblclick({ timeout: 1000 }); } catch {}

          // 等待内容区域或 iframe 加载
          await page.waitForTimeout(1500);

          const code = await extractCodeFromOpenedMail();
          if (code) {
            console.log('✅ 获取到验证码:', code);
            return code;
          }
          console.log(`[LOG] 邮件 ${i + 1} 中未找到验证码`);
        } catch (error) {
          console.log(`[LOG] 读取邮件 ${i + 1} 失败:`, error.message);
        }
      }

      console.log('⚠️ 所有邮件中未找到验证码，继续等待...');
    }
  }

  throw new Error('超时：未能在规定时间内收到验证码');
}