diff --git a/src/components/DriverView/DriverPropositions.vue b/src/components/DriverView/DriverPropositions.vue index c59934b..6bd8757 100644 --- a/src/components/DriverView/DriverPropositions.vue +++ b/src/components/DriverView/DriverPropositions.vue @@ -1,7 +1,5 @@ @@ -145,6 +158,27 @@ const chosenCategory = computed(() => { return availableCategories.value[chosenCategoryIndex.value]; }); +const getCargoWarnings = computed(() => { + const stockList = props.chosenTrain.stockList; + + let warnings: Set = new Set(); + + stockList.forEach((stockVehicle) => { + const [vehicleName, vehicleCargo] = stockVehicle.split(':'); + + if (vehicleName.startsWith('WB117')) warnings.add(vehicleCargo ? 'twr-un1965' : 'tn-un1965'); + else if (vehicleName.startsWith('445Rb')) warnings.add('tn-un1202'); + else if (vehicleName.startsWith('EDK80')) warnings.add('pn-edk80'); + + if (vehicleCargo) { + if (vehicleCargo.startsWith('wt_20')) warnings.add('pn-innofreight'); + else if (/^(tank|vehicles_01|truck)/.test(vehicleCargo)) warnings.add('pn-military'); + } + }); + + return warnings; +}); + const availableCategories = computed(() => { const stockList = props.chosenTrain.stockList; const headVehicle = stockList[0]?.split('-')[0] ?? ''; @@ -221,6 +255,7 @@ function selectCategory(i: number) {