@kkmila/cpc 1.0.1 → 1.0.5

package/src/lib/env.mjs

@@ -0,0 +1,115 @@
+import { homedir } from "node:os";
+import {
+	existsSync,
+	readFileSync,
+	writeFileSync,
+	appendFileSync,
+} from "node:fs";
+import { join } from "node:path";
+import { execSync } from "node:child_process";
+
+/**
+ * 跨平台环境变量持久化模块。
+ *
+ * codex 的 model_providers 只能用 `env_key` 引用环境变量，
+ * 不能内联 key。cpc 安装到 codex 时，需要把 provider.apiKey
+ * 落地到一个环境变量里，并保证新开的终端/codex 进程能读到。
+ *
+ * 平台策略：
+ * - Linux / macOS: 按 $SHELL 检测写 ~/.zshrc 或 ~/.bashrc（幂等）
+ * - Windows: 用 setx 写入用户环境变量（注册表 HKCU\Environment）
+ */
+
+/**
+ * 推断当前 shell 的 rc 文件路径。
+ * @returns {string|null} rc 文件绝对路径；Windows 返回 null
+ */
+function detectShellRc() {
+	const platform = process.platform;
+	if (platform === "win32") return null;
+
+	const shell = process.env.SHELL || "";
+	const home = homedir();
+	if (shell.includes("zsh")) return join(home, ".zshrc");
+	if (shell.includes("bash")) return join(home, ".bashrc");
+
+	// $SHELL 缺失时按平台兜底：macOS 默认 zsh，其余默认 bash
+	if (platform === "darwin") return join(home, ".zshrc");
+	return join(home, ".bashrc");
+}
+
+/**
+ * 幂等地把 `export KEY="value"` 写入 rc 文件。
+ * 已存在该 export 行则替换值，不存在则追加。
+ */
+function persistToRc(rcPath, key, value) {
+	const exportLine = `export ${key}="${value}"`;
+	let content = "";
+	if (existsSync(rcPath)) {
+		content = readFileSync(rcPath, "utf8");
+	}
+	// 精确匹配 `export KEY=...` 整行
+	const regex = new RegExp(`^export ${key}=.*$`, "m");
+	if (regex.test(content)) {
+		const updated = content.replace(regex, exportLine);
+		if (updated !== content) {
+			writeFileSync(rcPath, updated);
+		}
+		return false; // 不是新增
+	}
+	const sep = content && !content.endsWith("\n") ? "\n" : "";
+	const block = `${sep}# Added by cpc (codex provider)\n${exportLine}\n`;
+	appendFileSync(rcPath, block);
+	return true; // 新增
+}
+
+/**
+ * Windows: 用 setx 持久化用户环境变量。
+ * 注意 setx 设置的变量只在新进程生效，当前进程不会自动有。
+ */
+function persistWithSetx(key, value) {
+	// setx 对带空格的值需要引号；API key 一般是 token 串，安全起见仍加引号
+	execSync(`setx ${key} "${value}"`, {
+		shell: "cmd.exe",
+		stdio: "ignore",
+	});
+}
+
+/**
+ * 跨平台持久化环境变量。
+ *
+ * @param {string} key 环境变量名
+ * @param {string} value 值
+ * @returns {{method: "rc"|"setx"|"skip", target: string, created: boolean}}
+ */
+export function persistEnvVar(key, value) {
+	if (!key || value == null) {
+		return { method: "skip", target: "", created: false };
+	}
+
+	const rc = detectShellRc();
+	if (rc) {
+		const created = persistToRc(rc, key, value);
+		return { method: "rc", target: rc, created };
+	}
+
+	// Windows
+	try {
+		persistWithSetx(key, value);
+		return { method: "setx", target: "HKCU\\Environment", created: true };
+	} catch (err) {
+		throw new Error(`Failed to set env var on Windows: ${err.message}`);
+	}
+}
+
+/**
+ * 把 provider 名转成规范环境变量名：CPC_<NAME_UPPER>_API_KEY
+ * 非字母数字字符 -> _
+ */
+export function envKeyNameForProvider(providerName) {
+	const safe = String(providerName)
+		.toUpperCase()
+		.replace(/[^A-Z0-9]+/g, "_")
+		.replace(/^_+|_+$/g, "");
+	return `CPC_${safe}_API_KEY`;
+}