Refactor 'findNameInPattern()' in terms of 'findInPattern()'

dfinity · Aug 15, 2023 · 3ce64fe · 3ce64fe
1 parent 0c9d4c3
commit 3ce64fe
Showing 1 changed file with 3 additions and 37 deletions.
diff --git a/src/server/navigation.ts b/src/server/navigation.ts
@@ -1,7 +1,7 @@
 import { AST, Node, Span } from 'motoko/lib/ast';
 import { Location, Position, Range } from 'vscode-languageserver';
 import { Context, getContext } from './context';
-import { findNodes, matchNode, asNode } from './syntax';
+import { findNodes, matchNode, asNode, findInPattern } from './syntax';
 import { getAbsoluteUri } from './utils';
 
 interface Reference {
@@ -107,43 +107,9 @@ export function locationFromDefinition(definition: Definition) {
     return location;
 }
 
-// TODO: refactor to use `findInPattern()`
 function findNameInPattern(search: Search, pat: Node): Node | undefined {
-    const matchAny = (...args: Node[]) => {
-        for (const field of args) {
-            const result = findNameInPattern(search, field);
-            if (result) {
-                return result;
-            }
-        }
-        return;
-    };
-    const match = (arg: Node) => findNameInPattern(search, arg);
-    return (
-        matchNode(pat, 'VarP', (name: string) =>
-            name === search.name ? pat : undefined,
-        ) ||
-        matchNode(pat, 'ObjP', (...args: Node[]) => {
-            for (const field of args) {
-                const aliasNode = field.args![0] as Node;
-                const alias = matchNode(
-                    aliasNode,
-                    'VarP',
-                    (alias) => alias,
-                    field.name,
-                );
-                if (alias === search.name) {
-                    return aliasNode || field;
-                }
-            }
-            return;
-        }) ||
-        matchNode(pat, 'TupP', matchAny) ||
-        matchNode(pat, 'AltP', matchAny) ||
-        matchNode(pat, 'AnnotP', match) ||
-        matchNode(pat, 'ParP', match) ||
-        matchNode(pat, 'OptP', match) ||
-        matchNode(pat, 'TagP', (_tag, arg: Node) => match(arg))
+    return findInPattern(pat, (name, node) =>
+        name === search.name ? node : undefined,
     );
 }