feat: support compile option

Author: zhiyuanzmj

src/core/compiler/compile.ts

@@ -5,11 +5,11 @@ import {
   defaultOnError,
 } from '@vue-vapor/compiler-dom'
 import { extend, isString } from '@vue-vapor/shared'
-import { babelParse } from '@vue-macros/common'
 import {
   type VaporCodegenResult as BaseVaporCodegenResult,
   generate,
 } from '@vue-vapor/compiler-vapor'
+import { parse } from '@babel/parser'
 import {
   type DirectiveTransform,
   type NodeTransform,
@@ -62,11 +62,26 @@ export function compile(
 
   const resolvedOptions = extend({}, options, {
     prefixIdentifiers,
+    expressionPlugins: options.expressionPlugins || ['jsx'],
   })
+  if (!__BROWSER__ && options.isTS) {
+    const { expressionPlugins } = resolvedOptions
+    if (!expressionPlugins.includes('typescript')) {
+      resolvedOptions.expressionPlugins = [
+        ...(expressionPlugins || []),
+        'typescript',
+      ]
+    }
+  }
 
   const {
     body: [statement],
-  } = isString(source) ? babelParse(source) : source
+  } = isString(source)
+    ? parse(source, {
+        sourceType: 'module',
+        plugins: resolvedOptions.expressionPlugins,
+      }).program
+    : source
   let children!: JSXElement[] | JSXFragment['children']
   if (statement.type === 'ExpressionStatement') {
     children =
@@ -88,17 +103,6 @@ export function compile(
   const [nodeTransforms, directiveTransforms] =
     getBaseTransformPreset(prefixIdentifiers)
 
-  if (!__BROWSER__ && options.isTS) {
-    const { expressionPlugins } = options
-    if (!expressionPlugins || !expressionPlugins.includes('typescript')) {
-      resolvedOptions.expressionPlugins = [
-        ...(expressionPlugins || []),
-        'typescript',
-        'jsx',
-      ]
-    }
-  }
-
   const ir = transform(
     ast,
     extend({}, resolvedOptions, {

src/core/compiler/index.ts

@@ -1,13 +1,10 @@
 export { generate } from '@vue-vapor/compiler-vapor'
 
-export {
-  // wrapTemplate,
-  compile,
-  type CompilerOptions,
-  type TransformPreset,
-} from './compile'
+export { compile, type CompilerOptions, type TransformPreset } from './compile'
 export * from './transform'
 
+export { resolveDirectiveNode, resolveNode } from './utils'
+
 export * from './ir'
 
 export { transformText } from './transforms/transformText'

src/core/transform.ts

@@ -17,7 +17,7 @@ export type RootNodes = {
 export function transformVueJsxVapor(
   code: string,
   id: string,
-  _options?: Options,
+  options?: Options,
 ) {
   const s = new MagicStringAST(code)
   const rootNodes: RootNodes = []
@@ -41,8 +41,10 @@ export function transformVueJsxVapor(
     if (isJSXElement(node)) {
       let { code, vaporHelpers, preamble } = compile(s.sliceNode(node), {
         mode: 'module',
-        prefixIdentifiers: false,
         inline: true,
+        isTS: id.endsWith('tsx'),
+        filename: id,
+        ...options?.compile,
       })
       vaporHelpers.forEach((helper) => importSet.add(helper))
       preamble = preamble.replace(/^[^\n]*;\n?/, '')

src/types.ts

@@ -1,9 +1,9 @@
+import type { CompilerOptions } from './core/compiler'
 import type { FilterPattern } from 'vite'
-import type { compile } from '@vue-vapor/compiler-dom'
 
 export interface Options {
   // define your plugin options here
   include?: FilterPattern
   exclude?: FilterPattern
-  compile?: typeof compile
+  compile?: CompilerOptions
 }

test/transforms/_utils.ts

@@ -1,4 +1,4 @@
-import { babelParse } from '@vue-macros/common'
+import { parse } from '@babel/parser'
 import {
   type CompilerOptions,
   generate,
@@ -11,7 +11,10 @@ export function makeCompile(options: CompilerOptions = {}) {
   return (source: string, overrideOptions: CompilerOptions = {}) => {
     const {
       body: [statement],
-    } = babelParse(source)
+    } = parse(source, {
+      sourceType: 'module',
+      plugins: ['jsx', 'typescript'],
+    }).program
     let children!: JSXElement[] | JSXFragment['children']
     if (statement.type === 'ExpressionStatement') {
       children =