feat(utils): 新增 devOrProd 开发环境和生产环境返回不同的值或调用不同的函数

.vscode/launch.json

@@ -0,0 +1,15 @@
+{
+  "configurations": [
+    {
+      "type": "node",
+      "name": "vscode-jest-tests",
+      "request": "launch",
+      "console": "integratedTerminal",
+      "internalConsoleOptions": "neverOpen",
+      "disableOptimisticBPs": true,
+      "program": "${workspaceFolder}/node_modules/jest/bin/jest.js",
+      "cwd": "${workspaceFolder}",
+      "args": ["--runInBand", "--watchAll=false"]
+    }
+  ]
+}

src/utils/devOrProd.test.ts

@@ -0,0 +1,45 @@
+import { devOrProd } from './devOrProd'
+
+describe('devOrProd', () => {
+  beforeEach(() => {
+    delete process.env.NODE_ENV
+  })
+
+  test('空值被认为是生产环境', () => {
+    delete process.env.NODE_ENV
+    expect(devOrProd(1, 2)).toBe(2)
+    process.env.NODE_ENV = ''
+    expect(devOrProd(1, 2)).toBe(2)
+  })
+
+  test('production', () => {
+    process.env.NODE_ENV = 'production'
+    expect(devOrProd(1, 2)).toBe(2)
+  })
+
+  test('prod', () => {
+    process.env.NODE_ENV = 'prod'
+    expect(devOrProd(1, 2)).toBe(2)
+  })
+
+  test('development', () => {
+    process.env.NODE_ENV = 'development'
+    expect(devOrProd(1, 2)).toBe(1)
+  })
+
+  test('dev', () => {
+    process.env.NODE_ENV = 'dev'
+    expect(devOrProd(1, 2)).toBe(1)
+  })
+
+  test('函数', () => {
+    process.env.NODE_ENV = 'prod'
+    expect(devOrProd(1, () => 2)).toBe(2)
+    expect(
+      devOrProd(
+        () => 100,
+        () => 4,
+      ),
+    ).toBe(4)
+  })
+})

src/utils/devOrProd.ts

@@ -0,0 +1,21 @@
+/**
+ * 开发环境和生产环境返回不同的值或调用不同的函数。
+ *
+ * `process.env.NODE_ENV` 为空值、`production`、`prod` 时被认为是生产环境，否则是开发环境。
+ *
+ * @param devValue 开发环境返回的值或调用的函数
+ * @param prodValue 生产环境返回的值或调用的函数
+ */
+export function devOrProd<T>(
+  devValue: T | (() => T),
+  prodValue: T | (() => T),
+): T {
+  const nodeEnv = process.env.NODE_ENV
+  return !nodeEnv || nodeEnv === 'production' || nodeEnv === 'prod'
+    ? typeof prodValue === 'function'
+      ? (prodValue as any)()
+      : prodValue
+    : typeof devValue === 'function'
+    ? (devValue as any)()
+    : devValue
+}

src/utils/index.ts

@@ -20,6 +20,7 @@ export * from './createUrlQueryString'
 export * from './DataPacker'
 export * from './dedent'
 export * from './defaultIndexTo'
+export * from './devOrProd'
 export * from './EventBus'
 export * from './formatBytes'
 export * from './formatNumber'