init

worklenz-backend/src/controllers/project-members-controller.ts

@@ -7,6 +7,9 @@ import WorklenzControllerBase from "./worklenz-controller-base";
 import HandleExceptions from "../decorators/handle-exceptions";
 import {getColor} from "../shared/utils";
 import TeamMembersController from "./team-members-controller";
+import {checkTeamSubscriptionStatus} from "../shared/paddle-utils";
+import {updateUsers} from "../shared/paddle-requests";
+import {statusExclude} from "../shared/constants";
 import {NotificationsService} from "../services/notifications/notifications.service";
 
 export default class ProjectMembersController extends WorklenzControllerBase {
@@ -69,6 +72,70 @@ export default class ProjectMembersController extends WorklenzControllerBase {
 
     if (!req.user?.team_id) return res.status(200).send(new ServerResponse(false, "Required fields are missing."));
 
+    // check the subscription status
+    const subscriptionData = await checkTeamSubscriptionStatus(req.user?.team_id);
+
+    const userExists = await this.checkIfUserAlreadyExists(req.user?.owner_id as string, req.body.email);
+
+    // Return error if user already exists
+    if (userExists) {
+      return res.status(200).send(new ServerResponse(false, null, "User already exists in the team."));
+    }
+
+    // Handle self-hosted subscriptions differently
+    if (subscriptionData.subscription_type === 'SELF_HOSTED') {
+      // Adding as a team member
+      const teamMemberReq: { team_id?: string; emails: string[], project_id?: string; } = {
+        team_id: req.user?.team_id,
+        emails: [req.body.email]
+      };
+
+      if (req.body.project_id)
+        teamMemberReq.project_id = req.body.project_id;
+
+      const [member] = await TeamMembersController.createOrInviteMembers(teamMemberReq, req.user);
+
+      if (!member)
+        return res.status(200).send(new ServerResponse(true, null, "Failed to add the member to the project. Please try again."));
+
+      // Adding to the project
+      const projectMemberReq = {
+        team_member_id: member.team_member_id,
+        team_id: req.user?.team_id,
+        project_id: req.body.project_id,
+        user_id: req.user?.id,
+        access_level: req.body.access_level ? req.body.access_level : "MEMBER"
+      };
+      const data = await this.createOrInviteMembers(projectMemberReq);
+      return res.status(200).send(new ServerResponse(true, data.member));
+    }
+
+    if (statusExclude.includes(subscriptionData.subscription_status)) {
+      return res.status(200).send(new ServerResponse(false, null, "Unable to add user! Please check your subscription status."));
+    }
+
+    if (!userExists && subscriptionData.is_ltd && subscriptionData.current_count && (parseInt(subscriptionData.current_count) + 1 > parseInt(subscriptionData.ltd_users))) {
+      return res.status(200).send(new ServerResponse(false, null, "Maximum number of life time users reached."));
+    }
+
+    // if (subscriptionData.status === "trialing") break;
+    if (!userExists && !subscriptionData.is_credit && !subscriptionData.is_custom && subscriptionData.subscription_status !== "trialing") {
+      // if (subscriptionData.subscription_status === "active") {
+      //   const response = await updateUsers(subscriptionData.subscription_id, (subscriptionData.quantity + 1));
+      //   if (!response.body.subscription_id) return res.status(200).send(new ServerResponse(false, null, response.message || "Unable to add user! Please check your subscription."));
+      // }
+      const updatedCount = parseInt(subscriptionData.current_count) + 1;
+      const requiredSeats = updatedCount - subscriptionData.quantity;
+      if (updatedCount > subscriptionData.quantity) {
+        const obj = { 
+          seats_enough: false, 
+          required_count: requiredSeats,
+          current_seat_amount: subscriptionData.quantity
+        };
+        return res.status(200).send(new ServerResponse(false, obj, null));
+      }
+    }
+
     // Adding as a team member
     const teamMemberReq: { team_id?: string; emails: string[], project_id?: string; } = {
       team_id: req.user?.team_id,