feat: space limit is now fixed

migrations/20240716131219-add-number-of-seats-to-workspaces-table.js

@@ -0,0 +1,16 @@
+'use strict';
+
+const tableName = 'workspaces';
+const newColumn = 'number_of_seats';
+
+module.exports = {
+  async up(queryInterface, Sequelize) {
+    await queryInterface.addColumn(tableName, newColumn, {
+      type: Sequelize.INTEGER,
+    });
+  },
+
+  async down(queryInterface) {
+    await queryInterface.removeColumn(tableName, newColumn);
+  },
+};

src/modules/gateway/dto/initialize-workspace.dto.ts

@@ -17,10 +17,18 @@ export class InitializeWorkspaceDto {
   address?: string;
 
   @ApiProperty({
-    example: '312321312',
+    example: 312321312,
     description: 'Workspace max space in bytes',
   })
   @IsNotEmpty()
   @IsNumber()
   maxSpaceBytes: number;
+
+  @ApiProperty({
+    example: 20,
+    description: 'Workspace max number of users',
+  })
+  @IsNotEmpty()
+  @IsNumber()
+  numberOfSeats: number;
 }

src/modules/gateway/gateway.controller.spec.ts

@@ -27,6 +27,7 @@ describe('Gateway Controller', () => {
       ownerId: owner.uuid,
       maxSpaceBytes: 1000000,
       address: '123 Main St',
+      numberOfSeats: 20,
     };
 
     it('When workspace is created successfully, then return', async () => {

src/modules/gateway/gateway.usecase.spec.ts

@@ -56,6 +56,7 @@ describe('GatewayUseCases', () => {
         ownerId: owner.uuid,
         maxSpaceBytes,
         address: workspaceAddress,
+        numberOfSeats: 20,
       };
 
       await expect(
@@ -79,6 +80,7 @@ describe('GatewayUseCases', () => {
         ownerId: owner.uuid,
         maxSpaceBytes,
         address: workspaceAddress,
+        numberOfSeats: 20,
       };
       await expect(
         service.initializeWorkspace(initializeWorkspaceDto),

src/modules/gateway/gateway.usecase.ts

@@ -22,10 +22,12 @@ export class GatewayUseCases {
     Logger.log(
       `Initializing workspace with owner id: ${initializeWorkspaceDto.ownerId}`,
     );
-    const { ownerId, maxSpaceBytes, address } = initializeWorkspaceDto;
+    const { ownerId, maxSpaceBytes, address, numberOfSeats } =
+      initializeWorkspaceDto;
 
     return this.workspaceUseCases.initiateWorkspace(ownerId, maxSpaceBytes, {
       address,
+      numberOfSeats,
     });
   }
 

src/modules/workspaces/attributes/workspace.attributes.ts

@@ -8,6 +8,7 @@ export interface WorkspaceAttributes {
   defaultTeamId: string;
   workspaceUserId: string;
   setupCompleted: boolean;
+  numberOfSeats: number;
   rootFolderId?: string;
   createdAt: Date;
   updatedAt: Date;

src/modules/workspaces/domains/workspaces.domain.ts

@@ -13,6 +13,7 @@ export class Workspace implements WorkspaceAttributes {
   defaultTeamId: string;
   workspaceUserId: string;
   setupCompleted: boolean;
+  numberOfSeats: number;
   createdAt: Date;
   updatedAt: Date;
 
@@ -27,6 +28,7 @@ export class Workspace implements WorkspaceAttributes {
     workspaceUserId,
     setupCompleted,
     avatar,
+    numberOfSeats,
     createdAt,
     updatedAt,
   }: WorkspaceAttributes) {
@@ -40,6 +42,7 @@ export class Workspace implements WorkspaceAttributes {
     this.workspaceUserId = workspaceUserId;
     this.setupCompleted = setupCompleted;
     this.rootFolderId = rootFolderId;
+    this.numberOfSeats = numberOfSeats;
     this.createdAt = createdAt;
     this.updatedAt = updatedAt;
   }
@@ -56,6 +59,10 @@ export class Workspace implements WorkspaceAttributes {
     return this.setupCompleted === true;
   }
 
+  isWorkspaceFull(currentUsersCount: number) {
+    return this.numberOfSeats <= currentUsersCount;
+  }
+
   isDefaultTeam(team: WorkspaceTeam) {
     return this.defaultTeamId === team.id;
   }

src/modules/workspaces/dto/create-workspace-invite.dto.ts

@@ -1,5 +1,5 @@
 import { ApiProperty } from '@nestjs/swagger';
-import { IsNotEmpty, IsPositive } from 'class-validator';
+import { IsNotEmpty } from 'class-validator';
 import { User } from '../../user/user.domain';
 import { WorkspaceInvite } from '../domains/workspace-invite.domain';
 
@@ -11,13 +11,7 @@ export class CreateWorkspaceInviteDto {
   @IsNotEmpty()
   invitedUser: User['email'];
 
-  @ApiProperty({
-    example: '1073741824',
-    description: 'Space assigned to user in bytes',
-  })
-  @IsNotEmpty()
-  @IsPositive()
-  spaceLimit: WorkspaceInvite['spaceLimit'];
+  spaceLimit?: WorkspaceInvite['spaceLimit'];
 
   @ApiProperty({
     example: 'encrypted encryption key',

src/modules/workspaces/models/workspace.model.ts

@@ -61,6 +61,9 @@ export class WorkspaceModel extends Model implements WorkspaceAttributes {
   @Column(DataType.UUID)
   rootFolderId?: string;
 
+  @Column(DataType.INTEGER)
+  numberOfSeats: number;
+
   @HasOne(() => FolderModel, 'uuid')
   rootFolder: FolderModel;