From d9782b3218e60f195ae68fb1d972ff18c1329f98 Mon Sep 17 00:00:00 2001
From: faris Aljohari <83524184+farisaljohari@users.noreply.github.com>
Date: Tue, 16 Apr 2024 17:22:47 +0300
Subject: [PATCH] Add functionality to add user to a unit

---
 src/unit/controllers/unit.controller.ts | 18 ++++++++++++++++--
 src/unit/dtos/add.unit.dto.ts           | 19 +++++++++++++++++++
 src/unit/services/unit.service.ts       | 22 +++++++++++++++++++++-
 src/unit/unit.module.ts                 |  4 ++++
 4 files changed, 60 insertions(+), 3 deletions(-)

diff --git a/src/unit/controllers/unit.controller.ts b/src/unit/controllers/unit.controller.ts
index 5289200..a3e59e9 100644
--- a/src/unit/controllers/unit.controller.ts
+++ b/src/unit/controllers/unit.controller.ts
@@ -13,10 +13,11 @@ import {
 } from '@nestjs/common';
 import { ApiTags, ApiBearerAuth } from '@nestjs/swagger';
 import { JwtAuthGuard } from '../../../libs/common/src/guards/jwt.auth.guard';
-import { AddUnitDto } from '../dtos/add.unit.dto';
+import { AddUnitDto, AddUserUnitDto } from '../dtos/add.unit.dto';
 import { GetUnitChildDto } from '../dtos/get.unit.dto';
 import { UpdateUnitNameDto } from '../dtos/update.unit.dto';
 import { CheckFloorTypeGuard } from 'src/guards/floor.type.guard';
+import { CheckUserUnitGuard } from 'src/guards/user.unit.guard';
 
 @ApiTags('Unit Module')
 @Controller({
@@ -87,7 +88,20 @@ export class UnitController {
       );
     }
   }
-
+  @ApiBearerAuth()
+  @UseGuards(JwtAuthGuard, CheckUserUnitGuard)
+  @Post('user')
+  async addUserUnit(@Body() addUserUnitDto: AddUserUnitDto) {
+    try {
+      await this.unitService.addUserUnit(addUserUnitDto);
+      return { message: 'user unit added successfully' };
+    } catch (error) {
+      throw new HttpException(
+        error.message || 'Internal server error',
+        error.status || HttpStatus.INTERNAL_SERVER_ERROR,
+      );
+    }
+  }
   @ApiBearerAuth()
   @UseGuards(JwtAuthGuard)
   @Get('user/:userUuid')
diff --git a/src/unit/dtos/add.unit.dto.ts b/src/unit/dtos/add.unit.dto.ts
index 40f164a..e42d1bb 100644
--- a/src/unit/dtos/add.unit.dto.ts
+++ b/src/unit/dtos/add.unit.dto.ts
@@ -21,3 +21,22 @@ export class AddUnitDto {
     Object.assign(this, dto);
   }
 }
+export class AddUserUnitDto {
+  @ApiProperty({
+    description: 'unitUuid',
+    required: true,
+  })
+  @IsString()
+  @IsNotEmpty()
+  public unitUuid: string;
+  @ApiProperty({
+    description: 'userUuid',
+    required: true,
+  })
+  @IsString()
+  @IsNotEmpty()
+  public userUuid: string;
+  constructor(dto: Partial<AddUserUnitDto>) {
+    Object.assign(this, dto);
+  }
+}
diff --git a/src/unit/services/unit.service.ts b/src/unit/services/unit.service.ts
index 27bebbc..867300c 100644
--- a/src/unit/services/unit.service.ts
+++ b/src/unit/services/unit.service.ts
@@ -7,7 +7,7 @@ import {
   BadRequestException,
 } from '@nestjs/common';
 import { SpaceRepository } from '@app/common/modules/space/repositories';
-import { AddUnitDto } from '../dtos';
+import { AddUnitDto, AddUserUnitDto } from '../dtos';
 import {
   UnitChildInterface,
   UnitParentInterface,
@@ -228,6 +228,26 @@ export class UnitService {
       }
     }
   }
+
+  async addUserUnit(addUserUnitDto: AddUserUnitDto) {
+    try {
+      await this.userSpaceRepository.save({
+        user: { uuid: addUserUnitDto.userUuid },
+        space: { uuid: addUserUnitDto.unitUuid },
+      });
+    } catch (err) {
+      if (err.code === '23505') {
+        throw new HttpException(
+          'User already belongs to this unit',
+          HttpStatus.BAD_REQUEST,
+        );
+      }
+      throw new HttpException(
+        err.message || 'Internal Server Error',
+        HttpStatus.INTERNAL_SERVER_ERROR,
+      );
+    }
+  }
   async renameUnitByUuid(
     unitUuid: string,
     updateUnitNameDto: UpdateUnitNameDto,
diff --git a/src/unit/unit.module.ts b/src/unit/unit.module.ts
index 2f0234e..569ef39 100644
--- a/src/unit/unit.module.ts
+++ b/src/unit/unit.module.ts
@@ -8,6 +8,8 @@ import { SpaceTypeRepositoryModule } from '@app/common/modules/space-type/space.
 import { SpaceTypeRepository } from '@app/common/modules/space-type/repositories';
 import { UserSpaceRepositoryModule } from '@app/common/modules/user-space/user.space.repository.module';
 import { UserSpaceRepository } from '@app/common/modules/user-space/repositories';
+import { UserRepositoryModule } from '@app/common/modules/user/user.repository.module';
+import { UserRepository } from '@app/common/modules/user/repositories';
 
 @Module({
   imports: [
@@ -15,6 +17,7 @@ import { UserSpaceRepository } from '@app/common/modules/user-space/repositories
     SpaceRepositoryModule,
     SpaceTypeRepositoryModule,
     UserSpaceRepositoryModule,
+    UserRepositoryModule,
   ],
   controllers: [UnitController],
   providers: [
@@ -22,6 +25,7 @@ import { UserSpaceRepository } from '@app/common/modules/user-space/repositories
     SpaceRepository,
     SpaceTypeRepository,
     UserSpaceRepository,
+    UserRepository,
   ],
   exports: [UnitService],
 })