fix: updated the downgrade flow for restricting workspaces

Author: aakashreddy-p

src/modules/billing/repositories/stripe-subscription.repository.ts

@@ -303,7 +303,7 @@ export class StripeSubscriptionRepository {
       }
       const updateQuery: Record<string, any> = {};
       // Add workspaces if provided
-      if (workspaces && workspaces.length > 0) {
+      if (workspaces) {
         updateQuery["downgrade.workspaces"] = { $each: workspaces };
       }
       // Add users if provided

src/modules/billing/services/downgrade.service.ts

@@ -290,4 +290,38 @@ export class DownGradeService {
       console.log(error);
     }
   }
+
+  async validateWorkspaceRestrictions(
+    teamDetails: Team,
+    teamID: string,
+    newPlan: PlanName,
+  ): Promise<void> {
+    try {
+      const planData =
+        await this.stripeSubscriptionRepository.findPlanByName(newPlan);
+      if (!teamDetails || !Array.isArray(teamDetails.workspaces)) {
+        return;
+      }
+      // Get only unrestricted workspaces
+      // A workspace is unrestricted if isRestricted is not present OR isRestricted === false
+      const unrestrictedWorkspaces = teamDetails.workspaces.filter(
+        (w) => w?.isRestricted !== true,
+      );
+      const unRestrictedCount = unrestrictedWorkspaces.length;
+      const limit = planData?.limits?.workspacesPerHub?.value ?? 0;
+      // Only act when unrestricted workspaces exceed the plan limit
+      if (unRestrictedCount > limit) {
+        // Get excess workspaces (those beyond the limit)
+        const idsToRestrict = unrestrictedWorkspaces
+          .slice(limit) // Take all workspaces after the limit
+          .map((w) => w.id.toString());
+        await this.restrictTeamWorkspace(teamID, idsToRestrict);
+        for (const wid of idsToRestrict) {
+          await this.restrictWorkspace(wid);
+        }
+      }
+    } catch (error) {
+      console.log("Error in validateWorkspaceRestrictions:", error);
+    }
+  }
 }

src/modules/billing/services/stripe-subscription.service.ts

@@ -590,6 +590,16 @@ export class StripeSubscriptionService {
               newPlan,
               new Date(),
             );
+            if (
+              !team?.downgrade?.workspaces ||
+              team?.downgrade?.workspaces.length === 0
+            ) {
+              await this.downgradeService.validateWorkspaceRestrictions(
+                team,
+                metadata.hubId,
+                newPlan,
+              );
+            }
             await this.stripeSubscriptionRepo.removeDowngradeDetails(
               metadata.hubId,
             );