chore: auto fix 存储标记,确保fix只运行一次

This commit is contained in:
xiaojunnuo
2026-05-16 02:23:21 +08:00
parent c63745d1ba
commit c3baaf3ac7
3 changed files with 85 additions and 6 deletions
@@ -2,9 +2,24 @@ import assert from "assert";
import { AutoFix } from "./auto-fix.js";
describe("AutoFix", () => {
it("runs fix tasks in order", async () => {
it("runs unfinished fix tasks in order and marks them fixed", async () => {
const calls: string[] = [];
let savedSetting: any;
const autoFix = new AutoFix();
autoFix.sysSettingsService = {
async getSetting() {
return {
fixed: {
"oauth-subtype-bound-type": true,
},
};
},
async saveSetting(setting: any) {
savedSetting = {
fixed: { ...setting.fixed },
};
},
} as any;
autoFix.googleCommonEabAccountKeyFix = {
async init() {
calls.push("google");
@@ -28,6 +43,26 @@ describe("AutoFix", () => {
await autoFix.init();
assert.deepEqual(calls, ["google", "oauth", "cert", "suite"]);
assert.deepEqual(calls, ["google", "cert", "suite"]);
assert.equal(savedSetting.fixed["google-common-eab-account-key"], true);
assert.equal(savedSetting.fixed["oauth-subtype-bound-type"], true);
assert.equal(savedSetting.fixed["cert-info-wildcard-domain-count"], true);
assert.equal(savedSetting.fixed["suite-content-wildcard-domain-count"], true);
});
it("initializes missing fixed map", async () => {
const autoFix = new AutoFix();
autoFix.sysSettingsService = {
async getSetting() {
return {};
},
async saveSetting() {},
} as any;
autoFix.googleCommonEabAccountKeyFix = { async init() {} } as any;
autoFix.oauthSubtypeBoundTypeFix = { async init() {} } as any;
autoFix.certInfoWildcardDomainCountFix = { async init() {} } as any;
autoFix.suiteContentWildcardDomainCountFix = { async init() {} } as any;
await autoFix.init();
});
});
@@ -1,12 +1,23 @@
import { Inject, Provide, Scope, ScopeEnum } from "@midwayjs/core";
import { SysAutoFixSetting, SysSettingsService } from "@certd/lib-server";
import { GoogleCommonEabAccountKeyFix } from "./google-common-eab-account-key-fix.js";
import { OauthSubtypeBoundTypeFix } from "./oauth-subtype-bound-type-fix.js";
import { CertInfoWildcardDomainCountFix } from "./cert-info-wildcard-domain-count-fix.js";
import { SuiteContentWildcardDomainCountFix } from "./suite-content-wildcard-domain-count-fix.js";
type AutoFixTask = {
key: string;
fix: {
init(): Promise<void>;
};
};
@Provide()
@Scope(ScopeEnum.Request, { allowDowngrade: true })
export class AutoFix {
@Inject()
sysSettingsService: SysSettingsService;
@Inject()
googleCommonEabAccountKeyFix: GoogleCommonEabAccountKeyFix;
@@ -20,9 +31,34 @@ export class AutoFix {
suiteContentWildcardDomainCountFix: SuiteContentWildcardDomainCountFix;
async init() {
await this.googleCommonEabAccountKeyFix.init();
await this.oauthSubtypeBoundTypeFix.init();
await this.certInfoWildcardDomainCountFix.init();
await this.suiteContentWildcardDomainCountFix.init();
const setting = await this.sysSettingsService.getSetting<SysAutoFixSetting>(SysAutoFixSetting);
setting.fixed = setting.fixed || {};
const tasks: AutoFixTask[] = [
{
key: "google-common-eab-account-key",
fix: this.googleCommonEabAccountKeyFix,
},
{
key: "oauth-subtype-bound-type",
fix: this.oauthSubtypeBoundTypeFix,
},
{
key: "cert-info-wildcard-domain-count",
fix: this.certInfoWildcardDomainCountFix,
},
{
key: "suite-content-wildcard-domain-count",
fix: this.suiteContentWildcardDomainCountFix,
},
];
for (const task of tasks) {
if (setting.fixed?.[task.key]) {
continue;
}
await task.fix.init();
setting.fixed[task.key] = true;
await this.sysSettingsService.saveSetting(setting);
}
}
}