fix: support enum, close #35

Author: antfu

src/core/identifiers.ts

@@ -1,7 +1,12 @@
 import { PrivateName, Expression, Statement, SpreadElement, Node } from '@babel/types'
 
 export function getIdentifierDeclarations(nodes: Statement[], identifiers = new Set<string>()) {
-  for (const node of nodes) {
+  for (let node of nodes) {
+    if (node.type === 'ExportNamedDeclaration') {
+      node = node.declaration!
+      if (!node)
+        continue
+    }
     if (node.type === 'ImportDeclaration') {
       for (const specifier of node.specifiers)
         identifiers.add(specifier.local.name)
@@ -38,6 +43,10 @@ export function getIdentifierDeclarations(nodes: Statement[], identifiers = new
       if (node.id)
         identifiers.add(node.id.name)
     }
+    else if (node.type === 'TSEnumDeclaration') {
+      if (node.id)
+        identifiers.add(node.id.name)
+    }
     // else {
     //   console.log(node)
     // }

test/__snapshots__/transform.test.ts.snap

@@ -168,6 +168,28 @@ export default __sfc_main;
 
 exports[`transform fixtures test/fixtures/Empty.vue 1`] = `""`;
 
+exports[`transform fixtures test/fixtures/Enum.vue 1`] = `
+"<template>
+  <div>{{ MyEnum }}</div>
+</template>
+
+<script lang=\\"ts\\">
+enum MyEnum {
+  test = 'true',
+}
+const __sfc_main = {};
+
+__sfc_main.setup = (__props, __ctx) => {
+  return {
+    MyEnum
+  };
+};
+
+export default __sfc_main;
+</script>
+"
+`;
+
 exports[`transform fixtures test/fixtures/Macros.vue 1`] = `
 "<template>
   <div @click=\\"emit(props.msg)\\">{{ msg }}</div>

test/fixtures/Enum.vue

@@ -0,0 +1,9 @@
+<template>
+  <div>{{ MyEnum }}</div>
+</template>
+
+<script setup lang="ts">
+enum MyEnum {
+  test = 'true',
+}
+</script>