other: #192 🔹 restore getImportedName (#194)

chibivue-land · Dec 29, 2023 · 38bcb6e · 38bcb6e
1 parent 4706599
commit 38bcb6e
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/packages/compiler-core/src/babelUtils.ts b/packages/compiler-core/src/babelUtils.ts
@@ -1,4 +1,11 @@
-import { Identifier, Node, Function } from "@babel/types";
+import {
+  Identifier,
+  ImportDefaultSpecifier,
+  ImportNamespaceSpecifier,
+  ImportSpecifier,
+  Function,
+  Node,
+} from "@babel/types";
 
 import { walk } from "estree-walker";
 
@@ -106,6 +113,17 @@ export const isFunctionType = (node: Node): node is Function => {
   return /Function(?:Expression|Declaration)$|Method$/.test(node.type);
 };
 
+export function getImportedName(
+  specifier: ImportSpecifier | ImportDefaultSpecifier | ImportNamespaceSpecifier
+) {
+  if (specifier.type === "ImportSpecifier")
+    return specifier.imported.type === "Identifier"
+      ? specifier.imported.name
+      : specifier.imported.value;
+  else if (specifier.type === "ImportNamespaceSpecifier") return "*";
+  return "default";
+}
+
 export function isReferencedIdentifier(
   id: Identifier,
   parent: Node | null,