38 lines
1.3 KiB
TypeScript
38 lines
1.3 KiB
TypeScript
import { readFileSync, existsSync, writeFileSync } from "fs";
|
|
|
|
function createDefaultEnvFile(path: string) {
|
|
const defaultContent = ["# 环境变量配置", "NODE_ENV=dev"].join("\n");
|
|
writeFileSync(path, defaultContent, { encoding: "utf8" });
|
|
console.log(`[环境变量]: 已创建默认的 ${path}`);
|
|
}
|
|
|
|
function loadDotenvESM(envPath = ".env.local") {
|
|
let finalPath: string;
|
|
|
|
if (typeof process.versions?.electron !== "undefined") {
|
|
const { app } = require("electron");
|
|
finalPath = app.getPath("userData") + `/${envPath}`;
|
|
// 如果 userData 目录下的 env 文件不存在，则尝试当前目录
|
|
if (!existsSync(finalPath)) {
|
|
finalPath = envPath;
|
|
}
|
|
} else {
|
|
finalPath = envPath;
|
|
}
|
|
|
|
// 若文件不存在，自动创建一个带默认内容的环境变量文件
|
|
if (!existsSync(finalPath)) {
|
|
createDefaultEnvFile(finalPath);
|
|
}
|
|
|
|
const text = readFileSync(finalPath, "utf8");
|
|
for (const line of text.split("\n")) {
|
|
const idx = line.indexOf("=");
|
|
if (idx > 0) process.env[line.slice(0, idx).trim()] = line.slice(idx + 1).trim();
|
|
}
|
|
console.log(`[环境变量]: 已加载 ${finalPath}`);
|
|
}
|
|
|
|
// 若非 Electron 环境，则加载 .env.local
|
|
if (typeof process.versions?.electron == "undefined") loadDotenvESM(".env.local");
|