fix: upgrade plan correctly (#6543)

This commit is contained in:
darkskygit
2024-04-15 06:20:23 +00:00
parent 1217ef258b
commit 1a3d1a5421
2 changed files with 21 additions and 6 deletions

View File

@@ -1,7 +1,8 @@
import { PrismaClient } from '@prisma/client';
import { FeatureKind } from '../../../core/features';
import { Quota } from '../../../core/quota/types';
import { Quotas } from '../../../core/quota/schema';
import { Quota, QuotaType } from '../../../core/quota/types';
import { upsertFeature } from './user-features';
export async function upgradeQuotaVersion(
@@ -63,3 +64,14 @@ export async function upgradeQuotaVersion(
});
});
}
export async function upgradeLatestQuotaVersion(
db: PrismaClient,
type: QuotaType,
reason: string
) {
const quota = Quotas.filter(f => f.feature === type);
quota.sort((a, b) => b.version - a.version);
const latestQuota = quota[0];
await upgradeQuotaVersion(db, latestQuota, reason);
}