stacjownik/src/managers/trainFilterManager.ts

import { TrainFilter, TrainFilterId } from '../components/TrainsView/typings';
import { Train, TrainStop } from '../typings/common';

function confirmedPercentage(stops: TrainStop[] | undefined) {
  if (!stops) return -1;

  return Number(((stops.filter((stop) => stop.confirmed).length / stops.length) * 100).toFixed(0));
}

function currentDelay(stops: TrainStop[] | undefined) {
  if (!stops) return -Infinity;

  const delay =
    stops.find(
      (stop, i) =>
        (i == 0 && !stop.confirmed) || (i > 0 && stops[i - 1].confirmed && !stop.confirmed)
    )?.departureDelay || 0;

  return delay;
}

function filterTrainList(
  trainList: Train[],
  searchedTrain: string,
  searchedDriver: string,
  filters: TrainFilter[]
) {
  return trainList.filter((train) => {
    const isFiltered = filters.every((f) => {
      if (f.isActive) return true;

      switch (f.id) {
        case TrainFilterId.noTimetable:
          return train.timetableData;

        case TrainFilterId.withTimetable:
          return !train.timetableData;

        case TrainFilterId.withComments:
          return !train.timetableData?.followingStops.some((stop) => stop.comments);

        case TrainFilterId.noComments:
          return train.timetableData?.followingStops.some((stop) => stop.comments);

        case TrainFilterId.twr:
          return !train.timetableData?.twr;

        case TrainFilterId.pn:
          return !train.timetableData?.hasExtraDeliveries;

        case TrainFilterId.tn:
          return !train.timetableData?.hasDangerousCargo;

        case TrainFilterId.common:
          return train.timetableData?.twr;

        case TrainFilterId.passenger:
          return !/^[AMRE]\D{2}$/.test(train.timetableData?.category || '');

        case TrainFilterId.freight:
          return !train.timetableData?.category.startsWith('T');

        case TrainFilterId.other:
          return !/^[PXZL]\D{2}$/.test(train.timetableData?.category || '');

        default:
          return true;
      }
    });

    return (
      (searchedTrain.length > 0 ? train.trainNo.toString().startsWith(searchedTrain) : true) &&
      (searchedDriver.length > 0
        ? train.driverName.toLowerCase().startsWith(searchedDriver.toLowerCase())
        : true) &&
      isFiltered
    );
  });
}

function sortTrainList(trainList: Train[], sorterActive: { id: string; dir: number }) {
  return trainList.sort((a: Train, b: Train) => {
    switch (sorterActive.id) {
      case 'id':
        if ((a.timetableData?.timetableId || -1) > (b.timetableData?.timetableId || -1))
          return sorterActive.dir;

        return -sorterActive.dir;

      case 'mass':
        if (a.mass > b.mass) return sorterActive.dir;
        return -sorterActive.dir;

      case 'routeDistance':
        if ((a.timetableData?.routeDistance || -1) > (b.timetableData?.routeDistance || -1))
          return sorterActive.dir;

        return -sorterActive.dir;

      case 'progress':
        if (
          confirmedPercentage(a.timetableData?.followingStops) >
          confirmedPercentage(b.timetableData?.followingStops)
        )
          return sorterActive.dir;

        return -sorterActive.dir;

      case 'delay':
        if (
          currentDelay(a.timetableData?.followingStops) >
          currentDelay(b.timetableData?.followingStops)
        )
          return sorterActive.dir;

        return -sorterActive.dir;

      case 'speed':
        if (a.speed > b.speed) return sorterActive.dir;
        return -sorterActive.dir;

      case 'timetable':
        if (a.trainNo > b.trainNo) return sorterActive.dir;
        return -sorterActive.dir;

      case 'length':
        if (a.length > b.length) return sorterActive.dir;
        return -sorterActive.dir;

      default:
        break;
    }

    return 0;
  });
}

export function filteredTrainList(
  trainList: Train[],
  searchedTrain: string,
  searchedDriver: string,
  sorterActive: { id: string; dir: number },
  filters: TrainFilter[]
) {
  const filtered = filterTrainList(trainList, searchedTrain, searchedDriver, filters);
  return [...sortTrainList(filtered, sorterActive)];
}