Refactor controllers to return UUID of added entities

src/unit/controllers/unit.controller.ts

@@ -32,8 +32,8 @@ export class UnitController {
   @Post()
   async addUnit(@Body() addUnitDto: AddUnitDto) {
     try {
-      await this.unitService.addUnit(addUnitDto);
-      return { message: 'Unit added successfully' };
+      const unit = await this.unitService.addUnit(addUnitDto);
+      return { message: 'Unit added successfully', uuid: unit.uuid };
     } catch (error) {
       throw new HttpException(
         error.message || 'Internal server error',

src/unit/services/unit.service.ts

@@ -35,11 +35,12 @@ export class UnitService {
         },
       });
 
-      await this.spaceRepository.save({
+      const unit = await this.spaceRepository.save({
         spaceName: addUnitDto.unitName,
         parent: { uuid: addUnitDto.floorUuid },
         spaceType: { uuid: spaceType.uuid },
       });
+      return unit;
     } catch (err) {
       throw new HttpException(err.message, HttpStatus.INTERNAL_SERVER_ERROR);
     }