claude-controller 0.1.0

package/config.sh

@@ -0,0 +1,74 @@
+#!/usr/bin/env bash
+# ============================================================
+# Controller Service — Configuration
+# ============================================================
+
+# Claude CLI 경로
+# 1) 환경변수 CLAUDE_BIN이 있으면 사용
+# 2) PATH에서 claude를 찾음
+# 3) macOS 앱 기본 경로
+CLAUDE_BIN="${CLAUDE_BIN:-$(command -v claude 2>/dev/null || echo "/Applications/cmux.app/Contents/Resources/bin/claude")}"
+
+# 기본 출력 형식 (stream-json: 실시간 토큰 스트리밍)
+DEFAULT_OUTPUT_FORMAT="${DEFAULT_OUTPUT_FORMAT:-stream-json}"
+
+# 모든 도구 권한 허용
+DEFAULT_ALLOWED_TOOLS="${DEFAULT_ALLOWED_TOOLS:-Bash,Read,Write,Edit,Glob,Grep,Agent,NotebookEdit,WebFetch,WebSearch}"
+
+# 모델 설정
+DEFAULT_MODEL="${DEFAULT_MODEL:-}"
+
+# 디렉토리 경로
+CONTROLLER_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+LOGS_DIR="${CONTROLLER_DIR}/logs"
+SESSIONS_DIR="${CONTROLLER_DIR}/sessions"
+QUEUE_DIR="${CONTROLLER_DIR}/queue"
+
+# FIFO 파이프 경로 — 외부에서 이 파이프에 쓰면 서비스가 수신
+FIFO_PATH="${CONTROLLER_DIR}/queue/controller.pipe"
+
+# PID 파일
+PID_FILE="${CONTROLLER_DIR}/service/controller.pid"
+
+# 최대 동시 백그라운드 작업 수
+MAX_BACKGROUND_JOBS="${MAX_BACKGROUND_JOBS:-10}"
+
+# 시스템 프롬프트 추가
+APPEND_SYSTEM_PROMPT="${APPEND_SYSTEM_PROMPT:-}"
+
+# 작업 디렉토리 — claude -p 실행 시 --cwd
+WORKING_DIR="${WORKING_DIR:-}"
+
+# ── Worktree 설정 ─────────────────────────────────────────
+# 대상 Git 저장소 (worktree 생성 원본)
+TARGET_REPO="${TARGET_REPO:-}"
+
+# 워크트리 기준 브랜치
+BASE_BRANCH="${BASE_BRANCH:-main}"
+
+# 워크트리 저장 디렉토리
+WORKTREES_DIR="${CONTROLLER_DIR}/worktrees"
+
+# ── 권한 설정 ──────────────────────────────────────────────
+# true로 설정 시 --dangerously-skip-permissions 사용 (모든 도구 무제한 허용)
+SKIP_PERMISSIONS="${SKIP_PERMISSIONS:-true}"
+
+# ── Checkpoint 설정 ────────────────────────────────────────
+# 체크포인트 감시 주기 (초) — 이 간격으로 worktree 변경을 확인
+CHECKPOINT_INTERVAL="${CHECKPOINT_INTERVAL:-5}"
+
+# ── settings.json 오버라이드 ────────────────────────────────
+# data/settings.json이 존재하면 해당 값으로 기본값을 덮어씀
+SETTINGS_FILE="${CONTROLLER_DIR}/data/settings.json"
+if [[ -f "$SETTINGS_FILE" ]] && command -v jq &>/dev/null; then
+  _s() { jq -r "$1 // empty" "$SETTINGS_FILE" 2>/dev/null; }
+  _v=$(_s '.skip_permissions');       [[ -n "$_v" ]] && SKIP_PERMISSIONS="$_v"
+  _v=$(_s '.allowed_tools');          [[ -n "$_v" ]] && DEFAULT_ALLOWED_TOOLS="$_v"
+  _v=$(_s '.model');                  [[ -n "$_v" ]] && DEFAULT_MODEL="$_v"
+  _v=$(_s '.max_jobs');               [[ -n "$_v" ]] && MAX_BACKGROUND_JOBS="$_v"
+  _v=$(_s '.append_system_prompt');   [[ -n "$_v" ]] && APPEND_SYSTEM_PROMPT="$_v"
+  _v=$(_s '.target_repo');            [[ -n "$_v" ]] && TARGET_REPO="$_v"
+  _v=$(_s '.base_branch');            [[ -n "$_v" ]] && BASE_BRANCH="$_v"
+  _v=$(_s '.checkpoint_interval');    [[ -n "$_v" ]] && CHECKPOINT_INTERVAL="$_v"
+  unset -f _s; unset _v
+fi

package/lib/checkpoint.sh

@@ -0,0 +1,237 @@
+#!/usr/bin/env bash
+# ============================================================
+# Checkpoint 관리 모듈
+# Worktree에서 실행 중인 job의 파일 변경을 자동 커밋(checkpoint)하고
+# 특정 체크포인트로 되돌리는(rewind) 기능을 제공합니다.
+#
+# 의존: jobs.sh (_get_meta_field), config.sh (LOGS_DIR)
+# ============================================================
+
+# ── 설정 ────────────────────────────────────────────────────
+CHECKPOINT_INTERVAL="${CHECKPOINT_INTERVAL:-5}"
+CHECKPOINT_PREFIX="ckpt"
+
+# ── 체크포인트 워처 루프 ────────────────────────────────────
+# 백그라운드에서 worktree를 감시하며 파일 변경이 안정화되면 자동 커밋.
+# 호출 시 반드시 & (백그라운드)로 실행할 것.
+#
+# Usage: checkpoint_watcher_loop <worktree_path> <job_id> <meta_file> &
+checkpoint_watcher_loop() {
+  local wt_path="$1"
+  local job_id="$2"
+  local meta_file="$3"
+  local turn=0
+  local prev_hash=""
+
+  cd "$wt_path" 2>/dev/null || return 1
+
+  # worktree 로컬 git 설정 (커밋용)
+  git config user.email "checkpoint@controller.local" 2>/dev/null
+  git config user.name "Controller Checkpoint" 2>/dev/null
+
+  # 초기 상태에 untracked 파일이 있으면 커밋
+  git add -A 2>/dev/null
+  if ! git diff --cached --quiet 2>/dev/null; then
+    git commit -m "${CHECKPOINT_PREFIX}:${job_id}:0:init" --no-verify 2>/dev/null || true
+  fi
+
+  while true; do
+    sleep "$CHECKPOINT_INTERVAL"
+
+    # job 상태 확인 — running이 아니면 최종 커밋 후 종료
+    local status=""
+    [[ -f "$meta_file" ]] && status=$(grep '^STATUS=' "$meta_file" 2>/dev/null | tail -1 | sed 's/^STATUS=//')
+    if [[ "$status" != "running" ]]; then
+      _ckpt_commit_if_dirty "$wt_path" "$job_id" "$turn" "final"
+      break
+    fi
+
+    # 현재 변경사항 해시 (diff + untracked files)
+    local curr_hash
+    curr_hash=$( { git diff 2>/dev/null; git ls-files --others --exclude-standard 2>/dev/null; } | md5 2>/dev/null || echo "none" )
+
+    # 빈 해시 = 변경 없음
+    local empty_hash
+    empty_hash=$(echo -n | md5 2>/dev/null || echo "none")
+    if [[ "$curr_hash" == "$empty_hash" ]]; then
+      prev_hash=""
+      continue
+    fi
+
+    # 안정화 대기: 이전 체크와 해시가 다르면 아직 변경 진행 중
+    if [[ "$curr_hash" != "$prev_hash" ]]; then
+      prev_hash="$curr_hash"
+      continue
+    fi
+
+    # 2회 연속 동일 해시 = 변경 완료, 커밋 수행
+    (( turn++ )) || true
+    _ckpt_commit_if_dirty "$wt_path" "$job_id" "$turn" ""
+    prev_hash=""
+  done
+}
+
+# ── 내부: 변경이 있으면 커밋 ────────────────────────────────
+_ckpt_commit_if_dirty() {
+  local wt_path="$1" job_id="$2" turn="$3" suffix="$4"
+
+  cd "$wt_path" 2>/dev/null || return 1
+
+  git add -A 2>/dev/null
+
+  if ! git diff --cached --quiet 2>/dev/null; then
+    local msg="${CHECKPOINT_PREFIX}:${job_id}:${turn}"
+    [[ -n "$suffix" ]] && msg="${msg}:${suffix}"
+
+    # 변경 파일 목록 (최대 5개)
+    local changed
+    changed=$(git diff --cached --name-only 2>/dev/null | head -5 | tr '\n' ', ')
+    [[ -n "$changed" ]] && msg="${msg} [${changed%,}]"
+
+    git commit -m "$msg" --no-verify 2>/dev/null || true
+  fi
+}
+
+# ── 체크포인트 목록 조회 ────────────────────────────────────
+# worktree의 git log에서 이 job의 checkpoint 커밋들을 JSON 배열로 반환.
+#
+# Usage: checkpoint_list <worktree_path> <job_id>
+# Output: JSON array
+checkpoint_list() {
+  local wt_path="$1"
+  local job_id="$2"
+
+  # worktree 유효성 확인
+  if [[ ! -d "$wt_path" ]] || ! (cd "$wt_path" && git rev-parse --git-dir >/dev/null 2>&1); then
+    echo "[]"
+    return
+  fi
+
+  cd "$wt_path" 2>/dev/null || { echo "[]"; return; }
+
+  local result="["
+  local first=true
+
+  while IFS='|' read -r hash ts msg; do
+    [[ -z "$hash" ]] && continue
+
+    # 턴 번호 추출
+    local turn_num
+    turn_num=$(echo "$msg" | sed -n "s/.*${CHECKPOINT_PREFIX}:${job_id}:\([0-9]*\).*/\1/p")
+    [[ -z "$turn_num" ]] && turn_num=0
+
+    # 변경 파일 수
+    local file_count
+    file_count=$(git diff-tree --no-commit-id --name-only -r "$hash" 2>/dev/null | wc -l | tr -d ' ')
+
+    # 변경 파일 목록
+    local file_list
+    file_list=$(git diff-tree --no-commit-id --name-only -r "$hash" 2>/dev/null | head -10 | jq -R . 2>/dev/null | jq -s . 2>/dev/null)
+    [[ -z "$file_list" ]] && file_list="[]"
+
+    [[ "$first" == "true" ]] && first=false || result="${result},"
+    result="${result}{\"hash\":\"${hash}\",\"turn\":${turn_num},\"timestamp\":\"${ts}\",\"message\":$(echo "$msg" | jq -R .),\"files_changed\":${file_count},\"files\":${file_list}}"
+  done < <(git log --format='%H|%aI|%s' --grep="${CHECKPOINT_PREFIX}:${job_id}:" 2>/dev/null)
+
+  result="${result}]"
+  echo "$result"
+}
+
+# ── 대화 컨텍스트 추출 ──────────────────────────────────────
+# stream-json .out 파일에서 assistant 텍스트 + tool_use 요약을 추출.
+# 최대 max_chars 글자까지만 반환하여 프롬프트 크기를 제한.
+#
+# Usage: checkpoint_extract_context <out_file> [max_chars]
+# Output: plain text
+checkpoint_extract_context() {
+  local out_file="$1"
+  local max_chars="${2:-4000}"
+
+  [[ -f "$out_file" ]] || return 0
+
+  # jq로 한 번에 추출 (성능 + 안정성)
+  jq -r '
+    if .type == "assistant" then
+      [.message.content[]? |
+        if .type == "text" then "[\(.type)] \(.text[0:300])"
+        elif .type == "tool_use" then "[tool: \(.name)] \(.input | tostring[0:150])"
+        else empty end
+      ] | join("\n")
+    elif .type == "result" then
+      "[result] \(.result[0:300] // "")"
+    else empty end
+  ' "$out_file" 2>/dev/null | head -c "$max_chars"
+}
+
+# ── Rewind 실행 ──────────────────────────────────────────────
+# 1. 실행 중인 job 종료
+# 2. worktree를 지정된 checkpoint로 git reset --hard
+# 3. 대화 컨텍스트 추출
+# 4. 새 프롬프트 생성 (FIFO 전송은 호출자가 담당)
+#
+# Usage: checkpoint_rewind <job_id> <checkpoint_hash> <new_prompt>
+# Output: 생성된 full_prompt (stdout), 실패 시 exit 1
+checkpoint_rewind() {
+  local job_id="$1"
+  local ckpt_hash="$2"
+  local new_prompt="$3"
+
+  local meta_file="${LOGS_DIR}/job_${job_id}.meta"
+  local out_file="${LOGS_DIR}/job_${job_id}.out"
+
+  # meta 파일 확인
+  if [[ ! -f "$meta_file" ]]; then
+    echo "[오류] Job #${job_id}를 찾을 수 없습니다." >&2
+    return 1
+  fi
+
+  # 실행 중이면 종료
+  local status
+  status=$(_get_meta_field "$meta_file" "STATUS")
+  if [[ "$status" == "running" ]]; then
+    job_kill "$job_id" >/dev/null 2>&1
+    sleep 1
+  fi
+
+  # worktree 경로 확인
+  local wt_path
+  wt_path=$(_get_meta_field "$meta_file" "WORKTREE")
+  if [[ -z "$wt_path" || ! -d "$wt_path" ]]; then
+    echo "[오류] 워크트리를 찾을 수 없습니다." >&2
+    return 1
+  fi
+
+  # checkpoint 커밋 유효성 확인
+  if ! (cd "$wt_path" && git cat-file -t "$ckpt_hash" >/dev/null 2>&1); then
+    echo "[오류] 유효하지 않은 체크포인트: $ckpt_hash" >&2
+    return 1
+  fi
+
+  # 체크포인트로 reset
+  (cd "$wt_path" && git reset --hard "$ckpt_hash" 2>/dev/null) || {
+    echo "[오류] git reset 실패" >&2
+    return 1
+  }
+
+  # 대화 컨텍스트 추출
+  local context=""
+  if [[ -f "$out_file" ]]; then
+    context=$(checkpoint_extract_context "$out_file" 4000)
+  fi
+
+  # 새 프롬프트 구성
+  if [[ -n "$context" ]]; then
+    cat <<REWIND_PROMPT
+[이전 작업 컨텍스트 — 아래는 이전 세션에서 수행된 작업 요약입니다]
+${context}
+
+[Rewind 지시사항]
+파일 상태가 위 작업 중간의 체크포인트 시점으로 복원되었습니다.
+이전 작업 내용을 참고하되, 이어서 다음을 수행하세요:
+
+${new_prompt}
+REWIND_PROMPT
+  else
+    echo "$new_prompt"
+  fi
+}

package/lib/executor.sh

@@ -0,0 +1,183 @@
+#!/usr/bin/env bash
+# ============================================================
+# 실행 엔진 — 서비스 데몬 패턴
+# FIFO로 수신된 JSON 메시지를 파싱하여 claude -p 를 백그라운드로 실행합니다.
+# 모든 실행은 비동기이며, 결과는 $LOGS_DIR/job_<id>.out 에 저장됩니다.
+# ============================================================
+
+# ── claude -p 실행 (내부 전용) ─────────────────────────────
+# JSON에서 파싱한 개별 값을 받아 claude CLI를 호출합니다.
+# eval 대신 인자 배열을 직접 사용하여 글로브 확장·인젝션 방지
+_run_claude() {
+  local prompt="$1"
+  local cwd="$2"
+  local out_file="$3"
+  local session_mode="${4:-}"
+  local session_id="${5:-}"
+
+  local args=()
+  args+=(-p "$prompt")
+  args+=(--output-format json)
+
+  # 모든 도구 권한 허용
+  if [[ -n "${DEFAULT_ALLOWED_TOOLS:-}" ]]; then
+    args+=(--allowedTools "$DEFAULT_ALLOWED_TOOLS")
+  fi
+
+  # 모델 지정
+  if [[ -n "${DEFAULT_MODEL:-}" ]]; then
+    args+=(--model "$DEFAULT_MODEL")
+  fi
+
+  # 시스템 프롬프트 추가
+  if [[ -n "${APPEND_SYSTEM_PROMPT:-}" ]]; then
+    args+=(--append-system-prompt "$APPEND_SYSTEM_PROMPT")
+  fi
+
+  # 작업 디렉토리 — JSON에서 받은 cwd 우선, 없으면 글로벌 WORKING_DIR
+  local effective_cwd="${cwd:-${WORKING_DIR:-}}"
+  if [[ -n "$effective_cwd" ]]; then
+    args+=(--cwd "$effective_cwd")
+  fi
+
+  # 세션 이어가기 플래그
+  case "${session_mode}" in
+    continue)
+      args+=(--continue)
+      ;;
+    resume)
+      if [[ -n "$session_id" ]]; then
+        args+=(--resume "$session_id")
+      fi
+      ;;
+  esac
+
+  # 직접 실행 — 배열로 안전하게 호출
+  "$CLAUDE_BIN" "${args[@]}" > "$out_file" 2>&1
+}
+
+# ── JSON 기반 실행 (서비스 데몬의 진입점) ──────────────────
+# FIFO에서 수신한 JSON 문자열을 파싱하여 백그라운드로 claude -p 실행
+#
+# JSON 필드:
+#   id      (필수) — 작업 식별자. 외부에서 부여한 고유 ID
+#   prompt  (필수) — claude에 전달할 프롬프트
+#   cwd     (선택) — 작업 디렉토리. 미지정 시 WORKING_DIR 사용
+#   session (선택) — "continue" | "resume:<session_id>"
+#
+# 반환: stdout에 job_id 출력
+execute_from_json() {
+  local json_string="$1"
+
+  # ── JSON 유효성 검사 ──
+  if ! echo "$json_string" | jq empty 2>/dev/null; then
+    echo "[오류] 유효하지 않은 JSON: $json_string" >&2
+    return 1
+  fi
+
+  # ── 필수 필드 파싱 ──
+  local job_id prompt cwd session_raw images_json
+  job_id=$(echo "$json_string" | jq -r '.id // empty')
+  prompt=$(echo "$json_string" | jq -r '.prompt // empty')
+  cwd=$(echo "$json_string" | jq -r '.cwd // empty')
+  session_raw=$(echo "$json_string" | jq -r '.session // empty')
+  images_json=$(echo "$json_string" | jq -r '.images // empty')
+
+  if [[ -z "$job_id" ]]; then
+    echo "[오류] JSON에 'id' 필드가 없습니다." >&2
+    return 1
+  fi
+
+  if [[ -z "$prompt" ]]; then
+    echo "[오류] JSON에 'prompt' 필드가 없습니다." >&2
+    return 1
+  fi
+
+  # ── 첨부 이미지가 있으면 프롬프트에 경로 삽입 ──
+  if [[ -n "$images_json" && "$images_json" != "null" ]]; then
+    local img_count
+    img_count=$(echo "$images_json" | jq -r 'length')
+    if [[ "$img_count" -gt 0 ]]; then
+      local img_lines=""
+      for i in $(seq 0 $(( img_count - 1 ))); do
+        local img_path
+        img_path=$(echo "$images_json" | jq -r ".[$i]")
+        img_lines="${img_lines}
+- ${img_path}"
+      done
+      prompt="[첨부 파일 — Read 도구로 확인하세요]${img_lines}
+
+${prompt}"
+    fi
+  fi
+
+  # ── 세션 모드 분리 (예: "resume:abc-123") ──
+  local session_mode="" session_id=""
+  if [[ -n "$session_raw" ]]; then
+    case "$session_raw" in
+      resume:*)
+        session_mode="resume"
+        session_id="${session_raw#resume:}"
+        ;;
+      continue)
+        session_mode="continue"
+        ;;
+      *)
+        echo "[경고] 알 수 없는 session 값: $session_raw" >&2
+        ;;
+    esac
+  fi
+
+  # ── 최대 동시 작업 수 확인 ──
+  local running_count=0
+  for meta_file in "${LOGS_DIR}"/job_*.meta; do
+    [[ -f "$meta_file" ]] || continue
+    local STATUS=""
+    STATUS=$(_get_meta_field "$meta_file" "STATUS")
+    if [[ "$STATUS" == "running" ]]; then
+      (( running_count++ )) || true
+    fi
+  done
+
+  if [[ $running_count -ge $MAX_BACKGROUND_JOBS ]]; then
+    echo "[오류] 최대 동시 작업 수($MAX_BACKGROUND_JOBS)에 도달. job_id=$job_id 거부됨" >&2
+    return 1
+  fi
+
+  # ── Job 등록 (jobs.sh 모듈 사용) ──
+  # job_register는 내부 카운터 ID를 반환하지만,
+  # 외부 ID(job_id)를 파일명에 직접 사용하여 추적 일관성 확보
+  local internal_id
+  internal_id=$(job_register "$prompt")
+
+  # 외부 ID ↔ 내부 ID 매핑 저장
+  echo "$job_id" > "${LOGS_DIR}/job_${internal_id}.ext_id"
+
+  local out_file="${LOGS_DIR}/job_${internal_id}.out"
+
+  # ── 서브쉘에서 백그라운드 실행 ──
+  (
+    _run_claude "$prompt" "$cwd" "$out_file" "$session_mode" "$session_id"
+    local exit_code=$?
+
+    # 세션 ID 추출 (JSON 출력에서)
+    if [[ -f "$out_file" ]]; then
+      local sid
+      sid=$(jq -r '.session_id // empty' "$out_file" 2>/dev/null)
+      [[ -n "$sid" ]] && job_set_session "$internal_id" "$sid"
+    fi
+
+    # 완료 상태 갱신
+    if [[ $exit_code -eq 0 ]]; then
+      job_mark_done "$internal_id"
+    else
+      job_mark_failed "$internal_id"
+    fi
+  ) &
+
+  local bg_pid=$!
+  job_set_pid "$internal_id" "$bg_pid"
+
+  # 호출자(서비스 데몬)에게 내부 ID 반환
+  echo "$internal_id"
+}