72 lines
2.0 KiB
TypeScript
72 lines
2.0 KiB
TypeScript
import { existsSync, readFileSync } from "node:fs";
|
|
import { dirname, resolve } from "node:path";
|
|
import { fileURLToPath } from "node:url";
|
|
|
|
function resolveWorkspaceRoot() {
|
|
const currentDir = dirname(fileURLToPath(import.meta.url));
|
|
return resolve(currentDir, "../../../");
|
|
}
|
|
|
|
export function resolveWorkspaceEnvPaths(options: { workspaceRoot?: string; nodeEnv?: string | undefined } = {}) {
|
|
const workspaceRoot = options.workspaceRoot ?? resolveWorkspaceRoot();
|
|
const nodeEnv = options.nodeEnv ?? process.env.NODE_ENV;
|
|
const candidates = [".env"];
|
|
|
|
if (nodeEnv) {
|
|
candidates.push(`.env.${nodeEnv}`);
|
|
}
|
|
|
|
candidates.push(".env.local");
|
|
|
|
if (nodeEnv) {
|
|
candidates.push(`.env.${nodeEnv}.local`);
|
|
}
|
|
|
|
return [...new Set(candidates.map((candidate) => resolve(workspaceRoot, candidate)))];
|
|
}
|
|
|
|
export function loadWorkspaceEnv(options: { workspaceRoot?: string; nodeEnv?: string | undefined } = {}) {
|
|
const envPaths = resolveWorkspaceEnvPaths(options);
|
|
const originalKeys = new Set(Object.keys(process.env));
|
|
const loadedPaths: string[] = [];
|
|
|
|
for (const envPath of envPaths) {
|
|
if (!existsSync(envPath)) {
|
|
continue;
|
|
}
|
|
|
|
const contents = readFileSync(envPath, "utf8");
|
|
|
|
for (const rawLine of contents.split(/\r?\n/u)) {
|
|
const line = rawLine.trim();
|
|
if (!line || line.startsWith("#")) {
|
|
continue;
|
|
}
|
|
|
|
const separatorIndex = line.indexOf("=");
|
|
if (separatorIndex <= 0) {
|
|
continue;
|
|
}
|
|
|
|
const key = line.slice(0, separatorIndex).trim();
|
|
const rawValue = line.slice(separatorIndex + 1).trim();
|
|
const quoted =
|
|
(rawValue.startsWith("\"") && rawValue.endsWith("\"")) ||
|
|
(rawValue.startsWith("'") && rawValue.endsWith("'"));
|
|
const value = quoted ? rawValue.slice(1, -1) : rawValue;
|
|
|
|
if (!originalKeys.has(key)) {
|
|
process.env[key] = value;
|
|
}
|
|
}
|
|
|
|
loadedPaths.push(envPath);
|
|
}
|
|
|
|
return loadedPaths;
|
|
}
|
|
|
|
export function resolveWorkspacePath(...segments: string[]) {
|
|
return resolve(resolveWorkspaceRoot(), ...segments);
|
|
}
|