1
import fs from 'fs';
nebv
authored
5 years ago
2
import path from 'path';
3
4
import { networkInterfaces } from 'os';
import dotenv from 'dotenv';
nebv
authored
5 years ago
5
import chalk from 'chalk';
vben
authored
5 years ago
6
// import execa from 'execa';
nebv
authored
5 years ago
7
8
9
export const isFunction = (arg: unknown): arg is (...args: any[]) => any =>
typeof arg === 'function';
vben
authored
5 years ago
10
11
12
13
export const isRegExp = (arg: unknown): arg is RegExp =>
Object.prototype.toString.call(arg) === '[object RegExp]';
vben
authored
5 years ago
14
15
16
/**
* get client ip address
*/
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
export function getIPAddress() {
let interfaces = networkInterfaces();
for (let devName in interfaces) {
let iFace = interfaces[devName];
if (!iFace) return;
for (let i = 0; i < iFace.length; i++) {
let alias = iFace[i];
if (alias.family === 'IPv4' && alias.address !== '127.0.0.1' && !alias.internal) {
return alias.address;
}
}
}
return '';
}
vben
authored
5 years ago
33
export function isDevFn(mode: string): boolean {
vben
authored
5 years ago
34
return mode === 'development';
35
36
}
vben
authored
5 years ago
37
export function isProdFn(mode: string): boolean {
vben
authored
5 years ago
38
return mode === 'production';
39
40
}
vben
authored
5 years ago
41
42
43
/**
* Whether to generate package preview
*/
44
45
46
export function isReportMode(): boolean {
return process.env.REPORT === 'true';
}
vben
authored
5 years ago
47
48
49
50
/**
* Whether to generate gzip for packaging
*/
vben
authored
5 years ago
51
52
53
export function isBuildGzip(): boolean {
return process.env.VITE_BUILD_GZIP === 'true';
}
vben
authored
5 years ago
54
55
56
57
export interface ViteEnv {
VITE_PORT: number;
VITE_USE_MOCK: boolean;
vben
authored
5 years ago
58
VITE_USE_PWA: boolean;
59
60
VITE_PUBLIC_PATH: string;
VITE_PROXY: [string, string][];
nebv
authored
5 years ago
61
VITE_GLOB_APP_TITLE: string;
vben
authored
5 years ago
62
VITE_GLOB_APP_SHORT_NAME: string;
nebv
authored
5 years ago
63
VITE_USE_CDN: boolean;
vben
authored
5 years ago
64
65
VITE_DROP_CONSOLE: boolean;
VITE_BUILD_GZIP: boolean;
vben
authored
5 years ago
66
VITE_DYNAMIC_IMPORT: boolean;
vben
authored
5 years ago
67
VITE_LEGACY: boolean;
68
69
}
vben
authored
5 years ago
70
// Read all environment variable configuration files to process.env
vben
authored
5 years ago
71
export function wrapperEnv(envConf: any): ViteEnv {
72
73
const ret: any = {};
vben
authored
5 years ago
74
75
for (const envName of Object.keys(envConf)) {
let realName = envConf[envName].replace(/\\n/g, '\n');
76
77
78
79
80
81
82
83
84
realName = realName === 'true' ? true : realName === 'false' ? false : realName;
if (envName === 'VITE_PORT') {
realName = Number(realName);
}
if (envName === 'VITE_PROXY') {
try {
realName = JSON.parse(realName);
} catch (error) {}
}
85
86
87
88
89
ret[envName] = realName;
process.env[envName] = realName;
}
return ret;
}
nebv
authored
5 years ago
90
vben
authored
5 years ago
91
92
93
94
95
/**
* Get the environment variables starting with the specified prefix
* @param match prefix
* @param confFiles ext
*/
nebv
authored
5 years ago
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
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) {}
});
Object.keys(envConfig).forEach((key) => {
const reg = new RegExp(`^(${match})`);
if (!reg.test(key)) {
Reflect.deleteProperty(envConfig, key);
}
});
return envConfig;
}
vben
authored
5 years ago
114
function consoleFn(color: string, message: any) {
nebv
authored
5 years ago
115
116
console.log(
chalk.blue.bold('**************** ') +
vben
authored
5 years ago
117
(chalk as any)[color].bold(message) +
nebv
authored
5 years ago
118
119
120
121
chalk.blue.bold(' ****************')
);
}
vben
authored
5 years ago
122
123
124
125
/**
* warnConsole
* @param message
*/
vben
authored
5 years ago
126
127
128
129
export function successConsole(message: any) {
consoleFn('green', '✨ ' + message);
}
vben
authored
5 years ago
130
131
132
133
/**
* warnConsole
* @param message
*/
nebv
authored
5 years ago
134
export function errorConsole(message: any) {
vben
authored
5 years ago
135
consoleFn('red', '✨ ' + message);
nebv
authored
5 years ago
136
137
}
vben
authored
5 years ago
138
139
140
141
/**
* warnConsole
* @param message message
*/
nebv
authored
5 years ago
142
export function warnConsole(message: any) {
vben
authored
5 years ago
143
consoleFn('yellow', '✨ ' + message);
nebv
authored
5 years ago
144
145
}
vben
authored
5 years ago
146
147
148
149
/**
* Get user root directory
* @param dir file path
*/
nebv
authored
5 years ago
150
151
152
export function getCwdPath(...dir: string[]) {
return path.resolve(process.cwd(), ...dir);
}