Add unit module with controller, service, dtos, and middleware

src/unit/controllers/unit.controller.ts

@@ -0,0 +1,99 @@
+import { UnitService } from '../services/unit.service';
+import {
+  Body,
+  Controller,
+  Get,
+  HttpException,
+  HttpStatus,
+  Param,
+  Post,
+  Query,
+  UseGuards,
+} 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 { GetUnitChildDto } from '../dtos/get.unit.dto';
+
+@ApiTags('Unit Module')
+@Controller({
+  version: '1',
+  path: 'unit',
+})
+export class UnitController {
+  constructor(private readonly unitService: UnitService) {}
+
+  @ApiBearerAuth()
+  @UseGuards(JwtAuthGuard)
+  @Post()
+  async addUnit(@Body() addUnitDto: AddUnitDto) {
+    try {
+      await this.unitService.addUnit(addUnitDto);
+      return { message: 'Unit added successfully' };
+    } catch (error) {
+      throw new HttpException(
+        error.message || 'Internal server error',
+        HttpStatus.INTERNAL_SERVER_ERROR,
+      );
+    }
+  }
+
+  @ApiBearerAuth()
+  @UseGuards(JwtAuthGuard)
+  @Get(':unitUuid')
+  async getUnitByUuid(@Param('unitUuid') unitUuid: string) {
+    try {
+      const unit = await this.unitService.getUnitByUuid(unitUuid);
+      return unit;
+    } catch (error) {
+      if (error.status === 404) {
+        throw new HttpException('Unit not found', HttpStatus.NOT_FOUND);
+      } else {
+        throw new HttpException(
+          error.message || 'Internal server error',
+          HttpStatus.INTERNAL_SERVER_ERROR,
+        );
+      }
+    }
+  }
+
+  @ApiBearerAuth()
+  @UseGuards(JwtAuthGuard)
+  @Get('child/:unitUuid')
+  async getUnitChildByUuid(
+    @Param('unitUuid') unitUuid: string,
+    @Query() query: GetUnitChildDto,
+  ) {
+    try {
+      const unit = await this.unitService.getUnitChildByUuid(unitUuid, query);
+      return unit;
+    } catch (error) {
+      if (error.status === 404) {
+        throw new HttpException('Unit not found', HttpStatus.NOT_FOUND);
+      } else {
+        throw new HttpException(
+          error.message || 'Internal server error',
+          HttpStatus.INTERNAL_SERVER_ERROR,
+        );
+      }
+    }
+  }
+  @ApiBearerAuth()
+  @UseGuards(JwtAuthGuard)
+  @Get('parent/:unitUuid')
+  async getUnitParentByUuid(@Param('unitUuid') unitUuid: string) {
+    try {
+      const unit = await this.unitService.getUnitParentByUuid(unitUuid);
+      return unit;
+    } catch (error) {
+      if (error.status === 404) {
+        throw new HttpException('Unit not found', HttpStatus.NOT_FOUND);
+      } else {
+        throw new HttpException(
+          error.message || 'Internal server error',
+          HttpStatus.INTERNAL_SERVER_ERROR,
+        );
+      }
+    }
+  }
+}