Blame view

build/utils.ts 2.13 KB
陈文彬 authored
1
import fs from 'fs';
2
import path from 'path';
陈文彬 authored
3
import dotenv from 'dotenv';
4
vben authored
5
export function isDevFn(mode: string): boolean {
6
  return mode === 'development';
陈文彬 authored
7
8
}
vben authored
9
export function isProdFn(mode: string): boolean {
10
  return mode === 'production';
陈文彬 authored
11
12
}
vben authored
13
14
15
/**
 * Whether to generate package preview
 */
陈文彬 authored
16
17
18
export function isReportMode(): boolean {
  return process.env.REPORT === 'true';
}
vben authored
19
20
21
22
export interface ViteEnv {
  VITE_PORT: number;
  VITE_USE_MOCK: boolean;
vben authored
23
  VITE_USE_PWA: boolean;
24
25
  VITE_PUBLIC_PATH: string;
  VITE_PROXY: [string, string][];
26
  VITE_GLOB_APP_TITLE: string;
vben authored
27
  VITE_GLOB_APP_SHORT_NAME: string;
28
  VITE_USE_CDN: boolean;
vben authored
29
  VITE_DROP_CONSOLE: boolean;
30
  VITE_BUILD_COMPRESS: 'gzip' | 'brotli' | 'none';
vben authored
31
  VITE_LEGACY: boolean;
32
  VITE_USE_IMAGEMIN: boolean;
33
34
}
vben authored
35
// Read all environment variable configuration files to process.env
Vben authored
36
export function wrapperEnv(envConf: Recordable): ViteEnv {
陈文彬 authored
37
38
  const ret: any = {};
39
40
  for (const envName of Object.keys(envConf)) {
    let realName = envConf[envName].replace(/\\n/g, '\n');
41
    realName = realName === 'true' ? true : realName === 'false' ? false : realName;
Vben authored
42
43
44
45
46
47
48
49
50
    if (envName === 'VITE_PORT') {
      realName = Number(realName);
    }
    if (envName === 'VITE_PROXY') {
      try {
        realName = JSON.parse(realName);
      } catch (error) {}
    }
陈文彬 authored
51
52
53
54
55
    ret[envName] = realName;
    process.env[envName] = realName;
  }
  return ret;
}
56
vben authored
57
58
59
60
61
/**
 * Get the environment variables starting with the specified prefix
 * @param match prefix
 * @param confFiles ext
 */
62
63
64
65
66
67
68
69
export function getEnvConfig(match = 'VITE_GLOB_', confFiles = ['.env', '.env.production']) {
  let envConfig = {};
  confFiles.forEach((item) => {
    try {
      const env = dotenv.parse(fs.readFileSync(path.resolve(process.cwd(), item)));
      envConfig = { ...envConfig, ...env };
    } catch (error) {}
  });
Vben authored
70
71
72
73
74
75
76
77
78
79
  Object.keys(envConfig).forEach((key) => {
    const reg = new RegExp(`^(${match})`);
    if (!reg.test(key)) {
      Reflect.deleteProperty(envConfig, key);
    }
  });
  return envConfig;
}
vben authored
80
81
82
83
/**
 * Get user root directory
 * @param dir file path
 */
Vben authored
84
export function getRootPath(...dir: string[]) {
85
86
  return path.resolve(process.cwd(), ...dir);
}