feat(be): departures v2 minutesAfter

krystxf · Nov 16, 2024 · 04ae847 · 04ae847
1 parent 0a3522e
commit 04ae847
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/apps/backend/src/modules/departure/departure-v2.service.ts b/apps/backend/src/modules/departure/departure-v2.service.ts
@@ -23,6 +23,7 @@ export class DepartureServiceV2 {
         limit: number | null;
         totalLimit: number | null;
         minutesBefore: number;
+        minutesAfter: number;
     }): Promise<DepartureSchema[]> {
         const vehicleTypeWhere =
             args.vehicleType === "metro"
@@ -71,7 +72,7 @@ export class DepartureServiceV2 {
                         mode: "departures",
                         order: "real",
                         minutesBefore: String(args.minutesBefore),
-                        minutesAfter: String(10 * 60),
+                        minutesAfter: String(args.minutesAfter),
                         limit: String(args.totalLimit ?? 1_000),
                     }),
                 ),

diff --git a/apps/backend/src/modules/departure/departure.controller.ts b/apps/backend/src/modules/departure/departure.controller.ts
@@ -184,18 +184,16 @@ export class DepartureController {
         {
             name: "platform[]",
             description: "Platform IDs",
-            type: String,
-            isArray: true,
+            type: [String],
             example: ["U1040Z101P", "U1040Z102P"],
             allowEmptyValue: true,
             required: false,
         },
         {
             name: "stop[]",
             description: "Stop IDs",
-            type: String,
+            type: [String],
             example: ["U1040"],
-            isArray: true,
             allowEmptyValue: true,
             required: false,
         },
@@ -227,6 +225,13 @@ export class DepartureController {
             example: 0,
             required: false,
         },
+        {
+            name: "minutesAfter",
+            description: "Minutes After",
+            type: "integer",
+            example: 2 * 60,
+            required: false,
+        },
         {
             name: "limit",
             description: "Limit of results per platform and route",
@@ -260,6 +265,7 @@ export class DepartureController {
                 .nullable()
                 .default(null), // total limit of results (departures)
             minutesBefore: z.coerce.number().optional().nullable().default(0),
+            minutesAfter: z.coerce.number().optional().nullable().default(null),
         });
         const parsed = schema.safeParse(query);
         if (!parsed.success) {
@@ -285,6 +291,7 @@ export class DepartureController {
             limit: parsedQuery.limit ?? null,
             totalLimit: parsedQuery.totalLimit ?? null,
             minutesBefore: parsedQuery.minutesBefore ?? 0,
+            minutesAfter: parsedQuery.minutesAfter ?? 2 * 60,
         });
 
         departureSchema.array().parse(departures);