feat: add plugins option for babel parser

Author: sxzz

src/core/convert.ts

@@ -15,6 +15,7 @@ import {
 } from '@babel/types'
 import MagicString from 'magic-string'
 import { escapeString, parseObject, styleToString } from './utils'
+import type { OptionsResolved } from './options'
 import type {
   Expression,
   JSX,
@@ -36,10 +37,10 @@ import type {
   TemplateLiteral,
 } from '@babel/types'
 
-export const convert = (code: string, debug?: boolean) => {
+export const convert = (code: string, { debug, plugins }: OptionsResolved) => {
   const ast = parse(code, {
     sourceType: 'module',
-    plugins: ['typescript', 'jsx'],
+    plugins,
   })
 
   const nodes: [JSX, Expression][] = []

src/core/options.ts

@@ -0,0 +1,24 @@
+import type { ParserPlugin } from '@babel/parser'
+import type { FilterPattern } from '@rollup/pluginutils'
+
+export interface Options {
+  include?: FilterPattern
+  exclude?: FilterPattern | undefined
+  debug?: boolean
+  /**
+   * Plugins for `@babel/parser`
+   * @default `['typescript', 'jsx']`
+   */
+  plugins?: ParserPlugin[]
+}
+
+export type OptionsResolved = Required<Options>
+
+export function resolveOption(options: Options): OptionsResolved {
+  return {
+    include: options.include || [/\.[jt]sx$/],
+    exclude: options.exclude || undefined,
+    debug: options.debug ?? false,
+    plugins: ['typescript', 'jsx'],
+  }
+}

src/index.ts

@@ -1,28 +1,13 @@
 import { createUnplugin } from 'unplugin'
 import { createFilter } from '@rollup/pluginutils'
 import { convert } from './core/convert'
-import type { FilterPattern } from '@rollup/pluginutils'
+import { resolveOption } from './core/options'
+import type { Options } from './core/options'
 
 declare global {
   const jsxToString: (element: JSX.Element) => string
 }
 
-export interface Options {
-  include?: FilterPattern
-  exclude?: FilterPattern | undefined
-  debug?: boolean
-}
-
-export type OptionsResolved = Required<Options>
-
-function resolveOption(options: Options): OptionsResolved {
-  return {
-    include: options.include || [/\.[jt]sx$/],
-    exclude: options.exclude || undefined,
-    debug: options.debug ?? false,
-  }
-}
-
 export default createUnplugin<Options>((options = {}) => {
   const opt = resolveOption(options)
   const filter = createFilter(opt.include, opt.exclude)
@@ -38,7 +23,7 @@ export default createUnplugin<Options>((options = {}) => {
 
     transform(code) {
       try {
-        return convert(code, opt.debug)
+        return convert(code, opt)
       } catch (err: unknown) {
         this.error(`${name} ${err}`)
       }