edr-platform/apps/edr-freight-api/src/modules/warehouses/warehouse-yards.repository.ts

import { BaseRepository } from '@edr/api-common';
import { Injectable } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { DeepPartial, Repository } from 'typeorm';

import { CargoType } from '../rule-engine/entities/cargo-type.entity';
import { WarehouseYard } from './entities/warehouse-yard.entity';

@Injectable()
export class WarehouseYardsRepository extends BaseRepository<WarehouseYard> {
  constructor(@InjectRepository(WarehouseYard) repository: Repository<WarehouseYard>) {
    super(repository);
  }

  /** The cargoTypes relation can't ride a column UPDATE — sync it via entity save, like the plain columns. */
  async update(id: string, data: DeepPartial<WarehouseYard>): Promise<WarehouseYard | null> {
    const { cargoTypes, ...columns } = data;
    if (Object.keys(columns).length) {
      await this.repository.update(id, columns as never);
    }
    if (cargoTypes) {
      const entity = await this.repository.findOne({ where: { id } as never });
      if (entity) {
        entity.cargoTypes = cargoTypes as CargoType[];
        await this.repository.save(entity);
      }
    }
    return this.findById(id);
  }
}