jiek 2.3.1 → 2.3.2

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "jiek",
   "type": "module",
-  "version": "2.3.1",
+  "version": "2.3.2",
   "description": "A lightweight toolkit for compiling and managing libraries based on `package.json` metadata and suitable for `Monorepo`.",
   "author": "YiJie <yijie4188@gmail.com>",
   "homepage": "https://github.com/NWYLZW/jiek/tree/master/packages/jiek#readme",
@@ -35,6 +35,11 @@
       "jiek/__source__": "./src/rollup/index.ts",
       "require": "./dist/rollup/index.cjs",
       "default": "./dist/rollup/index.js"
+    },
+    "./rollup-plugin-utils": {
+      "jiek/__source__": "./src/rollup-plugin-utils.ts",
+      "require": "./dist/rollup-plugin-utils.cjs",
+      "default": "./dist/rollup-plugin-utils.js"
     }
   },
   "imports": {
@@ -74,13 +79,11 @@
     "js-yaml": "^4.1.0",
     "jsonc-parser": "^3.2.1",
     "koa": "^2.15.3",
+    "magic-string": "^0.30.17",
     "rollup": "^4.0.0",
     "@jiek/pkger": "^0.2.2",
     "@jiek/utils": "^0.2.3"
   },
-  "publishConfig": {
-    "directory": "./dist/.internal"
-  },
   "peerDependenciesMeta": {
     "@pnpm/filter-workspace-packages": {
       "optional": true

package/rollup-plugin-utils/package.json

@@ -0,0 +1 @@
+{"type":"module","module":"../dist/rollup-plugin-utils.js","main":"../dist/rollup-plugin-utils.cjs"}

package/src/rollup/base.ts

@@ -1,4 +1,5 @@
 import type { InputPluginOption, OutputOptions } from 'rollup'
+import { Replacements } from './plugins/replace'
 
 export type Mapping2ROO<K extends keyof OutputOptions> = OutputOptions[K] | {
   js?: OutputOptions[K]
@@ -134,4 +135,17 @@ export interface TemplateOptions {
    * ```
    */
   injects?: Record<string, string | [string, string]>
+  /**
+   * Replace the specified content in the code.
+   *
+   * @example
+   * ```js
+   * {
+   *   'process.env.DEBUG': 'false',
+   *   'process.env.NODE_ENV': JSON.stringify('production'),
+   *   'process.env.BUILD_PATH': ctx => JSON.stringify(ctx.id)
+   * }
+   * ```
+   */
+  replacements?: Replacements
 }

package/src/rollup/index.ts

@@ -27,6 +27,7 @@ import { getCompilerOptionsByFilePath } from '#~/utils/ts'
 import type { ConfigGenerateContext, TemplateOptions } from './base'
 import createRequire, { CREATE_REQUIRE_VIRTUAL_MODULE_NAME } from './plugins/create-require'
 import progress from './plugins/progress'
+import replace from './plugins/replace'
 import skip from './plugins/skip'
 import withExternal from './plugins/with-external.ts'
 import type { PackageJSON } from './utils/externalResolver'
@@ -423,7 +424,13 @@ const generateConfigs = (
   const { js: jsOutput, dts: dtsOutput } = resolveOutputControls(context, build.output)
   const rollupOptions: RollupOptions[] = []
 
-  const commonPlugins: InputPluginOption[] = [
+  const commonPlugins = (
+    { sourcemap }: { sourcemap?: string | boolean }
+  ): InputPluginOption[] => [
+    replace({
+      sourcemap: sourcemap === 'hidden' ? false : !!sourcemap,
+      values: build.replacements
+    }),
     ...inputCommonPlugins,
     withExternal(),
     !disableCollectInternalModule && {
@@ -570,7 +577,7 @@ const generateConfigs = (
         )
       ],
       plugins: [
-        ...commonPlugins,
+        ...commonPlugins({ sourcemap }),
         nodeResolvePluginInstance,
         import('rollup-plugin-postcss')
           .then(({ default: postcss }) =>
@@ -630,7 +637,7 @@ const generateConfigs = (
         }
       ],
       plugins: [
-        ...commonPlugins,
+        ...commonPlugins({ sourcemap }),
         skip({ patterns: [STYLE_REGEXP] }),
         dts({
           respectExternal: true,

package/src/rollup/plugins/replace.ts

@@ -0,0 +1,96 @@
+import type { FilterOptions } from 'jiek/rollup-plugin-utils'
+import { createFilter, definePlugin } from 'jiek/rollup-plugin-utils'
+import MagicString from 'magic-string'
+
+export interface ReplacementFuncCtx {
+  type: 'transform' | 'renderChunk'
+  id: string
+  code: string
+  start: number
+  end: number
+}
+
+export type ReplacementFunc = (ctx: ReplacementFuncCtx) => string
+
+export type Replacements = Record<
+  string,
+  string | ReplacementFunc
+>
+
+export type Options =
+  & FilterOptions
+  & {
+    sourcemap?: boolean
+    values?: Replacements
+  }
+
+export default definePlugin((options: Options = {}) => {
+  const {
+    include = [/\.[cm]?[tj]sx?$/],
+    exclude = [/node_modules/],
+    values = {},
+    sourcemap
+  } = options
+  const allValues = { ...values }
+  const allKeys = Object.keys(allValues)
+  const filter = createFilter({ include, exclude })
+
+  const replaceAll = (ctx: Pick<ReplacementFuncCtx, 'type' | 'id'>, code: string) => {
+    const ms = new MagicString(code)
+    allKeys.forEach(key => {
+      const reg = new RegExp(key, 'g')
+      let match
+      // eslint-disable-next-line no-cond-assign
+      while ((match = reg.exec(code))) {
+        const start = match.index
+        const end = start + key.length
+        ms.overwrite(
+          match.index,
+          match.index + key.length,
+          typeof allValues[key] === 'function'
+            ? allValues[key]({
+              ...ctx,
+              code,
+              start,
+              end
+            })
+            : allValues[key]
+        )
+      }
+    })
+    return ms
+  }
+
+  return {
+    name: 'jiek:replace',
+    transform: {
+      order: 'pre',
+      handler(code, id) {
+        if (allKeys.length === 0) return
+        if (filter(id)) return
+        const ms = replaceAll({ type: 'transform', id }, code)
+        if (ms == null) return
+
+        return {
+          code: ms.toString(),
+          map: sourcemap ? ms.generateMap({ hires: true }) : null
+        }
+      }
+    },
+    renderChunk: {
+      order: 'post',
+      handler(code, { fileName: id }) {
+        if (allKeys.length === 0) return
+        if (filter(id)) return
+
+        const ms = replaceAll({ type: 'renderChunk', id }, code)
+        if (ms == null) return
+
+        return {
+          code: ms.toString(),
+          map: sourcemap ? ms.generateMap({ hires: true }) : null
+        }
+      }
+    }
+  }
+})

package/src/rollup-plugin-utils.ts

@@ -0,0 +1,32 @@
+import { isMatch } from 'micromatch'
+import type { PluginImpl } from 'rollup'
+
+export const definePlugin = <O extends object>(plugin: PluginImpl<O>) => plugin
+
+export interface FilterOptions {
+  include?: string | RegExp | (string | RegExp)[]
+  exclude?: string | RegExp | (string | RegExp)[]
+}
+
+export function createFilter(options: FilterOptions) {
+  const { include = [], exclude = [] } = options
+
+  const resolvedInclude = Array.isArray(include) ? include : [include]
+  const resolvedExclude = Array.isArray(exclude) ? exclude : [exclude]
+
+  return (id: string) => {
+    if (typeof id !== 'string') return false
+    const isInclude = resolvedInclude.length === 0 || resolvedInclude.some(filter => {
+      return filter instanceof RegExp
+        ? filter.test(id)
+        : isMatch(id, filter)
+    })
+    const isExclude = resolvedExclude.length > 0 && resolvedExclude.some(filter => {
+      return filter instanceof RegExp
+        ? filter.test(id)
+        : isMatch(id, filter)
+    })
+
+    return !isInclude || isExclude
+  }
+}