diff --git a/src/resolvers/SystemResolvers.ts b/src/resolvers/SystemResolvers.ts index a270798..dbb98ca 100644 --- a/src/resolvers/SystemResolvers.ts +++ b/src/resolvers/SystemResolvers.ts @@ -14,6 +14,8 @@ export const SystemResolvers: Resolvers = { const stop = await contextValue.repository.getStopById(args.id); if (stop === null) return null; + if (stop.systemId !== parent.id) return null; + return { id: stop.id, name: stop.name, @@ -25,6 +27,8 @@ export const SystemResolvers: Resolvers = { const route = await contextValue.repository.getRouteById(args.id); if (route === null) return null; + if (route.systemId !== parent.id) return null; + return { color: route.color, id: route.id, @@ -37,6 +41,8 @@ export const SystemResolvers: Resolvers = { const shuttle = await contextValue.repository.getShuttleById(args.id); if (shuttle === null) return null; + if (shuttle.systemId !== parent.id) return null; + return shuttle; }, shuttles: async (parent, args, contextValue, info) => {