fix(macros): prevent return defineComponent for generic component

Author: zhiyuanzmj

packages/macros/src/volar/define-component.ts

@@ -8,7 +8,9 @@ export function transformDefineComponent(
 ): void {
   const { codes, ast, ts } = options
 
-  codes.replaceRange(node.arguments[0].end, node.end - 1)
+  const [comp, compOptions] = node.arguments
+
+  codes.replaceRange(comp.end, node.end - 1)
 
   codes.replaceRange(
     node.getStart(ast),
@@ -18,27 +20,26 @@ export function transformDefineComponent(
 const __setup = `,
   )
 
-  const compOptions = node.arguments[1]
-  codes.replaceRange(
-    node.end,
-    node.end,
-    '\n  return ',
-    [node.expression.getText(ast), node.expression.getStart(ast)],
-    `({
+  const result =
+    (ts.isArrowFunction(comp) || ts.isFunctionExpression(comp)) &&
+    comp.typeParameters?.length
+      ? ['__setup']
+      : ([
+          [node.expression.getText(ast), node.expression.getStart(ast)],
+          `({
     __typeProps: {} as Parameters<typeof __setup>[0],
-    setup:() => {},
     ...{} as Parameters<typeof __setup>[1] extends { slots?: infer S, expose?: infer E } | undefined ? {
       setup: E extends (exposed: infer Exposed) => any ? () => Exposed : never,
       slots: S extends Record<string, any> ? import('vue').SlotsType<S> : never
     } : {},`,
-    ...(compOptions
-      ? ([
-          '\n    ...',
-          [compOptions.getText(ast), compOptions.getStart(ast)],
+          ...(compOptions
+            ? [
+                '\n    ...',
+                [compOptions.getText(ast), compOptions.getStart(ast)],
+              ]
+            : []),
+          `
+  })`,
         ] as Code[])
-      : []),
-    `
-  })
-})()`,
-  )
+  codes.replaceRange(node.end, node.end, '\n  return ', ...result, `\n})()`)
 }