import { Pipe, PipeTransform } from '@angular/core';

import { PMTrailerShortResponse } from '../../../models/pMTrailerShortResponse';
import { PMTruckShortResponse } from '../../..//models/pMTruckShortResponse';

@Pipe({
    name: 'pmItem',
})
export class PmItemPipe implements PipeTransform {
    transform<
        T extends {
            pmTruck: PMTruckShortResponse;
            pmTrailer: PMTrailerShortResponse;
        },
    >(item: T): string {
        const { pmTruck, pmTrailer } = item;

        const pmIcon = !!pmTruck ? pmTruck.logoName : pmTrailer.logoName;

        return `assets/svg/common/repair-pm/${pmIcon}`;
    }
}