fixes

git-town · Nov 27, 2024 · 3c97339 · 3c97339
1 parent b917c0a
commit 3c97339
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 46 deletions.
diff --git a/dist/index.js b/dist/index.js
@@ -42891,6 +42891,22 @@ async function main({
       isCurrent: true,
       ...pullRequest
     });
+    (0, import_graphology_traversal.dfsFromNode)(
+      repoGraph,
+      pullRequest.head.ref,
+      (_, attributes) => {
+        if (attributes.type !== "pull-request") {
+          return false;
+        }
+        stackGraph2.mergeNode(attributes.head.ref, {
+          ...attributes,
+          shouldPrint: true
+        });
+        stackGraph2.mergeDirectedEdge(attributes.base.ref, attributes.head.ref);
+        return true;
+      },
+      { mode: "outbound" }
+    );
     (0, import_graphology_traversal.bfsFromNode)(
       repoGraph,
       pullRequest.head.ref,
@@ -42911,27 +42927,6 @@ async function main({
       },
       { mode: "inbound" }
     );
-    (0, import_graphology_traversal.dfsFromNode)(
-      repoGraph,
-      pullRequest.head.ref,
-      (_, attributes) => {
-        if (attributes.type !== "pull-request") {
-          return false;
-        }
-        stackGraph2.mergeNode(attributes.head.ref, {
-          ...attributes,
-          shouldPrint: true
-        });
-        stackGraph2.mergeDirectedEdge(attributes.base.ref, attributes.head.ref);
-        return true;
-      },
-      { mode: "outbound" }
-    );
-    stackGraph2.forEachNode((ref, stackNode) => {
-      if (!stackNode.shouldPrint) {
-        stackGraph2.dropNode(ref);
-      }
-    });
     return stackGraph2;
   };
   const getOutput = (graph) => {

diff --git a/src/main.ts b/src/main.ts
@@ -76,6 +76,25 @@ export async function main({
       ...pullRequest,
     })
 
+    dfsFromNode(
+      repoGraph,
+      pullRequest.head.ref,
+      (_, attributes) => {
+        if (attributes.type !== 'pull-request') {
+          return false
+        }
+
+        stackGraph.mergeNode(attributes.head.ref, {
+          ...attributes,
+          shouldPrint: true,
+        })
+        stackGraph.mergeDirectedEdge(attributes.base.ref, attributes.head.ref)
+
+        return true
+      },
+      { mode: 'outbound' }
+    )
+
     bfsFromNode(
       repoGraph,
       pullRequest.head.ref,
@@ -98,31 +117,6 @@ export async function main({
       { mode: 'inbound' }
     )
 
-    dfsFromNode(
-      repoGraph,
-      pullRequest.head.ref,
-      (_, attributes) => {
-        if (attributes.type !== 'pull-request') {
-          return false
-        }
-
-        stackGraph.mergeNode(attributes.head.ref, {
-          ...attributes,
-          shouldPrint: true,
-        })
-        stackGraph.mergeDirectedEdge(attributes.base.ref, attributes.head.ref)
-
-        return true
-      },
-      { mode: 'outbound' }
-    )
-
-    stackGraph.forEachNode((ref, stackNode) => {
-      if (!stackNode.shouldPrint) {
-        stackGraph.dropNode(ref)
-      }
-    })
-
     return stackGraph
   }