refactor typów danych

src/managers/trainFilterManager.ts

@@ -0,0 +1,145 @@
+import { TrainFilter, TrainFilterId } from '../components/TrainsView/typings';
+import Train from '../scripts/interfaces/Train';
+import { TrainStop } from '../store/typings';
+
+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.skr:
+          return !train.timetableData?.SKR;
+
+        case TrainFilterId.common:
+          return train.timetableData?.SKR || 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)];
+}