refactor(utilities/ast): minor improvements

Author: Rel1cx

packages/utilities/ast/src/function-init-path.ts

@@ -104,75 +104,30 @@ export type FunctionInitPath =
   ];
 
 export function getFunctionInitPath(node: TSESTreeFunction): FunctionInitPath | _ {
-  const { parent } = node;
   if (node.type === T.FunctionDeclaration) {
     return [node] as const;
   }
-  if (parent.type === T.VariableDeclarator) {
-    return [
-      parent.parent,
-      parent,
-      node,
-    ] as const;
-  }
-  if (
-    parent.type === T.CallExpression
-    && parent.parent.type === T.VariableDeclarator
-  ) {
-    return [
-      parent.parent.parent,
-      parent.parent,
-      parent,
-      node,
-    ] as const;
-  }
-  if (
-    parent.type === T.CallExpression
-    && parent.parent.type === T.CallExpression
-    && parent.parent.parent.type === T.VariableDeclarator
-  ) {
-    return [
-      parent.parent.parent.parent,
-      parent.parent.parent,
-      parent.parent,
-      parent,
-      node,
-    ] as const;
-  }
-  if (
-    parent.type === T.Property
-    && parent.parent.type === T.ObjectExpression
-    && parent.parent.parent.type === T.VariableDeclarator
-  ) {
-    return [
-      parent.parent.parent.parent,
-      parent.parent.parent,
-      parent.parent,
-      parent,
-      node,
-    ] as const;
-  }
-  if (
-    parent.type === T.MethodDefinition
-    && parent.parent.parent.type === T.ClassDeclaration
-  ) {
-    return [
-      parent.parent.parent,
-      parent.parent,
-      parent,
-      node,
-    ] as const;
-  }
-  if (
-    parent.type === T.PropertyDefinition
-    && parent.parent.parent.type === T.ClassDeclaration
-  ) {
-    return [
-      parent.parent.parent,
-      parent.parent,
-      parent,
-      node,
-    ] as const;
+  const { parent } = node;
+  switch (true) {
+    case parent.type === T.VariableDeclarator:
+      return [parent.parent, parent, node] as const;
+    case parent.type === T.CallExpression
+      && parent.parent.type === T.VariableDeclarator:
+      return [parent.parent.parent, parent.parent, parent, node] as const;
+    case parent.type === T.CallExpression
+      && parent.parent.type === T.CallExpression
+      && parent.parent.parent.type === T.VariableDeclarator:
+      return [parent.parent.parent.parent, parent.parent.parent, parent.parent, parent, node] as const;
+    case parent.type === T.Property
+      && parent.parent.type === T.ObjectExpression
+      && parent.parent.parent.type === T.VariableDeclarator:
+      return [parent.parent.parent.parent, parent.parent.parent, parent.parent, parent, node] as const;
+    case parent.type === T.MethodDefinition
+      && parent.parent.parent.type === T.ClassDeclaration:
+      return [parent.parent.parent, parent.parent, parent, node] as const;
+    case parent.type === T.PropertyDefinition
+      && parent.parent.parent.type === T.ClassDeclaration:
+      return [parent.parent.parent, parent.parent, parent, node] as const;
   }
   return _;
 }

packages/utilities/ast/src/get-ecma-expression.ts

@@ -4,9 +4,9 @@ import { isTsOnlyExpression } from "./is";
 import type { TSESTreeTSOnlyExpression } from "./types";
 
 /**
- * Gets the ECMAScript expression from a TS expression.
- * @param node The TS expression.
- * @returns The ECMAScript expression within the TS expression.
+ * Recursively get the inner expression until it's not a TSOnlyExpression
+ * @param node - The node to get the expression from
+ * @returns The inner expression
  */
 export function getEcmaExpression(node: TSESTree.Node): Exclude<
   TSESTree.Node,