merge: sync dev and finalize sub2api localhost callback flow

This commit is contained in:
hisen666
2026-04-12 16:30:28 +08:00
11 changed files with 879 additions and 85 deletions
+148 -62
View File
@@ -26,7 +26,6 @@ const PERSISTED_SETTING_DEFAULTS = {
sub2apiEmail: '', // SUB2API 登录邮箱。
sub2apiPassword: '', // SUB2API 登录密码。
sub2apiGroupName: DEFAULT_SUB2API_GROUP_NAME, // SUB2API 创建账号时绑定的分组名。
sub2apiRedirectUri: DEFAULT_SUB2API_REDIRECT_URI, // SUB2API OpenAI Auth 回调地址。
customPassword: '', // 自定义账号密码;留空时由程序自动生成随机密码。
autoRunSkipFailures: false, // 自动运行遇到失败步骤后,是否继续执行后续流程。
mailProvider: '163', // 验证码邮箱来源,当前支持 163 / inbucket。
@@ -243,19 +242,6 @@ function normalizeSub2ApiUrl(rawUrl) {
return parsed.toString();
}
function normalizeSub2ApiRedirectUri(rawUrl) {
const input = (rawUrl || '').trim() || DEFAULT_SUB2API_REDIRECT_URI;
const withProtocol = /^https?:\/\//i.test(input) ? input : `http://${input}`;
const parsed = new URL(withProtocol);
if (!parsed.pathname || parsed.pathname === '/') {
parsed.pathname = '/auth/callback';
}
if (parsed.pathname !== '/auth/callback') {
throw new Error('SUB2API 回调地址必须是 /auth/callback,例如 http://localhost:1455/auth/callback');
}
return parsed.toString();
}
function getPanelMode(state = {}) {
return state.panelMode === 'sub2api' ? 'sub2api' : 'cpa';
}
@@ -280,18 +266,22 @@ function isLocalhostOAuthCallbackUrl(rawUrl) {
if (!parsed) return false;
if (!['http:', 'https:'].includes(parsed.protocol)) return false;
if (!['localhost', '127.0.0.1'].includes(parsed.hostname)) return false;
if (parsed.pathname !== '/auth/callback') return false;
if (!['/auth/callback', '/codex/callback'].includes(parsed.pathname)) return false;
const code = (parsed.searchParams.get('code') || '').trim();
const state = (parsed.searchParams.get('state') || '').trim();
return Boolean(code && state);
}
function buildLocalhostCleanupPrefix(rawUrl) {
if (!isLocalhostOAuthCallbackUrl(rawUrl)) return '';
function isLocalCpaUrl(rawUrl) {
const parsed = parseUrlSafely(rawUrl);
return parsed ? `${parsed.origin}/auth` : '';
if (!parsed) return false;
if (!['http:', 'https:'].includes(parsed.protocol)) return false;
return ['localhost', '127.0.0.1'].includes(parsed.hostname);
}
function shouldBypassStep9ForLocalCpa(state) {
return Boolean(state?.localhostUrl) && isLocalCpaUrl(state?.vpsUrl);
}
function matchesSourceUrlFamily(source, candidateUrl, referenceUrl) {
@@ -366,21 +356,43 @@ async function closeConflictingTabsForSource(source, currentUrl, options = {}) {
await addLog(`已关闭 ${matchedIds.length} 个旧的${getSourceLabel(source)}标签页。`, 'info');
}
async function closeTabsByUrlPrefix(prefix, options = {}) {
if (!prefix) return 0;
function isLocalhostOAuthCallbackTabMatch(callbackUrl, candidateUrl) {
if (!isLocalhostOAuthCallbackUrl(callbackUrl) || !isLocalhostOAuthCallbackUrl(candidateUrl)) {
return false;
}
const callback = parseUrlSafely(callbackUrl);
const candidate = parseUrlSafely(candidateUrl);
if (!callback || !candidate) return false;
return callback.origin === candidate.origin
&& callback.pathname === candidate.pathname
&& callback.searchParams.get('code') === candidate.searchParams.get('code')
&& callback.searchParams.get('state') === candidate.searchParams.get('state');
}
async function closeLocalhostCallbackTabs(callbackUrl, options = {}) {
if (!isLocalhostOAuthCallbackUrl(callbackUrl)) return 0;
const { excludeTabIds = [] } = options;
const excluded = new Set(excludeTabIds.filter(id => Number.isInteger(id)));
const tabs = await chrome.tabs.query({});
const matchedIds = tabs
.filter((tab) => Number.isInteger(tab.id) && !excluded.has(tab.id))
.filter((tab) => typeof tab.url === 'string' && tab.url.startsWith(prefix))
.filter((tab) => isLocalhostOAuthCallbackTabMatch(callbackUrl, tab.url))
.map((tab) => tab.id);
if (!matchedIds.length) return 0;
await chrome.tabs.remove(matchedIds).catch(() => { });
await addLog(`已关闭 ${matchedIds.length} 个匹配 ${prefix} 的 localhost 残留标签页。`, 'info');
const registry = await getTabRegistry();
if (registry['signup-page']?.tabId && matchedIds.includes(registry['signup-page'].tabId)) {
registry['signup-page'] = null;
await setState({ tabRegistry: registry });
}
await addLog(`已关闭 ${matchedIds.length} 个匹配当前 OAuth callback 的 localhost 残留标签页。`, 'info');
return matchedIds.length;
}
@@ -925,6 +937,26 @@ async function addLog(message, level = 'info') {
chrome.runtime.sendMessage({ type: 'LOG_ENTRY', payload: entry }).catch(() => { });
}
function getStep8CallbackUrlFromNavigation(details, signupTabId) {
if (!Number.isInteger(signupTabId) || !details) return '';
if (details.tabId !== signupTabId) return '';
if (details.frameId !== 0) return '';
return isLocalhostOAuthCallbackUrl(details.url) ? details.url : '';
}
function getStep8CallbackUrlFromTabUpdate(tabId, changeInfo, tab, signupTabId) {
if (!Number.isInteger(signupTabId) || tabId !== signupTabId) return '';
const candidates = [changeInfo?.url, tab?.url];
for (const candidate of candidates) {
if (isLocalhostOAuthCallbackUrl(candidate)) {
return candidate;
}
}
return '';
}
function getSourceLabel(source) {
const labels = {
'sidepanel': '侧边栏',
@@ -1387,7 +1419,6 @@ async function handleMessage(message, sender) {
if (message.payload.sub2apiEmail !== undefined) updates.sub2apiEmail = message.payload.sub2apiEmail;
if (message.payload.sub2apiPassword !== undefined) updates.sub2apiPassword = message.payload.sub2apiPassword;
if (message.payload.sub2apiGroupName !== undefined) updates.sub2apiGroupName = message.payload.sub2apiGroupName;
if (message.payload.sub2apiRedirectUri !== undefined) updates.sub2apiRedirectUri = message.payload.sub2apiRedirectUri;
if (message.payload.customPassword !== undefined) updates.customPassword = message.payload.customPassword;
if (message.payload.autoRunSkipFailures !== undefined) updates.autoRunSkipFailures = Boolean(message.payload.autoRunSkipFailures);
if (message.payload.mailProvider !== undefined) updates.mailProvider = message.payload.mailProvider;
@@ -1468,9 +1499,8 @@ async function handleStepData(step, payload) {
}
break;
case 9: {
const localhostPrefix = buildLocalhostCleanupPrefix(payload.localhostUrl);
if (localhostPrefix) {
await closeTabsByUrlPrefix(localhostPrefix);
if (payload.localhostUrl) {
await closeLocalhostCallbackTabs(payload.localhostUrl);
}
break;
}
@@ -1549,10 +1579,8 @@ async function requestStop(options = {}) {
stopRequested = true;
cancelPendingCommands();
if (webNavListener) {
chrome.webNavigation.onBeforeNavigate.removeListener(webNavListener);
webNavListener = null;
}
cleanupStep8NavigationListeners();
rejectPendingStep8(new Error(STOP_ERROR_MESSAGE));
await addLog(logMessage, 'warn');
await broadcastStopToContentScripts();
@@ -2123,7 +2151,6 @@ async function executeCpaStep1(state) {
async function executeSub2ApiStep1(state) {
const sub2apiUrl = normalizeSub2ApiUrl(state.sub2apiUrl);
const sub2apiRedirectUri = normalizeSub2ApiRedirectUri(state.sub2apiRedirectUri);
const groupName = (state.sub2apiGroupName || DEFAULT_SUB2API_GROUP_NAME).trim() || DEFAULT_SUB2API_GROUP_NAME;
if (!state.sub2apiEmail) {
@@ -2169,7 +2196,6 @@ async function executeSub2ApiStep1(state) {
sub2apiEmail: state.sub2apiEmail,
sub2apiPassword: state.sub2apiPassword,
sub2apiGroupName: groupName,
sub2apiRedirectUri,
},
}, {
timeoutMs: 40000,
@@ -2696,6 +2722,9 @@ async function executeStep7(state) {
// ============================================================
let webNavListener = null;
let webNavCommittedListener = null;
let step8TabUpdatedListener = null;
let step8PendingReject = null;
const STEP8_CLICK_EFFECT_TIMEOUT_MS = 3500;
const STEP8_CLICK_RETRY_DELAY_MS = 500;
const STEP8_READY_WAIT_TIMEOUT_MS = 30000;
@@ -2707,6 +2736,34 @@ const STEP8_STRATEGIES = [
{ mode: 'debugger', label: 'debugger click retry' },
];
function cleanupStep8NavigationListeners() {
if (webNavListener) {
chrome.webNavigation.onBeforeNavigate.removeListener(webNavListener);
webNavListener = null;
}
if (webNavCommittedListener) {
chrome.webNavigation.onCommitted.removeListener(webNavCommittedListener);
webNavCommittedListener = null;
}
if (step8TabUpdatedListener) {
chrome.tabs.onUpdated.removeListener(step8TabUpdatedListener);
step8TabUpdatedListener = null;
}
}
function rejectPendingStep8(error) {
if (!step8PendingReject) return;
const reject = step8PendingReject;
step8PendingReject = null;
reject(error);
}
function throwIfStep8SettledOrStopped(isSettled = false) {
if (isSettled || stopRequested) {
throw new Error(STOP_ERROR_MESSAGE);
}
}
async function getStep8PageState(tabId, responseTimeoutMs = 1500) {
try {
const result = await sendTabMessageWithTimeout(tabId, 'signup-page', {
@@ -2841,13 +2898,11 @@ async function executeStep8(state) {
let signupTabId = null;
const cleanupListener = () => {
if (webNavListener) {
chrome.webNavigation.onBeforeNavigate.removeListener(webNavListener);
webNavListener = null;
}
cleanupStep8NavigationListeners();
step8PendingReject = null;
};
const failStep8 = (error) => {
const rejectStep8 = (error) => {
if (resolved) return;
resolved = true;
clearTimeout(timeout);
@@ -2855,44 +2910,67 @@ async function executeStep8(state) {
reject(error);
};
const finalizeStep8Callback = (callbackUrl) => {
if (resolved || !callbackUrl) return;
resolved = true;
cleanupListener();
clearTimeout(timeout);
addLog(`步骤 8:已捕获 localhost 地址:${callbackUrl}`, 'ok').then(() => {
return completeStepFromBackground(8, { localhostUrl: callbackUrl });
}).then(() => {
resolve();
}).catch((err) => {
reject(err);
});
};
const timeout = setTimeout(() => {
failStep8(new Error('120 秒内未捕获到 localhost 回调跳转,步骤 8 已持续重试点击“继续”但页面仍未完成授权。'));
rejectStep8(new Error('120 秒内未捕获到 localhost 回调跳转,步骤 8 的点击可能被拦截了。'));
}, 120000);
step8PendingReject = (error) => {
rejectStep8(error);
};
webNavListener = (details) => {
if (resolved || !signupTabId) return;
if (details.tabId !== signupTabId) return;
if (details.frameId !== 0) return;
if (isLocalhostOAuthCallbackUrl(details.url)) {
console.log(LOG_PREFIX, `已捕获 localhost OAuth 回调:${details.url}`);
resolved = true;
cleanupListener();
clearTimeout(timeout);
addLog(`步骤 8:已捕获 localhost 地址:${details.url}`, 'ok').then(() => {
return completeStepFromBackground(8, { localhostUrl: details.url });
}).then(() => {
resolve();
}).catch((err) => {
reject(err);
});
}
const callbackUrl = getStep8CallbackUrlFromNavigation(details, signupTabId);
finalizeStep8Callback(callbackUrl);
};
webNavCommittedListener = (details) => {
const callbackUrl = getStep8CallbackUrlFromNavigation(details, signupTabId);
finalizeStep8Callback(callbackUrl);
};
step8TabUpdatedListener = (tabId, changeInfo, tab) => {
const callbackUrl = getStep8CallbackUrlFromTabUpdate(tabId, changeInfo, tab, signupTabId);
finalizeStep8Callback(callbackUrl);
};
(async () => {
try {
throwIfStep8SettledOrStopped(resolved);
signupTabId = await getTabId('signup-page');
if (signupTabId) {
throwIfStep8SettledOrStopped(resolved);
if (signupTabId && await isTabAlive('signup-page')) {
await chrome.tabs.update(signupTabId, { active: true });
await addLog('步骤 8:已切回认证页,准备循环确认“继续”按钮直到页面真正跳转...');
await addLog('步骤 8:已切回认证页,正在准备调试器点击...');
} else {
signupTabId = await reuseOrCreateTab('signup-page', state.oauthUrl);
await addLog('步骤 8:已重新打开认证页,准备循环确认“继续”按钮直到页面真正跳转...');
await addLog('步骤 8:已重新打开认证页,正在准备调试器点击...');
}
throwIfStep8SettledOrStopped(resolved);
chrome.webNavigation.onBeforeNavigate.addListener(webNavListener);
chrome.webNavigation.onCommitted.addListener(webNavCommittedListener);
chrome.tabs.onUpdated.addListener(step8TabUpdatedListener);
let attempt = 0;
while (!resolved) {
throwIfStep8SettledOrStopped(resolved);
const pageState = await waitForStep8Ready(signupTabId);
if (!pageState?.consentReady) {
await sleepWithStop(STEP8_CLICK_RETRY_DELAY_MS);
@@ -2907,9 +2985,8 @@ async function executeStep8(state) {
if (strategy.mode === 'debugger') {
const clickTarget = await prepareStep8DebuggerClick();
if (!resolved) {
await clickWithDebugger(signupTabId, clickTarget?.rect);
}
throwIfStep8SettledOrStopped(resolved);
await clickWithDebugger(signupTabId, clickTarget?.rect);
} else {
await triggerStep8ContentStrategy(strategy.strategy);
}
@@ -2932,7 +3009,7 @@ async function executeStep8(state) {
await sleepWithStop(STEP8_CLICK_RETRY_DELAY_MS);
}
} catch (err) {
failStep8(err);
rejectStep8(err);
}
})();
});
@@ -2960,6 +3037,15 @@ async function executeCpaStep9(state) {
throw new Error('尚未填写 CPA 地址,请先在侧边栏输入。');
}
if (shouldBypassStep9ForLocalCpa(state)) {
await addLog('步骤 9:检测到本地 CPA,步骤 8 完成后已自动添加,无需重复提交回调地址。', 'info');
await completeStepFromBackground(9, {
localhostUrl: state.localhostUrl,
verifiedStatus: 'local-auto',
});
return;
}
await addLog('步骤 9:正在打开 CPA 面板...');
const injectFiles = ['content/utils.js', 'content/vps-panel.js'];
+3 -3
View File
@@ -42,8 +42,8 @@ function getSub2ApiOrigin(payload = {}) {
}
}
function normalizeRedirectUri(rawUrl) {
const input = (rawUrl || '').trim() || SUB2API_DEFAULT_REDIRECT_URI;
function normalizeRedirectUri() {
const input = SUB2API_DEFAULT_REDIRECT_URI;
const withProtocol = /^https?:\/\//i.test(input) ? input : `http://${input}`;
const parsed = new URL(withProtocol);
if (!parsed.pathname || parsed.pathname === '/') {
@@ -288,7 +288,7 @@ function openAccountsPageSoon(origin) {
}
async function step1_generateOpenAiAuthUrl(payload = {}) {
const redirectUri = normalizeRedirectUri(payload.sub2apiRedirectUri);
const redirectUri = normalizeRedirectUri();
const groupName = (payload.sub2apiGroupName || SUB2API_DEFAULT_GROUP_NAME).trim() || SUB2API_DEFAULT_GROUP_NAME;
const { origin, token } = await loginSub2Api(payload);
+1 -1
View File
@@ -169,7 +169,7 @@ function isLocalhostOAuthCallbackUrl(rawUrl) {
if (!parsed) return false;
if (!['http:', 'https:'].includes(parsed.protocol)) return false;
if (!['localhost', '127.0.0.1'].includes(parsed.hostname)) return false;
if (parsed.pathname !== '/auth/callback') return false;
if (!['/auth/callback', '/codex/callback'].includes(parsed.pathname)) return false;
const code = (parsed.searchParams.get('code') || '').trim();
const state = (parsed.searchParams.get('state') || '').trim();
-4
View File
@@ -78,10 +78,6 @@
<span class="data-label">分组</span>
<input type="text" id="input-sub2api-group" class="data-input" placeholder="默认 codex" />
</div>
<div class="data-row" id="row-sub2api-redirect-uri" style="display:none;">
<span class="data-label">回调</span>
<input type="text" id="input-sub2api-redirect-uri" class="data-input" placeholder="默认 http://localhost:1455/auth/callback" />
</div>
<div class="data-row">
<span class="data-label">邮箱服务</span>
<select id="select-mail-provider" class="data-select">
-15
View File
@@ -41,8 +41,6 @@ const rowSub2ApiPassword = document.getElementById('row-sub2api-password');
const inputSub2ApiPassword = document.getElementById('input-sub2api-password');
const rowSub2ApiGroup = document.getElementById('row-sub2api-group');
const inputSub2ApiGroup = document.getElementById('input-sub2api-group');
const rowSub2ApiRedirectUri = document.getElementById('row-sub2api-redirect-uri');
const inputSub2ApiRedirectUri = document.getElementById('input-sub2api-redirect-uri');
const selectMailProvider = document.getElementById('select-mail-provider');
const rowInbucketHost = document.getElementById('row-inbucket-host');
const inputInbucketHost = document.getElementById('input-inbucket-host');
@@ -296,7 +294,6 @@ function collectSettingsPayload() {
sub2apiEmail: inputSub2ApiEmail.value.trim(),
sub2apiPassword: inputSub2ApiPassword.value,
sub2apiGroupName: inputSub2ApiGroup.value.trim(),
sub2apiRedirectUri: inputSub2ApiRedirectUri.value.trim(),
customPassword: inputPassword.value,
mailProvider: selectMailProvider.value,
inbucketHost: inputInbucketHost.value.trim(),
@@ -476,9 +473,6 @@ async function restoreState() {
if (state.sub2apiGroupName) {
inputSub2ApiGroup.value = state.sub2apiGroupName;
}
if (state.sub2apiRedirectUri) {
inputSub2ApiRedirectUri.value = state.sub2apiRedirectUri;
}
if (state.mailProvider) {
selectMailProvider.value = state.mailProvider;
}
@@ -532,7 +526,6 @@ function updatePanelModeUI() {
rowSub2ApiEmail.style.display = useSub2Api ? '' : 'none';
rowSub2ApiPassword.style.display = useSub2Api ? '' : 'none';
rowSub2ApiGroup.style.display = useSub2Api ? '' : 'none';
rowSub2ApiRedirectUri.style.display = useSub2Api ? '' : 'none';
const step9Btn = document.querySelector('.step-btn[data-step="9"]');
if (step9Btn) {
@@ -1045,14 +1038,6 @@ inputSub2ApiGroup.addEventListener('blur', () => {
saveSettings({ silent: true }).catch(() => { });
});
inputSub2ApiRedirectUri.addEventListener('input', () => {
markSettingsDirty(true);
scheduleSettingsAutoSave();
});
inputSub2ApiRedirectUri.addEventListener('blur', () => {
saveSettings({ silent: true }).catch(() => { });
});
inputInbucketMailbox.addEventListener('input', () => {
markSettingsDirty(true);
scheduleSettingsAutoSave();
+104
View File
@@ -0,0 +1,104 @@
const assert = require('assert');
const fs = require('fs');
const source = fs.readFileSync('background.js', 'utf8');
function extractFunction(name) {
const start = source.indexOf(`function ${name}(`);
if (start < 0) {
throw new Error(`missing function ${name}`);
}
const braceStart = source.indexOf('{', start);
let depth = 0;
let end = braceStart;
for (; end < source.length; end++) {
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('parseUrlSafely'),
extractFunction('isLocalhostOAuthCallbackUrl'),
extractFunction('getStep8CallbackUrlFromNavigation'),
extractFunction('getStep8CallbackUrlFromTabUpdate'),
].join('\n');
const api = new Function(`${bundle}; return { getStep8CallbackUrlFromNavigation, getStep8CallbackUrlFromTabUpdate };`)();
const callbackUrl = 'http://127.0.0.1:8317/codex/callback?code=abc&state=xyz';
assert.strictEqual(
api.getStep8CallbackUrlFromNavigation({
tabId: 123,
frameId: 0,
url: callbackUrl,
}, 123),
callbackUrl,
'应识别 onCommitted/onBeforeNavigate 命中的 callback'
);
assert.strictEqual(
api.getStep8CallbackUrlFromNavigation({
tabId: 123,
frameId: 1,
url: callbackUrl,
}, 123),
'',
'子 frame 不应命中 callback'
);
assert.strictEqual(
api.getStep8CallbackUrlFromNavigation({
tabId: 999,
frameId: 0,
url: callbackUrl,
}, 123),
'',
'非 signup tab 不应命中 callback'
);
assert.strictEqual(
api.getStep8CallbackUrlFromTabUpdate(
123,
{ url: callbackUrl },
{ url: callbackUrl },
123
),
callbackUrl,
'tabs.onUpdated 应能从 changeInfo.url 捕获 callback'
);
assert.strictEqual(
api.getStep8CallbackUrlFromTabUpdate(
123,
{},
{ url: callbackUrl },
123
),
callbackUrl,
'tabs.onUpdated 应能从 tab.url 兜底捕获 callback'
);
assert.strictEqual(
api.getStep8CallbackUrlFromTabUpdate(
999,
{ url: callbackUrl },
{ url: callbackUrl },
123
),
'',
'非 signup tab 的 tabs.onUpdated 不应命中 callback'
);
console.log('step8 callback handling tests passed');
+109
View File
@@ -0,0 +1,109 @@
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++) {
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++) {
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('throwIfStopped'),
extractFunction('clickWithDebugger'),
].join('\n');
const api = new Function(`
let stopRequested = false;
const STOP_ERROR_MESSAGE = '流程已被用户停止。';
const commands = [];
let attachCount = 0;
let detachCount = 0;
const chrome = {
debugger: {
async attach(target, version) {
attachCount += 1;
},
async sendCommand(target, command, payload) {
commands.push(command);
if (command === 'Page.bringToFront') {
stopRequested = true;
}
},
async detach(target) {
detachCount += 1;
},
},
};
${bundle}
return {
clickWithDebugger,
snapshot() {
return {
commands,
attachCount,
detachCount,
};
},
};
`)();
(async () => {
const error = await api.clickWithDebugger(123, { centerX: 10, centerY: 20 }).catch((err) => err);
const state = api.snapshot();
assert.strictEqual(error?.message, '流程已被用户停止。', 'debugger 点击过程中收到 Stop 后应终止');
assert.deepStrictEqual(state.commands, ['Page.bringToFront'], 'Stop 后不应继续发送鼠标事件');
assert.strictEqual(state.attachCount, 1, '应先附加 debugger');
assert.strictEqual(state.detachCount, 1, '即使停止也应在 finally 中释放 debugger');
console.log('step8 debugger stop tests passed');
})().catch((error) => {
console.error(error);
process.exit(1);
});
+54
View File
@@ -0,0 +1,54 @@
const assert = require('assert');
const fs = require('fs');
const source = fs.readFileSync('background.js', 'utf8');
function extractFunction(name) {
const start = source.indexOf(`function ${name}(`);
if (start < 0) {
throw new Error(`missing function ${name}`);
}
const braceStart = source.indexOf('{', start);
let depth = 0;
let end = braceStart;
for (; end < source.length; end++) {
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('isRetryableContentScriptTransportError'),
].join('\n');
const api = new Function(`${bundle}; return { isRetryableContentScriptTransportError };`)();
assert.strictEqual(
api.isRetryableContentScriptTransportError(new Error('Content script on signup-page did not respond in 2s. Try refreshing the tab and retry.')),
true,
'Step 8 状态探测短超时应被视为可重试错误'
);
assert.strictEqual(
api.isRetryableContentScriptTransportError(new Error('Content script on signup-page did not respond in 30s. Try refreshing the tab and retry.')),
true,
'普通内容脚本超时也应沿用可重试分支'
);
assert.strictEqual(
api.isRetryableContentScriptTransportError(new Error('按钮不存在')),
false,
'真实业务错误不应被误判为可重试传输错误'
);
console.log('step8 state timeout retry tests passed');
+209
View File
@@ -0,0 +1,209 @@
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++) {
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++) {
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('throwIfStopped'),
extractFunction('cleanupStep8NavigationListeners'),
extractFunction('rejectPendingStep8'),
extractFunction('throwIfStep8SettledOrStopped'),
extractFunction('requestStop'),
extractFunction('executeStep8'),
].join('\n');
const api = new Function(`
let stopRequested = false;
const STOP_ERROR_MESSAGE = '流程已被用户停止。';
let webNavListener = null;
let webNavCommittedListener = null;
let step8TabUpdatedListener = null;
let step8PendingReject = null;
let autoRunActive = true;
let autoRunCurrentRun = 2;
let autoRunTotalRuns = 3;
let autoRunAttemptRun = 4;
const added = {
beforeNavigate: 0,
committed: 0,
tabUpdated: 0,
};
const removed = {
beforeNavigate: 0,
committed: 0,
tabUpdated: 0,
};
const sentMessages = [];
let clickCount = 0;
let resolveTabId = null;
const chrome = {
webNavigation: {
onBeforeNavigate: {
addListener(listener) {
added.beforeNavigate += 1;
},
removeListener(listener) {
removed.beforeNavigate += 1;
},
},
onCommitted: {
addListener(listener) {
added.committed += 1;
},
removeListener(listener) {
removed.committed += 1;
},
},
},
tabs: {
onUpdated: {
addListener(listener) {
added.tabUpdated += 1;
},
removeListener(listener) {
removed.tabUpdated += 1;
},
},
async update() {},
},
};
const stepWaiters = new Map();
let resumeWaiter = null;
function cancelPendingCommands() {}
async function addLog() {}
async function broadcastStopToContentScripts() {}
async function markRunningStepsStopped() {}
async function broadcastAutoRunStatus() {}
function getStep8CallbackUrlFromNavigation() { return ''; }
function getStep8CallbackUrlFromTabUpdate() { return ''; }
async function completeStepFromBackground() {}
async function getTabId() {
return await new Promise((resolve) => {
resolveTabId = resolve;
});
}
async function reuseOrCreateTab() {
return 999;
}
async function isTabAlive() {
return true;
}
async function sendToContentScript(source, message) {
sentMessages.push({ source, type: message.type });
return { rect: { centerX: 10, centerY: 20 } };
}
async function clickWithDebugger() {
clickCount += 1;
}
${bundle}
return {
executeStep8,
requestStop,
resolveTabId(tabId) {
if (!resolveTabId) {
throw new Error('resolveTabId is not ready');
}
resolveTabId(tabId);
},
snapshot() {
return {
stopRequested,
webNavListener,
webNavCommittedListener,
step8TabUpdatedListener,
step8PendingReject,
added,
removed,
sentMessages,
clickCount,
autoRunActive,
};
},
};
`)();
(async () => {
const step8Promise = api.executeStep8({ oauthUrl: 'https://example.com/oauth' });
const settledStep8Promise = step8Promise.catch((err) => err);
await new Promise((resolve) => setImmediate(resolve));
await api.requestStop();
await new Promise((resolve) => setImmediate(resolve));
api.resolveTabId(123);
await new Promise((resolve) => setImmediate(resolve));
await new Promise((resolve) => setImmediate(resolve));
const error = await settledStep8Promise;
const state = api.snapshot();
assert.strictEqual(error?.message, '流程已被用户停止。', 'Stop 后 Step 8 promise 应被拒绝为停止错误');
assert.deepStrictEqual(
state.added,
{ beforeNavigate: 0, committed: 0, tabUpdated: 0 },
'Stop 先发生时,不应再注册 Step 8 监听'
);
assert.strictEqual(state.sentMessages.length, 0, 'Stop 后不应再发送 STEP8_FIND_AND_CLICK 命令');
assert.strictEqual(state.clickCount, 0, 'Stop 后不应再触发 debugger 点击');
assert.strictEqual(state.webNavListener, null, 'Stop 后 onBeforeNavigate 引用应为空');
assert.strictEqual(state.webNavCommittedListener, null, 'Stop 后 onCommitted 引用应为空');
assert.strictEqual(state.step8TabUpdatedListener, null, 'Stop 后 tabs.onUpdated 引用应为空');
assert.strictEqual(state.step8PendingReject, null, 'Stop 后不应保留 Step 8 挂起 reject');
console.log('step8 stop cleanup tests passed');
})().catch((error) => {
console.error(error);
process.exit(1);
});
+58
View File
@@ -0,0 +1,58 @@
const assert = require('assert');
const fs = require('fs');
const source = fs.readFileSync('background.js', 'utf8');
function extractFunction(name) {
const start = source.indexOf(`function ${name}(`);
if (start < 0) {
throw new Error(`missing function ${name}`);
}
const braceStart = source.indexOf('{', start);
let depth = 0;
let end = braceStart;
for (; end < source.length; end++) {
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('parseUrlSafely'),
extractFunction('isLocalCpaUrl'),
extractFunction('shouldBypassStep9ForLocalCpa'),
].join('\n');
const api = new Function(`${bundle}; return { isLocalCpaUrl, shouldBypassStep9ForLocalCpa };`)();
assert.strictEqual(api.isLocalCpaUrl('http://127.0.0.1:8317/management.html#/oauth'), true, '127.0.0.1 应视为本地 CPA');
assert.strictEqual(api.isLocalCpaUrl('http://localhost:1455/management.html#/oauth'), true, 'localhost 应视为本地 CPA');
assert.strictEqual(api.isLocalCpaUrl('https://example.com/management.html#/oauth'), false, '远程域名不应视为本地 CPA');
assert.strictEqual(api.isLocalCpaUrl('notaurl'), false, '非法 URL 不应视为本地 CPA');
assert.strictEqual(api.shouldBypassStep9ForLocalCpa({
vpsUrl: 'http://127.0.0.1:8317/management.html#/oauth',
localhostUrl: 'http://127.0.0.1:8317/codex/callback?code=abc&state=xyz',
}), true, '本地 CPA 且已有 callback 时应跳过远程提交流程');
assert.strictEqual(api.shouldBypassStep9ForLocalCpa({
vpsUrl: 'https://example.com/management.html#/oauth',
localhostUrl: 'http://127.0.0.1:8317/codex/callback?code=abc&state=xyz',
}), false, '远程 CPA 不应跳过步骤 9');
assert.strictEqual(api.shouldBypassStep9ForLocalCpa({
vpsUrl: 'http://127.0.0.1:8317/management.html#/oauth',
localhostUrl: '',
}), false, '没有 callback 时不应跳过步骤 9');
console.log('step9 cpa mode tests passed');
+193
View File
@@ -0,0 +1,193 @@
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++) {
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++) {
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('getTabRegistry'),
extractFunction('parseUrlSafely'),
extractFunction('isLocalhostOAuthCallbackUrl'),
extractFunction('isLocalhostOAuthCallbackTabMatch'),
extractFunction('closeLocalhostCallbackTabs'),
extractFunction('handleStepData'),
].join('\n');
const api = new Function(`
let currentState = {
tabRegistry: {
'signup-page': { tabId: 1, ready: true },
'vps-panel': { tabId: 99, ready: true },
},
};
let currentTabs = [];
const removedBatches = [];
const logMessages = [];
const chrome = {
tabs: {
async query() {
return currentTabs;
},
async remove(ids) {
removedBatches.push(ids);
currentTabs = currentTabs.filter((tab) => !ids.includes(tab.id));
},
},
};
async function getState() {
return currentState;
}
async function setState(updates) {
currentState = { ...currentState, ...updates };
}
async function setEmailState(email) {
currentState = { ...currentState, email };
}
function broadcastDataUpdate() {}
async function addLog(message) {
logMessages.push(message);
}
${bundle}
return {
handleStepData,
closeLocalhostCallbackTabs,
isLocalhostOAuthCallbackTabMatch,
reset({ tabs, tabRegistry }) {
currentTabs = tabs;
removedBatches.length = 0;
logMessages.length = 0;
currentState = {
tabRegistry: tabRegistry || {},
};
},
snapshot() {
return {
currentState,
removedBatches,
logMessages,
};
},
};
`)();
(async () => {
const codexCallbackUrl = 'http://127.0.0.1:8317/codex/callback?code=abc&state=xyz';
const authCallbackUrl = 'http://localhost:1455/auth/callback?code=def&state=uvw';
assert.strictEqual(
api.isLocalhostOAuthCallbackTabMatch(codexCallbackUrl, codexCallbackUrl),
true,
'真实 callback 页应命中清理规则'
);
assert.strictEqual(
api.isLocalhostOAuthCallbackTabMatch(codexCallbackUrl, authCallbackUrl),
false,
'/codex/callback 不应误伤 /auth/callback'
);
assert.strictEqual(
api.isLocalhostOAuthCallbackTabMatch(authCallbackUrl, codexCallbackUrl),
false,
'/auth/callback 不应误伤 /codex/callback'
);
api.reset({
tabs: [
{ id: 1, url: codexCallbackUrl },
{ id: 2, url: 'http://127.0.0.1:8317/codex/dashboard' },
{ id: 3, url: 'http://127.0.0.1:8317/codex/callback?code=other&state=xyz' },
{ id: 4, url: authCallbackUrl },
],
tabRegistry: {
'signup-page': { tabId: 1, ready: true },
'vps-panel': { tabId: 99, ready: true },
},
});
await api.handleStepData(9, { localhostUrl: codexCallbackUrl });
let snapshot = api.snapshot();
assert.deepStrictEqual(snapshot.removedBatches, [[1]], 'handleStepData(9) 只应关闭当前 callback 页');
assert.strictEqual(
snapshot.currentState.tabRegistry['signup-page'],
null,
'关闭 callback 页后应同步清理 signup-page 的 tabRegistry'
);
assert.deepStrictEqual(
snapshot.currentState.tabRegistry['vps-panel'],
{ tabId: 99, ready: true },
'不相关的 tabRegistry 项不应被误清理'
);
api.reset({
tabs: [
{ id: 1, url: codexCallbackUrl },
{ id: 4, url: authCallbackUrl },
{ id: 5, url: 'http://localhost:1455/auth/dashboard' },
],
tabRegistry: {},
});
const closedCount = await api.closeLocalhostCallbackTabs(authCallbackUrl);
snapshot = api.snapshot();
assert.strictEqual(closedCount, 1, 'auth callback 也应只关闭当前命中的 callback 页');
assert.deepStrictEqual(snapshot.removedBatches, [[4]], '不应按 /auth 前缀批量清理页面');
assert.strictEqual(snapshot.logMessages.length, 1, '发生清理时应记录一条日志');
console.log('step9 localhost cleanup scope tests passed');
})().catch((error) => {
console.error(error);
process.exit(1);
});