base_parts_ai 1.0.58 → 1.0.59

package/README.md

@@ -38,7 +38,7 @@ ucx
 jgrok
 ```
 
-`ucx` 会扫描用户主目录下一层的 `~/.codex` 与 `~/.codex_*`。账号名由 `config.toml` 顶层 `model_provider`、`auth.json` 中非空 `OPENAI_API_KEY` 的脱敏短名、JWT email 组合成 `配置名_Key_Email`，缺段去掉；不会自动创建空的 `~/.codex`。选中当前 `~/.codex` 会进入二级菜单，只在该目录内用 `juser_*` 切换 `config.toml`、`auth.json`、`.env`、`models.json`，不改其它 `~/.codex_*` 账号目录。
+`ucx` 会扫描用户主目录下一层的 `~/.codex` 与 `~/.codex_*`。账号名由 `config.toml` 顶层 `model_provider`、`auth.json` 中非空 `OPENAI_API_KEY` 的脱敏短名、JWT email 组合成 `配置名_Key_Email`，缺段去掉；不会自动创建空的 `~/.codex`。选中当前 `~/.codex` 会进入二级菜单，只在该目录内用 `juser_*` 切换 `config.toml`、`auth.json`、`.env`、`models.json`，不改其它 `~/.codex_*` 账号目录。二级选中子账户后（含选中当前无需切换）会进入下一级选择：还原最后一次会话并进入 codex（读 `~/.codex/session_index.jsonl` 取最后一次会话 id 后执行 `codex fork <会话id> --yolo`）、直接进入 codex（`codex --yolo`）、退出 ucx；启动 codex 时会等待其执行完成后一起退出并透传其退出码，`~/.codex/.env` 存在时会注入子进程环境。
 
 `jgrok` 若 `~/.grok/.env` 不存在则写入代理示例并打印绝对路径，请编辑后再运行。已存在则把该文件注入环境变量：本机没有 `grok` 时通过腾讯云 npm 源安装 `@xai-official/grok`（安装不注入环境变量），然后启动 `grok`，用户参数原样透传。
 

package/bin/ucx.js

@@ -29,45 +29,51 @@ function loadLib(libName) {
 
 /**
  * 拉取远程配置（含自动升级）后执行账号切换
- * @returns {Promise<void>}
+ * @returns {Promise<number|undefined>} 子账户三级菜单启动过 codex 时返回其退出码
  */
 async function runSwitch() {
     var buildCfg = getBuildCfg();
     var utils = loadLib('ai_utils');
     await utils.fetchConfig(buildCfg);
     await utils.showNotice();
-    await loadLib('ucx')({}, buildCfg);
+    return loadLib('ucx')({}, buildCfg);
 }
 
 /**
  * 运行 ucx CLI，可供命令行直接执行或被 base_parts_ai require 调用
  * @param {Array<string>} args 用户参数，不包含 node 和脚本路径
- * @returns {Promise<void>}
+ * @returns {Promise<number|undefined>} 透传 codex 退出码
  */
 async function runCli(args) {
     var cliArgs = (args || []).slice();
     var program = new Command();
+    var exitCode;
 
     program
         .name('ucx')
         .description('通过目录改名切换 Codex 账号，选中当前目录可切换其子账户')
         .version(packageJson.version)
         .action(async function () {
-            await runSwitch();
+            exitCode = await runSwitch();
         });
 
     // 无参数时直接进入切换，避免 commander 空参数不触发根 action
     if (cliArgs.length <= 0) {
-        await runSwitch();
-        return;
+        return runSwitch();
     }
 
     // 使用 from:user 避免读取或修改全局 process.argv，便于被主入口 require 调用
     await program.parseAsync(cliArgs, { from: 'user' });
+    return exitCode;
 }
 
 if (require.main === module) {
-    runCli(process.argv.slice(2)).catch(function (err) {
+    runCli(process.argv.slice(2)).then(function (code) {
+        // 启动过 codex 时与其共用退出码，便于脚本感知 codex 结果
+        if (typeof code === 'number') {
+            process.exit(code);
+        }
+    }).catch(function (err) {
         console.error('ucx 执行失败:', err.message);
         process.exit(1);
     });

package/lib/ucx.js

@@ -27,6 +27,15 @@ var REMOVE_CURRENT_SUB = '__remove_current_sub__';
 /** 二级菜单：返回一级账号列表 */
 var BACK_TO_PARENT = '__back_to_parent__';
 
+/** 三级菜单：还原最后一次会话并进入 codex */
+var POST_SWITCH_RESUME = '__post_switch_resume__';
+
+/** 三级菜单：直接进入 codex */
+var POST_SWITCH_DIRECT = '__post_switch_direct__';
+
+/** 三级菜单：退出 ucx，不执行任何命令 */
+var POST_SWITCH_EXIT = '__post_switch_exit__';
+
 /** 子账户目录前缀，只出现在当前 ~/.codex 内 */
 var SUB_DIR_PREFIX = 'juser_';
 
@@ -407,6 +416,102 @@ function killCodexProcesses(execImpl) {
     });
 }
 
+/**
+ * 解析 .env 文本为键值对象：跳过注释与空行，去掉成对引号
+ * @param {string} content .env 文件内容
+ * @returns {object}
+ */
+function parseCodexEnvContent(content) {
+    var result = {};
+    String(content || '').split(/\r?\n/).forEach(function (line) {
+        var trimmed = line.trim();
+        if (!trimmed || trimmed.indexOf('#') === 0) {
+            return;
+        }
+        var eqIndex = trimmed.indexOf('=');
+        if (eqIndex <= 0) {
+            return;
+        }
+        var key = trimmed.slice(0, eqIndex).trim();
+        var value = trimmed.slice(eqIndex + 1).trim();
+        if (value.length >= 2 && ((value[0] === '"' && value[value.length - 1] === '"') || (value[0] === '\'' && value[value.length - 1] === '\''))) {
+            value = value.slice(1, -1);
+        }
+        result[key] = value;
+    });
+    return result;
+}
+
+/**
+ * 构建 codex 子进程环境：在当前进程环境之上合并 ~/.codex/.env，确保子账户切换后代理与 Key 立即生效
+ * @param {string} homeDir 用户主目录
+ * @returns {object}
+ */
+function buildCodexChildEnv(homeDir) {
+    var envPath = path.join(homeDir, '.codex', '.env');
+    var childEnv = Object.assign({}, process.env);
+    if (!fs.existsSync(envPath)) {
+        return childEnv;
+    }
+    try {
+        var parsed = parseCodexEnvContent(fs.readFileSync(envPath, 'utf8'));
+        return Object.assign(childEnv, parsed);
+    } catch (e) {
+        // .env 读取失败时退回当前进程环境，不阻断启动
+        return childEnv;
+    }
+}
+
+/**
+ * 检测本机是否存在 codex 命令；测试环境或注入 exec 时避免触碰宿主机
+ * @param {Function} [execImpl] 可注入的 execSync，便于单测
+ * @returns {boolean}
+ */
+function codexCommandExists(execImpl) {
+    var exec = execImpl || childProcess.execSync;
+    var cmd = process.platform === 'win32' ? 'where codex' : 'which codex';
+    try {
+        exec(cmd, { stdio: 'ignore' });
+        return true;
+    } catch (e) {
+        return false;
+    }
+}
+
+/**
+ * 读取 ~/.codex/session_index.jsonl 的最后一次会话记录：倒序跳过坏行，取第一条含非空 id 的记录
+ * @param {string} homeDir 用户主目录
+ * @returns {{ id: string, threadName: string }|null} 无有效记录时返回 null
+ */
+function readLastSessionRecord(homeDir) {
+    var indexPath = path.join(homeDir, '.codex', 'session_index.jsonl');
+    var content = '';
+    try {
+        content = fs.readFileSync(indexPath, 'utf8');
+    } catch (e) {
+        return null;
+    }
+    var lines = content.split(/\r?\n/);
+    for (var i = lines.length - 1; i >= 0; i -= 1) {
+        var line = lines[i].trim();
+        if (!line) {
+            continue;
+        }
+        try {
+            var record = JSON.parse(line);
+            if (record && typeof record.id === 'string' && record.id.trim()) {
+                return {
+                    id: record.id.trim(),
+                    threadName: typeof record.thread_name === 'string' ? record.thread_name : '',
+                };
+            }
+        } catch (e) {
+            // 半行写入等坏行跳过，继续向前找
+        }
+    }
+    return null;
+}
+
 /**
  * 为当前账号分配不冲突的停放目录名；同名已存在时追加 _2、_3，不覆盖
  * @param {string} homeDir 用户主目录
@@ -919,11 +1024,92 @@ async function removeCurrentSubAccount(codexDir, currentRecord, options) {
     return { cancelled: false };
 }
 
+/**
+ * 三级菜单选项：还原最后一次会话 / 直接进入 / 退出，固定顺序
+ * @returns {Array<object>}
+ */
+function buildPostSwitchChoices() {
+    return [
+        { name: '还原最后一次会话并进入 codex', value: POST_SWITCH_RESUME },
+        { name: '直接进入 codex', value: POST_SWITCH_DIRECT },
+        { name: '退出 ucx', value: POST_SWITCH_EXIT },
+    ];
+}
+
+/**
+ * 启动 codex 并等待其执行完成，返回子进程退出码；codex 缺失时提示并返回 null
+ * @param {Array<string>} args 传给 codex 的参数
+ * @param {string} homeDir 用户主目录
+ * @param {object} [options] 可选配置，spawnImpl/execImpl 用于单测注入
+ * @returns {number|null}
+ */
+function launchCodex(args, homeDir, options) {
+    options = options || {};
+    // 测试环境且未注入实现时跳过，避免真实启动 codex
+    if (process.env.JCC_TEST_HOME && !options.spawnImpl) {
+        console.log('测试环境跳过启动 codex。');
+        return null;
+    }
+    if (!codexCommandExists(options.execImpl)) {
+        console.error('未检测到 codex 命令，请先安装 Codex CLI 后重试。');
+        return null;
+    }
+    var spawn = options.spawnImpl || childProcess.spawnSync;
+    var result = spawn('codex', args || [], {
+        stdio: 'inherit',
+        shell: process.platform === 'win32',
+        env: buildCodexChildEnv(homeDir),
+    }) || {};
+    if (result.error) {
+        throw result.error;
+    }
+    if (result.signal) {
+        return 1;
+    }
+    return typeof result.status === 'number' ? result.status : 0;
+}
+
+/**
+ * 子账号切换后的三级菜单：还原最后一次会话 / 直接进入 codex / 退出 ucx
+ * 还原时读 ~/.codex/session_index.jsonl 取最后一次会话 id 后执行 codex fork <id> --yolo
+ * @param {string} homeDir 用户主目录
+ * @param {object} [options] 可选配置，spawnImpl/execImpl 用于单测注入
+ * @returns {Promise<number|undefined>} codex 退出码；未启动 codex 时为 undefined
+ */
+async function runPostSwitchMenu(homeDir, options) {
+    options = options || {};
+    for (;;) {
+        var answer = await inquirer.prompt([{
+            type: 'list',
+            name: 'action',
+            message: '请选择接下来的操作',
+            choices: buildPostSwitchChoices(),
+        }]);
+
+        if (answer.action === POST_SWITCH_EXIT) {
+            return undefined;
+        }
+        if (answer.action === POST_SWITCH_DIRECT) {
+            return launchCodex(['--yolo'], homeDir, options);
+        }
+
+        // 还原最后一次会话：无会话记录时提示并重新选择，用户可改选直接进入
+        var session = readLastSessionRecord(homeDir);
+        if (!session) {
+            console.log('未在 ~/.codex/session_index.jsonl 中找到会话记录，无法还原。');
+            continue;
+        }
+        var display = session.threadName || session.id;
+        console.log('正在还原最后一次会话：' + display);
+        return launchCodex(['fork', session.id, '--yolo'], homeDir, options);
+    }
+}
+
 /**
  * 当前空间的子账户二级菜单；所有改动限制在 codexDir 内
  * @param {string} codexDir 当前 ~/.codex
  * @param {object} [options] 可选配置
- * @returns {Promise<{ back: boolean }>}
+ * @returns {Promise<{ back: boolean, exitCode?: number }>}
  */
 async function runSubAccountMenu(codexDir, options) {
     options = options || {};
@@ -952,7 +1138,9 @@ async function runSubAccountMenu(codexDir, options) {
 
     var selected = findRecordByDirPath(records, answer.dirPath);
     await switchSubAccount(codexDir, selected, records, options);
-    return { back: false };
+    // 切换完成（含“已是当前子账户”的跳过场景）后进入三级菜单，可选择还原会话/直接进入 codex/退出
+    var exitCode = await runPostSwitchMenu(path.dirname(codexDir), options);
+    return { back: false, exitCode: exitCode };
 }
 
 /**
@@ -976,11 +1164,13 @@ function findRecordByDirPath(records, dirPath) {
  * 选中当前 ~/.codex 进入子账户二级菜单；返回上级时重新弹出一级列表
  * @param {object} cmd commander 选项（当前无业务字段）
  * @param {object} buildCfg 含 homeDir 的配置
- * @returns {Promise<void>}
+ * @param {object} [options] 可选配置，spawnImpl/execImpl 用于单测注入
+ * @returns {Promise<number|undefined>} 启动过 codex 时返回其退出码
  */
-async function runUcx(cmd, buildCfg) {
+async function runUcx(cmd, buildCfg, options) {
+    options = options || {};
     var homeDir = getHomeDir(buildCfg);
-    while (true) {
+    for (;;) {
         var records = buildAccountRecords(homeDir);
         var current = findRecordByDirPath(records, path.join(homeDir, '.codex'));
         var answer = await inquirer.prompt([{
@@ -992,19 +1182,19 @@ async function runUcx(cmd, buildCfg) {
         }]);
 
         if (answer.dirPath === REMOVE_CURRENT_ACCOUNT) {
-            await removeCurrentAccount(homeDir, current || null, {});
+            await removeCurrentAccount(homeDir, current || null, options);
             return;
         }
 
         var selected = findRecordByDirPath(records, answer.dirPath);
         if (selected && selected.isCurrent) {
-            var subResult = await runSubAccountMenu(selected.dirPath, {});
+            var subResult = await runSubAccountMenu(selected.dirPath, options);
             if (subResult && subResult.back) {
                 continue;
             }
-            return;
+            return subResult ? subResult.exitCode : undefined;
         }
-        await switchCodexAccount(homeDir, selected, records, {});
+        await switchCodexAccount(homeDir, selected, records, options);
         return;
     }
 }
@@ -1014,6 +1204,9 @@ runUcx._private = {
     REMOVE_CURRENT_ACCOUNT: REMOVE_CURRENT_ACCOUNT,
     REMOVE_CURRENT_SUB: REMOVE_CURRENT_SUB,
     BACK_TO_PARENT: BACK_TO_PARENT,
+    POST_SWITCH_RESUME: POST_SWITCH_RESUME,
+    POST_SWITCH_DIRECT: POST_SWITCH_DIRECT,
+    POST_SWITCH_EXIT: POST_SWITCH_EXIT,
     SUB_DIR_PREFIX: SUB_DIR_PREFIX,
     SUB_FILE_NAMES: SUB_FILE_NAMES,
     CODEX_ENV_DEMO: CODEX_ENV_DEMO,
@@ -1050,6 +1243,13 @@ runUcx._private = {
     switchSubAccount: switchSubAccount,
     removeCurrentSubAccount: removeCurrentSubAccount,
     runSubAccountMenu: runSubAccountMenu,
+    parseCodexEnvContent: parseCodexEnvContent,
+    buildCodexChildEnv: buildCodexChildEnv,
+    codexCommandExists: codexCommandExists,
+    readLastSessionRecord: readLastSessionRecord,
+    buildPostSwitchChoices: buildPostSwitchChoices,
+    launchCodex: launchCodex,
+    runPostSwitchMenu: runPostSwitchMenu,
 };
 
 module.exports = runUcx;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "base_parts_ai",
-  "version": "1.0.58",
+  "version": "1.0.59",
   "description": "jaskle base_parts_ai",
   "main": "./main.js",
   "registry": true,