Refactor findNodesOfType using walkTreeFlatMapping

neuroevolutus · neuroevolutus · commit 3f46409312eb · 2025-07-08T01:36:39.000-05:00
diff --git a/server/src/server.ts b/server/src/server.ts
@@ -1206,13 +1206,9 @@ export function getDocumentSymbols(document: TextDocument): DocumentSymbol[] {
 }
 
 function findNodesOfType(rootNode: SyntaxNode, type: string): SyntaxNode[] {
-  const nodes: SyntaxNode[] = [];
-  walkTree(rootNode, (node) => {
-    if (node.type === type) {
-      nodes.push(node);
-    }
+  return walkTreeFlatMapping(rootNode, (node) => {
+    return (node.type === type) ? [node] : [];
   });
-  return nodes;
 }
 
 function findWhereClauseDeclarations(rootNode: SyntaxNode): SyntaxNode[] {

Original file line number	Diff line number	Diff line change
`@@ -1206,13 +1206,9 @@ export function getDocumentSymbols(document: TextDocument): DocumentSymbol[] {`
`1206`	`1206`	`}`
`1207`	`1207`
`1208`	`1208`	`function findNodesOfType(rootNode: SyntaxNode, type: string): SyntaxNode[] {`
`1209`		`- const nodes: SyntaxNode[] = [];`
`1210`		`- walkTree(rootNode, (node) => {`
`1211`		`- if (node.type === type) {`
`1212`		`- nodes.push(node);`
`1213`		`- }`
	`1209`	`+ return walkTreeFlatMapping(rootNode, (node) => {`
	`1210`	`+ return (node.type === type) ? [node] : [];`
`1214`	`1211`	`});`
`1215`		`- return nodes;`
`1216`	`1212`	`}`
`1217`	`1213`
`1218`	`1214`	`function findWhereClauseDeclarations(rootNode: SyntaxNode): SyntaxNode[] {`