feat!: support factory for defineDirective

Author: KeJunMao

README.md

@@ -206,11 +206,11 @@ Taking the built-in directive as an example:
 ```ts
 /** @see https://xregexp.com/ */
 import type { NamedGroupsArray } from 'xregexp'
-import { defineDirective } from 'unplugin-preprocessor-directives'
+import { defineDirective } from '../directive'
 
-export default defineDirective({
-  name: '#define',
+export default defineDirective<undefined>(() => ({
   nested: false,
+  name: '#define',
   pattern: /.*?#(?<directive>(?:un)?def(?:ine)?)\s*(?<key>[\w]*)\s/gm,
   processor({ ctx }) {
     return (...args) => {
@@ -225,7 +225,7 @@ export default defineDirective({
       return ''
     }
   },
-})
+}))
 ```
 
 ### `name: string`

README.zh-cn.md

@@ -207,11 +207,11 @@ class MyClass {
 ```ts
 /** @see https://xregexp.com/ */
 import type { NamedGroupsArray } from 'xregexp'
-import { defineDirective } from 'unplugin-preprocessor-directives'
+import { defineDirective } from '../directive'
 
-export default defineDirective({
-  name: '#define',
+export default defineDirective<undefined>(() => ({
   nested: false,
+  name: '#define',
   pattern: /.*?#(?<directive>(?:un)?def(?:ine)?)\s*(?<key>[\w]*)\s/gm,
   processor({ ctx }) {
     return (...args) => {
@@ -226,7 +226,7 @@ export default defineDirective({
       return ''
     }
   },
-})
+}))
 ```
 
 ### `name: string`

src/core/context.ts

@@ -56,13 +56,17 @@ export class Context {
       '',
     )
 
+    if (directives.length === 0) {
+      directives = [...builtinDirectives.map(v =>
+        typeof v === 'function' ? v(this) : v,
+      )]
+    }
+
     const [preDirectives, normalDirectives, postDirectives] = sortUserDirectives(directives.map(v =>
       typeof v === 'function' ? v(this) : v,
     ))
 
-    this.directives = [...builtinDirectives.map(v =>
-      typeof v === 'function' ? v(this) : v,
-    ), ...preDirectives, ...normalDirectives, ...postDirectives]
+    this.directives = [...preDirectives, ...normalDirectives, ...postDirectives]
 
     this.logger = createLogger(undefined, {
       prefix: 'unplugin-preprocessor-directives',

src/core/directive.ts

@@ -1,7 +1,9 @@
-import type { Directive, FunctionDirective } from '../types'
+import type { DirectiveFactory } from '../types'
 
-export function defineDirective(
-  directive: Directive | FunctionDirective,
+export function defineDirective<UserOptions>(
+  factory: DirectiveFactory<UserOptions>,
 ) {
-  return directive
+  return (userOptions?: UserOptions) => {
+    return factory(userOptions!)
+  }
 }

src/core/directives/define.ts

@@ -1,7 +1,7 @@
 import type { NamedGroupsArray } from 'xregexp'
 import { defineDirective } from '../directive'
 
-export default defineDirective({
+export default defineDirective<undefined>(() => ({
   nested: false,
   name: '#define',
   pattern: /.*?#(?<directive>(?:un)?def(?:ine)?)\s*(?<key>[\w]*)\s/gm,
@@ -18,4 +18,4 @@ export default defineDirective({
       return ''
     }
   },
-})
+}))

src/core/directives/error.ts

@@ -2,7 +2,7 @@ import { relative } from 'node:path'
 import type { NamedGroupsArray } from 'xregexp'
 import { defineDirective } from '../directive'
 
-export default defineDirective({
+export default defineDirective<undefined>(() => ({
   nested: false,
   name: '#error',
   pattern: /.*?#(?<directive>(?:warning)|(?:error)|(?:info))\s*(?<message>[\u4E00-\u9FA5 \t\w]*).*?$/gm,
@@ -20,4 +20,4 @@ export default defineDirective({
       return ''
     }
   },
-})
+}))

src/core/directives/if.ts

@@ -25,7 +25,7 @@ function resolveConditional(test: string, env = process.env) {
   }
 }
 
-export default defineDirective({
+export default defineDirective<undefined>(() => ({
   name: '#if',
   nested: true,
   pattern: {
@@ -53,4 +53,4 @@ export default defineDirective({
     }
     return ''
   },
-})
+}))

src/core/directives/index.ts

@@ -2,4 +2,8 @@ import IfDirective from './if'
 import DefineDirective from './define'
 import ErrorDirective from './error'
 
-export const builtinDirectives = [DefineDirective, IfDirective, ErrorDirective]
+export { default as IfDirective } from './if'
+export { default as DefineDirective } from './define'
+export { default as ErrorDirective } from './error'
+
+export const builtinDirectives = [DefineDirective(), IfDirective(), ErrorDirective()]

src/index.ts

@@ -1,3 +1,4 @@
 export * from './types'
 export * from './core/directive'
 export { default } from './core/unplugin'
+export * from './core/directives'

src/types.ts

@@ -62,3 +62,5 @@ export interface Options {
 }
 
 export interface UserOptions extends Partial<Options> { }
+
+export type DirectiveFactory<UserOptions> = (options: UserOptions) => FunctionDirective | Directive