fix(be): routeId in departures

krystxf · Dec 5, 2024 · 3abfd5e · 3abfd5e · vercel · Dec 5, 2024
1 parent b9aadd3
commit 3abfd5e
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/apps/backend/src/modules/departure/departure-v2.service.ts b/apps/backend/src/modules/departure/departure-v2.service.ts
@@ -95,13 +95,28 @@ export class DepartureServiceV2 {
             throw new Error(parsed.error.message);
         }
 
+        const gtfsRoutes = await this.prisma.gtfsRoute.findMany({
+            where: {
+                shortName: {
+                    in: parsed.data.departures.map(
+                        (departure) => departure.route.short_name,
+                    ),
+                },
+            },
+        });
+
         const parsedDepartures = parsed.data.departures.map((departure) => {
             return {
                 id: departure.trip.id,
                 departure: departure.departure_timestamp,
                 delay: getDelayInSeconds(departure.delay),
                 headsign: departure.trip.headsign,
                 route: departure.route.short_name,
+                routeId:
+                    gtfsRoutes.find(
+                        (gtfsRoute) =>
+                            gtfsRoute.shortName === departure.route.short_name,
+                    )?.id ?? null,
                 platformId: departure.stop.id,
                 platformCode: departure.stop.platform_code,
             };