50 lines
1.2 KiB
TypeScript
50 lines
1.2 KiB
TypeScript
|
import { join } from 'node:path';
|
||
|
|
|
||
|
|
import { fs } from '@vben/node-utils';
|
||
|
|
import dotenv from 'dotenv';
|
||
|
|
|
||
|
|
/**
|
||
|
|
* 获取当前环境下生效的配置文件名
|
||
|
|
*/
|
||
|
|
function getConfFiles() {
|
||
|
|
const script = process.env.npm_lifecycle_script as string;
|
||
|
|
const reg = /--mode ([\d_a-z]+)/;
|
||
|
|
const result = reg.exec(script);
|
||
|
|
if (result) {
|
||
|
|
const mode = result[1];
|
||
|
|
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_',
|
||
|
|
confFiles = getConfFiles(),
|
||
|
|
) {
|
||
|
|
let envConfig = {};
|
||
|
|
|
||
|
|
for (const confFile of confFiles) {
|
||
|
|
try {
|
||
|
|
const envPath = await fs.readFile(join(process.cwd(), confFile), {
|
||
|
|
encoding: 'utf8',
|
||
|
|
});
|
||
|
|
const env = dotenv.parse(envPath);
|
||
|
|
envConfig = { ...envConfig, ...env };
|
||
|
|
} catch (error) {
|
||
|
|
console.error(`Error in parsing ${confFile}`, error);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
const reg = new RegExp(`^(${match})`);
|
||
|
|
Object.keys(envConfig).forEach((key) => {
|
||
|
|
if (!reg.test(key)) {
|
||
|
|
Reflect.deleteProperty(envConfig, key);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
return envConfig;
|
||
|
|
}
|