refactor typów danych

src/store/utils.ts

@@ -0,0 +1,267 @@
+import Station from '../scripts/interfaces/Station';
+import Train from '../scripts/interfaces/Train';
+import { API } from '../typings/api';
+import { ScheduledTrain, StationTrain, StopStatus, TrainStop } from './typings';
+
+export function getLocoURL(locoType: string): string {
+  return `https://rj.td2.info.pl/dist/img/thumbnails/${
+    locoType.includes('EN') ? locoType + 'rb' : locoType
+  }.png`;
+}
+
+export function getStatusTimestamp(stationStatus: any): number {
+  if (!stationStatus) return -2;
+
+  const statusCode = stationStatus[2];
+  const statusTimestamp = stationStatus[3];
+
+  switch (statusCode) {
+    case 0:
+    case 1:
+    case 3:
+      return statusTimestamp;
+
+    case 2:
+      if (statusTimestamp == 0) return 0;
+      break;
+
+    default:
+      break;
+  }
+
+  return -1;
+}
+
+export function parseSpawns(spawnString: string | null) {
+  if (!spawnString) return [];
+  if (spawnString === 'NO_SPAWN') return [];
+
+  return spawnString.split(';').map((spawn) => {
+    const spawnArray = spawn.split(',');
+    const spawnName = spawnArray[6] ? spawnArray[6] : spawnArray[0];
+    const spawnLength = parseInt(spawnArray[2]);
+    const isElectrified = spawnArray[3] == 'True';
+
+    return { spawnName, spawnLength, isElectrified };
+  });
+}
+
+export function getTimestamp(date: string | null): number {
+  return date ? new Date(date).getTime() : 0;
+}
+
+export function getTrainStopStatus(
+  stopInfo: TrainStop,
+  currentStationName: string,
+  sceneryName: string
+) {
+  let stopStatus = StopStatus.ARRIVING,
+    stopLabel = '',
+    stopStatusID = -1;
+
+  if (stopInfo.terminatesHere && stopInfo.confirmed) {
+    stopStatus = StopStatus.TERMINATED;
+    stopLabel = 'Skończył bieg';
+    stopStatusID = 5;
+  } else if (!stopInfo.terminatesHere && stopInfo.confirmed && currentStationName == sceneryName) {
+    stopStatus = StopStatus.DEPARTED;
+    stopLabel = 'Odprawiony';
+    stopStatusID = 2;
+  } else if (!stopInfo.terminatesHere && stopInfo.confirmed && currentStationName != sceneryName) {
+    stopStatus = StopStatus.DEPARTED_AWAY;
+    stopLabel = 'Odjechał';
+    stopStatusID = 4;
+  } else if (currentStationName == sceneryName && !stopInfo.stopped) {
+    stopStatus = StopStatus.ONLINE;
+    stopLabel = 'Na stacji';
+    stopStatusID = 0;
+  } else if (currentStationName == sceneryName && stopInfo.stopped) {
+    stopStatus = StopStatus.STOPPED;
+    stopLabel = 'Postój';
+    stopStatusID = 1;
+  } else if (currentStationName != sceneryName) {
+    stopStatus = StopStatus.ARRIVING;
+    stopLabel = 'W drodze';
+    stopStatusID = 3;
+  }
+
+  return { stopStatus, stopLabel, stopStatusID };
+}
+
+export function getCheckpointTrain(
+  train: Train,
+  trainStopIndex: number,
+  sceneryName: string
+): ScheduledTrain {
+  const timetable = train.timetableData!;
+  const followingStops = timetable.followingStops;
+  const trainStop = followingStops[trainStopIndex];
+
+  const trainStopStatus = getTrainStopStatus(trainStop, train.currentStationName, sceneryName);
+
+  let prevStationName = '',
+    nextStationName = '';
+
+  let prevDepartureLine: string | null = null,
+    nextArrivalLine: string | null = null;
+
+  for (let i = trainStopIndex - 1; i >= 0; i--) {
+    if (/strong|podg/g.test(followingStops[i].stopName)) {
+      prevStationName = followingStops[i].stopNameRAW.replace(/,.*/g, '');
+
+      break;
+    }
+  }
+
+  for (let i = trainStopIndex + 1; i < followingStops.length; i++) {
+    if (/strong|podg/g.test(followingStops[i].stopName)) {
+      nextStationName = followingStops[i].stopNameRAW.replace(/,.*/g, '');
+
+      break;
+    }
+  }
+
+  let departureLine: string | null = null;
+  let arrivingLine: string | null = null;
+
+  for (let i = trainStopIndex; i < followingStops.length; i++) {
+    const currentStop = followingStops[i];
+
+    if (currentStop.departureLine == null) continue;
+
+    if (!/-|_|it|sbl/gi.test(currentStop.departureLine)) {
+      departureLine = currentStop.departureLine;
+      nextArrivalLine = followingStops[i + 1]?.arrivalLine || null;
+
+      break;
+    }
+  }
+
+  for (let i = trainStopIndex; i >= 0; i--) {
+    const currentStop = followingStops[i];
+
+    if (currentStop.arrivalLine == null) continue;
+
+    if (!/-|_|it|sbl/gi.test(currentStop.arrivalLine)) {
+      arrivingLine = currentStop.arrivalLine;
+      prevDepartureLine = followingStops[i - 1]?.departureLine || null;
+
+      break;
+    }
+  }
+
+  return {
+    checkpointName: trainStop.stopNameRAW,
+
+    trainNo: train.trainNo,
+    trainId: train.trainId,
+
+    signal: train.signal,
+    connectedTrack: train.connectedTrack,
+
+    driverName: train.driverName,
+    driverId: train.driverId,
+    currentStationName: train.currentStationName,
+    currentStationHash: train.currentStationHash,
+    category: timetable.category,
+    beginsAt: timetable.followingStops[0].stopNameRAW,
+    terminatesAt: timetable.followingStops[timetable.followingStops.length - 1].stopNameRAW,
+
+    nextStationName,
+    prevStationName,
+
+    stopInfo: trainStop,
+    stopLabel: trainStopStatus.stopLabel,
+    stopStatus: trainStopStatus.stopStatus,
+    stopStatusID: trainStopStatus.stopStatusID,
+
+    arrivingLine,
+    departureLine,
+
+    nextArrivalLine,
+    prevDepartureLine
+  };
+}
+
+export function getScheduledTrains(
+  trainList: Train[],
+  sceneryData: API.ActiveSceneries.Data,
+  stationGeneralInfo: Station['generalInfo']
+): ScheduledTrain[] {
+  const stationName = sceneryData.stationName.toLocaleLowerCase();
+
+  stationGeneralInfo?.checkpoints.forEach((cp) => (cp.scheduledTrains.length = 0));
+
+  return trainList.reduce((acc: ScheduledTrain[], train) => {
+    if (!train.timetableData) return acc;
+
+    const timetable = train.timetableData;
+    if (!timetable.sceneries.includes(sceneryData.stationHash)) return acc;
+
+    const stopInfoIndex = timetable.followingStops.findIndex((stop) => {
+      const stopName = stop.stopNameRAW.toLowerCase();
+
+      return (
+        stationName == stopName ||
+        (!/(po\.|podg\.)/.test(stopName) && stopName.includes(stationName)) ||
+        (!/(po\.|podg\.)/.test(stationName) && stationName.includes(stopName)) ||
+        (stopName.split(', podg.')[0] !== undefined &&
+          stationName.startsWith(stopName.split(', podg.')[0]))
+      );
+    });
+
+    const checkpointScheduledTrains: ScheduledTrain[] = [];
+
+    if (stopInfoIndex != -1) {
+      checkpointScheduledTrains.push(
+        getCheckpointTrain(train, stopInfoIndex, sceneryData.stationName)
+      );
+    }
+
+    stationGeneralInfo?.checkpoints?.forEach((checkpoint) => {
+      if (checkpoint.checkpointName.toLocaleLowerCase() == stationName) return;
+
+      if (
+        checkpointScheduledTrains.findIndex(
+          (cpTrain) =>
+            cpTrain.checkpointName.toLocaleLowerCase() ==
+            checkpoint.checkpointName.toLocaleLowerCase()
+        ) != -1
+      )
+        return;
+
+      const index = timetable.followingStops.findIndex(
+        (stop) => stop.stopNameRAW.toLowerCase() == checkpoint.checkpointName.toLowerCase()
+      );
+
+      if (index > -1)
+        checkpointScheduledTrains.push(getCheckpointTrain(train, index, sceneryData.stationName));
+    });
+
+    acc.push(...checkpointScheduledTrains);
+    return acc;
+  }, []) as ScheduledTrain[];
+}
+
+export function getStationTrains(
+  trainList: Train[],
+  scheduledTrainList: ScheduledTrain[],
+  region: string,
+  sceneryData: API.ActiveSceneries.Data
+): StationTrain[] {
+  return trainList
+    .filter(
+      (train) =>
+        train?.region === region &&
+        train.online &&
+        train.currentStationName === sceneryData.stationName
+    )
+    .map((train) => ({
+      driverName: train.driverName,
+      driverId: train.driverId,
+      trainNo: train.trainNo,
+      trainId: train.trainId,
+      stopStatus:
+        scheduledTrainList.find((st) => st.trainNo === train.trainNo)?.stopStatus || 'no-timetable'
+    }));
+}