wip: working impl

metatypedev · Dec 10, 2024 · 56829e9 · 56829e9
1 parent 2ea7a48
commit 56829e9
Show file tree

Hide file tree

Showing 8 changed files with 247 additions and 151 deletions.
diff --git a/src/typegate/src/engine/planner/mod.ts b/src/typegate/src/engine/planner/mod.ts
@@ -507,30 +507,32 @@ export class Planner {
     stages.push(stage);
 
     this.policiesBuilder.push(stage.id(), node.typeIdx, collected.policies);
-    const types = this.policiesBuilder.setReferencedTypes(
-      stage.id(),
-      node.typeIdx,
-      outputIdx,
-      inputIdx,
-    );
-
-    // nested quantifiers
-    let wrappedTypeIdx = outputIdx;
-    let wrappedType = this.tg.type(wrappedTypeIdx);
-    while (isQuantifier(wrappedType)) {
-      wrappedTypeIdx = getWrappedType(wrappedType);
-      wrappedType = this.tg.type(wrappedTypeIdx);
-      types.push(wrappedTypeIdx);
+    {
+      const types = this.policiesBuilder.setReferencedTypes(
+        stage.id(),
+        node.typeIdx,
+        outputIdx,
+        inputIdx,
+      );
+
+      // nested quantifiers
+      let wrappedTypeIdx = outputIdx;
+      let wrappedType = this.tg.type(wrappedTypeIdx);
+      while (isQuantifier(wrappedType)) {
+        wrappedTypeIdx = getWrappedType(wrappedType);
+        wrappedType = this.tg.type(wrappedTypeIdx);
+        types.push(wrappedTypeIdx);
+      }
+
+      stages.push(
+        ...this.traverse(
+          { ...node, typeIdx: wrappedTypeIdx, parentStage: stage },
+          stage,
+        ),
+      );
     }
-
-    stages.push(
-      ...this.traverse(
-        { ...node, typeIdx: wrappedTypeIdx, parentStage: stage },
-        stage,
-      ),
-    );
-
     this.policiesBuilder.pop(stage.id());
+
     return stages;
   }