chore: refresh existed user

toeverything · Apr 16, 2024 · a9c3914 · a9c3914
1 parent 9d43095
commit a9c3914
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/packages/backend/server/src/data/migrations/1713258139569-refresh-new-plan.ts b/packages/backend/server/src/data/migrations/1713258139569-refresh-new-plan.ts
@@ -1,12 +1,25 @@
+import { ModuleRef } from '@nestjs/core';
 import { PrismaClient } from '@prisma/client';
 
-import { QuotaType } from '../../core/quota/types';
+import { QuotaService, QuotaType } from '../../core/quota';
+import { UserService } from '../../core/user';
+import { Config } from '../../fundamentals';
 import { upgradeLatestQuotaVersion } from './utils/user-quotas';
 
 export class RefreshNewPlan1713258139569 {
   // do the migration
-  static async up(db: PrismaClient) {
-    await upgradeLatestQuotaVersion(db, QuotaType.SelfHostedPlanV1, '');
+  static async up(db: PrismaClient, ref: ModuleRef) {
+    const config = ref.get(Config, { strict: false });
+    const { AFFINE_ADMIN_EMAIL } = process.env;
+    if (config.isSelfhosted && AFFINE_ADMIN_EMAIL) {
+      const user = ref.get(UserService, { strict: false });
+      const quota = ref.get(QuotaService, { strict: false });
+      const { id } = (await user.findUserByEmail(AFFINE_ADMIN_EMAIL)) || {};
+      if (id) {
+        await upgradeLatestQuotaVersion(db, QuotaType.SelfHostedPlanV1, '');
+        await quota.switchUserQuota(id, QuotaType.SelfHostedPlanV1);
+      }
+    }
   }
 
   // revert the migration