76 lines
2.1 KiB
TypeScript
76 lines
2.1 KiB
TypeScript
/**
|
|
* Copyright (c) 2013-Now http://jeesite.com All rights reserved.
|
|
* No deletion without permission, or be held responsible to law.
|
|
* @author ThinkGem
|
|
*/
|
|
import fs from 'fs-extra';
|
|
import dotenv from 'dotenv';
|
|
import { join } from 'node:path';
|
|
|
|
// Read all environment variable configuration files to process.env
|
|
export function wrapperEnv(envConf: Recordable): ViteEnv {
|
|
const ret: any = {};
|
|
|
|
for (const envName of Object.keys(envConf)) {
|
|
let realName = envConf[envName].replace(/\\n/g, '\n');
|
|
realName = realName === 'true' ? true : realName === 'false' ? false : realName;
|
|
|
|
if (envName === 'VITE_PORT') {
|
|
realName = Number(realName);
|
|
}
|
|
if (envName === 'VITE_PROXY' && realName) {
|
|
try {
|
|
realName = JSON.parse(realName.replace(/'/g, '"'));
|
|
} catch (error) {
|
|
realName = '';
|
|
}
|
|
}
|
|
ret[envName] = realName;
|
|
if (typeof realName === 'string') {
|
|
process.env[envName] = realName;
|
|
} else if (typeof realName === 'object') {
|
|
process.env[envName] = JSON.stringify(realName);
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
* 获取当前环境下生效的配置文件名
|
|
*/
|
|
function getEnvConfigFiles() {
|
|
const script = process.env.npm_lifecycle_script;
|
|
const reg = new RegExp('--mode ([a-z_\\d]+)');
|
|
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'];
|
|
}
|
|
|
|
/**
|
|
* Get the environment variables starting with the specified prefix
|
|
* @param match prefix
|
|
* @param confFiles ext
|
|
*/
|
|
export async function getEnvConfig(match = 'VITE_(GLOB|FILE)_', confFiles = getEnvConfigFiles()) {
|
|
let envConfig = {};
|
|
for (const item of confFiles) {
|
|
try {
|
|
const envPath = await fs.readFile(join(process.cwd(), item), { encoding: 'utf8' });
|
|
const env = dotenv.parse(envPath);
|
|
envConfig = { ...envConfig, ...env };
|
|
} catch (e) {
|
|
console.error(`Error in parsing ${item}`, e);
|
|
}
|
|
}
|
|
const reg = new RegExp(`^(${match})`);
|
|
Object.keys(envConfig).forEach((key) => {
|
|
if (!reg.test(key)) {
|
|
Reflect.deleteProperty(envConfig, key);
|
|
}
|
|
});
|
|
return envConfig;
|
|
}
|