booking operations and trains scheduling also allocations

apps/edr-freight-api/src/modules/wagons/wagons.service.ts

@@ -1,3 +1,4 @@
+import { WagonReadiness, WagonStatus } from '@edr/types';
 import { Injectable, NotFoundException, ConflictException } from '@nestjs/common';
 import { InjectRepository } from '@nestjs/typeorm';
 import { Repository, DataSource, FindOptionsOrder, FindOptionsWhere, ILike } from 'typeorm';
@@ -19,7 +20,11 @@ export class WagonsService {
   ) {}
 
   async create(dto: CreateWagonDto): Promise<Wagon> {
-    const wagon = this.wagonRepo.create(dto);
+    const wagon = this.wagonRepo.create({
+      ...dto,
+      status: dto.status ?? WagonStatus.Available,
+      readiness: dto.readiness ?? WagonReadiness.ImportReady,
+    });
     // Convert undefined to null for nullable fields
     if (dto.trainId === undefined) wagon.trainId = null;
     if (dto.sequenceNumber === undefined) wagon.sequenceNumber = null;
@@ -30,23 +35,28 @@ export class WagonsService {
     const where: FindOptionsWhere<Wagon>[] | FindOptionsWhere<Wagon> = [];
     const search = query.search?.trim();
     const status = query.status?.trim();
+    const readiness = query.readiness?.trim();
     const trainId = query.trainId?.trim();
+    const filters = {
+      ...(status ? { status: status as Wagon['status'] } : {}),
+      ...(readiness ? { readiness: readiness as Wagon['readiness'] } : {}),
+      ...(trainId ? { trainId } : {}),
+    };
 
     if (search) {
       where.push({
         wagonNumber: ILike(`%${search}%`),
-        ...(status ? { status } : {}),
-        ...(trainId ? { trainId } : {}),
+        ...filters,
       });
     }
 
-    const sortBy = ['wagonNumber', 'tareWeight', 'maxPayloadWeight', 'status', 'sequenceNumber'].includes(query.sortBy ?? '')
+    const sortBy = ['wagonNumber', 'tareWeight', 'maxPayloadWeight', 'status', 'readiness', 'sequenceNumber'].includes(query.sortBy ?? '')
       ? (query.sortBy as keyof Wagon)
       : 'wagonNumber';
     const sortOrder = query.sortOrder?.toUpperCase() === 'DESC' ? 'DESC' : 'ASC';
 
     return this.wagonRepo.find({
-      where: search ? where : { ...(status ? { status } : {}), ...(trainId ? { trainId } : {}) },
+      where: search ? where : filters,
       order: { [sortBy]: sortOrder } as FindOptionsOrder<Wagon>,
       skip: query.page && query.limit ? (Number(query.page) - 1) * Number(query.limit) : undefined,
       take: query.limit ? Number(query.limit) : undefined,
@@ -72,7 +82,7 @@ export class WagonsService {
 
   async assignToTrain(wagonId: string, dto: AssignWagonToTrainDto): Promise<Wagon> {
     const wagon = await this.findById(wagonId);
-    if (wagon.status === 'ASSIGNED') {
+    if (wagon.status === WagonStatus.Assigned) {
       throw new ConflictException('Wagon already assigned to a train');
     }
 
@@ -91,7 +101,7 @@ export class WagonsService {
 
     wagon.trainId = train.id;
     wagon.sequenceNumber = sequence;
-    wagon.status = 'ASSIGNED';
+    wagon.status = WagonStatus.Assigned;
     return this.wagonRepo.save(wagon);
   }
 
@@ -99,7 +109,7 @@ export class WagonsService {
     const wagon = await this.findById(wagonId);
     wagon.trainId = null;
     wagon.sequenceNumber = null;
-    wagon.status = 'AVAILABLE';
+    wagon.status = WagonStatus.Available;
     return this.wagonRepo.save(wagon);
   }