Add functionality to add user to a unit

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')