195 lines
6.9 KiB
JavaScript
195 lines
6.9 KiB
JavaScript
|
// provider_sync.js —— 一键把历史会话同步到「当前账号」的 provider，使其在 Codex 侧栏重新显示。
|
|||
|
|
// 复刻 CodexPlusPlus 的 provider-sync：改 rollout 首行 provider + state DB(provider/has_user_event/cwd)。
|
|||
|
|
// 必须在 Codex 已退出时运行。数据库用纯 JS 的 sql.js（免原生编译，跨平台）。
|
|||
|
|
const fs = require("fs");
|
|||
|
|
const path = require("path");
|
|||
|
|
const os = require("os");
|
|||
|
|
const { execFileSync } = require("child_process");
|
|||
|
|
const initSqlJs = require("sql.js");
|
|||
|
|
|
|||
|
|
const CODEX_HOME = path.join(os.homedir(), ".codex");
|
|||
|
|
|
|||
|
|
// 检测 Codex 是否在运行（尽力而为，跨平台）
|
|||
|
|
function codexRunning() {
|
|||
|
|
try {
|
|||
|
|
if (process.platform === "darwin") {
|
|||
|
|
return execFileSync("pgrep", ["-f", "Codex.app/Contents/MacOS/Codex"], { encoding: "utf-8" }).trim().length > 0;
|
|||
|
|
}
|
|||
|
|
if (process.platform === "win32") {
|
|||
|
|
return /Codex\.exe/i.test(execFileSync("tasklist", [], { encoding: "utf-8" }));
|
|||
|
|
}
|
|||
|
|
} catch {
|
|||
|
|
/* pgrep 无匹配时退出码非 0，视为未运行 */
|
|||
|
|
}
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 目标 provider = config.toml 根级 model_provider，缺省 openai
|
|||
|
|
function targetProvider() {
|
|||
|
|
try {
|
|||
|
|
const head = fs.readFileSync(path.join(CODEX_HOME, "config.toml"), "utf-8").split("\n[")[0];
|
|||
|
|
const m = head.match(/^\s*model_provider\s*=\s*["']([^"']+)/m);
|
|||
|
|
if (m) return m[1];
|
|||
|
|
} catch {}
|
|||
|
|
return "openai";
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function findRollouts() {
|
|||
|
|
const out = [];
|
|||
|
|
for (const root of [path.join(CODEX_HOME, "sessions"), path.join(CODEX_HOME, "archived_sessions")]) {
|
|||
|
|
const walk = (dir) => {
|
|||
|
|
let ents;
|
|||
|
|
try {
|
|||
|
|
ents = fs.readdirSync(dir, { withFileTypes: true });
|
|||
|
|
} catch {
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
for (const e of ents) {
|
|||
|
|
const full = path.join(dir, e.name);
|
|||
|
|
if (e.isDirectory()) walk(full);
|
|||
|
|
else if (e.isFile() && e.name.startsWith("rollout-") && e.name.endsWith(".jsonl")) out.push(full);
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
walk(root);
|
|||
|
|
}
|
|||
|
|
return out;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
function ts() {
|
|||
|
|
const d = new Date();
|
|||
|
|
const p = (n) => String(n).padStart(2, "0");
|
|||
|
|
return `${d.getFullYear()}${p(d.getMonth() + 1)}${p(d.getDate())}-${p(d.getHours())}${p(d.getMinutes())}${p(d.getSeconds())}`;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async function syncProviderToCurrent(log = () => {}, onProgress = () => {}) {
|
|||
|
|
if (codexRunning()) throw new Error("检测到 Codex 正在运行，请先完全退出 Codex 再同步。");
|
|||
|
|
const target = targetProvider();
|
|||
|
|
log(`目标 provider：${target}`);
|
|||
|
|
onProgress({ phase: "备份中", done: 0, total: 0 });
|
|||
|
|
|
|||
|
|
// 1) 备份
|
|||
|
|
const backup = path.join(CODEX_HOME, "backups_state", "provider-sync", ts());
|
|||
|
|
fs.mkdirSync(backup, { recursive: true });
|
|||
|
|
for (const name of ["config.toml", ".codex-global-state.json", ".codex-global-state.json.bak"]) {
|
|||
|
|
const src = path.join(CODEX_HOME, name);
|
|||
|
|
if (fs.existsSync(src)) fs.copyFileSync(src, path.join(backup, name));
|
|||
|
|
}
|
|||
|
|
for (const db of fs.readdirSync(CODEX_HOME).filter((f) => /^state_.*\.sqlite/.test(f))) {
|
|||
|
|
fs.copyFileSync(path.join(CODEX_HOME, db), path.join(backup, db));
|
|||
|
|
}
|
|||
|
|
log(`已备份到 ${backup}`);
|
|||
|
|
|
|||
|
|
// 2) 遍历 rollout：改首行 provider，收集 thread_id / cwd / has_user_event
|
|||
|
|
const userEventIds = new Set();
|
|||
|
|
const cwdById = {};
|
|||
|
|
let changedFiles = 0;
|
|||
|
|
const rolloutFiles = findRollouts();
|
|||
|
|
const rtotal = rolloutFiles.length;
|
|||
|
|
let ri = 0;
|
|||
|
|
for (const file of rolloutFiles) {
|
|||
|
|
ri++;
|
|||
|
|
if (ri % 5 === 0 || ri === rtotal) onProgress({ phase: "改写会话文件", done: ri, total: rtotal });
|
|||
|
|
let lines;
|
|||
|
|
try {
|
|||
|
|
lines = fs.readFileSync(file, "utf-8").split(/(?<=\n)/); // 保留换行
|
|||
|
|
} catch {
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
if (!lines.length) continue;
|
|||
|
|
const blob = lines.join("");
|
|||
|
|
const hasUE = blob.includes('"user_message"') || blob.includes('"user_input"');
|
|||
|
|
let tid = null;
|
|||
|
|
let rewrote = false;
|
|||
|
|
for (let i = 0; i < lines.length; i++) {
|
|||
|
|
const line = lines[i];
|
|||
|
|
if (!line.trim()) continue;
|
|||
|
|
let rec;
|
|||
|
|
try {
|
|||
|
|
rec = JSON.parse(line);
|
|||
|
|
} catch {
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
if (rec.type === "session_meta" && rec.payload && typeof rec.payload === "object") {
|
|||
|
|
tid = rec.payload.id || rec.payload.session_id || null;
|
|||
|
|
const cwd = rec.payload.cwd;
|
|||
|
|
if (tid && cwd) cwdById[tid] = cwd;
|
|||
|
|
if (rec.payload.model_provider !== target) {
|
|||
|
|
rec.payload.model_provider = target;
|
|||
|
|
lines[i] = JSON.stringify(rec) + "\n";
|
|||
|
|
rewrote = true;
|
|||
|
|
}
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if (rewrote) {
|
|||
|
|
fs.writeFileSync(file, lines.join(""), "utf-8");
|
|||
|
|
changedFiles++;
|
|||
|
|
}
|
|||
|
|
if (tid && hasUE) userEventIds.add(tid);
|
|||
|
|
}
|
|||
|
|
log(`改写 rollout 文件：${changedFiles} 个；含用户消息的会话：${userEventIds.size} 条`);
|
|||
|
|
|
|||
|
|
// 3) 更新所有 state_*.sqlite
|
|||
|
|
onProgress({ phase: "更新数据库", done: 0, total: 0 });
|
|||
|
|
const SQL = await initSqlJs({
|
|||
|
|
// wasm 恒在【包内】node_modules（热更包不含 node_modules），故用 BUNDLE_ROOT 而非 __dirname
|
|||
|
|
locateFile: (f) =>
|
|||
|
|
path.join(global.__BUNDLE_ROOT__ || path.join(__dirname, "..", ".."),
|
|||
|
|
"node_modules", "sql.js", "dist", f),
|
|||
|
|
});
|
|||
|
|
let provRows = 0;
|
|||
|
|
let ueRows = 0;
|
|||
|
|
let cwdRows = 0;
|
|||
|
|
for (const name of fs.readdirSync(CODEX_HOME).filter((f) => /^state_.*\.sqlite$/.test(f))) {
|
|||
|
|
const dbPath = path.join(CODEX_HOME, name);
|
|||
|
|
let db;
|
|||
|
|
try {
|
|||
|
|
db = new SQL.Database(fs.readFileSync(dbPath));
|
|||
|
|
const tbl = db.exec("SELECT name FROM sqlite_master WHERE type='table' AND name='threads'");
|
|||
|
|
if (!tbl.length) {
|
|||
|
|
db.close();
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
const cols = new Set(db.exec("PRAGMA table_info(threads)")[0].values.map((r) => r[1]));
|
|||
|
|
if (!cols.has("model_provider")) {
|
|||
|
|
db.close();
|
|||
|
|
continue;
|
|||
|
|
}
|
|||
|
|
db.run("UPDATE threads SET model_provider=? WHERE COALESCE(model_provider,'')<>?", [target, target]);
|
|||
|
|
provRows += db.getRowsModified();
|
|||
|
|
if (cols.has("has_user_event")) {
|
|||
|
|
for (const tid of userEventIds) {
|
|||
|
|
db.run("UPDATE threads SET has_user_event=1 WHERE id=? AND COALESCE(has_user_event,0)<>1", [tid]);
|
|||
|
|
ueRows += db.getRowsModified();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if (cols.has("cwd")) {
|
|||
|
|
for (const tid of Object.keys(cwdById)) {
|
|||
|
|
db.run("UPDATE threads SET cwd=? WHERE id=? AND COALESCE(cwd,'')<>?", [cwdById[tid], tid, cwdById[tid]]);
|
|||
|
|
cwdRows += db.getRowsModified();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
fs.writeFileSync(dbPath, Buffer.from(db.export()));
|
|||
|
|
db.close();
|
|||
|
|
} catch (e) {
|
|||
|
|
if (db) try { db.close(); } catch {}
|
|||
|
|
log(` 数据库跳过 ${name}：${e.message || e}`);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
log(`数据库更新：provider ${provRows} 行、has_user_event ${ueRows} 行、cwd ${cwdRows} 行`);
|
|||
|
|
onProgress({ phase: "完成", done: 0, total: 0 });
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
target,
|
|||
|
|
totalConversations: rtotal,
|
|||
|
|
changedFiles,
|
|||
|
|
userEvents: userEventIds.size,
|
|||
|
|
provRows,
|
|||
|
|
ueRows,
|
|||
|
|
cwdRows,
|
|||
|
|
backup,
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
module.exports = { syncProviderToCurrent, codexRunning, targetProvider };
|