✨ removing extension nodes

Author: aexol

package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphql-js-tree",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "private": false,
   "license": "MIT",
   "description": "GraphQL Parser providing simplier structure",

src/TreeOperations/shared.ts

@@ -1,4 +1,13 @@
-import { AllTypes, FieldType, Options, ParserField, ScalarTypes, TypeExtension } from '@/Models';
+import {
+  AllTypes,
+  FieldType,
+  Options,
+  ParserField,
+  ParserTree,
+  ScalarTypes,
+  TypeDefinition,
+  TypeExtension,
+} from '@/Models';
 import { generateNodeId, getTypeName } from '@/shared';
 
 export function filterNotNull<T>(t: T | null): t is T {
@@ -102,6 +111,26 @@ export const ChangeAllRelatedNodes = ({
 }) => {
   nodes.forEach((n) => ChangeRelatedNode({ oldName, newName, node: n }));
 };
+export const RemoveRelatedExtensionNodes = ({ node, tree }: { tree: ParserTree; node: ParserField }) => {
+  const {
+    data: { type },
+  } = node;
+  if (
+    type === TypeDefinition.EnumTypeDefinition ||
+    type === TypeDefinition.InputObjectTypeDefinition ||
+    type === TypeDefinition.InterfaceTypeDefinition ||
+    type === TypeDefinition.ObjectTypeDefinition ||
+    type === TypeDefinition.ScalarTypeDefinition ||
+    type === TypeDefinition.UnionTypeDefinition
+  ) {
+    [...tree.nodes].forEach((n) => {
+      if (n.name === node.name) {
+        const nodeToBeRemoved = tree.nodes.findIndex((fn) => fn.id === n.id);
+        tree.nodes.splice(nodeToBeRemoved, 1);
+      }
+    });
+  }
+};
 
 export const isScalarArgument = (field: ParserField, scalarTypes: string[]) => {
   const typeName = getTypeName(field.type.fieldType);

src/TreeOperations/tree.ts

@@ -28,6 +28,7 @@ import {
   isExtensionNode,
   isScalarArgument,
   regenerateId,
+  RemoveRelatedExtensionNodes,
 } from '@/TreeOperations/shared';
 
 export const mutate = (tree: ParserTree, allNodes: ParserField[]) => {
@@ -172,6 +173,7 @@ export const mutate = (tree: ParserTree, allNodes: ParserField[]) => {
     if (node.data.type === TypeExtension.InterfaceTypeExtension) {
     }
     tree.nodes.splice(deletedNode, 1);
+    RemoveRelatedExtensionNodes({ node, tree });
     tree.nodes.forEach((n) => {
       n.args = n.args
         .filter((a) => {

src/__tests__/TreeOperations/mocks.ts

@@ -1,4 +1,4 @@
-import { ParserTree, TypeDefinition, ScalarTypes, ValueDefinition, Options, Value } from '@/Models';
+import { ParserTree, TypeDefinition, ScalarTypes, ValueDefinition, Options, Value, TypeExtension } from '@/Models';
 import {
   createRootField,
   createPlainField,
@@ -9,6 +9,7 @@ import {
   createPlainArgument,
   createPlainEnumValue,
   createUnionMember,
+  createRootExtensionField,
 } from '@/shared';
 
 const mainMock: ParserTree = {
@@ -135,6 +136,26 @@ const mainMock: ParserTree = {
       name: 'model',
       args: [createPlainInputValue({ name: 'maxAge', type: ScalarTypes.Int })],
     }),
+    createRootExtensionField({
+      name: 'Person',
+      type: TypeExtension.ObjectTypeExtension,
+      args: [
+        createPlainField({
+          name: 'extendedName',
+          type: ScalarTypes.String,
+        }),
+      ],
+    }),
+    createRootExtensionField({
+      name: 'Person',
+      type: TypeExtension.ObjectTypeExtension,
+      args: [
+        createPlainField({
+          name: 'extendedName2',
+          type: ScalarTypes.String,
+        }),
+      ],
+    }),
   ],
 };
 

src/__tests__/TreeOperations/tree.remove.spec.ts

@@ -74,4 +74,12 @@ describe('Tree Operations - node removal tests', () => {
     expect(treeMock.nodes[5].args).not.toContainEqual(oldArgument);
     expect(treeMock.nodes[0].directives[0].args).toHaveLength(0);
   });
+  test('Delete node extensions when node is removed', () => {
+    const treeMock = createMock();
+    const oldExtendNode = JSON.parse(JSON.stringify(treeMock.nodes[6]));
+    const oldExtendNode2 = JSON.parse(JSON.stringify(treeMock.nodes[7]));
+    mutate(treeMock, treeMock.nodes).removeNode(treeMock.nodes[0]);
+    expect(treeMock.nodes).not.toContainEqual(oldExtendNode);
+    expect(treeMock.nodes).not.toContainEqual(oldExtendNode2);
+  });
 });