13907292e9
- codex-async-notify.sh: tmux paste-buffer方式で安定化 - codex-keep-two.sh: 正確なプロセスカウント実装 - デフォルト2プロセス制限で暴走防止 - 日本語プロンプト「まだタスクがあれば次のタスクお願いします」で無限ループ実現 - count_running_codex_display()でcomm+argsによる正確な検出 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
335 lines
13 KiB
Bash
335 lines
13 KiB
Bash
#!/bin/bash
|
||
# codex-async-notify.sh - Codexを非同期実行してtmuxセッションに通知
|
||
|
||
# 使い方を表示
|
||
if [ $# -eq 0 ]; then
|
||
echo "Usage: $0 <task description> [tmux_session]"
|
||
echo "Examples:"
|
||
echo " $0 'Refactor MIR builder to 13 instructions'"
|
||
echo " $0 'Write paper introduction' gemini-session"
|
||
echo " $0 'Review code quality' chatgpt"
|
||
echo ""
|
||
echo "Default tmux session: claude"
|
||
exit 1
|
||
fi
|
||
|
||
# 引数解析
|
||
TASK="$1"
|
||
TARGET_SESSION="${2:-claude}" # デフォルトは "claude"
|
||
# 通知用ウィンドウ名（既定: codex-notify）。存在しなければ作成する
|
||
NOTIFY_WINDOW_NAME="${CODEX_NOTIFY_WINDOW:-codex-notify}"
|
||
|
||
# 設定
|
||
WORK_DIR="$HOME/.codex-async-work"
|
||
LOG_DIR="$WORK_DIR/logs"
|
||
RUN_DIR="$WORK_DIR/running"
|
||
WORK_ID=$(date +%s%N)
|
||
LOG_FILE="$LOG_DIR/codex-${WORK_ID}.log"
|
||
|
||
# 作業ディレクトリ準備
|
||
mkdir -p "$LOG_DIR" "$RUN_DIR"
|
||
|
||
# === オプショナル並列制御 ===
|
||
# - CODEX_MAX_CONCURRENT: 許容最大同時実行数（デフォルト2）
|
||
# - CODEX_CONCURRENCY_MODE: "block"(既定) or "drop"（上限超過時に起動を諦める）
|
||
# - CODEX_DEDUP: 1 で同一 TASK が実行中なら重複起動を避ける
|
||
MAX_CONCURRENT=${CODEX_MAX_CONCURRENT:-2}
|
||
CONC_MODE=${CODEX_CONCURRENCY_MODE:-block}
|
||
DEDUP=${CODEX_DEDUP:-0}
|
||
# 検出パターン（上書き可）。例: export CODEX_PROC_PATTERN='codex .* exec'
|
||
CODEX_PROC_PATTERN=${CODEX_PROC_PATTERN:-'codex .* exec'}
|
||
FLOCK_WAIT=${CODEX_FLOCK_WAIT:-5}
|
||
LOG_RETENTION_DAYS=${CODEX_LOG_RETENTION_DAYS:-0}
|
||
LOG_MAX_BYTES=${CODEX_LOG_MAX_BYTES:-0}
|
||
# 実行中カウント方式: proc(プロセス数) / pgid(プロセスグループ数) / sentinel(将来のための拡張)
|
||
CODEX_COUNT_MODE=${CODEX_COUNT_MODE:-sentinel}
|
||
|
||
list_running_codex() {
|
||
# 実ジョブのみを検出: "codex ... exec ..." を含むコマンドライン
|
||
if command -v pgrep >/dev/null 2>&1; then
|
||
# pgrep -af は PID と引数を出力
|
||
pgrep -af -- "$CODEX_PROC_PATTERN" || true
|
||
else
|
||
# フォールバック: ps+grep（grep 自身は除外）
|
||
ps -eo pid=,args= | grep -E -- "$CODEX_PROC_PATTERN" | grep -v grep || true
|
||
fi
|
||
}
|
||
|
||
count_running_codex() {
|
||
case "$CODEX_COUNT_MODE" in
|
||
sentinel)
|
||
# Count sentinel files for tasks we launched; clean up stale ones
|
||
local cnt=0
|
||
if [ -d "$RUN_DIR" ]; then
|
||
for f in "$RUN_DIR"/codex-*.run; do
|
||
[ -e "$f" ] || continue
|
||
# Optional liveness check by stored pid
|
||
pid=$(awk -F': ' '/^pid:/{print $2; exit}' "$f" 2>/dev/null || true)
|
||
if [ -n "$pid" ] && ! kill -0 "$pid" 2>/dev/null; then
|
||
rm -f "$f" 2>/dev/null || true
|
||
continue
|
||
fi
|
||
cnt=$((cnt+1))
|
||
done
|
||
fi
|
||
# Fallback: if 0 (older jobs without sentinel), attempt pgid-based count
|
||
if [ "${cnt:-0}" -eq 0 ]; then
|
||
if command -v pgrep >/dev/null 2>&1; then
|
||
pgc=$(pgrep -f -- "$CODEX_PROC_PATTERN" \
|
||
| xargs -r -I {} sh -c 'ps -o pgid= -p "$1" 2>/dev/null' _ {} \
|
||
| awk '{print $1}' | grep -E '^[0-9]+$' | sort -u | wc -l | tr -d ' ' || echo 0)
|
||
echo "${pgc:-0}"
|
||
else
|
||
list_running_codex | wc -l | tr -d ' ' || echo 0
|
||
fi
|
||
else
|
||
echo "$cnt"
|
||
fi
|
||
;;
|
||
pgid)
|
||
# pgrepで対象PIDsを取得 → 各PIDのPGIDを集計（自分自身のawk/ps行は拾わない）
|
||
if command -v pgrep >/dev/null 2>&1; then
|
||
pgrep -f -- "$CODEX_PROC_PATTERN" \
|
||
| xargs -r -I {} sh -c 'ps -o pgid= -p "$1" 2>/dev/null' _ {} \
|
||
| awk '{print $1}' | grep -E '^[0-9]+$' | sort -u | wc -l | tr -d ' ' || echo 0
|
||
else
|
||
# フォールバック: プロセス数
|
||
list_running_codex | wc -l | tr -d ' ' || echo 0
|
||
fi
|
||
;;
|
||
proc|*)
|
||
list_running_codex | wc -l | tr -d ' ' || echo 0
|
||
;;
|
||
esac
|
||
}
|
||
|
||
# Display-oriented counter: count real 'codex exec' processes by comm+args
|
||
count_running_codex_display() {
|
||
if ps -eo comm=,args= >/dev/null 2>&1; then
|
||
ps -eo comm=,args= \
|
||
| awk '($1 ~ /^codex$/) && ($0 ~ / exec[ ]/) {print $0}' \
|
||
| wc -l | tr -d ' ' || echo 0
|
||
else
|
||
# Fallback to pattern match
|
||
list_running_codex | wc -l | tr -d ' ' || echo 0
|
||
fi
|
||
}
|
||
|
||
is_same_task_running() {
|
||
# ざっくり: 引数列に TASK 文字列（1行化）を含む行があれば重複とみなす
|
||
local oneline
|
||
oneline=$(echo "$TASK" | tr '\n' ' ' | sed 's/ */ /g')
|
||
list_running_codex | grep -F -- "$oneline" >/dev/null 2>&1
|
||
}
|
||
|
||
maybe_wait_for_slot() {
|
||
[ "${MAX_CONCURRENT}" -gt 0 ] || return 0
|
||
|
||
# 起動判定〜起動直前までをクリティカルセクションにする
|
||
mkdir -p "$WORK_DIR"
|
||
exec 9>"$WORK_DIR/concurrency.lock"
|
||
if ! flock -w "$FLOCK_WAIT" 9; then
|
||
echo "⚠️ concurrency.lock の取得に失敗（${FLOCK_WAIT}s）。緩やかに続行。" >&2
|
||
else
|
||
export CODEX_LOCK_HELD=1
|
||
fi
|
||
|
||
if [ "$DEDUP" = "1" ] && is_same_task_running; then
|
||
echo "⚠️ 同一タスクが実行中のため起動をスキップ: $TASK"
|
||
exit 0
|
||
fi
|
||
|
||
local current
|
||
current=$(count_running_codex)
|
||
if [ "$current" -lt "$MAX_CONCURRENT" ]; then
|
||
return 0
|
||
fi
|
||
|
||
case "$CONC_MODE" in
|
||
drop)
|
||
echo "⚠️ 上限(${MAX_CONCURRENT})到達のため起動をスキップ。現在: $current"
|
||
exit 3
|
||
;;
|
||
block|*)
|
||
echo "⏳ スロット待機: 現在 $current / 上限 $MAX_CONCURRENT"
|
||
while :; do
|
||
sleep 1
|
||
current=$(count_running_codex)
|
||
if [ "$current" -lt "$MAX_CONCURRENT" ]; then
|
||
echo "✅ 空きスロット確保: 現在 $current / 上限 $MAX_CONCURRENT"
|
||
break
|
||
fi
|
||
done
|
||
;;
|
||
esac
|
||
}
|
||
|
||
# 上限が定義されていれば起動前に調整
|
||
maybe_wait_for_slot
|
||
|
||
# 非同期実行関数
|
||
run_codex_async() {
|
||
{
|
||
# Create sentinel to track running task (removed on exit)
|
||
SEN_FILE="$RUN_DIR/codex-${WORK_ID}.run"
|
||
{
|
||
echo "work_id: $WORK_ID"
|
||
echo "task: $TASK"
|
||
echo "started: $(date -Is)"
|
||
echo "pid: $$"
|
||
} > "$SEN_FILE" 2>/dev/null || true
|
||
# Ensure sentinel (and dedup file if any) are always cleaned up on exit
|
||
trap 'rm -f "$SEN_FILE" "${CODEX_DEDUP_FILE:-}" >/dev/null 2>&1 || true' EXIT
|
||
# Dedupロック: 子プロセス側で握る（同一TASKの多重起動回避）
|
||
if [ "${CODEX_DEDUP_FILE:-}" != "" ]; then
|
||
exec 8>"${CODEX_DEDUP_FILE}"
|
||
if ! flock -n 8; then
|
||
echo "⚠️ Duplicate task detected (dedup lock busy). Skipping: $TASK" | tee -a "$LOG_FILE"
|
||
rm -f "$SEN_FILE" >/dev/null 2>&1 || true
|
||
exit 0
|
||
fi
|
||
# (cleanup handled by global EXIT trap above)
|
||
fi
|
||
# Detach: silence this background subshell's stdout/stderr while still tee-ing to log
|
||
if [ "${CODEX_ASYNC_DETACH:-0}" = "1" ]; then
|
||
exec >/dev/null 2>&1
|
||
fi
|
||
echo "=====================================" | tee "$LOG_FILE"
|
||
echo "🚀 Codex Task Started" | tee -a "$LOG_FILE"
|
||
echo "Work ID: $WORK_ID" | tee -a "$LOG_FILE"
|
||
echo "Task: $TASK" | tee -a "$LOG_FILE"
|
||
echo "Start: $(date)" | tee -a "$LOG_FILE"
|
||
echo "=====================================" | tee -a "$LOG_FILE"
|
||
echo "" | tee -a "$LOG_FILE"
|
||
|
||
# Codex実行
|
||
START_TIME=$(date +%s)
|
||
codex exec "$TASK" 2>&1 | tee -a "$LOG_FILE"
|
||
EXIT_CODE=${PIPESTATUS[0]}
|
||
END_TIME=$(date +%s)
|
||
DURATION=$((END_TIME - START_TIME))
|
||
|
||
echo "" | tee -a "$LOG_FILE"
|
||
echo "=====================================" | tee -a "$LOG_FILE"
|
||
echo "✅ Codex Task Completed" | tee -a "$LOG_FILE"
|
||
echo "Exit Code: $EXIT_CODE" | tee -a "$LOG_FILE"
|
||
echo "Duration: ${DURATION}s" | tee -a "$LOG_FILE"
|
||
echo "End: $(date)" | tee -a "$LOG_FILE"
|
||
echo "=====================================" | tee -a "$LOG_FILE"
|
||
|
||
# 末尾表示行数（環境変数で上書き可）/ ミニマル通知モード
|
||
TAIL_N=${CODEX_NOTIFY_TAIL:-20}
|
||
MINIMAL=${CODEX_NOTIFY_MINIMAL:-1}
|
||
|
||
# 通知内容を一時ファイルに組み立て（空行も保持）
|
||
TASK_ONELINE=$(echo "$TASK" | tr '\n' ' ' | sed 's/ */ /g')
|
||
NOTIFY_FILE="$WORK_DIR/notify-${WORK_ID}.tmp"
|
||
if [ "$MINIMAL" = "1" ]; then
|
||
{
|
||
echo ""
|
||
echo "# 🤖 Codex作業完了通知 [$(date +%H:%M:%S)]"
|
||
echo "# Work ID: $WORK_ID"
|
||
echo "# Status: $([ $EXIT_CODE -eq 0 ] && echo '✅ Success' || echo '❌ Failed')"
|
||
echo "# Log: $LOG_FILE"
|
||
echo ""
|
||
} > "$NOTIFY_FILE"
|
||
else
|
||
{
|
||
echo ""
|
||
echo "# 🤖 Codex作業完了通知 [$(date +%H:%M:%S)]"
|
||
echo "# Work ID: $WORK_ID"
|
||
echo "# Task: $TASK_ONELINE"
|
||
echo "# Status: $([ $EXIT_CODE -eq 0 ] && echo '✅ Success' || echo '❌ Failed')"
|
||
echo "# Duration: ${DURATION}秒"
|
||
echo "# Log: $LOG_FILE"
|
||
echo "# === 最後の出力 (tail -n ${TAIL_N}) ==="
|
||
tail -n "$TAIL_N" "$LOG_FILE" | sed -e 's/^/# > /'
|
||
echo "# =================="
|
||
echo ""
|
||
} > "$NOTIFY_FILE"
|
||
fi
|
||
|
||
# 共通のシンプル通知（4行）を指定の tmux セッションのアクティブペインへ送る（Enter= C-m）
|
||
if tmux has-session -t "$TARGET_SESSION" 2>/dev/null; then
|
||
STATUS_MARK=$([ $EXIT_CODE -eq 0 ] && echo '✅ Success' || echo '❌ Failed')
|
||
CHAT_FILE="$WORK_DIR/chat-${WORK_ID}.tmp"
|
||
TASK_ONELINE=$(echo "$TASK" | tr '\n' ' ' | sed 's/ */ /g')
|
||
# 表示用: 実プロセス（comm=codex かつ args に exec を含む）の数を採用
|
||
CURRENT_RUNNING=$(count_running_codex_display)
|
||
case "$CURRENT_RUNNING" in (*[!0-9]*) CURRENT_RUNNING=0;; esac
|
||
{
|
||
echo "# 🤖 Codex: Done [$(date +%H:%M:%S)] (実行中: $CURRENT_RUNNING)"
|
||
echo "# Work ID: $WORK_ID"
|
||
echo "# Status: $STATUS_MARK"
|
||
echo "# Log: $LOG_FILE"
|
||
echo "# Task: $TASK_ONELINE — まだタスクがあれば次のタスクお願いします"
|
||
echo ""
|
||
} > "$CHAT_FILE"
|
||
# アクティブペインを取得
|
||
TARGET_PANE=$(tmux list-panes -t "$TARGET_SESSION" -F '#{pane_id} #{pane_active}' 2>/dev/null | awk '$2=="1"{print $1; exit}')
|
||
[ -z "$TARGET_PANE" ] && TARGET_PANE="$TARGET_SESSION"
|
||
BUF_NAME="codex-chat-$WORK_ID"
|
||
tmux load-buffer -b "$BUF_NAME" "$CHAT_FILE" 2>/dev/null || true
|
||
tmux paste-buffer -b "$BUF_NAME" -t "$TARGET_PANE" 2>/dev/null || true
|
||
tmux delete-buffer -b "$BUF_NAME" 2>/dev/null || true
|
||
# Small delay to ensure paste completes before sending Enter
|
||
sleep 0.2
|
||
tmux send-keys -t "$TARGET_PANE" C-m 2>/dev/null || true
|
||
sleep 0.05
|
||
tmux send-keys -t "$TARGET_PANE" C-m 2>/dev/null || true
|
||
rm -f "$NOTIFY_FILE" "$CHAT_FILE" 2>/dev/null || true
|
||
else
|
||
echo "⚠️ Target tmux session '$TARGET_SESSION' not found"
|
||
echo " Notification was not sent, but work completed."
|
||
echo " Available sessions:"
|
||
tmux list-sessions 2>/dev/null || echo " No tmux sessions running"
|
||
fi
|
||
# 古いログの自動クリーン（任意）
|
||
if [ "$LOG_RETENTION_DAYS" -gt 0 ] 2>/dev/null; then
|
||
find "$LOG_DIR" -type f -name 'codex-*.log' -mtime +"$LOG_RETENTION_DAYS" -delete 2>/dev/null || true
|
||
fi
|
||
if [ "$LOG_MAX_BYTES" -gt 0 ] 2>/dev/null; then
|
||
# 容量超過時に古い順で間引く
|
||
CUR=$(du -sb "$LOG_DIR" 2>/dev/null | awk '{print $1}' || echo 0)
|
||
while [ "${CUR:-0}" -gt "$LOG_MAX_BYTES" ]; do
|
||
oldest=$(ls -1t "$LOG_DIR"/codex-*.log 2>/dev/null | tail -n 1)
|
||
[ -n "$oldest" ] || break
|
||
rm -f "$oldest" 2>/dev/null || true
|
||
CUR=$(du -sb "$LOG_DIR" 2>/dev/null | awk '{print $1}' || echo 0)
|
||
done
|
||
fi
|
||
} &
|
||
}
|
||
|
||
# Dedupファイルパス（子へ受け渡し）
|
||
if [ "$DEDUP" = "1" ]; then
|
||
# TASK 正規化→SHA1
|
||
TASK_ONELINE=$(echo "$TASK" | tr '\n' ' ' | sed 's/ */ /g')
|
||
TASK_SHA=$(printf "%s" "$TASK_ONELINE" | sha1sum | awk '{print $1}')
|
||
export CODEX_DEDUP_FILE="$WORK_DIR/dedup-${TASK_SHA}.lock"
|
||
fi
|
||
|
||
# バックグラウンドで実行（必要ならロック解放は起動直後に）
|
||
run_codex_async
|
||
ASYNC_PID=$!
|
||
|
||
# すぐにロックが残っていれば解放（他の起動を待たせない）
|
||
if [ "${CODEX_LOCK_HELD:-0}" = "1" ]; then
|
||
flock -u 9 2>/dev/null || true
|
||
fi
|
||
|
||
# 実行開始メッセージ
|
||
echo ""
|
||
echo "✅ Codex started asynchronously!"
|
||
echo " PID: $ASYNC_PID"
|
||
echo " Work ID: $WORK_ID"
|
||
echo " Log file: $LOG_FILE"
|
||
echo ""
|
||
echo "📝 Monitor progress:"
|
||
echo " tail -f $LOG_FILE"
|
||
echo ""
|
||
echo "🔍 Check status:"
|
||
echo " ps -p $ASYNC_PID"
|
||
echo ""
|
||
echo "Codex is now working in the background..."
|