From 5b775dfec948eeeaf684c045f9ddc487faebc360 Mon Sep 17 00:00:00 2001
From: Spythere <spythere@protonmail.com>
Date: Sat, 4 Nov 2023 17:00:50 +0100
Subject: [PATCH] fix: filtry RJ

---
 src/scripts/utils/filterUtils.ts | 20 ++++++--------------
 1 file changed, 6 insertions(+), 14 deletions(-)

diff --git a/src/scripts/utils/filterUtils.ts b/src/scripts/utils/filterUtils.ts
index c55c3ae..6535f38 100644
--- a/src/scripts/utils/filterUtils.ts
+++ b/src/scripts/utils/filterUtils.ts
@@ -53,28 +53,20 @@ export const sortStations = (
 
     case 'timetableConfirmed':
       diff =
-        (a.onlineInfo?.scheduledTrains
-          ? a.onlineInfo.scheduledTrains.filter((train) => train.stopInfo.confirmed).length
-          : -1) -
-        (b.onlineInfo?.scheduledTrains
-          ? b.onlineInfo.scheduledTrains.filter((train) => train.stopInfo.confirmed).length
-          : -1);
+        (a.onlineInfo?.scheduledTrainCount.confirmed ?? -1) -
+        (b.onlineInfo?.scheduledTrainCount.confirmed ?? -1);
       break;
 
     case 'timetableUnconfirmed':
       diff =
-        (a.onlineInfo?.scheduledTrains
-          ? a.onlineInfo.scheduledTrains.filter((train) => !train.stopInfo.confirmed).length
-          : -1) -
-        (b.onlineInfo?.scheduledTrains
-          ? b.onlineInfo.scheduledTrains.filter((train) => !train.stopInfo.confirmed).length
-          : -1);
+        (a.onlineInfo?.scheduledTrainCount.unconfirmed ?? -1) -
+        (b.onlineInfo?.scheduledTrainCount.unconfirmed ?? -1);
       break;
 
     case 'timetableAll':
       diff =
-        (a.onlineInfo?.scheduledTrains ? a.onlineInfo.scheduledTrains.length : -1) -
-        (b.onlineInfo?.scheduledTrains ? b.onlineInfo.scheduledTrains.length : -1);
+        (a.onlineInfo?.scheduledTrainCount.all ?? -1) -
+        (b.onlineInfo?.scheduledTrainCount.all ?? -1);
       break;
 
     default: