feat: 添加 Gmail 分类支持，优化邮件轮询逻辑

content/gmail-mail.js

@@ -204,6 +204,102 @@ function findInboxLink() {
   }) || null;
 }
 
+const GMAIL_CATEGORY_LABELS = {
+  primary: [/^primary$/i, /^inbox$/i, /^主要$/],
+  updates: [/^updates$/i, /^更新$/],
+  promotions: [/^promotions$/i, /^推广$/],
+  social: [/^social$/i, /^社交$/],
+};
+
+function getCategoryKeyFromText(text) {
+  const normalizedText = normalizeText(text);
+  if (!normalizedText) return '';
+
+  for (const [key, patterns] of Object.entries(GMAIL_CATEGORY_LABELS)) {
+    if (patterns.some((pattern) => pattern.test(normalizedText))) {
+      return key;
+    }
+  }
+
+  return '';
+}
+
+function getCategoryTabLabel(tab) {
+  const text = normalizeText(
+    tab?.getAttribute?.('aria-label')
+    || tab?.getAttribute?.('data-tooltip')
+    || tab?.getAttribute?.('title')
+    || tab?.textContent
+    || ''
+  );
+  return text;
+}
+
+function collectCategoryTabs() {
+  const tabs = Array.from(document.querySelectorAll('[role="tab"], [data-tooltip-align][role="link"]'));
+  const categoryTabs = [];
+  const seenKeys = new Set();
+
+  tabs.forEach((tab) => {
+    if (!isVisibleElement(tab)) return;
+    const label = getCategoryTabLabel(tab);
+    const key = getCategoryKeyFromText(label);
+    if (!key || seenKeys.has(key)) return;
+
+    seenKeys.add(key);
+    categoryTabs.push({
+      key,
+      label,
+      selected: tab.getAttribute('aria-selected') === 'true' || /\bTO\b/.test(tab.className || ''),
+      tab,
+    });
+  });
+
+  return categoryTabs;
+}
+
+function getCategoryScanOrder() {
+  const categoryTabs = collectCategoryTabs();
+  if (!categoryTabs.length) {
+    return [{ key: 'primary', label: 'Primary', selected: true, tab: null }];
+  }
+
+  const ordered = ['updates', 'primary']
+    .map((key) => categoryTabs.find((item) => item.key === key))
+    .filter(Boolean);
+
+  return ordered.length
+    ? ordered
+    : [{ key: 'primary', label: 'Primary', selected: true, tab: null }];
+}
+
+async function activateCategoryTab(step, categoryKey) {
+  const categoryTabs = collectCategoryTabs();
+  const target = categoryTabs.find((item) => item.key === categoryKey);
+  if (!target?.tab) {
+    return { key: categoryKey, label: categoryKey, switched: false };
+  }
+
+  if (target.selected) {
+    return { key: target.key, label: target.label, switched: false };
+  }
+
+  simulateClick(target.tab);
+  for (let i = 0; i < 20; i++) {
+    await sleep(200);
+    const refreshed = collectCategoryTabs().find((item) => item.key === categoryKey);
+    if (refreshed?.selected) {
+      await sleep(500);
+      log(`步骤 ${step}：已切换到 Gmail 分类 ${refreshed.label}。`);
+      return { key: refreshed.key, label: refreshed.label, switched: true };
+    }
+  }
+
+  await sleep(600);
+  log(`步骤 ${step}：已尝试切换到 Gmail 分类 ${target.label}。`, 'info');
+  return { key: target.key, label: target.label, switched: true };
+}
+
 function findRefreshButton() {
   const selectors = [
     'div[role="button"][data-tooltip="刷新"]',
@@ -473,8 +569,15 @@ async function handlePollEmail(step, payload) {
     throw new Error('Gmail 收件箱列表未加载完成，请确认当前已打开 Gmail 收件箱。');
   }
 
-  const existingMailIds = getCurrentMailIds(initialRows);
+  const categoryOrder = getCategoryScanOrder();
-  log(`步骤 ${step}：已记录当前 ${existingMailIds.size} 封旧邮件快照`);
+  const existingMailIdsByCategory = new Map();
+
+  for (const category of categoryOrder) {
+    const activeCategory = await activateCategoryTab(step, category.key);
+    const rows = collectThreadRows();
+    existingMailIdsByCategory.set(activeCategory.key, getCurrentMailIds(rows));
+    log(`步骤 ${step}：已记录 Gmail 分类 ${activeCategory.label} 的 ${rows.length} 封旧邮件快照`);
+  }
 
   for (let attempt = 1; attempt <= maxAttempts; attempt++) {
     log(`步骤 ${step}：正在轮询 Gmail，第 ${attempt}/${maxAttempts} 次`);
@@ -483,9 +586,13 @@ async function handlePollEmail(step, payload) {
       await refreshInbox(step);
     }
 
-    const rows = collectThreadRows();
     const useFallback = attempt > GMAIL_FALLBACK_AFTER;
 
+    for (const category of categoryOrder) {
+      const activeCategory = await activateCategoryTab(step, category.key);
+      const rows = collectThreadRows();
+      const existingMailIds = existingMailIdsByCategory.get(activeCategory.key) || new Set();
+
       for (let index = 0; index < rows.length; index++) {
         const row = rows[index];
         const rowId = getRowFingerprint(row, index);
@@ -523,7 +630,7 @@ async function handlePollEmail(step, payload) {
           const source = useFallback && existingMailIds.has(rowId) ? '回退匹配邮件' : '新邮件';
           const timeLabel = rowTimestamp ? `，时间：${new Date(rowTimestamp).toLocaleString('zh-CN', { hour12: false })}` : '';
           const targetLabel = previewTargetState.matches ? '，目标邮箱命中' : '';
-        log(`步骤 ${step}：已在 Gmail 找到验证码：${previewCode}（来源：${source}${timeLabel}${targetLabel}）`, 'ok');
+          log(`步骤 ${step}：已在 Gmail ${activeCategory.label} 分类找到验证码：${previewCode}（来源：${source}${timeLabel}${targetLabel}）`, 'ok');
           return {
             ok: true,
             code: previewCode,
@@ -551,7 +658,7 @@ async function handlePollEmail(step, payload) {
         const source = useFallback && existingMailIds.has(rowId) ? '回退匹配邮件正文' : '新邮件正文';
         const timeLabel = rowTimestamp ? `，时间：${new Date(rowTimestamp).toLocaleString('zh-CN', { hour12: false })}` : '';
         const targetLabel = openedTargetState.matches ? '，目标邮箱命中' : '';
-      log(`步骤 ${step}：已在 Gmail 正文中找到验证码：${bodyCode}（来源：${source}${timeLabel}${targetLabel}）`, 'ok');
+        log(`步骤 ${step}：已在 Gmail ${activeCategory.label} 分类正文中找到验证码：${bodyCode}（来源：${source}${timeLabel}${targetLabel}）`, 'ok');
         return {
           ok: true,
           code: bodyCode,
@@ -559,6 +666,7 @@ async function handlePollEmail(step, payload) {
           mailId: rowId,
         };
       }
+    }
 
     if (attempt === GMAIL_FALLBACK_AFTER + 1) {
       log(`步骤 ${step}：连续 ${GMAIL_FALLBACK_AFTER} 次未发现新邮件，开始回退到首封匹配邮件`, 'warn');