From f1579904c787448cc0512525e6f8e51705ff00b4 Mon Sep 17 00:00:00 2001
From: Brendan Chen <me@bchen.dev>
Date: Wed, 29 Jan 2025 00:03:31 -0800
Subject: [PATCH] check for system association in system resolvers

---
 src/resolvers/SystemResolvers.ts | 6 ++++++
 1 file changed, 6 insertions(+)

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<ServerContext> = {
       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<ServerContext> = {
       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<ServerContext> = {
       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) => {