booking operations and trains scheduling also allocations

apps/edr-freight-api/src/modules/wagons/dto/create-wagon.dto.ts

@@ -1,4 +1,5 @@
-import { IsString, IsUUID, IsOptional, IsInt, Min, IsNumber, IsIn } from 'class-validator';
+import { WagonReadiness, WagonStatus } from '@edr/types';
+import { IsString, IsUUID, IsOptional, IsInt, Min, IsNumber, IsEnum } from 'class-validator';
 
 export class CreateWagonDto {
   @IsString()
@@ -25,10 +26,14 @@ export class CreateWagonDto {
   maxPayloadWeight!: number;
 
   @IsOptional()
-  @IsIn(['AVAILABLE', 'ASSIGNED', 'MAINTENANCE', 'RETIRED'])
-  status?: string;
+  @IsEnum(WagonStatus)
+  status?: WagonStatus;
+
+  @IsOptional()
+  @IsEnum(WagonReadiness)
+  readiness?: WagonReadiness;
 
   @IsOptional()
   @IsString()
   notes?: string;
-}
+}

apps/edr-freight-api/src/modules/wagons/entities/wagon.entity.ts

@@ -1,10 +1,29 @@
 // apps/edr-freight-api/src/modules/wagons/entities/wagon.entity.ts
-import { Entity, Column, ManyToOne, OneToMany, JoinColumn } from 'typeorm';
+import { WagonReadiness, WagonStatus } from '@edr/types';
+import { Entity, Column, ManyToOne, OneToMany, JoinColumn, Index } from 'typeorm';
 import { BaseEntity } from '@edr/api-common';
 import { Train } from '../../trains/entities/train.entity';
+import { TrainSchedule } from '../../train-schedules/entities/train-schedule.entity';
+import { TrainSetWagon } from '../../train-sets/entities/train-set-wagon.entity';
 import { Container } from '../../container-management/entities/container.entity';
 
+export const WAGON_STATUSES = [
+  WagonStatus.Available,
+  WagonStatus.Assigned,
+  WagonStatus.Maintenance,
+  WagonStatus.Retired,
+] as const;
+
+export const WAGON_READINESS_VALUES = [
+  WagonReadiness.ImportReady,
+  WagonReadiness.ExportReady,
+] as const;
+
+export type WagonStatusType = (typeof WAGON_STATUSES)[number];
+export type WagonReadinessType = (typeof WAGON_READINESS_VALUES)[number];
+
 @Entity({ name: 'wagons', schema: 'freight' })
+@Index(['readiness'])
 export class Wagon extends BaseEntity {
   @Column({ unique: true, name: 'wagon_number' })
   wagonNumber!: string;
@@ -24,13 +43,30 @@ export class Wagon extends BaseEntity {
   @Column({ name: 'max_payload_weight', type: 'decimal', precision: 10, scale: 2 })
   maxPayloadWeight!: number;
 
-  @Column({ type: 'varchar', default: 'AVAILABLE' })
-  status!: string; // AVAILABLE, ASSIGNED, MAINTENANCE, RETIRED
+  @Column({ type: 'varchar', length: 20, default: WagonStatus.Available })
+  status!: WagonStatusType;
+
+  @Column({ type: 'varchar', length: 20, default: WagonReadiness.ImportReady })
+  readiness!: WagonReadinessType;
 
   @Column({ type: 'text', nullable: true })
   notes!: string | null;
 
-  // Relationship to Train
+  @Column({ name: 'train_set_wagon_id', type: 'uuid', nullable: true })
+  trainSetWagonId!: string | null;
+
+  @ManyToOne(() => TrainSetWagon, { nullable: true, onDelete: 'SET NULL' })
+  @JoinColumn({ name: 'train_set_wagon_id' })
+  trainSetWagon?: TrainSetWagon | null;
+
+  @Column({ name: 'current_train_schedule_id', type: 'uuid', nullable: true })
+  currentTrainScheduleId!: string | null;
+
+  @ManyToOne(() => TrainSchedule, { nullable: true, onDelete: 'SET NULL' })
+  @JoinColumn({ name: 'current_train_schedule_id' })
+  currentTrainSchedule?: TrainSchedule | null;
+
+  /** Fleet master consist grouping — separate from operational train_schedules. */
   @ManyToOne(() => Train, (train) => train.wagons, { onDelete: 'SET NULL' })
   @JoinColumn({ name: 'train_id' })
   train!: Train | null;

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);
   }