feat: 新增 isSignupEntryHost 函数以支持 chatgpt.com 和 chat.openai.com 的注册页面识别,更新相关测试用例
This commit is contained in:
+5
-1
@@ -3210,6 +3210,10 @@ function isSignupPageHost(hostname = '') {
|
|||||||
return ['auth0.openai.com', 'auth.openai.com', 'accounts.openai.com'].includes(hostname);
|
return ['auth0.openai.com', 'auth.openai.com', 'accounts.openai.com'].includes(hostname);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function isSignupEntryHost(hostname = '') {
|
||||||
|
return ['chatgpt.com', 'chat.openai.com'].includes(hostname);
|
||||||
|
}
|
||||||
|
|
||||||
function isSignupPasswordPageUrl(rawUrl) {
|
function isSignupPasswordPageUrl(rawUrl) {
|
||||||
const parsed = parseUrlSafely(rawUrl);
|
const parsed = parseUrlSafely(rawUrl);
|
||||||
if (!parsed) return false;
|
if (!parsed) return false;
|
||||||
@@ -3261,7 +3265,7 @@ function matchesSourceUrlFamily(source, candidateUrl, referenceUrl) {
|
|||||||
|
|
||||||
switch (source) {
|
switch (source) {
|
||||||
case 'signup-page':
|
case 'signup-page':
|
||||||
return isSignupPageHost(candidate.hostname);
|
return isSignupPageHost(candidate.hostname) || isSignupEntryHost(candidate.hostname);
|
||||||
case 'duck-mail':
|
case 'duck-mail':
|
||||||
return candidate.hostname === 'duckduckgo.com' && candidate.pathname.startsWith('/email/');
|
return candidate.hostname === 'duckduckgo.com' && candidate.pathname.startsWith('/email/');
|
||||||
case 'qq-mail':
|
case 'qq-mail':
|
||||||
|
|||||||
@@ -125,7 +125,12 @@ return {
|
|||||||
return { logs, listCalls };
|
return { logs, listCalls };
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
`)(options);
|
`)({
|
||||||
|
...options,
|
||||||
|
receiveMailbox,
|
||||||
|
messages,
|
||||||
|
deleteShouldFail,
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
test('pollCloudflareTempEmailVerificationCode returns code even if delete fails', async () => {
|
test('pollCloudflareTempEmailVerificationCode returns code even if delete fails', async () => {
|
||||||
|
|||||||
@@ -0,0 +1,206 @@
|
|||||||
|
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('getTabRegistry'),
|
||||||
|
extractFunction('parseUrlSafely'),
|
||||||
|
extractFunction('isSignupPageHost'),
|
||||||
|
extractFunction('isSignupEntryHost'),
|
||||||
|
extractFunction('matchesSourceUrlFamily'),
|
||||||
|
extractFunction('closeConflictingTabsForSource'),
|
||||||
|
].join('\n');
|
||||||
|
|
||||||
|
const api = new Function(`
|
||||||
|
let currentState = {
|
||||||
|
sourceLastUrls: {},
|
||||||
|
tabRegistry: {},
|
||||||
|
};
|
||||||
|
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 addLog(message, level = 'info') {
|
||||||
|
logMessages.push({ message, level });
|
||||||
|
}
|
||||||
|
|
||||||
|
function getSourceLabel(source) {
|
||||||
|
return source;
|
||||||
|
}
|
||||||
|
|
||||||
|
${bundle}
|
||||||
|
|
||||||
|
return {
|
||||||
|
matchesSourceUrlFamily,
|
||||||
|
closeConflictingTabsForSource,
|
||||||
|
reset({ tabs, state }) {
|
||||||
|
currentTabs = tabs;
|
||||||
|
removedBatches.length = 0;
|
||||||
|
logMessages.length = 0;
|
||||||
|
currentState = {
|
||||||
|
sourceLastUrls: {},
|
||||||
|
tabRegistry: {},
|
||||||
|
...(state || {}),
|
||||||
|
};
|
||||||
|
},
|
||||||
|
snapshot() {
|
||||||
|
return {
|
||||||
|
currentState,
|
||||||
|
currentTabs,
|
||||||
|
removedBatches,
|
||||||
|
logMessages,
|
||||||
|
};
|
||||||
|
},
|
||||||
|
};
|
||||||
|
`)();
|
||||||
|
|
||||||
|
(async () => {
|
||||||
|
assert.strictEqual(
|
||||||
|
api.matchesSourceUrlFamily('signup-page', 'https://chatgpt.com/', 'https://chatgpt.com/'),
|
||||||
|
true,
|
||||||
|
'signup-page family should include chatgpt.com'
|
||||||
|
);
|
||||||
|
assert.strictEqual(
|
||||||
|
api.matchesSourceUrlFamily('signup-page', 'https://chat.openai.com/', 'https://auth.openai.com/authorize'),
|
||||||
|
true,
|
||||||
|
'signup-page family should include legacy chat.openai.com'
|
||||||
|
);
|
||||||
|
|
||||||
|
api.reset({
|
||||||
|
tabs: [
|
||||||
|
{ id: 1, url: 'https://chatgpt.com/' },
|
||||||
|
{ id: 2, url: 'https://chat.openai.com/' },
|
||||||
|
{ id: 3, url: 'https://auth.openai.com/authorize?client_id=test' },
|
||||||
|
{ id: 4, url: 'https://example.com/' },
|
||||||
|
],
|
||||||
|
state: {
|
||||||
|
sourceLastUrls: {
|
||||||
|
'signup-page': 'https://chatgpt.com/',
|
||||||
|
},
|
||||||
|
tabRegistry: {
|
||||||
|
'signup-page': { tabId: 3, ready: true },
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
await api.closeConflictingTabsForSource('signup-page', 'https://auth.openai.com/authorize', {
|
||||||
|
excludeTabIds: [3],
|
||||||
|
});
|
||||||
|
|
||||||
|
let snapshot = api.snapshot();
|
||||||
|
assert.deepStrictEqual(
|
||||||
|
snapshot.removedBatches,
|
||||||
|
[[1, 2]],
|
||||||
|
'opening auth page should clean up stale ChatGPT entry tabs'
|
||||||
|
);
|
||||||
|
assert.deepStrictEqual(
|
||||||
|
snapshot.currentTabs,
|
||||||
|
[
|
||||||
|
{ id: 3, url: 'https://auth.openai.com/authorize?client_id=test' },
|
||||||
|
{ id: 4, url: 'https://example.com/' },
|
||||||
|
],
|
||||||
|
'non-signup tabs and excluded current tab should remain'
|
||||||
|
);
|
||||||
|
|
||||||
|
api.reset({
|
||||||
|
tabs: [
|
||||||
|
{ id: 11, url: 'https://chatgpt.com/' },
|
||||||
|
{ id: 12, url: 'https://auth.openai.com/authorize?client_id=test' },
|
||||||
|
],
|
||||||
|
state: {
|
||||||
|
sourceLastUrls: {
|
||||||
|
'signup-page': 'https://auth.openai.com/authorize?client_id=test',
|
||||||
|
},
|
||||||
|
tabRegistry: {
|
||||||
|
'signup-page': { tabId: 11, ready: true },
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
await api.closeConflictingTabsForSource('signup-page', 'https://chatgpt.com/');
|
||||||
|
|
||||||
|
snapshot = api.snapshot();
|
||||||
|
assert.deepStrictEqual(
|
||||||
|
snapshot.removedBatches,
|
||||||
|
[[11, 12]],
|
||||||
|
'opening ChatGPT entry should remove older signup-family tabs'
|
||||||
|
);
|
||||||
|
assert.strictEqual(
|
||||||
|
snapshot.currentState.tabRegistry['signup-page'],
|
||||||
|
null,
|
||||||
|
'registry should be cleared when the tracked signup tab is removed'
|
||||||
|
);
|
||||||
|
|
||||||
|
console.log('signup page tab cleanup tests passed');
|
||||||
|
})().catch((error) => {
|
||||||
|
console.error(error);
|
||||||
|
process.exit(1);
|
||||||
|
});
|
||||||
@@ -70,6 +70,7 @@ let autoRunActive = true;
|
|||||||
let autoRunCurrentRun = 2;
|
let autoRunCurrentRun = 2;
|
||||||
let autoRunTotalRuns = 3;
|
let autoRunTotalRuns = 3;
|
||||||
let autoRunAttemptRun = 4;
|
let autoRunAttemptRun = 4;
|
||||||
|
const AUTO_RUN_TIMER_KIND_SCHEDULED_START = 'scheduled_start';
|
||||||
|
|
||||||
const added = {
|
const added = {
|
||||||
beforeNavigate: 0,
|
beforeNavigate: 0,
|
||||||
@@ -128,6 +129,9 @@ async function broadcastAutoRunStatus() {}
|
|||||||
async function getState() {
|
async function getState() {
|
||||||
return { autoRunning: false };
|
return { autoRunning: false };
|
||||||
}
|
}
|
||||||
|
function getPendingAutoRunTimerPlan() {
|
||||||
|
return null;
|
||||||
|
}
|
||||||
function isAutoRunScheduledState() {
|
function isAutoRunScheduledState() {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -72,6 +72,7 @@ const api = new Function(`
|
|||||||
const HOTMAIL_PROVIDER = 'hotmail-api';
|
const HOTMAIL_PROVIDER = 'hotmail-api';
|
||||||
const CLOUDFLARE_TEMP_EMAIL_PROVIDER = 'cloudflare-temp-email';
|
const CLOUDFLARE_TEMP_EMAIL_PROVIDER = 'cloudflare-temp-email';
|
||||||
const CLOUDFLARE_TEMP_EMAIL_GENERATOR = 'cloudflare-temp-email';
|
const CLOUDFLARE_TEMP_EMAIL_GENERATOR = 'cloudflare-temp-email';
|
||||||
|
const GMAIL_PROVIDER = 'gmail';
|
||||||
const MAIL_2925_MODE_PROVIDE = 'provide';
|
const MAIL_2925_MODE_PROVIDE = 'provide';
|
||||||
const MAIL_2925_MODE_RECEIVE = 'receive';
|
const MAIL_2925_MODE_RECEIVE = 'receive';
|
||||||
const DEFAULT_MAIL_2925_MODE = MAIL_2925_MODE_PROVIDE;
|
const DEFAULT_MAIL_2925_MODE = MAIL_2925_MODE_PROVIDE;
|
||||||
|
|||||||
Reference in New Issue
Block a user