94 lines
3.1 KiB
TypeScript
94 lines
3.1 KiB
TypeScript
import { execFileSync } from "node:child_process";
|
|
import fs from "node:fs";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
|
|
type RestoreEntry = { key: string; value: string | undefined };
|
|
|
|
function restoreEnv(entries: RestoreEntry[]): void {
|
|
for (const { key, value } of entries) {
|
|
if (value === undefined) delete process.env[key];
|
|
else process.env[key] = value;
|
|
}
|
|
}
|
|
|
|
function loadProfileEnv(): void {
|
|
const profilePath = path.join(os.homedir(), ".profile");
|
|
if (!fs.existsSync(profilePath)) return;
|
|
try {
|
|
const output = execFileSync(
|
|
"/bin/bash",
|
|
["-lc", `set -a; source "${profilePath}" >/dev/null 2>&1; env -0`],
|
|
{ encoding: "utf8" },
|
|
);
|
|
const entries = output.split("\0");
|
|
let applied = 0;
|
|
for (const entry of entries) {
|
|
if (!entry) continue;
|
|
const idx = entry.indexOf("=");
|
|
if (idx <= 0) continue;
|
|
const key = entry.slice(0, idx);
|
|
if (!key || (process.env[key] ?? "") !== "") continue;
|
|
process.env[key] = entry.slice(idx + 1);
|
|
applied += 1;
|
|
}
|
|
if (applied > 0) {
|
|
console.log(`[live] loaded ${applied} env vars from ~/.profile`);
|
|
}
|
|
} catch {
|
|
// ignore profile load failures
|
|
}
|
|
}
|
|
|
|
export function installTestEnv(): { cleanup: () => void; tempHome: string } {
|
|
const live =
|
|
process.env.LIVE === "1" ||
|
|
process.env.CLAWDBOT_LIVE_TEST === "1" ||
|
|
process.env.CLAWDBOT_LIVE_GATEWAY === "1";
|
|
|
|
// Live tests must use the real user environment (keys, profiles, config).
|
|
// The default test env isolates HOME to avoid touching real state.
|
|
if (live) {
|
|
loadProfileEnv();
|
|
return { cleanup: () => {}, tempHome: process.env.HOME ?? "" };
|
|
}
|
|
|
|
const restore: RestoreEntry[] = [
|
|
{ key: "HOME", value: process.env.HOME },
|
|
{ key: "USERPROFILE", value: process.env.USERPROFILE },
|
|
{ key: "XDG_CONFIG_HOME", value: process.env.XDG_CONFIG_HOME },
|
|
{ key: "XDG_DATA_HOME", value: process.env.XDG_DATA_HOME },
|
|
{ key: "XDG_STATE_HOME", value: process.env.XDG_STATE_HOME },
|
|
{ key: "XDG_CACHE_HOME", value: process.env.XDG_CACHE_HOME },
|
|
{ key: "CLAWDBOT_STATE_DIR", value: process.env.CLAWDBOT_STATE_DIR },
|
|
{ key: "CLAWDBOT_TEST_HOME", value: process.env.CLAWDBOT_TEST_HOME },
|
|
];
|
|
|
|
const tempHome = fs.mkdtempSync(path.join(os.tmpdir(), "clawdbot-test-home-"));
|
|
|
|
process.env.HOME = tempHome;
|
|
process.env.USERPROFILE = tempHome;
|
|
process.env.CLAWDBOT_TEST_HOME = tempHome;
|
|
|
|
// Windows: prefer the legacy default state dir so auth/profile tests match real paths.
|
|
if (process.platform === "win32") {
|
|
process.env.CLAWDBOT_STATE_DIR = path.join(tempHome, ".clawdbot");
|
|
}
|
|
|
|
process.env.XDG_CONFIG_HOME = path.join(tempHome, ".config");
|
|
process.env.XDG_DATA_HOME = path.join(tempHome, ".local", "share");
|
|
process.env.XDG_STATE_HOME = path.join(tempHome, ".local", "state");
|
|
process.env.XDG_CACHE_HOME = path.join(tempHome, ".cache");
|
|
|
|
const cleanup = () => {
|
|
restoreEnv(restore);
|
|
try {
|
|
fs.rmSync(tempHome, { recursive: true, force: true });
|
|
} catch {
|
|
// ignore cleanup errors
|
|
}
|
|
};
|
|
|
|
return { cleanup, tempHome };
|
|
}
|