fix: use options for webpack loader, close #3

package.json

@@ -9,6 +9,9 @@
     ".": {
       "import": "./dist/index.mjs",
       "require": "./dist/index.cjs"
+    },
+    "./dist/webpack/loader": {
+      "require": "./dist/webpack/loader.cjs"
     }
   },
   "main": "dist/index.cjs",

siroc.config.ts

@@ -0,0 +1,5 @@
+import { defineSirocConfig } from 'siroc'
+
+export default defineSirocConfig({
+
+})

src/rollup.ts → src/rollup/index.ts

@@ -1,5 +1,5 @@
 import { Plugin as RollupPlugin } from 'rollup'
-import { UnpluginInstance, UnpluginFactory } from './types'
+import { UnpluginInstance, UnpluginFactory } from '../types'
 
 export function getRollupPlugin <UserOptions = {}> (
   factory: UnpluginFactory<UserOptions>

src/webpack/index.ts

@@ -0,0 +1,44 @@
+import { resolve } from 'path'
+import { Compiler } from 'webpack'
+import { UnpluginInstance, UnpluginFactory } from '../types'
+
+export function getWebpackPlugin<UserOptions = {}> (
+  factory: UnpluginFactory<UserOptions>
+): UnpluginInstance<UserOptions>['webpack'] {
+  class UnpluginWebpackPlugin {
+    // eslint-disable-next-line no-useless-constructor
+    constructor (public userOptions?: UserOptions) {}
+    apply (compiler: Compiler) {
+      const rawPlugin = factory(this.userOptions)
+
+      // @ts-expect-error
+      if (!compiler.$unpluginContext) {
+      // @ts-expect-error
+        compiler.$unpluginContext = {}
+      }
+      // @ts-expect-error
+      compiler.$unpluginContext[rawPlugin.name] = rawPlugin
+
+      if (rawPlugin.transform) {
+        compiler.options.module.rules.push({
+          include (id: string) {
+            if (rawPlugin.transformInclude) {
+              return rawPlugin.transformInclude(id)
+            } else {
+              return true
+            }
+          },
+          enforce: rawPlugin.enforce,
+          use: [{
+            loader: resolve(__dirname, '..', 'dist/webpack/loader.cjs'),
+            options: {
+              unpluginName: rawPlugin.name
+            }
+          }]
+        })
+      }
+    }
+  }
+
+  return UserOptions => new UnpluginWebpackPlugin(UserOptions)
+}

src/webpack/loader.ts

@@ -0,0 +1,18 @@
+import type { LoaderContext } from 'webpack'
+
+export default async function (this: LoaderContext<any>, source: string, map: any) {
+  const callback = this.async()
+  const { unpluginName } = this.query
+  // @ts-expect-error
+  const plugin = this._compiler.$unpluginContext[unpluginName]
+
+  const res = await plugin.transform(source, this.resource)
+
+  if (res == null) {
+    callback(null, source, map)
+  } else if (typeof res !== 'string') {
+    callback(null, res.code, res.map)
+  } else {
+    callback(null, res, map)
+  }
+}