feat: add business blob limit (#5734)

packages/backend/server/src/data/migrations/1705395933447-new-free-plan.ts

@@ -1,65 +1,14 @@
 import { PrismaClient } from '@prisma/client';
 
-import { FeatureKind } from '../../core/features';
 import { Quotas } from '../../core/quota';
-import { upsertFeature } from './utils/user-features';
+import { upgradeQuotaVersion } from './utils/user-quotas';
 
 export class NewFreePlan1705395933447 {
   // do the migration
   static async up(db: PrismaClient) {
-    // add new free plan
-    await upsertFeature(db, Quotas[3]);
-    // migrate all free plan users to new free plan
-    await db.$transaction(async tx => {
-      const latestFreePlan = await tx.features.findFirstOrThrow({
-        where: { feature: Quotas[3].feature },
-        orderBy: { version: 'desc' },
-        select: { id: true },
-      });
-
-      // find all users that have old free plan
-      const userIds = await db.user.findMany({
-        where: {
-          features: {
-            every: {
-              feature: {
-                type: FeatureKind.Quota,
-                feature: Quotas[3].feature,
-                version: { lt: Quotas[3].version },
-              },
-              activated: true,
-            },
-          },
-        },
-        select: { id: true },
-      });
-
-      // deactivate all old quota for the user
-      await tx.userFeatures.updateMany({
-        where: {
-          id: undefined,
-          userId: {
-            in: userIds.map(({ id }) => id),
-          },
-          feature: {
-            type: FeatureKind.Quota,
-          },
-          activated: true,
-        },
-        data: {
-          activated: false,
-        },
-      });
-
-      await tx.userFeatures.createMany({
-        data: userIds.map(({ id: userId }) => ({
-          userId,
-          featureId: latestFreePlan.id,
-          reason: 'free plan 1.0 migration',
-          activated: true,
-        })),
-      });
-    });
+    // free plan 1.0
+    const quota = Quotas[3];
+    await upgradeQuotaVersion(db, quota, 'free plan 1.0 migration');
   }
 
   // revert the migration

packages/backend/server/src/data/migrations/1706513866287-business-blob-limit.ts

@@ -0,0 +1,16 @@
+import { PrismaClient } from '@prisma/client';
+
+import { Quotas } from '../../core/quota';
+import { upgradeQuotaVersion } from './utils/user-quotas';
+
+export class BusinessBlobLimit1706513866287 {
+  // do the migration
+  static async up(db: PrismaClient) {
+    // free plan 1.1
+    const quota = Quotas[4];
+    await upgradeQuotaVersion(db, quota, 'free plan 1.1 migration');
+  }
+
+  // revert the migration
+  static async down(_db: PrismaClient) {}
+}

packages/backend/server/src/data/migrations/utils/user-quotas.ts

@@ -0,0 +1,65 @@
+import { PrismaClient } from '@prisma/client';
+
+import { FeatureKind } from '../../../core/features';
+import { Quota } from '../../../core/quota/types';
+import { upsertFeature } from './user-features';
+
+export async function upgradeQuotaVersion(
+  db: PrismaClient,
+  quota: Quota,
+  reason: string
+) {
+  // add new quota
+  await upsertFeature(db, quota);
+  // migrate all users that using old quota to new quota
+  await db.$transaction(async tx => {
+    const latestQuotaVersion = await tx.features.findFirstOrThrow({
+      where: { feature: quota.feature },
+      orderBy: { version: 'desc' },
+      select: { id: true },
+    });
+
+    // find all users that have old free plan
+    const userIds = await db.user.findMany({
+      where: {
+        features: {
+          every: {
+            feature: {
+              type: FeatureKind.Quota,
+              feature: quota.feature,
+              version: { lt: quota.version },
+            },
+            activated: true,
+          },
+        },
+      },
+      select: { id: true },
+    });
+
+    // deactivate all old quota for the user
+    await tx.userFeatures.updateMany({
+      where: {
+        id: undefined,
+        userId: {
+          in: userIds.map(({ id }) => id),
+        },
+        feature: {
+          type: FeatureKind.Quota,
+        },
+        activated: true,
+      },
+      data: {
+        activated: false,
+      },
+    });
+
+    await tx.userFeatures.createMany({
+      data: userIds.map(({ id: userId }) => ({
+        userId,
+        featureId: latestQuotaVersion.id,
+        reason,
+        activated: true,
+      })),
+    });
+  });
+}