feat: 添加 Plus 模式下的自动运行风险提示逻辑,支持超过 3 轮的警告和确认模态框

This commit is contained in:
QLHazyCoder
2026-04-26 06:25:11 +08:00
parent 9a63cba994
commit 6329407b95
3 changed files with 147 additions and 2 deletions
@@ -48,12 +48,23 @@ function extractFunction(name) {
return sidepanelSource.slice(start, end);
}
function createApi({ refreshImpl, confirmImpl, dismissImpl } = {}) {
function createApi({
refreshImpl,
confirmImpl,
dismissImpl,
runCount = 3,
plusModeEnabled = false,
plusRiskEnabled = false,
plusRiskConfirmed = true,
plusRiskDismissPrompt = false,
} = {}) {
const bundle = extractFunction('startAutoRunFromCurrentSettings');
return new Function(`
const events = [];
const latestState = { contributionMode: false };
const currentPlusModeEnabled = ${JSON.stringify(Boolean(plusModeEnabled))};
const inputPlusModeEnabled = { checked: ${JSON.stringify(Boolean(plusModeEnabled))} };
const inputAutoSkipFailures = { checked: false };
const inputContributionNickname = { value: 'tester' };
const inputContributionQq = { value: '123456' };
@@ -75,7 +86,7 @@ const console = {
events.push({ type: 'warn', args });
},
};
function getRunCountValue() { return 3; }
function getRunCountValue() { return ${Math.max(1, Number(runCount) || 1)}; }
function normalizeAutoRunThreadIntervalMinutes(value) { return Number(value) || 0; }
function shouldOfferAutoModeChoice() { return false; }
async function openAutoStartChoiceDialog() { throw new Error('should not be called'); }
@@ -85,6 +96,20 @@ function shouldWarnAutoRunFallbackRisk() { return false; }
function isAutoRunFallbackRiskPromptDismissed() { return false; }
async function openAutoRunFallbackRiskConfirmModal() { throw new Error('should not be called'); }
function setAutoRunFallbackRiskPromptDismissed() {}
function shouldWarnPlusAutoRunRisk(totalRuns, plusModeEnabled) {
return ${JSON.stringify(Boolean(plusRiskEnabled))} && Boolean(plusModeEnabled) && Number(totalRuns) > 3;
}
function isAutoRunPlusRiskPromptDismissed() { return false; }
async function openPlusAutoRunRiskConfirmModal(totalRuns) {
events.push({ type: 'plus-risk-modal', totalRuns });
return {
confirmed: ${JSON.stringify(Boolean(plusRiskConfirmed))},
dismissPrompt: ${JSON.stringify(Boolean(plusRiskDismissPrompt))},
};
}
function setAutoRunPlusRiskPromptDismissed(dismissed) {
events.push({ type: 'plus-risk-dismiss', dismissed });
}
function normalizeAutoDelayMinutes(value) { return Number(value) || 30; }
async function refreshContributionContentHint() {
events.push({ type: 'refresh' });
@@ -155,3 +180,39 @@ test('startAutoRunFromCurrentSettings aborts when contribution update confirmati
['refresh', 'confirm-check']
);
});
test('startAutoRunFromCurrentSettings shows Plus risk warning before starting more than 3 runs', async () => {
const api = createApi({
runCount: 4,
plusModeEnabled: true,
plusRiskEnabled: true,
});
const result = await api.startAutoRunFromCurrentSettings();
const events = api.getEvents();
assert.equal(result, true);
assert.deepEqual(
events.map((entry) => entry.type),
['refresh', 'confirm-check', 'plus-risk-modal', 'send']
);
assert.equal(events[2].totalRuns, 4);
assert.equal(events[3].message.payload.totalRuns, 4);
});
test('startAutoRunFromCurrentSettings aborts when Plus risk warning is declined', async () => {
const api = createApi({
runCount: 4,
plusModeEnabled: true,
plusRiskEnabled: true,
plusRiskConfirmed: false,
});
const result = await api.startAutoRunFromCurrentSettings();
assert.equal(result, false);
assert.deepEqual(
api.getEvents().map((entry) => entry.type),
['refresh', 'confirm-check', 'plus-risk-modal']
);
});