From 0289bbbfdb027572fe35897a5d24c87f9d13dcc7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Anton=20Jyrki=C3=A4inen?= <anton.jyrkiainen@helsinki.fi>
Date: Wed, 8 Jul 2026 14:34:29 +0300
Subject: [PATCH] fix route variant trunk data

---
 src/creators/createRoutesResponse.ts | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/src/creators/createRoutesResponse.ts b/src/creators/createRoutesResponse.ts
index e71b60a..39ca0de 100644
--- a/src/creators/createRoutesResponse.ts
+++ b/src/creators/createRoutesResponse.ts
@@ -127,6 +127,26 @@ export async function createRoutesResponse(
 
     const filteredRoutes = Object.values(filteredRouteSet)
 
+    const routesByDirectionAndId = new Map<string, JoreRoute>()
+
+    filteredRoutes.forEach((route) => {
+      routesByDirectionAndId.set(`${route.direction}.${route.route_id}`, route)
+    })
+
+    filteredRoutes.forEach((route) => {
+      const [mainRouteId] = route.route_id.split(' ')
+
+      if (mainRouteId === route.route_id) {
+        return
+      }
+
+      const mainRoute = routesByDirectionAndId.get(`${route.direction}.${mainRouteId}`)
+
+      if (mainRoute?.trunk_route === '1') {
+        route.trunk_route = '1'
+      }
+    })
+
     return filteredRoutes.map((route) => {
       const routeCancellations = cancellations.filter(
         (cancellation) =>