feat: support for emit transform only in setup functions

a145789 · Dec 23, 2022 · f273601 · f273601
1 parent 8742ea4
commit f273601
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 34 deletions.
diff --git a/src/transform/emits.ts b/src/transform/emits.ts
@@ -18,7 +18,7 @@ import { Visitor } from "@swc/core/Visitor.js";
 import type MagicString from "magic-string";
 
 function transformEmits(
-  emitsAst: ArrayExpression | ObjectExpression | Identifier,
+  emitsAst: ArrayExpression | ObjectExpression | Identifier | null,
   setupAst: SetupAst,
   config: Config,
 ) {
@@ -78,23 +78,31 @@ function transformEmits(
     }
   }
 
-  if (emitsAst.type === "ObjectExpression") {
-    const { start, end } = getRealSpan(emitsAst.span, offset);
-    str = `${preCode}defineEmits(${script.slice(start, end)});\n`;
+  let keys: string[] = [];
+  if (emitsAst) {
+    if (emitsAst.type === "ObjectExpression") {
+      const { start, end } = getRealSpan(emitsAst.span, offset);
+      str = `${preCode}defineEmits(${script.slice(start, end)});\n`;
 
-    return MyVisitor;
-  }
+      return MyVisitor;
+    }
+
+    if (emitsAst.type === "Identifier") {
+      if (name !== emitsAst.value) {
+        str = `${preCode}defineEmits(${emitsAst.value});\n`;
+      } else {
+        str = `${preCode}defineEmits($${emitsAst.value});\n`;
+        isSameEmitsName = true;
+      }
 
-  if (emitsAst.type === "Identifier") {
-    if (name !== emitsAst.value) {
-      str = `${preCode}defineEmits(${emitsAst.value});\n`;
-    } else {
-      str = `${preCode}defineEmits($${emitsAst.value});\n`;
-      isSameEmitsName = true;
+      return MyVisitor;
     }
-    console.log(str);
 
-    return MyVisitor;
+    keys = emitsAst.elements.map((ast) => {
+      const { span } = ast!.expression as Identifier;
+      const { start, end } = getRealSpan(span, offset);
+      return script.slice(start, end);
+    });
   }
 
   let emitNames: string[] = [];
@@ -103,21 +111,12 @@ function transformEmits(
     visitor.visitFn(setupAst);
 
     const setupOffset = setupAst.span.start;
-
     emitNames = (visitor.firstArgAst as Identifier[]).map((ast) => {
-      const {
-        span: { start, end },
-      } = ast;
-      return setupScript.slice(start - setupOffset, end - setupOffset);
+      const { start, end } = getRealSpan(ast.span, setupOffset);
+      return setupScript.slice(start, end);
     });
   }
 
-  const keys = emitsAst.elements.map((ast) => {
-    const { span } = ast!.expression as Identifier;
-    const { start, end } = getRealSpan(span, offset);
-    return script.slice(start, end);
-  });
-
   str = `${preCode}defineEmits([${[...new Set([...keys, ...emitNames])].join(
     ", ",
   )}]);\n`;

diff --git a/src/transform/expose.ts b/src/transform/expose.ts
@@ -32,14 +32,9 @@ function transformExpose(setupAst: SetupAst, config: Config) {
           return "";
         }
 
-        const {
-          span: { start, end },
-        } = ast;
+        const { start, end } = getRealSpan(ast.span, setupOffset);
 
-        return setupScript
-          .slice(start - setupOffset, end - setupOffset)
-          .replace(/{|}/g, "")
-          .split(",");
+        return setupScript.slice(start, end).replace(/{|}/g, "").split(",");
       })
       .filter((s) => Boolean(s.trim()));
   }

diff --git a/src/transform/script.ts b/src/transform/script.ts
@@ -138,9 +138,10 @@ function transformScript(config: Config) {
       ? setupAst
       : (setupAst.value as ArrowFunctionExpression);
 
+  const setupFnAstSpan = getRealSpan(setupFnAst.span, config.offset);
   config.setupScript = config.script.slice(
-    setupFnAst.span.start - config.offset,
-    setupFnAst.span.end - config.offset,
+    setupFnAstSpan.start,
+    setupFnAstSpan.end,
   );
 
   const transformOption: TransformOption = {};
@@ -216,6 +217,9 @@ function transformScript(config: Config) {
   }
 
   try {
+    if (!transformOption.emits) {
+      transformOption.emits = transformEmits(null, setupFnAst, config);
+    }
     transformOption.expose = transformExpose(setupFnAst, config);
     transformOption.attrsAndSlots = transformAttrsAndSlots(setupFnAst, config);
   } catch (error) {