MIM-589: Add endpoint for getting maintenance units by property code (#82)

* Add endpoint for getting maintenance units by property code

* Simplify query and improve error handling

---------

Co-authored-by: Henrik Persson <henrik.persson@iteam.com>

Author: henrikhenrikpersson

packages/property-base/src/adapters/maintenance-units-adapter.ts

@@ -1,5 +1,6 @@
 import { trimStrings } from '@src/utils/data-conversion'
 import { prisma } from './db'
+import { MaintenanceUnit } from '@src/types/maintenance-unit'
 
 export const getMaintenanceUnitsByRentalId = async (rentalId: string) => {
   /**
@@ -55,11 +56,52 @@ export const getMaintenanceUnitsByRentalId = async (rentalId: string) => {
       rentalPropertyId: rentalPropertyInfoTrimmed.rentalId,
       code: item.maintenanceUnit?.code,
       caption: item?.maintenanceUnit?.name,
-      type: item.maintenanceUnit?.maintenanceUnitType?.name,
-      estateCode: rentalPropertyInfoTrimmed.propertyCode,
-      estate: rentalPropertyInfoTrimmed.propertyName,
+      type: item.maintenanceUnit?.maintenanceUnitType?.name ?? null,
+      propertyCode: rentalPropertyInfoTrimmed.propertyCode,
+      propertyName: rentalPropertyInfoTrimmed.propertyName,
     }
   })
 
   return maintenanceUnitPropertyStructuresMapped
 }
+
+export const getMaintenanceUnitsByPropertyCode = async (
+  propertyCode: string
+): Promise<MaintenanceUnit[]> => {
+  const maintenanceUnits = await prisma.maintenanceUnit.findMany({
+    where: {
+      propertyStructures: {
+        some: {
+          propertyCode: propertyCode,
+        },
+      },
+    },
+    select: {
+      id: true,
+      code: true,
+      name: true,
+      maintenanceUnitType: {
+        select: {
+          name: true,
+        },
+      },
+      propertyStructures: {
+        select: {
+          propertyCode: true,
+          propertyName: true,
+        },
+      },
+    },
+  })
+
+  return trimStrings(maintenanceUnits).map((item) => {
+    return {
+      id: item.id,
+      code: item.code,
+      caption: item.name,
+      type: item.maintenanceUnitType?.name ?? null,
+      propertyCode: item.propertyStructures[0]?.propertyCode ?? null,
+      propertyName: item.propertyStructures[0]?.propertyName ?? null,
+    }
+  })
+}

packages/property-base/src/routes/maintenance-units.ts

@@ -1,6 +1,12 @@
 import KoaRouter from '@koa/router'
-import { getMaintenanceUnitsByRentalId } from '@src/adapters/maintenance-units-adapter'
-import { MaintenanceUnitSchema } from '@src/types/maintenance-unit'
+import {
+  getMaintenanceUnitsByRentalId,
+  getMaintenanceUnitsByPropertyCode,
+} from '@src/adapters/maintenance-units-adapter'
+import {
+  MaintenanceUnit,
+  MaintenanceUnitSchema,
+} from '@src/types/maintenance-unit'
 import { generateRouteMetadata, logger } from 'onecore-utilities'
 
 /**
@@ -70,4 +76,64 @@ export const routes = (router: KoaRouter) => {
       ctx.body = { reason: errorMessage, ...metadata }
     }
   })
+
+  /**
+   * @swagger
+   * /maintenance-units/by-property-code/{code}:
+   *   get:
+   *     summary: Get all maintenance units for a specific property code
+   *     description: |
+   *       Retrieves all maintenance units associated with a given property code.
+   *     tags:
+   *       - Maintenance units
+   *     parameters:
+   *       - in: path
+   *         name: code
+   *         required: true
+   *         schema:
+   *           type: string
+   *         description: The code of the property or which to retrieve maintenance units.
+   *     responses:
+   *       200:
+   *         description: Successfully retrieved the maintenance units.
+   *         content:
+   *           application/json:
+   *             schema:
+   *               type: object
+   *               properties:
+   *                 content:
+   *                   type: array
+   *                   items:
+   *                     $ref: '#/components/schemas/MaintenanceUnit'
+   *       400:
+   *         description: Invalid query parameters.
+   *       500:
+   *         description: Internal server error.
+   */
+  router.get('(.*)/maintenance-units/by-property-code/:code', async (ctx) => {
+    const metadata = generateRouteMetadata(ctx)
+    const { code } = ctx.params
+    logger.info(`GET /maintenance-units/by-property-code/${code}`, metadata)
+
+    try {
+      const response = await getMaintenanceUnitsByPropertyCode(code)
+
+      if (!response) {
+        ctx.status = 404
+        return
+      }
+
+      ctx.body = {
+        content: response satisfies MaintenanceUnit[],
+        ...metadata,
+      }
+    } catch (err) {
+      logger.error(
+        `Error fetching maintenance units for property code ${code}: ${err}`
+      )
+      ctx.status = 500
+      const errorMessage = err instanceof Error ? err.message : 'unknown error'
+      ctx.body = { reason: errorMessage, ...metadata }
+    }
+  })
 }

packages/property-base/src/types/maintenance-unit.ts

@@ -2,12 +2,12 @@ import { z } from 'zod'
 
 export const MaintenanceUnitSchema = z.object({
   id: z.string(),
-  rentalPropertyId: z.string(),
+  rentalPropertyId: z.string().optional(),
   code: z.string(),
-  caption: z.string(),
+  caption: z.string().nullable(),
   type: z.string().nullable(),
-  estateCode: z.string(),
-  estate: z.string(),
+  propertyCode: z.string().nullable(),
+  propertyName: z.string().nullable(),
 })
 
-export type MaintenanceUnitInfo = z.infer<typeof MaintenanceUnitSchema>
+export type MaintenanceUnit = z.infer<typeof MaintenanceUnitSchema>