Add new method to check if node is null or not (#13940)

Co-authored-by: Kristoffer K. <merceyz@users.noreply.github.com>
babel · Feb 1, 2022 · b91c3e1 · b91c3e1
1 parent b092bd0
commit b91c3e1
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/packages/babel-traverse/src/path/index.ts b/packages/babel-traverse/src/path/index.ts
@@ -122,6 +122,10 @@ class NodePath<T extends t.Node = t.Node> {
     return val;
   }
 
+  hasNode(): this is NodePath<NonNullable<this["node"]>> {
+    return this.node != null;
+  }
+
   buildCodeFrameError(
     msg: string,
     Error: new () => Error = SyntaxError,

diff --git a/packages/babel-traverse/test/path/index.js b/packages/babel-traverse/test/path/index.js
@@ -47,5 +47,20 @@ describe("NodePath", () => {
 
       expect(path.getData(symbol)).toBe(42);
     });
+
+    describe("hasNode", () => {
+      it("returns false if node is null", () => {
+        const path = new NodePath({}, {});
+
+        expect(path.hasNode()).toBe(false);
+      });
+
+      it("returns true if node is not null", () => {
+        const path = new NodePath({}, {});
+        path.node = {};
+
+        expect(path.hasNode()).toBe(true);
+      });
+    });
   });
 });