Merge pull request #7 from PerimeterX/improve/parentKey

Improve getParentKey to include parsing the name of grouped nodes

Author: BenBaryoPX

README.md

@@ -188,6 +188,8 @@ inside the arborist.
 To contribute to this project see our [contribution guide](CONTRIBUTING.md)
 
 ## Changes
+### v1.1.1
+- Improve getParentKey to include parsing the name of grouped nodes such as 'arguments' or 'body'.
 ### v1.1.0
  - Added parentKey property.
  - Improved ASTNode definition (useful for intellisense).

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "flast",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "description": "Flatten JS AST",
   "main": "src/index.js",
   "scripts": {

package.json

@@ -40,6 +40,11 @@ function getParentKey(parent, targetChildNodeId) {
 	if (parent) {
 		for (const key of Object.keys(parent)) {
 			if (parent[key]?.nodeId === targetChildNodeId) return key;
+			else if (Array.isArray(parent[key])) {
+				for (const item of parent[key]) {
+					if (item.nodeId === targetChildNodeId) return key;
+				}
+			}
 		}
 	}
 	return null;

src/flast.js

@@ -64,7 +64,7 @@ function testNodesStructureIntegrity() {
 	const ast = generateFlatAST(code);
 	const expectedBreakdown = [
 		{nodeId: 0, type: 'Program', start: 0, end: 3, src: 'a=3', parentNode: null, parentKey: null},
-		{nodeId: 1, type: 'ExpressionStatement', start: 0, end: 3, src: 'a=3', parentKey: null},
+		{nodeId: 1, type: 'ExpressionStatement', start: 0, end: 3, src: 'a=3', parentKey: 'body'},
 		{nodeId: 2, type: 'AssignmentExpression', start: 0, end: 3, src: 'a=3', operator: '=', parentKey: 'expression'},
 		{nodeId: 3, type: 'Identifier', start: 0, end: 1, src: 'a', parentKey: 'left'},
 		{nodeId: 4, type: 'Literal', start: 2, end: 3, src: '3', value: 3, raw: '3', parentKey: 'right'},