1
import fs from 'fs';
nebv
authored
5 years ago
2
import path from 'path';
3
import dotenv from 'dotenv';
nebv
authored
5 years ago
4
vben
authored
5 years ago
5
export function isDevFn(mode: string): boolean {
vben
authored
5 years ago
6
return mode === 'development';
7
8
}
vben
authored
5 years ago
9
export function isProdFn(mode: string): boolean {
vben
authored
5 years ago
10
return mode === 'production';
11
12
}
vben
authored
5 years ago
13
14
15
/**
* Whether to generate package preview
*/
16
17
18
export function isReportMode(): boolean {
return process.env.REPORT === 'true';
}
vben
authored
5 years ago
19
20
// Read all environment variable configuration files to process.env
Vben
authored
4 years ago
21
export function wrapperEnv(envConf: Recordable): ViteEnv {
22
23
const ret: any = {};
vben
authored
5 years ago
24
25
for (const envName of Object.keys(envConf)) {
let realName = envConf[envName].replace(/\\n/g, '\n');
26
realName = realName === 'true' ? true : realName === 'false' ? false : realName;
Vben
authored
4 years ago
27
28
29
30
31
32
33
34
35
if (envName === 'VITE_PORT') {
realName = Number(realName);
}
if (envName === 'VITE_PROXY') {
try {
realName = JSON.parse(realName);
} catch (error) {}
}
36
37
38
39
40
ret[envName] = realName;
process.env[envName] = realName;
}
return ret;
}
nebv
authored
5 years ago
41
vben
authored
5 years ago
42
/**
43
44
45
46
* 获取当前环境下生效的配置文件名
*/
function getConfFiles() {
const script = process.env.npm_lifecycle_script;
47
const reg = new RegExp('--mode ([a-z]+)');
48
49
50
51
52
53
54
55
56
const result = reg.exec(script as string) as any;
if (result) {
const mode = result[1] as string;
return ['.env', `.env.${mode}`];
}
return ['.env', '.env.production'];
}
/**
vben
authored
5 years ago
57
58
59
60
* Get the environment variables starting with the specified prefix
* @param match prefix
* @param confFiles ext
*/
61
export function getEnvConfig(match = 'VITE_GLOB_', confFiles = getConfFiles()) {
nebv
authored
5 years ago
62
63
64
65
66
let envConfig = {};
confFiles.forEach((item) => {
try {
const env = dotenv.parse(fs.readFileSync(path.resolve(process.cwd(), item)));
envConfig = { ...envConfig, ...env };
67
68
69
} catch (e) {
console.error(`Error in parsing ${item}`, e);
}
nebv
authored
5 years ago
70
});
71
const reg = new RegExp(`^(${match})`);
nebv
authored
5 years ago
72
73
74
75
76
77
78
79
Object.keys(envConfig).forEach((key) => {
if (!reg.test(key)) {
Reflect.deleteProperty(envConfig, key);
}
});
return envConfig;
}
vben
authored
5 years ago
80
81
82
83
/**
* Get user root directory
* @param dir file path
*/
Vben
authored
4 years ago
84
export function getRootPath(...dir: string[]) {
nebv
authored
5 years ago
85
86
return path.resolve(process.cwd(), ...dir);
}