|
1
|
import fs from 'fs';
|
nebv
authored
|
2
|
import path from 'path';
|
|
3
|
import dotenv from 'dotenv';
|
nebv
authored
|
4
|
|
vben
authored
|
5
|
export function isDevFn(mode: string): boolean {
|
vben
authored
|
6
|
return mode === 'development';
|
|
7
8
|
}
|
vben
authored
|
9
|
export function isProdFn(mode: string): boolean {
|
vben
authored
|
10
|
return mode === 'production';
|
|
11
12
|
}
|
vben
authored
|
13
14
15
|
/**
* Whether to generate package preview
*/
|
|
16
17
18
|
export function isReportMode(): boolean {
return process.env.REPORT === 'true';
}
|
vben
authored
|
19
20
|
// Read all environment variable configuration files to process.env
|
Vben
authored
|
21
|
export function wrapperEnv(envConf: Recordable): ViteEnv {
|
|
22
23
|
const ret: any = {};
|
vben
authored
|
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
|
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
|
41
|
|
vben
authored
|
42
43
44
45
46
|
/**
* Get the environment variables starting with the specified prefix
* @param match prefix
* @param confFiles ext
*/
|
nebv
authored
|
47
48
49
50
51
52
53
54
|
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
|
55
|
|
nebv
authored
|
56
57
58
59
60
61
62
63
64
|
Object.keys(envConfig).forEach((key) => {
const reg = new RegExp(`^(${match})`);
if (!reg.test(key)) {
Reflect.deleteProperty(envConfig, key);
}
});
return envConfig;
}
|
vben
authored
|
65
66
67
68
|
/**
* Get user root directory
* @param dir file path
*/
|
Vben
authored
|
69
|
export function getRootPath(...dir: string[]) {
|
nebv
authored
|
70
71
|
return path.resolve(process.cwd(), ...dir);
}
|