diff --git a/background.js b/background.js index 67d0765..a906b28 100644 --- a/background.js +++ b/background.js @@ -7433,6 +7433,27 @@ async function executeStep4(state) { // Step 5: Fill Name & Birthday (via signup-page.js) // ============================================================ +async function waitForStep5ChatgptRedirect(tabId, timeoutMs = 15000) { + if (!Number.isInteger(tabId)) { + return null; + } + + const matchedTab = await waitForTabUrlMatch(tabId, (url) => /chatgpt\.com/i.test(url || ''), { + timeoutMs, + retryDelayMs: 300, + }); + if (matchedTab) { + return matchedTab; + } + + const currentTab = await chrome.tabs.get(tabId).catch(() => null); + if (currentTab && /chatgpt\.com/i.test(currentTab.url || '')) { + return currentTab; + } + + return null; +} + async function executeStep5(state) { const { firstName, lastName } = generateRandomName(); const { year, month, day } = generateRandomBirthday(); @@ -7468,16 +7489,19 @@ async function executeStep5(state) { return; } + if (step5Result?.chatgptHome) { + await addLog('步骤 5:检测到已进入 ChatGPT 页面,注册成功。', 'ok'); + return; + } + // Case 2: content script died due to cross-origin navigation — check tab URL if (step5TransportError) { const signupTabId = await getTabId('signup-page'); - if (signupTabId) { - const tab = await chrome.tabs.get(signupTabId).catch(() => null); - if (tab && /chatgpt\.com/i.test(tab.url || '')) { - await addLog('步骤 5:内容脚本因页面跳转到 ChatGPT 而断开,正在处理引导页跳过...'); - await handleChatgptOnboardingSkip(signupTabId); - return; - } + const redirectedTab = await waitForStep5ChatgptRedirect(signupTabId); + if (redirectedTab) { + await addLog('步骤 5:内容脚本因页面跳转到 ChatGPT 而断开,正在处理引导页跳过...'); + await handleChatgptOnboardingSkip(redirectedTab.id); + return; } // Not on chatgpt.com — re-throw the original error throw step5TransportError; @@ -7534,6 +7558,12 @@ async function handleChatgptOnboardingSkip(knownTabId) { throw new Error(result.error); } + if (result?.alreadyCompleted) { + await addLog('步骤 5:已进入 ChatGPT 页面,无需继续跳过引导,注册成功。', 'ok'); + await completeStepFromBackground(5, { chatgptHome: true }); + return; + } + await addLog('步骤 5:ChatGPT 引导页跳过完成,注册成功。', 'ok'); await completeStepFromBackground(5, { chatgptOnboarding: true }); } diff --git a/content/signup-page.js b/content/signup-page.js index 39c79e8..baff50c 100644 --- a/content/signup-page.js +++ b/content/signup-page.js @@ -103,6 +103,8 @@ const VERIFICATION_CODE_INPUT_SELECTOR = [ const ONE_TIME_CODE_LOGIN_PATTERN = /使用一次性验证码登录|改用(?:一次性)?验证码(?:登录)?|使用验证码登录|一次性验证码|验证码登录|one[-\s]*time\s*(?:passcode|password|code)|use\s+(?:a\s+)?one[-\s]*time\s*(?:passcode|password|code)(?:\s+instead)?|use\s+(?:a\s+)?code(?:\s+instead)?|sign\s+in\s+with\s+(?:email|code)|email\s+(?:me\s+)?(?:a\s+)?code/i; const RESEND_VERIFICATION_CODE_PATTERN = /重新发送(?:验证码)?|再次发送(?:验证码)?|重发(?:验证码)?|未收到(?:验证码|邮件)|resend(?:\s+code)?|send\s+(?:a\s+)?new\s+code|send\s+(?:it\s+)?again|request\s+(?:a\s+)?new\s+code|didn'?t\s+receive/i; +const CHATGPT_ONBOARDING_TEXT_PATTERN = /welcome\s+to\s+chatgpt|what\s+should\s+chatgpt\s+call\s+you|how\s+do\s+you\s+want\s+chatgpt\s+to\s+respond|tell\s+chatgpt\s+what\s+traits|personal(?:ize|ise)\s+your\s+experience|介绍一下你自己|ChatGPT 应该如何称呼你|你希望 ChatGPT 如何回应/i; +const CHATGPT_HOME_TEXT_PATTERN = /new\s+chat|temporary\s+chat|message\s+chatgpt|send\s+a\s+message|chatgpt\s+can\s+make\s+mistakes|新建聊天|临时聊天|给\s*ChatGPT\s*发消息|ChatGPT\s*可能会犯错/i; function isVisibleElement(el) { if (!el) return false; @@ -779,7 +781,67 @@ function getStep5ErrorText() { } function isChatgptOnboardingPage() { - return /chatgpt\.com/i.test(location.href); + if (!isChatgptUrl()) { + return false; + } + + if (findChatgptSkipButton()) { + return true; + } + + return CHATGPT_ONBOARDING_TEXT_PATTERN.test(getPageTextSnapshot()); +} + +function isChatgptUrl(rawUrl = location.href) { + return /(^https?:\/\/)?([^.]+\.)?chatgpt\.com(?::\d+)?/i.test(rawUrl || ''); +} + +function hasVisibleElementMatchingSelector(selector) { + return Array.from(document.querySelectorAll(selector)).some(isVisibleElement); +} + +function isChatgptAuthenticatedHomePage() { + if (!isChatgptUrl()) { + return false; + } + + if (isChatgptOnboardingPage()) { + return false; + } + + const appShellSelectors = [ + 'textarea[placeholder*="Message" i]', + 'textarea[aria-label*="Message" i]', + '[data-testid="composer-root"]', + '[data-testid="history-and-skills"]', + 'button[aria-label*="New chat" i]', + 'a[aria-label*="New chat" i]', + 'a[href^="/c/"]', + ]; + if (appShellSelectors.some((selector) => hasVisibleElementMatchingSelector(selector))) { + return true; + } + + return CHATGPT_HOME_TEXT_PATTERN.test(getPageTextSnapshot()); +} + +async function waitForChatgptPostSignupState(timeout = 15000) { + const start = Date.now(); + while (Date.now() - start < timeout) { + throwIfStopped(); + + if (isChatgptOnboardingPage()) { + return 'onboarding'; + } + + if (isChatgptAuthenticatedHomePage()) { + return 'home'; + } + + await sleep(300); + } + + return null; } async function waitForStep5SubmitOutcome(timeout = 15000) { @@ -801,6 +863,10 @@ async function waitForStep5SubmitOutcome(timeout = 15000) { return { success: true, chatgptOnboarding: true }; } + if (isChatgptAuthenticatedHomePage()) { + return { success: true, chatgptHome: true }; + } + if (isStep8Ready()) { return { success: true }; } @@ -2155,10 +2221,20 @@ async function step5_fillNameBirthday(payload) { throw new Error(`步骤 5:${outcome.errorText}`); } - if (outcome.chatgptOnboarding) { - log(`步骤 5:资料已通过,页面已跳转到 ChatGPT 引导页。`, 'ok'); - reportComplete(5, { chatgptOnboarding: true }); - return { chatgptOnboarding: true }; + if (outcome.chatgptOnboarding || outcome.chatgptHome) { + if (outcome.chatgptOnboarding) { + log('步骤 5:资料已通过,页面已跳转到 ChatGPT 引导页。', 'ok'); + } else { + log('步骤 5:资料已通过,页面已进入已登录的 ChatGPT 页面。', 'ok'); + } + reportComplete(5, { + chatgptOnboarding: Boolean(outcome.chatgptOnboarding), + chatgptHome: Boolean(outcome.chatgptHome), + }); + return { + chatgptOnboarding: Boolean(outcome.chatgptOnboarding), + chatgptHome: Boolean(outcome.chatgptHome), + }; } log(`步骤 5:资料已通过。`, 'ok'); @@ -2173,6 +2249,7 @@ function findChatgptSkipButton() { // Look for buttons containing "Skip" or "跳过" text with btn-ghost class const buttons = document.querySelectorAll('button'); for (const btn of buttons) { + if (!isVisibleElement(btn)) continue; const text = (btn.textContent || '').trim(); if (/^(skip|跳过)$/i.test(text) && /btn-ghost/i.test(btn.className)) { return btn; @@ -2180,6 +2257,7 @@ function findChatgptSkipButton() { } // Fallback: look for any button matching skip text without class constraint for (const btn of buttons) { + if (!isVisibleElement(btn)) continue; const text = (btn.textContent || '').trim(); if (/^(skip|跳过)$/i.test(text)) { return btn; @@ -2202,6 +2280,18 @@ async function waitForChatgptSkipButton(timeout = 15000) { } async function skipChatgptOnboarding() { + log('ChatGPT 页面:正在确认是引导页还是已登录主页...'); + + const initialState = await waitForChatgptPostSignupState(15000); + if (initialState === 'home') { + log('ChatGPT 页面:未出现可跳过引导,已进入登录后的 ChatGPT 页面,按注册成功处理。', 'ok'); + return { success: true, alreadyCompleted: true }; + } + + if (initialState !== 'onboarding') { + throw new Error('ChatGPT 页面:未检测到可跳过引导,也未确认进入登录后的主页。URL: ' + location.href); + } + log('ChatGPT 引导页:正在查找第一个"Skip/跳过"按钮...'); // Find first Skip button: sibling of "Next" button, with btn-ghost class @@ -2220,7 +2310,11 @@ async function skipChatgptOnboarding() { // Find second Skip button: sibling of "Continue" button, with btn-ghost class const secondSkipBtn = await waitForChatgptSkipButton(5000); if (!secondSkipBtn) { - log('ChatGPT 引导页:5秒内未找到第二个"Skip/跳过"按钮,判定注册已完成。', 'ok'); + if (isChatgptAuthenticatedHomePage()) { + log('ChatGPT 引导页:未找到第二个"Skip/跳过"按钮,但页面已进入登录后的 ChatGPT 页面。', 'ok'); + } else { + log('ChatGPT 引导页:5秒内未找到第二个"Skip/跳过"按钮,判定注册已完成。', 'ok'); + } return { success: true }; } diff --git a/tests/step5-chatgpt-onboarding.test.js b/tests/step5-chatgpt-onboarding.test.js new file mode 100644 index 0000000..5bc8d57 --- /dev/null +++ b/tests/step5-chatgpt-onboarding.test.js @@ -0,0 +1,172 @@ +const assert = require('assert'); +const fs = require('fs'); + +const source = fs.readFileSync('content/signup-page.js', 'utf8'); + +function extractFunction(name) { + const markers = [`async function ${name}(`, `function ${name}(`]; + const start = markers + .map((marker) => source.indexOf(marker)) + .find((index) => index >= 0); + if (start < 0) { + throw new Error(`missing function ${name}`); + } + + let parenDepth = 0; + let signatureEnded = false; + let braceStart = -1; + for (let i = start; i < source.length; i += 1) { + const ch = source[i]; + if (ch === '(') { + parenDepth += 1; + } else if (ch === ')') { + parenDepth -= 1; + if (parenDepth === 0) { + signatureEnded = true; + } + } else if (ch === '{' && signatureEnded) { + braceStart = i; + break; + } + } + if (braceStart < 0) { + throw new Error(`missing body for function ${name}`); + } + + let depth = 0; + let end = braceStart; + for (; end < source.length; end += 1) { + const ch = source[end]; + if (ch === '{') depth += 1; + if (ch === '}') { + depth -= 1; + if (depth === 0) { + end += 1; + break; + } + } + } + + return source.slice(start, end); +} + +const bundle = [ + extractFunction('getPageTextSnapshot'), + extractFunction('findChatgptSkipButton'), + extractFunction('waitForChatgptSkipButton'), + extractFunction('isChatgptOnboardingPage'), + extractFunction('isChatgptUrl'), + extractFunction('hasVisibleElementMatchingSelector'), + extractFunction('isChatgptAuthenticatedHomePage'), + extractFunction('waitForChatgptPostSignupState'), + extractFunction('skipChatgptOnboarding'), +].join('\n'); + +const api = new Function(` +const CHATGPT_ONBOARDING_TEXT_PATTERN = /welcome\\s+to\\s+chatgpt|what\\s+should\\s+chatgpt\\s+call\\s+you|how\\s+do\\s+you\\s+want\\s+chatgpt\\s+to\\s+respond|tell\\s+chatgpt\\s+what\\s+traits|personal(?:ize|ise)\\s+your\\s+experience|介绍一下你自己|ChatGPT 应该如何称呼你|你希望 ChatGPT 如何回应/i; +const CHATGPT_HOME_TEXT_PATTERN = /new\\s+chat|temporary\\s+chat|message\\s+chatgpt|send\\s+a\\s+message|chatgpt\\s+can\\s+make\\s+mistakes|新建聊天|临时聊天|给\\s*ChatGPT\\s*发消息|ChatGPT\\s*可能会犯错/i; + +let buttonList = []; +let selectorMap = {}; +let pageText = ''; +let clickedButtons = []; +let logs = []; +const location = { href: 'https://chatgpt.com/' }; +const document = { + body: { innerText: '', textContent: '' }, + querySelectorAll(selector) { + if (selector === 'button') { + return buttonList; + } + return selectorMap[selector] || []; + }, +}; + +function isVisibleElement(el) { + return Boolean(el) && !el.hidden; +} + +function throwIfStopped() {} +async function sleep() {} +async function humanPause() {} +function log(message, level = 'info') { + logs.push({ message, level }); +} +function simulateClick(button) { + clickedButtons.push(button.textContent || button.id || 'button'); + button.hidden = true; +} + +${bundle} + +return { + setPage({ href = 'https://chatgpt.com/', text = '', buttons = [], selectors = {} }) { + location.href = href; + pageText = text; + buttonList = buttons.map((button, index) => ({ + id: button.id || \`button-\${index}\`, + textContent: button.textContent || '', + className: button.className || '', + hidden: Boolean(button.hidden), + })); + selectorMap = {}; + for (const [selector, count] of Object.entries(selectors)) { + selectorMap[selector] = Array.from({ length: count }, (_, index) => ({ id: \`\${selector}-\${index}\`, hidden: false })); + } + document.body.innerText = pageText; + document.body.textContent = pageText; + clickedButtons = []; + logs = []; + }, + isChatgptOnboardingPage() { + return isChatgptOnboardingPage(); + }, + isChatgptAuthenticatedHomePage() { + return isChatgptAuthenticatedHomePage(); + }, + async skipChatgptOnboarding() { + return skipChatgptOnboarding(); + }, + snapshot() { + return { clickedButtons, logs }; + }, +}; +`)(); + +(async () => { + api.setPage({ + href: 'https://chatgpt.com/', + text: 'New chat ChatGPT can make mistakes', + selectors: { + 'textarea[placeholder*="Message" i]': 1, + }, + }); + + assert.strictEqual(api.isChatgptOnboardingPage(), false, '已登录主页不应仅因 chatgpt.com URL 被误判为 onboarding'); + assert.strictEqual(api.isChatgptAuthenticatedHomePage(), true, '主页特征存在时应识别为已登录 ChatGPT 页面'); + + let result = await api.skipChatgptOnboarding(); + let snapshot = api.snapshot(); + assert.deepStrictEqual(result, { success: true, alreadyCompleted: true }, '无 Skip 按钮但已进入主页时应按成功处理'); + assert.deepStrictEqual(snapshot.clickedButtons, [], '已登录主页场景不应再误点按钮'); + + api.setPage({ + href: 'https://chatgpt.com/', + text: 'Welcome to ChatGPT', + buttons: [ + { textContent: 'Skip', className: 'btn-ghost' }, + { textContent: 'Skip', className: 'btn-ghost' }, + ], + }); + + assert.strictEqual(api.isChatgptOnboardingPage(), true, '存在 Skip 按钮时应识别为 onboarding'); + result = await api.skipChatgptOnboarding(); + snapshot = api.snapshot(); + assert.deepStrictEqual(result, { success: true }, '真实 onboarding 仍应继续执行跳过逻辑'); + assert.deepStrictEqual(snapshot.clickedButtons, ['Skip', 'Skip'], '双 Skip onboarding 应依次点击两个按钮'); + + console.log('step5 chatgpt onboarding tests passed'); +})().catch((error) => { + console.error(error); + process.exit(1); +}); diff --git a/tests/step5-chatgpt-redirect-race.test.js b/tests/step5-chatgpt-redirect-race.test.js new file mode 100644 index 0000000..797b39c --- /dev/null +++ b/tests/step5-chatgpt-redirect-race.test.js @@ -0,0 +1,123 @@ +const assert = require('assert'); +const fs = require('fs'); + +const source = fs.readFileSync('background.js', 'utf8'); + +function extractFunction(name) { + const markers = [`async function ${name}(`, `function ${name}(`]; + const start = markers + .map((marker) => source.indexOf(marker)) + .find((index) => index >= 0); + if (start < 0) { + throw new Error(`missing function ${name}`); + } + + let parenDepth = 0; + let signatureEnded = false; + let braceStart = -1; + for (let i = start; i < source.length; i += 1) { + const ch = source[i]; + if (ch === '(') { + parenDepth += 1; + } else if (ch === ')') { + parenDepth -= 1; + if (parenDepth === 0) { + signatureEnded = true; + } + } else if (ch === '{' && signatureEnded) { + braceStart = i; + break; + } + } + if (braceStart < 0) { + throw new Error(`missing body for function ${name}`); + } + + let depth = 0; + let end = braceStart; + for (; end < source.length; end += 1) { + const ch = source[end]; + if (ch === '{') depth += 1; + if (ch === '}') { + depth -= 1; + if (depth === 0) { + end += 1; + break; + } + } + } + + return source.slice(start, end); +} + +const bundle = [ + extractFunction('waitForStep5ChatgptRedirect'), +].join('\n'); + +const api = new Function(` +let waitArgs = null; +let waitResult = null; +let currentTab = null; + +const chrome = { + tabs: { + async get() { + return currentTab; + }, + }, +}; + +async function waitForTabUrlMatch(tabId, matcher, options = {}) { + waitArgs = { tabId, matcher, options }; + return waitResult; +} + +${bundle} + +return { + async run(tabId, timeoutMs) { + return waitForStep5ChatgptRedirect(tabId, timeoutMs); + }, + setWaitResult(value) { + waitResult = value; + }, + setCurrentTab(value) { + currentTab = value; + }, + snapshot() { + return { waitArgs }; + }, +}; +`)(); + +(async () => { + const redirected = { id: 86, url: 'https://chatgpt.com/' }; + api.setWaitResult(redirected); + api.setCurrentTab({ id: 86, url: 'https://auth.openai.com/' }); + + let result = await api.run(86, 22000); + let snapshot = api.snapshot(); + assert.deepStrictEqual(result, redirected, '等待命中 chatgpt.com 时应直接返回匹配到的标签页'); + assert.strictEqual(snapshot.waitArgs.tabId, 86, '应使用 signup-page 当前标签页等待跳转'); + assert.strictEqual(snapshot.waitArgs.options.timeoutMs, 22000, '应透传等待超时时间'); + assert.strictEqual(snapshot.waitArgs.options.retryDelayMs, 300, '应使用较短轮询间隔覆盖 URL 更新 race'); + assert.strictEqual(snapshot.waitArgs.matcher('https://chatgpt.com/?model=gpt-5'), true, 'matcher 应接受 chatgpt.com'); + assert.strictEqual(snapshot.waitArgs.matcher('https://auth.openai.com/u/signup'), false, 'matcher 不应把旧认证页误判为成功'); + + api.setWaitResult(null); + api.setCurrentTab({ id: 86, url: 'https://chatgpt.com/?temporary-chat=true' }); + result = await api.run(86, 15000); + assert.deepStrictEqual(result, { id: 86, url: 'https://chatgpt.com/?temporary-chat=true' }, '等待超时后仍应回读当前标签页 URL 兜底'); + + api.setCurrentTab({ id: 86, url: 'https://auth.openai.com/u/signup' }); + result = await api.run(86, 15000); + assert.strictEqual(result, null, '仍停留旧域名时不应误判为跳转成功'); + + result = await api.run(null, 15000); + assert.strictEqual(result, null, '无有效标签页时应直接返回 null'); + + console.log('step5 chatgpt redirect race tests passed'); +})().catch((error) => { + console.error(error); + process.exit(1); +});