diff --git a/src/resolvers.ts b/src/resolvers.ts index 5dc5691..80df9b2 100644 --- a/src/resolvers.ts +++ b/src/resolvers.ts @@ -88,6 +88,20 @@ export const resolvers: Resolvers = { } }, Route: { + shuttles: async (parent, args, contextValue, info) => { + const shuttles = await contextValue.repository.getShuttlesByRouteId(parent.id); + + return shuttles.map(({ + coordinates, + name, + id, + }) => ({ + coordinates: coordinates as Coordinates, + name, + route: parent, + id, + })); + } }, Shuttle: { eta: (parent, args, contextValue, info) => {