feat: 增强 Plus Checkout 创建逻辑,支持重用 ChatGPT 标签页并优化测试用例

This commit is contained in:
QLHazyCoder
2026-04-26 15:50:37 +08:00
parent 336fdb0c3a
commit 6c2bec6520
3 changed files with 126 additions and 3 deletions
+3
View File
@@ -7067,6 +7067,9 @@ const plusCheckoutCreateExecutor = self.MultiPageBackgroundPlusCheckoutCreate?.c
chrome,
completeStepFromBackground,
ensureContentScriptReadyOnTabUntilStopped,
getTabId,
isTabAlive,
registerTab,
reuseOrCreateTab,
sendTabMessageUntilStopped,
setState,
+52 -3
View File
@@ -2,8 +2,21 @@
root.MultiPageBackgroundPlusCheckoutCreate = factory();
})(typeof self !== 'undefined' ? self : globalThis, function createBackgroundPlusCheckoutCreateModule() {
const PLUS_CHECKOUT_SOURCE = 'plus-checkout';
const SIGNUP_PAGE_SOURCE = 'signup-page';
const PLUS_CHECKOUT_ENTRY_URL = 'https://chatgpt.com/';
const PLUS_CHECKOUT_INJECT_FILES = ['content/utils.js', 'content/plus-checkout.js'];
const PLUS_CHECKOUT_ONLY_INJECT_FILES = ['content/plus-checkout.js'];
function isReusableChatGptTabUrl(url = '') {
try {
const parsed = new URL(String(url || ''));
const hostname = parsed.hostname.toLowerCase();
return ['chatgpt.com', 'www.chatgpt.com', 'chat.openai.com'].includes(hostname)
&& !/^\/checkout(?:\/|$)/i.test(parsed.pathname || '');
} catch {
return false;
}
}
function createPlusCheckoutCreateExecutor(deps = {}) {
const {
@@ -11,6 +24,9 @@
chrome,
completeStepFromBackground,
ensureContentScriptReadyOnTabUntilStopped,
getTabId,
isTabAlive,
registerTab,
reuseOrCreateTab,
sendTabMessageUntilStopped,
setState,
@@ -18,16 +34,49 @@
waitForTabCompleteUntilStopped,
} = deps;
async function executePlusCheckoutCreate() {
await addLog('步骤 6:正在打开 ChatGPT 会话页,准备创建 Plus Checkout...', 'info');
async function getReusableSignupChatGptTabId() {
if (typeof getTabId !== 'function' || typeof isTabAlive !== 'function') {
return null;
}
const tabId = await getTabId(SIGNUP_PAGE_SOURCE);
if (!tabId || !(await isTabAlive(SIGNUP_PAGE_SOURCE))) {
return null;
}
const tab = await chrome.tabs.get(tabId).catch(() => null);
return tab && isReusableChatGptTabUrl(tab.url) ? tabId : null;
}
async function resolveChatGptTabForCheckoutCreate() {
const existingSignupTabId = await getReusableSignupChatGptTabId();
if (existingSignupTabId) {
await chrome.tabs.update(existingSignupTabId, { active: true });
if (typeof registerTab === 'function') {
await registerTab(PLUS_CHECKOUT_SOURCE, existingSignupTabId);
}
await addLog('步骤 6:检测到第 5 步已打开 ChatGPT 页面,直接接管当前标签页创建 Plus Checkout。', 'info');
return {
tabId: existingSignupTabId,
injectFiles: PLUS_CHECKOUT_ONLY_INJECT_FILES,
};
}
const tabId = await reuseOrCreateTab(PLUS_CHECKOUT_SOURCE, PLUS_CHECKOUT_ENTRY_URL, {
reloadIfSameUrl: false,
});
return {
tabId,
injectFiles: PLUS_CHECKOUT_INJECT_FILES,
};
}
async function executePlusCheckoutCreate() {
await addLog('步骤 6:正在打开 ChatGPT 会话页,准备创建 Plus Checkout...', 'info');
const { tabId, injectFiles } = await resolveChatGptTabForCheckoutCreate();
await waitForTabCompleteUntilStopped(tabId);
await sleepWithStop(1000);
await ensureContentScriptReadyOnTabUntilStopped(PLUS_CHECKOUT_SOURCE, tabId, {
inject: PLUS_CHECKOUT_INJECT_FILES,
inject: injectFiles,
injectSource: PLUS_CHECKOUT_SOURCE,
logMessage: '步骤 6ChatGPT 页面仍在加载,等待 Plus Checkout 脚本就绪...',
});
+71
View File
@@ -62,3 +62,74 @@ test('Plus checkout create does not wait 20 seconds after opening checkout page'
assert.equal(events.some((event) => event.type === 'sleep' && event.ms === 20000), false);
assert.equal(events.some((event) => event.type === 'log' && /固定等待 20 秒后继续下一步/.test(event.message)), false);
});
test('Plus checkout create reuses the ChatGPT tab from signup step 5', async () => {
const events = [];
const executor = api.createPlusCheckoutCreateExecutor({
addLog: async (message, level = 'info') => {
events.push({ type: 'log', message, level });
},
chrome: {
tabs: {
get: async (tabId) => {
events.push({ type: 'tab-get', tabId });
return { id: tabId, url: 'https://chatgpt.com/' };
},
update: async (tabId, payload) => {
events.push({ type: 'tab-update', tabId, payload });
},
},
},
completeStepFromBackground: async (step, payload) => {
events.push({ type: 'complete', step, payload });
},
ensureContentScriptReadyOnTabUntilStopped: async (source, tabId, options) => {
events.push({ type: 'ready', source, tabId, options });
},
getTabId: async (source) => {
events.push({ type: 'get-tab-id', source });
return source === 'signup-page' ? 55 : null;
},
isTabAlive: async (source) => {
events.push({ type: 'alive', source });
return source === 'signup-page';
},
registerTab: async (source, tabId) => {
events.push({ type: 'register', source, tabId });
},
reuseOrCreateTab: async () => {
events.push({ type: 'reuse-tab' });
return 42;
},
sendTabMessageUntilStopped: async () => ({
checkoutUrl: 'https://checkout.stripe.com/c/pay/session',
country: 'US',
currency: 'USD',
}),
setState: async (payload) => {
events.push({ type: 'set-state', payload });
},
sleepWithStop: async (ms) => {
events.push({ type: 'sleep', ms });
},
waitForTabCompleteUntilStopped: async (tabId) => {
events.push({ type: 'tab-complete', tabId });
},
});
await executor.executePlusCheckoutCreate();
assert.equal(events.some((event) => event.type === 'reuse-tab'), false);
assert.deepEqual(
events.find((event) => event.type === 'register'),
{ type: 'register', source: 'plus-checkout', tabId: 55 }
);
assert.deepEqual(
events.find((event) => event.type === 'ready').options.inject,
['content/plus-checkout.js']
);
assert.equal(
events.some((event) => event.type === 'log' && /直接接管当前标签页/.test(event.message)),
true
);
});