New: Integrate plugin for environment variable replacement in build process

Author: 1aron

packages/techor/package.json

@@ -68,6 +68,7 @@
     "dependencies": {
         "@rollup/plugin-commonjs": "^28.0.2",
         "@rollup/plugin-node-resolve": "^16.0.0",
+        "@rollup/plugin-replace": "^6.0.2",
         "@swc/core": "^1.10.1",
         "@techor/extend": "workspace:^",
         "@techor/fs": "workspace:^",

packages/techor/src/commands/build.ts

@@ -17,6 +17,7 @@ import getWideExternal from '../utils/get-wide-external'
 import { BuildCommonOptions, Config, default as defaultConfig } from '../config'
 import yargsParser, { Options as YargsParserOptions } from 'yargs-parser'
 import loadConfig from '../load-config'
+import replace from '@rollup/plugin-replace'
 
 // core plugins
 import esmShim from '../plugins/esm-shim'
@@ -97,6 +98,8 @@ export default async function build() {
                     extendedBuild.minify = true
                 }
                 if (fileBasenameSplits.includes('global') || fileBasenameSplits.includes('iife')) extendedBuild.output.format = 'iife'
+                extendedBuild.env = fileBasenameSplits.includes('development') ? 'development' : 'production'
+
                 for (const [eachInput, eachBuildOptions] of buildMap) {
                     for (const eachOutputOptions of eachBuildOptions.outputOptionsList) {
                         if (normalize(eachOutputOptions.output.file) === normalize(extendedBuild.output.file)) {
@@ -147,6 +150,10 @@ export default async function build() {
                 }
                 (buildOptions.input.plugins as RollupInputPluginOption[]).unshift(
                     ...[
+                        replace({
+                            preventAssignment: true,
+                            'process.env.NODE_ENV': JSON.stringify(extendedBuild.env)
+                        }),
                         swc(extendedSWCOptions),
                         config.build.commonjs && commonjs(config.build.commonjs),
                         config.build.nodeResolve && nodeResolve(config.build.nodeResolve),
@@ -208,6 +215,8 @@ export default async function build() {
                                     case 'import':
                                         handleExports(eachUnknowExports, 'esm')
                                         break
+                                    default:
+                                        handleExports(eachUnknowExports, eachFormat)
                                 }
                             }
                         }

packages/techor/src/config.ts

@@ -82,6 +82,7 @@ export interface BuildCommonOptions {
     external?: string[]
     tsconfig?: string
     formats?: RollupOutputOptions['format'][]
+    env?: 'development' | 'production'
 }
 
 export interface BuildOptions extends BuildCommonOptions {

packages/techor/tests/env/package.json

@@ -0,0 +1,21 @@
+{
+    "name": "@test/env",
+    "version": "1.0.0",
+    "private": true,
+    "files": [
+        "dist"
+    ],
+    "browser": "./dist/global.min.js",
+    "exports": {
+        ".": {
+            "import": {
+                "development": "./dist/index.development.mjs",
+                "default": "./dist/index.mjs"
+            },
+            "require": {
+                "development": "./dist/index.development.cjs",
+                "default": "./dist/index.cjs"
+            }
+        }
+    }
+}

packages/techor/tests/env/src/global.min.ts

@@ -0,0 +1 @@
+export * from './index'

packages/techor/tests/env/src/index.development.ts

@@ -0,0 +1 @@
+export * from './index'

packages/techor/tests/env/src/index.ts

@@ -0,0 +1,7 @@
+import startDebug from './start-debug'
+
+if (process.env.NODE_ENV === 'development') {
+    startDebug()
+}
+
+console.log('hello world')

packages/techor/tests/env/src/start-debug.ts

@@ -0,0 +1,3 @@
+export default function startDebug() {
+console.log('start debug')
+}

packages/techor/tests/env/test.ts

@@ -0,0 +1,46 @@
+import { execSync } from 'node:child_process'
+import { readFileSync } from 'node:fs'
+import { join } from 'node:path'
+import dedent from 'ts-dedent'
+
+beforeAll(() => {
+    execSync('tsx ../../src/bin build', { cwd: __dirname })
+})
+
+test('production', () => {
+    expect(readFileSync(join(__dirname, './dist/index.cjs'), 'utf-8')).toBe(dedent`
+        'use strict';
+
+        console.log('hello world');
+
+    `)
+    expect(readFileSync(join(__dirname, './dist/index.mjs'), 'utf-8')).toBe(`console.log('hello world');\n`)
+    expect(readFileSync(join(__dirname, './dist/global.min.js'), 'utf-8')).toBe(`console.log("hello world");\n`)
+})
+
+test('development', () => {
+    expect(readFileSync(join(__dirname, './dist/index.development.cjs'), 'utf-8')).toBe(dedent`
+        'use strict';
+
+        function startDebug() {
+            console.log('start debug');
+        }
+
+        {
+            startDebug();
+        }
+        console.log('hello world');
+
+    `)
+    expect(readFileSync(join(__dirname, './dist/index.development.mjs'), 'utf-8')).toBe(dedent`
+        function startDebug() {
+            console.log('start debug');
+        }
+
+        {
+            startDebug();
+        }
+        console.log('hello world');
+
+    `)
+})

packages/techor/tests/env/tsconfig.json

@@ -0,0 +1,13 @@
+{
+    "extends": "../../../../tsconfig.json",
+    "compilerOptions": {
+        "baseUrl": ".",
+        "outDir": "dist",
+    },
+    "include": [
+        "src/**/*"
+    ],
+    "exclude": [
+        "src/**/*.test.ts"
+    ]
+}