fix marking related nodes

graphql-editor · May 10, 2024 · d87adb6 · d87adb6
1 parent 7535584
commit d87adb6
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/packages/editor/src/state/containers/trees/index.ts b/packages/editor/src/state/containers/trees/index.ts
@@ -186,7 +186,12 @@ const useTreesStateContainer = createContainer(() => {
   };
 
   const relatedToSelectedTypes = (activeNode?: ParserField) => {
-    const parents = activeNode?.args.flatMap((ana) => ana.args);
+    const parents = allNodes.nodes.filter((node) =>
+      activeNode?.args
+        .flatMap((ana) => ana.args)
+        .map((ana) => getTypeName(ana.type.fieldType))
+        .includes(node.name)
+    );
     const inputs = allNodes.nodes.filter((an) =>
       an.args.find(
         (ana) =>
@@ -230,7 +235,7 @@ const useTreesStateContainer = createContainer(() => {
     );
 
     const notBaseTypes = argsTypeRelated
-      .concat(parents || [])
+      .concat(parents)
       .concat(inputs)
       .concat(interfacesRelatedToActiveNode)
       .concat(nodesRelatedToActiveInterface)