QLHazyCoder
300 lines
10 KiB
JavaScript
300 lines
10 KiB
JavaScript
// content/mail-163.js — Content script for 163 Mail (steps 4, 7)
|
||
// Injected on: mail.163.com
|
||
//
|
||
// DOM structure:
|
||
// Mail item: div[sign="letter"] with aria-label="你的 ChatGPT 代码为 479637 发件人 ： OpenAI ..."
|
||
// Sender: .nui-user (e.g., "OpenAI")
|
||
// Subject: span.da0 (e.g., "你的 ChatGPT 代码为 479637")
|
||
// Right-click menu: .nui-menu → .nui-menu-item with text "删除邮件"
|
||
|
||
const MAIL163_PREFIX = '[MultiPage:mail-163]';
|
||
const isTopFrame = window === window.top;
|
||
|
||
console.log(MAIL163_PREFIX, 'Content script loaded on', location.href, 'frame:', isTopFrame ? 'top' : 'child');
|
||
|
||
// Only operate in the top frame
|
||
if (!isTopFrame) {
|
||
console.log(MAIL163_PREFIX, 'Skipping child frame');
|
||
} else {
|
||
|
||
// Track codes we've already seen — persisted in chrome.storage.session to survive script re-injection
|
||
let seenCodes = new Set();
|
||
|
||
async function loadSeenCodes() {
|
||
try {
|
||
const data = await chrome.storage.session.get('seenCodes');
|
||
if (data.seenCodes && Array.isArray(data.seenCodes)) {
|
||
seenCodes = new Set(data.seenCodes);
|
||
console.log(MAIL163_PREFIX, `Loaded ${seenCodes.size} previously seen codes`);
|
||
}
|
||
} catch (err) {
|
||
console.warn(MAIL163_PREFIX, 'Session storage unavailable, using in-memory seen codes:', err?.message || err);
|
||
}
|
||
}
|
||
|
||
// Load previously seen codes on startup
|
||
loadSeenCodes();
|
||
|
||
async function persistSeenCodes() {
|
||
try {
|
||
await chrome.storage.session.set({ seenCodes: [...seenCodes] });
|
||
} catch (err) {
|
||
console.warn(MAIL163_PREFIX, 'Could not persist seen codes, continuing in-memory only:', err?.message || err);
|
||
}
|
||
}
|
||
|
||
// ============================================================
|
||
// Message Handler (top frame only)
|
||
// ============================================================
|
||
|
||
chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
|
||
if (message.type === 'POLL_EMAIL') {
|
||
resetStopState();
|
||
handlePollEmail(message.step, message.payload).then(result => {
|
||
sendResponse(result);
|
||
}).catch(err => {
|
||
if (isStopError(err)) {
|
||
log(`步骤 ${message.step}：已被用户停止。`, 'warn');
|
||
sendResponse({ stopped: true, error: err.message });
|
||
return;
|
||
}
|
||
log(`步骤 ${message.step}：邮箱轮询失败：${err.message}`, 'warn');
|
||
sendResponse({ error: err.message });
|
||
});
|
||
return true;
|
||
}
|
||
});
|
||
|
||
// ============================================================
|
||
// Find mail items
|
||
// ============================================================
|
||
|
||
function findMailItems() {
|
||
return document.querySelectorAll('div[sign="letter"]');
|
||
}
|
||
|
||
function getCurrentMailIds() {
|
||
const ids = new Set();
|
||
findMailItems().forEach(item => {
|
||
const id = item.getAttribute('id') || '';
|
||
if (id) ids.add(id);
|
||
});
|
||
return ids;
|
||
}
|
||
|
||
// ============================================================
|
||
// Email Polling
|
||
// ============================================================
|
||
|
||
async function handlePollEmail(step, payload) {
|
||
const { senderFilters, subjectFilters, maxAttempts, intervalMs, excludeCodes = [] } = payload;
|
||
const excludedCodeSet = new Set(excludeCodes.filter(Boolean));
|
||
|
||
log(`步骤 ${step}：开始轮询 163 邮箱（最多 ${maxAttempts} 次）`);
|
||
|
||
// Click inbox in sidebar to ensure we're in inbox view
|
||
log(`步骤 ${step}：正在等待侧边栏加载...`);
|
||
try {
|
||
const inboxLink = await waitForElement('.nui-tree-item-text[title="收件箱"]', 5000);
|
||
inboxLink.click();
|
||
log(`步骤 ${step}：已点击收件箱`);
|
||
} catch {
|
||
log(`步骤 ${step}：未找到收件箱入口，继续尝试后续流程...`, 'warn');
|
||
}
|
||
|
||
// Wait for mail list to appear
|
||
log(`步骤 ${step}：正在等待邮件列表加载...`);
|
||
let items = [];
|
||
for (let i = 0; i < 20; i++) {
|
||
items = findMailItems();
|
||
if (items.length > 0) break;
|
||
await sleep(500);
|
||
}
|
||
|
||
if (items.length === 0) {
|
||
await refreshInbox();
|
||
await sleep(2000);
|
||
items = findMailItems();
|
||
}
|
||
|
||
if (items.length === 0) {
|
||
throw new Error('163 邮箱列表未加载完成，请确认当前已打开收件箱。');
|
||
}
|
||
|
||
log(`步骤 ${step}：邮件列表已加载，共 ${items.length} 封邮件`);
|
||
|
||
// Snapshot existing mail IDs
|
||
const existingMailIds = getCurrentMailIds();
|
||
log(`步骤 ${step}：已记录当前 ${existingMailIds.size} 封旧邮件快照`);
|
||
|
||
const FALLBACK_AFTER = 3;
|
||
|
||
for (let attempt = 1; attempt <= maxAttempts; attempt++) {
|
||
log(`步骤 ${step}：正在轮询 163 邮箱，第 ${attempt}/${maxAttempts} 次`);
|
||
|
||
if (attempt > 1) {
|
||
await refreshInbox();
|
||
await sleep(1000);
|
||
}
|
||
|
||
const allItems = findMailItems();
|
||
const useFallback = attempt > FALLBACK_AFTER;
|
||
|
||
for (const item of allItems) {
|
||
const id = item.getAttribute('id') || '';
|
||
|
||
if (!useFallback && existingMailIds.has(id)) continue;
|
||
|
||
const senderEl = item.querySelector('.nui-user');
|
||
const sender = senderEl ? senderEl.textContent.toLowerCase() : '';
|
||
|
||
const subjectEl = item.querySelector('span.da0');
|
||
const subject = subjectEl ? subjectEl.textContent : '';
|
||
|
||
const ariaLabel = (item.getAttribute('aria-label') || '').toLowerCase();
|
||
|
||
const senderMatch = senderFilters.some(f => sender.includes(f.toLowerCase()) || ariaLabel.includes(f.toLowerCase()));
|
||
const subjectMatch = subjectFilters.some(f => subject.toLowerCase().includes(f.toLowerCase()) || ariaLabel.includes(f.toLowerCase()));
|
||
|
||
if (senderMatch || subjectMatch) {
|
||
const code = extractVerificationCode(subject + ' ' + ariaLabel);
|
||
if (code && excludedCodeSet.has(code)) {
|
||
log(`步骤 ${step}：跳过排除的验证码：${code}`, 'info');
|
||
} else if (code && !seenCodes.has(code)) {
|
||
seenCodes.add(code);
|
||
persistSeenCodes();
|
||
const source = useFallback && existingMailIds.has(id) ? '回退匹配邮件' : '新邮件';
|
||
log(`步骤 ${step}：已找到验证码：${code}（来源：${source}，主题：${subject.slice(0, 40)}）`, 'ok');
|
||
|
||
// Delete this email via right-click menu, WAIT for it to finish before returning
|
||
await deleteEmail(item, step);
|
||
// Extra wait to ensure deletion is processed
|
||
await sleep(1000);
|
||
|
||
return { ok: true, code, emailTimestamp: Date.now(), mailId: id };
|
||
} else if (code && seenCodes.has(code)) {
|
||
log(`步骤 ${step}：跳过已处理过的验证码：${code}`, 'info');
|
||
}
|
||
}
|
||
}
|
||
|
||
if (attempt === FALLBACK_AFTER + 1) {
|
||
log(`步骤 ${step}：连续 ${FALLBACK_AFTER} 次未发现新邮件，开始回退到首封匹配邮件`, 'warn');
|
||
}
|
||
|
||
if (attempt < maxAttempts) {
|
||
await sleep(intervalMs);
|
||
}
|
||
}
|
||
|
||
throw new Error(
|
||
`${(maxAttempts * intervalMs / 1000).toFixed(0)} 秒后仍未在 163 邮箱中找到新的匹配邮件。` +
|
||
'请手动检查收件箱。'
|
||
);
|
||
}
|
||
|
||
// ============================================================
|
||
// Delete Email via Right-Click Menu
|
||
// ============================================================
|
||
|
||
async function deleteEmail(item, step) {
|
||
try {
|
||
log(`步骤 ${step}：正在删除邮件...`);
|
||
|
||
// Strategy 1: Click the trash icon inside the mail item
|
||
// Each mail item has: <b class="nui-ico nui-ico-delete" title="删除邮件" sign="trash">
|
||
// These icons appear on hover, so we trigger mouseover first
|
||
item.dispatchEvent(new MouseEvent('mouseover', { bubbles: true }));
|
||
item.dispatchEvent(new MouseEvent('mouseenter', { bubbles: true }));
|
||
await sleep(300);
|
||
|
||
const trashIcon = item.querySelector('[sign="trash"], .nui-ico-delete, [title="删除邮件"]');
|
||
if (trashIcon) {
|
||
trashIcon.click();
|
||
log(`步骤 ${step}：已点击删除图标`, 'ok');
|
||
await sleep(1500);
|
||
|
||
// Check if item disappeared (confirm deletion)
|
||
const stillExists = document.getElementById(item.id);
|
||
if (!stillExists || stillExists.style.display === 'none') {
|
||
log(`步骤 ${step}：邮件已成功删除`);
|
||
} else {
|
||
log(`步骤 ${step}：邮件可能尚未删除，列表中仍可见`, 'warn');
|
||
}
|
||
return;
|
||
}
|
||
|
||
// Strategy 2: Select checkbox then click toolbar delete button
|
||
log(`步骤 ${step}：未找到删除图标，尝试使用复选框加工具栏删除...`);
|
||
const checkbox = item.querySelector('[sign="checkbox"], .nui-chk');
|
||
if (checkbox) {
|
||
checkbox.click();
|
||
await sleep(300);
|
||
|
||
// Click toolbar delete button
|
||
const toolbarBtns = document.querySelectorAll('.nui-btn .nui-btn-text');
|
||
for (const btn of toolbarBtns) {
|
||
if (btn.textContent.replace(/\s/g, '').includes('删除')) {
|
||
btn.closest('.nui-btn').click();
|
||
log(`步骤 ${step}：已点击工具栏删除`, 'ok');
|
||
await sleep(1500);
|
||
return;
|
||
}
|
||
}
|
||
}
|
||
|
||
log(`步骤 ${step}：无法删除邮件（未找到删除按钮）`, 'warn');
|
||
} catch (err) {
|
||
log(`步骤 ${step}：删除邮件失败：${err.message}`, 'warn');
|
||
}
|
||
}
|
||
|
||
// ============================================================
|
||
// Inbox Refresh
|
||
// ============================================================
|
||
|
||
async function refreshInbox() {
|
||
// Try toolbar "刷 新" button
|
||
const toolbarBtns = document.querySelectorAll('.nui-btn .nui-btn-text');
|
||
for (const btn of toolbarBtns) {
|
||
if (btn.textContent.replace(/\s/g, '') === '刷新') {
|
||
btn.closest('.nui-btn').click();
|
||
console.log(MAIL163_PREFIX, 'Clicked "刷新" button');
|
||
await sleep(800);
|
||
return;
|
||
}
|
||
}
|
||
|
||
// Fallback: click sidebar "收 信"
|
||
const shouXinBtns = document.querySelectorAll('.ra0');
|
||
for (const btn of shouXinBtns) {
|
||
if (btn.textContent.replace(/\s/g, '').includes('收信')) {
|
||
btn.click();
|
||
console.log(MAIL163_PREFIX, 'Clicked "收信" button');
|
||
await sleep(800);
|
||
return;
|
||
}
|
||
}
|
||
|
||
console.log(MAIL163_PREFIX, 'Could not find refresh button');
|
||
}
|
||
|
||
// ============================================================
|
||
// Verification Code Extraction
|
||
// ============================================================
|
||
|
||
function extractVerificationCode(text) {
|
||
const matchCn = text.match(/(?:代码为|验证码[^0-9]*?)[\s：:]*(\d{6})/);
|
||
if (matchCn) return matchCn[1];
|
||
|
||
const matchEn = text.match(/code[:\s]+is[:\s]+(\d{6})|code[:\s]+(\d{6})/i);
|
||
if (matchEn) return matchEn[1] || matchEn[2];
|
||
|
||
const match6 = text.match(/\b(\d{6})\b/);
|
||
if (match6) return match6[1];
|
||
|
||
return null;
|
||
}
|
||
|
||
} // end of isTopFrame else block
|