oh-my-codex 0.8.13 → 0.8.15

package/dist/team/__tests__/runtime.test.js

@@ -5,6 +5,7 @@ import { mkdtemp, rm, writeFile, readFile, mkdir, chmod } from 'fs/promises';
 import { join } from 'path';
 import { tmpdir } from 'os';
 import { existsSync } from 'fs';
+import { HUD_TMUX_TEAM_HEIGHT_LINES } from '../../hud/constants.js';
 import { initTeamState, createTask, readTeamConfig, saveTeamConfig, listMailboxMessages, listDispatchRequests, updateWorkerHeartbeat, writeAtomic, readTask, readMonitorSnapshot, claimTask, transitionTaskStatus, } from '../state.js';
 import { monitorTeam, shutdownTeam, resumeTeam, startTeam, assignTask, sendWorkerMessage, resolveWorkerLaunchArgsFromEnv, waitForWorkerStartupEvidence, waitForClaudeStartupEvidence, } from '../runtime.js';
 import { resolveTeamLowComplexityDefaultModel } from '../model-contract.js';
@@ -732,6 +733,130 @@ process.on('SIGTERM', () => process.exit(0));
             await rm(cwd, { recursive: true, force: true });
         }
     });
+    it('startTeam relaunch re-creates HUD pane and re-registers reconcile hooks after shutdown', async () => {
+        const cwd = await mkdtemp(join(tmpdir(), 'omx-runtime-relaunch-hud-'));
+        const fakeBinDir = await mkdtemp(join(tmpdir(), 'omx-runtime-relaunch-hud-bin-'));
+        const tmuxLogPath = join(fakeBinDir, 'tmux.log');
+        const tmuxStubPath = join(fakeBinDir, 'tmux');
+        const geminiStubPath = join(fakeBinDir, 'gemini');
+        const previousPath = process.env.PATH;
+        const previousTmux = process.env.TMUX;
+        const previousTmuxPane = process.env.TMUX_PANE;
+        const previousLaunchMode = process.env.OMX_TEAM_WORKER_LAUNCH_MODE;
+        const previousWorkerCli = process.env.OMX_TEAM_WORKER_CLI;
+        let runtime = null;
+        try {
+            await writeFile(tmuxStubPath, `#!/bin/sh
+set -eu
+printf '%s\\n' "$*" >> "${tmuxLogPath}"
+case "\${1:-}" in
+  -V)
+    echo "tmux 3.4"
+    exit 0
+    ;;
+  display-message)
+    case "$*" in
+      *"#{window_width}"*)
+        echo "120"
+        ;;
+      *)
+        echo "leader:0 %1"
+        ;;
+    esac
+    exit 0
+    ;;
+  list-panes)
+    case "$*" in
+      *"pane_current_command"* )
+        printf "%%1\\tnode\\t'codex'\\n"
+        ;;
+      *"#{pane_dead} #{pane_pid}"*)
+        echo "1 999999"
+        ;;
+      *"#{pane_pid}"*)
+        echo "999999"
+        ;;
+      *)
+        exit 0
+        ;;
+    esac
+    exit 0
+    ;;
+  split-window)
+    case "$*" in
+      *" -h "*)
+        echo "%2"
+        ;;
+      *)
+        echo "%3"
+        ;;
+    esac
+    exit 0
+    ;;
+  set-hook|run-shell|select-layout|set-window-option|select-pane|send-keys|kill-pane|kill-session)
+    exit 0
+    ;;
+  *)
+    exit 0
+    ;;
+esac
+`);
+            await chmod(tmuxStubPath, 0o755);
+            await writeFile(geminiStubPath, `#!/bin/sh
+exit 0
+`);
+            await chmod(geminiStubPath, 0o755);
+            process.env.PATH = `${fakeBinDir}:${previousPath ?? ''}`;
+            process.env.TMUX = 'leader-session,stub,0';
+            process.env.TMUX_PANE = '%1';
+            process.env.OMX_TEAM_WORKER_LAUNCH_MODE = 'interactive';
+            process.env.OMX_TEAM_WORKER_CLI = 'gemini';
+            runtime = await withoutTeamWorkerEnv(() => startTeam('team-rerun-hud', 'rerun hud restore', 'explore', 1, [{ subject: 'restore hud', description: 'restore hud', owner: 'worker-1' }], cwd));
+            assert.equal(runtime.config.hud_pane_id, '%3');
+            assert.ok(runtime.config.resize_hook_name);
+            await shutdownTeam(runtime.teamName, cwd, { force: true });
+            runtime = null;
+            runtime = await withoutTeamWorkerEnv(() => startTeam('team-rerun-hud', 'rerun hud restore', 'explore', 1, [{ subject: 'restore hud again', description: 'restore hud again', owner: 'worker-1' }], cwd));
+            assert.equal(runtime.config.hud_pane_id, '%3');
+            assert.ok(runtime.config.resize_hook_name);
+            const tmuxLog = await readFile(tmuxLogPath, 'utf-8');
+            const hudSplitRe = new RegExp(`split-window -v -l ${HUD_TMUX_TEAM_HEIGHT_LINES} -t %1 -d -P -F #\\{pane_id\\}`, 'g');
+            assert.equal(tmuxLog.match(hudSplitRe)?.length ?? 0, 3);
+            assert.equal(tmuxLog.match(/set-hook -t leader:0 client-resized\[\d+\]/g)?.length ?? 0, 2);
+            assert.equal(tmuxLog.match(/set-hook -t leader:0 client-attached\[\d+\]/g)?.length ?? 0, 2);
+            assert.equal(tmuxLog.match(/run-shell -b sleep \d+; tmux resize-pane -t %3 -y \d+ >/g)?.length ?? 0, 3);
+            assert.equal(tmuxLog.match(/run-shell tmux resize-pane -t %3 -y \d+ >/g)?.length ?? 0, 3);
+            assert.ok((tmuxLog.match(/select-layout -t leader:0 main-vertical/g)?.length ?? 0) >= 2);
+            assert.match(tmuxLog, /kill-pane -t %3/);
+        }
+        finally {
+            if (runtime) {
+                await shutdownTeam(runtime.teamName, cwd, { force: true }).catch(() => { });
+            }
+            if (typeof previousPath === 'string')
+                process.env.PATH = previousPath;
+            else
+                delete process.env.PATH;
+            if (typeof previousTmux === 'string')
+                process.env.TMUX = previousTmux;
+            else
+                delete process.env.TMUX;
+            if (typeof previousTmuxPane === 'string')
+                process.env.TMUX_PANE = previousTmuxPane;
+            else
+                delete process.env.TMUX_PANE;
+            if (typeof previousLaunchMode === 'string')
+                process.env.OMX_TEAM_WORKER_LAUNCH_MODE = previousLaunchMode;
+            else
+                delete process.env.OMX_TEAM_WORKER_LAUNCH_MODE;
+            if (typeof previousWorkerCli === 'string')
+                process.env.OMX_TEAM_WORKER_CLI = previousWorkerCli;
+            else
+                delete process.env.OMX_TEAM_WORKER_CLI;
+            await rm(cwd, { recursive: true, force: true });
+            await rm(fakeBinDir, { recursive: true, force: true });
+        }
+    });
     it('startTeam routes detached worktree worker inbox and mailbox triggers through leader-root state references', async () => {
         const repo = await initRepo();
         const binDir = join(repo, 'bin');
@@ -1511,6 +1636,69 @@ esac
             await rm(fakeBinDir, { recursive: true, force: true });
         }
     });
+    it('shutdownTeam restores a standalone HUD pane after tearing down the team HUD', async () => {
+        const cwd = await mkdtemp(join(tmpdir(), 'omx-runtime-shutdown-restore-hud-'));
+        const fakeBinDir = await mkdtemp(join(tmpdir(), 'omx-runtime-shutdown-restore-hud-bin-'));
+        const tmuxLogPath = join(fakeBinDir, 'tmux.log');
+        const tmuxStubPath = join(fakeBinDir, 'tmux');
+        const previousPath = process.env.PATH;
+        try {
+            await writeFile(tmuxStubPath, `#!/bin/sh
+set -eu
+printf '%s\n' "$*" >> "${tmuxLogPath}"
+case "$1" in
+  -V)
+    echo "tmux 3.4"
+    exit 0
+    ;;
+  list-panes)
+    exit 1
+    ;;
+  split-window)
+    printf '%%44\n'
+    exit 0
+    ;;
+  kill-pane|kill-session|select-pane)
+    exit 0
+    ;;
+  *)
+    exit 0
+    ;;
+esac
+`);
+            await chmod(tmuxStubPath, 0o755);
+            process.env.PATH = `${fakeBinDir}:${previousPath ?? ''}`;
+            await initTeamState('team-shutdown-restore-hud', 'shutdown restore hud test', 'executor', 2, cwd);
+            const config = await readTeamConfig('team-shutdown-restore-hud', cwd);
+            assert.ok(config);
+            if (!config)
+                return;
+            config.tmux_session = 'leader:0';
+            config.leader_pane_id = '%11';
+            config.hud_pane_id = '%12';
+            config.workers[0].pane_id = '%12';
+            config.workers[1].pane_id = '%13';
+            await saveTeamConfig(config, cwd);
+            await shutdownTeam('team-shutdown-restore-hud', cwd, { force: true });
+            const tmuxLog = await readFile(tmuxLogPath, 'utf-8');
+            assert.doesNotMatch(tmuxLog, /kill-pane -t %11/);
+            assert.match(tmuxLog, /kill-pane -t %12/);
+            assert.match(tmuxLog, /kill-pane -t %13/);
+            assert.match(tmuxLog, new RegExp(`split-window -v -l ${HUD_TMUX_TEAM_HEIGHT_LINES} -t %11 -d -P -F #\{pane_id\}`));
+            assert.match(tmuxLog, /run-shell -b sleep \d+; tmux resize-pane -t %44 -y \d+ >/);
+            assert.match(tmuxLog, /run-shell tmux resize-pane -t %44 -y \d+ >/);
+            assert.match(tmuxLog, /hud --watch/);
+            assert.match(tmuxLog, /select-pane -t %11/);
+        }
+        finally {
+            if (typeof previousPath === 'string')
+                process.env.PATH = previousPath;
+            else
+                delete process.env.PATH;
+            await rm(cwd, { recursive: true, force: true });
+            await rm(fakeBinDir, { recursive: true, force: true });
+        }
+    });
     it('shutdownTeam preserves leader exclusion while tearing down the hud pane', async () => {
         const cwd = await mkdtemp(join(tmpdir(), 'omx-runtime-shutdown-exclusions-'));
         const fakeBinDir = await mkdtemp(join(tmpdir(), 'omx-runtime-shutdown-exclusions-bin-'));