mac
95 lines
2.5 KiB
JavaScript
95 lines
2.5 KiB
JavaScript
// codex_data_dir.js —— 会话相关功能使用的 Codex 数据目录，可由用户覆盖默认 ~/.codex。
|
||
const { app } = require("electron");
|
||
const fs = require("fs");
|
||
const os = require("os");
|
||
const path = require("path");
|
||
|
||
const DEFAULT_DIR = path.join(os.homedir(), ".codex");
|
||
let loaded = false;
|
||
let configuredDir = null;
|
||
|
||
function configFile() {
|
||
return path.join(app.getPath("userData"), "codex-data-dir.json");
|
||
}
|
||
|
||
function normalize(dir) {
|
||
if (typeof dir !== "string" || !dir.trim() || !path.isAbsolute(dir.trim())) return null;
|
||
return path.resolve(dir.trim());
|
||
}
|
||
|
||
function load() {
|
||
if (loaded) return;
|
||
loaded = true;
|
||
try {
|
||
configuredDir = normalize(JSON.parse(fs.readFileSync(configFile(), "utf-8")).path);
|
||
} catch {}
|
||
}
|
||
|
||
function get() {
|
||
load();
|
||
return configuredDir || DEFAULT_DIR;
|
||
}
|
||
|
||
function isCodexDataDir(dir) {
|
||
try {
|
||
const names = fs.readdirSync(dir);
|
||
return names.includes("sessions") || names.includes("archived_sessions") || names.includes("config.toml") ||
|
||
names.includes("auth.json") || names.some((name) => /^state_.*\.sqlite(?:-(?:wal|shm))?$/.test(name));
|
||
} catch {
|
||
return false;
|
||
}
|
||
}
|
||
|
||
function isDirectory(dir) {
|
||
try {
|
||
return fs.statSync(dir).isDirectory();
|
||
} catch {
|
||
return false;
|
||
}
|
||
}
|
||
|
||
function writeAtomic(file, text) {
|
||
fs.mkdirSync(path.dirname(file), { recursive: true });
|
||
const tmp = file + ".tmp-" + process.pid + "-" + Date.now();
|
||
const fd = fs.openSync(tmp, "w", 0o600);
|
||
try {
|
||
fs.writeFileSync(fd, text, "utf-8");
|
||
fs.fsyncSync(fd);
|
||
} finally {
|
||
fs.closeSync(fd);
|
||
}
|
||
try {
|
||
fs.renameSync(tmp, file);
|
||
} finally {
|
||
try { fs.rmSync(tmp, { force: true }); } catch {}
|
||
}
|
||
}
|
||
|
||
function set(dir) {
|
||
const next = normalize(dir);
|
||
if (!next || !isDirectory(next)) {
|
||
throw new Error("请选择存在的 Codex 数据目录");
|
||
}
|
||
if (!isCodexDataDir(next)) {
|
||
throw new Error("所选目录未找到 Codex 数据（sessions、config.toml 或 state_*.sqlite）");
|
||
}
|
||
configuredDir = next === DEFAULT_DIR ? null : next;
|
||
loaded = true;
|
||
writeAtomic(configFile(), JSON.stringify({ path: configuredDir || "", updatedAt: new Date().toISOString() }, null, 2));
|
||
return info();
|
||
}
|
||
|
||
function reset() {
|
||
configuredDir = null;
|
||
loaded = true;
|
||
writeAtomic(configFile(), JSON.stringify({ path: "", updatedAt: new Date().toISOString() }, null, 2));
|
||
return info();
|
||
}
|
||
|
||
function info() {
|
||
const dir = get();
|
||
return { path: dir, isDefault: dir === DEFAULT_DIR, exists: isDirectory(dir) };
|
||
}
|
||
|
||
module.exports = { get, set, reset, info, isCodexDataDir };
|