51 lines
1.7 KiB
TypeScript
51 lines
1.7 KiB
TypeScript
import { readFileSync, existsSync, writeFileSync, mkdirSync } from "fs";
|
||
import path from "path";
|
||
|
||
// 默认环境变量（当 env 文件不存在时自动创建）
|
||
const defaultEnvValues: Record<string, string> = {
|
||
dev: `NODE_ENV=dev\nPORT=60000\nOSSURL=http://127.0.0.1:60000/`,
|
||
prod: `NODE_ENV=prod\nPORT=60000\nOSSURL=http://127.0.0.1:60000/`,
|
||
};
|
||
|
||
// 判断是否为打包后的 Electron 环境
|
||
const isElectron = typeof process.versions?.electron !== "undefined";
|
||
let isPackaged = false;
|
||
if (isElectron) {
|
||
const { app } = require("electron");
|
||
isPackaged = app.isPackaged;
|
||
}
|
||
|
||
//加载环境变量（打包环境默认使用 prod）
|
||
const env = process.env.NODE_ENV ?? (isPackaged ? "prod" : "dev");
|
||
if (!env) {
|
||
console.log("[环境变量为空]");
|
||
process.exit(1);
|
||
} else {
|
||
// Electron 打包环境使用 userData 目录，开发环境使用项目根目录
|
||
let envDir: string;
|
||
if (isElectron) {
|
||
const { app } = require("electron");
|
||
envDir = path.join(app.getPath("userData"), "env");
|
||
} else {
|
||
envDir = path.resolve("env");
|
||
}
|
||
const envFilePath = path.join(envDir, `.env.${env}`);
|
||
|
||
// 自动创建 env 目录和文件（.gitignore 可能忽略了这些文件）
|
||
if (!existsSync(envDir)) {
|
||
mkdirSync(envDir, { recursive: true });
|
||
}
|
||
if (!existsSync(envFilePath)) {
|
||
const content = defaultEnvValues[env] ?? defaultEnvValues.prod;
|
||
writeFileSync(envFilePath, content, "utf8");
|
||
console.log(`[环境变量] 自动创建 ${envFilePath}`);
|
||
}
|
||
|
||
const text = readFileSync(envFilePath, "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(`[环境变量] ${env}`);
|
||
}
|