fix: emit generates the name of the error

Author: a145789

src/constants.ts

@@ -30,7 +30,6 @@ export interface Config {
   offset: number;
   fileAbsolutePath: string;
   propsNotOnlyTs?: boolean;
-  setupScript?: string;
 }
 
 export const parseOption = {

src/transform/emits.ts

@@ -6,6 +6,7 @@ import type {
   NamedImportSpecifier,
   ImportDefaultSpecifier,
   ImportSpecifier,
+  BlockStatement,
 } from "@swc/core";
 
 import { Config, SetupAst } from "../constants";
@@ -22,7 +23,7 @@ function transformEmits(
   setupAst: SetupAst,
   config: Config,
 ) {
-  const { script, offset, setupScript, fileAbsolutePath } = config;
+  const { script, offset, fileAbsolutePath } = config;
   const name = getSetupSecondParams("emit", setupAst, fileAbsolutePath);
   if (!name) {
     return;
@@ -106,15 +107,15 @@ function transformEmits(
   }
 
   let emitNames: string[] = [];
-  if (setupScript) {
+  if ((setupAst.body as BlockStatement)?.stmts?.length) {
     const visitor = new GetCallExpressionFirstArg(name);
     visitor.visitFn(setupAst);
 
-    const setupOffset = setupAst.span.start;
     emitNames = (visitor.firstArgAst as Identifier[]).map((ast) => {
-      const { start, end } = getRealSpan(ast.span, setupOffset);
-      return setupScript.slice(start, end);
+      const { start, end } = getRealSpan(ast.span, offset);
+      return script.slice(start, end);
     });
+    console.log(emitNames);
   }
 
   str = `${preCode}defineEmits([${[...new Set([...keys, ...emitNames])].join(

src/transform/expose.ts

@@ -5,6 +5,7 @@ import {
   getSetupSecondParams,
 } from "../utils";
 import {
+  BlockStatement,
   ExportDefaultExpression,
   KeyValueProperty,
   MethodProperty,
@@ -14,27 +15,26 @@ import { Visitor } from "@swc/core/Visitor.js";
 import type MagicString from "magic-string";
 
 function transformExpose(setupAst: SetupAst, config: Config) {
-  const { setupScript, offset, fileAbsolutePath } = config;
+  const { script, offset, fileAbsolutePath } = config;
   const name = getSetupSecondParams("expose", setupAst, fileAbsolutePath);
   if (!name) {
     return;
   }
 
   let exposeArg: string[] = [];
-  if (setupScript) {
+  if ((setupAst.body as BlockStatement)?.stmts?.length) {
     const visitor = new GetCallExpressionFirstArg(name);
     visitor.visitFn(setupAst);
 
-    const setupOffset = setupAst.span.start;
     exposeArg = visitor.firstArgAst
       .flatMap((ast) => {
         if (ast.type !== "ObjectExpression") {
           return "";
         }
 
-        const { start, end } = getRealSpan(ast.span, setupOffset);
+        const { start, end } = getRealSpan(ast.span, offset);
 
-        return setupScript.slice(start, end).replace(/{|}/g, "").split(",");
+        return script.slice(start, end).replace(/{|}/g, "").split(",");
       })
       .filter((s) => Boolean(s.trim()));
   }

src/transform/script.ts

@@ -138,12 +138,6 @@ function transformScript(config: Config) {
       ? setupAst
       : (setupAst.value as ArrowFunctionExpression);
 
-  const setupFnAstSpan = getRealSpan(setupFnAst.span, config.offset);
-  config.setupScript = config.script.slice(
-    setupFnAstSpan.start,
-    setupFnAstSpan.end,
-  );
-
   const transformOption: TransformOption = {};
   for (const ast of optionAst.properties) {
     if (ast.type === "SpreadElement") {

src/transform/sfc.ts

@@ -23,7 +23,6 @@ export function transformSfc(path: string, option: CommandsOption) {
       script: script.content.trim(),
       offset: 0,
       fileAbsolutePath: path,
-      setupScript: "",
     });
   } catch (error) {
     output.error(`transform script failed in the ${path}`);

src/utils.ts

@@ -223,6 +223,8 @@ export class GetCallExpressionFirstArg extends Visitor {
     ) {
       this.firstArgAst.push(n.arguments[0].expression);
     }
+
+    super.visitCallExpression(n);
     return n;
   }
 

test/index.test.ts

@@ -86,7 +86,6 @@ describe("test transform script", () => {
           script: testScript1.code.trim(),
           offset: 0,
           fileAbsolutePath: "",
-          setupScript: "",
         }),
       ),
     ).toBe(transformToSingeLine(testScript1.transform.trim()));
@@ -98,7 +97,6 @@ describe("test transform script", () => {
           script: testScript2.code.trim(),
           offset: 0,
           fileAbsolutePath: "",
-          setupScript: "",
         }),
       ),
     ).toBe(transformToSingeLine(testScript2.transform.trim()));
@@ -110,7 +108,6 @@ describe("test transform script", () => {
           script: testScript3.code.trim(),
           offset: 0,
           fileAbsolutePath: "",
-          setupScript: "",
         }),
       ),
     ).toBe(transformToSingeLine(testScript3.transform.trim()));