Fix bug

CHANGELOG.md

@@ -1,6 +1,12 @@
 # Changelog
 
-## [1.0.0-beta8] - 2024-02-06
+## [1.0.0-beta9] - 2024-02-14
+
+### Bugfix
+
+* More scoping issues for binding
+
+## [1.0.0-beta8] - 2024-02-14
 
 ### Bugfix
 

package.json

@@ -1,6 +1,6 @@
 {
  "name": "babel-q",
- "version": "1.0.0-beta.8",
+ "version": "1.0.0-beta.9",
  "description": "offers a way to query an babel AST to find specific patterns using a syntax somewhat similar to XPath.",
  "scripts": {
  "lint": "eslint . --ext .ts --fix --ignore-path .gitignore",

src/traverse.ts

@@ -99,6 +99,7 @@ function createNodePath(node: Babel.Node, key: string | undefined, parentKey: st
 function registerBinding(node: Babel.Node, parentNode: Babel.Node, grandParentNode: Babel.Node | undefined, scope: Scope) {
  if (t.isBinding(node, parentNode, grandParentNode) && !t.isMemberExpression(node)) {
  if (t.isIdentifier(node) && !t.isAssignmentExpression(parentNode)) {
+ //A bit of a hack here as well. Needs some further investigation
  if (t.isFunctionDeclaration(parentNode) || t.isFunctionExpression(parentNode) || t.isScope(node, parentNode)) { 
  setBinding(scope, node.name, { path: createNodePath(node, undefined, undefined, scope) });
  } else {
@@ -117,7 +118,8 @@ function registerBindings(node: Babel.Node, parentNode: Babel.Node, grandParentN
  const keys = t.VISITOR_KEYS[node.type];
 
  let childScope = scope;
- if (t.isScopable(node)) {
+ // This is also buggy. Need to investigate what creates a new scope
+ if (t.isScopable(node) || t.isExportSpecifier(node)) {
  childScope = createScope(scope);
  }
  for (const key of keys) {

tests/query.test.ts

@@ -281,6 +281,19 @@ describe('testing index file', () => {
  const nodes1 = query(ast!, "//MemberExpression/$:object");
  const nodes2 = query(ast!, "//FunctionDeclaration//FunctionDeclaration/:id");
  expect(nodes1).toEqual(nodes2);
+ });
+ test("find correct binding when exported", () => {
+ const code = `
+ let a = 1;
+ let b = 2;
+ b = a;
+ export {
+ a
+ }`
+ const ast = babel.parseSync(code);
+ const nodes = query(ast!, "//AssignmentExpression/$:right/:init/:value");
+ console.log(nodes);
+ expect(nodes).toEqual([1]);
  })
 });