ralph.sh

#!/bin/bash
set -e

trap 'echo -e "\n  -> Прервано (Ctrl+C)"; exit 130' INT

TASKS_FILE="tasks/tasks.json"
PROMPT_FILE="prompt.md"
COOLDOWN=14400          # 4 часа в секундах — пауза при rate limit

# Agent selection:
# - Set RALPH_AGENT=claude or RALPH_AGENT=codex to force.
# - Otherwise auto-detect (prefers Claude if available).
resolve_agent() {
    if [[ -n "${RALPH_AGENT:-}" ]]; then
        echo "$RALPH_AGENT"
        return 0
    fi
    if command -v claude >/dev/null 2>&1; then
        echo "claude"
        return 0
    fi
    if command -v codex >/dev/null 2>&1; then
        echo "codex"
        return 0
    fi
    return 1
}

run_agent() {
    local agent="$1"
    local prompt="$2"

    case "$agent" in
        claude)
            claude --permission-mode acceptEdits -p "$prompt"
            ;;
        codex)
            local output_file
            output_file="$(mktemp -t ralph_codex.XXXXXX)"
            codex exec --full-auto --color never -C "$PWD" --output-last-message "$output_file" "$prompt" >/dev/null
            cat "$output_file"
            rm -f "$output_file"
            ;;
        *)
            echo "Unsupported agent: $agent" >&2
            return 1
            ;;
    esac
}

# Функция проверки наличия pending задач
has_pending_tasks() {
    pending_count=$(grep -c '"status": "pending"' "$TASKS_FILE" 2>/dev/null) || pending_count=0
    [ "$pending_count" -gt 0 ]
}

# Получить ID следующей pending задачи по приоритету (пропускает задачи с attempts >= MAX_ATTEMPTS)
get_next_task_id() {
    python3 -c "
import json
with open('$TASKS_FILE') as f:
    data = json.load(f)
priority_order = {'critical': 0, 'high': 1, 'medium': 2, 'low': 3}
pending = [t for t in data['tasks'] if t['status'] == 'pending' and t.get('attempts', 0) < $MAX_ATTEMPTS]
pending.sort(key=lambda t: priority_order.get(t.get('priority','low'), 4))
if pending:
    print(pending[0]['id'])
" 2>/dev/null || echo ""
}

get_task_info() {
    local task_id="$1"
    python3 -c "
import json
with open('$TASKS_FILE') as f:
    data = json.load(f)
for t in data['tasks']:
    if t['id'] == '${task_id}':
        print(f\"{t['id']}: {t['description']}\")
        break
" 2>/dev/null || echo "$task_id"
}

build_prompt() {
    local task_id="$1"
    sed "s/__TASK_ID__/${task_id}/g" "$PROMPT_FILE"
}

# Увеличить счётчик попыток (задача остаётся pending)
bump_attempts() {
    local task_id="$1"
    python3 -c "
import json
with open('$TASKS_FILE') as f:
    data = json.load(f)
for t in data['tasks']:
    if t['id'] == '${task_id}':
        t['attempts'] = t.get('attempts', 0) + 1
        break
with open('$TASKS_FILE', 'w') as f:
    json.dump(data, f, ensure_ascii=False, indent='\t')
" 2>/dev/null
}

MAX_ATTEMPTS=2

iteration=1
completed_tasks=""
completed_count=0
failed_tasks=""
failed_count=0

while has_pending_tasks; do
    next_id=$(get_next_task_id)

    if [[ -z "$next_id" ]]; then
        break
    fi

    next_info=$(get_task_info "$next_id")
    remaining=$(grep -c '"status": "pending"' "$TASKS_FILE" 2>/dev/null) || remaining=0

    echo "[$iteration] $next_info (осталось $remaining)"

    agent=$(resolve_agent) || {
        echo "Не найден поддерживаемый агент. Установите 'claude' или 'codex', либо задайте RALPH_AGENT." >&2
        exit 1
    }

    prompt=$(build_prompt "$next_id")

    # Запуск агента с обработкой rate limit
    result=""
    if ! result=$(run_agent "$agent" "$prompt" 2>&1); then
        if echo "$result" | grep -qi "rate\|limit\|usage\|capacity\|throttl\|429\|too many"; then
            echo "  -> Rate limit! Жду ${COOLDOWN}с (4 часа)..."
            echo "  -> Пауза до: $(date -v+${COOLDOWN}S '+%H:%M:%S' 2>/dev/null || date -d "+${COOLDOWN} seconds" '+%H:%M:%S' 2>/dev/null || echo '~4ч')"
            sleep "$COOLDOWN"
            echo "  -> Пауза окончена, продолжаю задачу $next_id"
            ((iteration++))
            continue
        else
            echo "  -> Агент упал с ошибкой"
            echo "  -> $result" | head -5
        fi
    fi

    if [[ "$result" == *"RALPH_COMPLETE"* ]]; then
        # Ensure task is marked done (agent may have failed to update JSON)
        python3 -c "
import json
with open('$TASKS_FILE') as f:
    data = json.load(f)
for t in data['tasks']:
    if t['id'] == '${next_id}':
        t['status'] = 'done'
        break
with open('$TASKS_FILE', 'w') as f:
    json.dump(data, f, ensure_ascii=False, indent='\t')
    f.write('\n')
" 2>/dev/null
        completed_tasks="${completed_tasks}  + ${next_info}
"
        ((completed_count++))
        echo "  -> done"
    elif [[ "$result" == *"RALPH_PARTIAL"* ]]; then
        echo "  -> partial (прогресс записан)"
    else
        bump_attempts "$next_id"
        attempts=$(python3 -c "
import json
with open('$TASKS_FILE') as f:
    data = json.load(f)
for t in data['tasks']:
    if t['id'] == '${next_id}':
        print(t.get('attempts', 0))
        break
" 2>/dev/null)
        echo "  -> нет результата, попытка ${attempts}/${MAX_ATTEMPTS}"
        if [ "${attempts:-0}" -ge "$MAX_ATTEMPTS" ]; then
            echo "  -> исчерпаны попытки, пропускаю"
            failed_tasks="${failed_tasks}  - ${next_info}
"
            ((failed_count++))
        fi
    fi

    ((iteration++))
done

# === Финальный отчёт ===
echo ""
echo "==========================================="
echo "  Ralph завершил работу. Итераций: $((iteration-1))"
echo "==========================================="

if [ "$completed_count" -gt 0 ]; then
    echo ""
    echo "Выполнено ($completed_count):"
    printf "%s" "$completed_tasks"
fi

if [ "$failed_count" -gt 0 ]; then
    echo ""
    echo "Не удалось завершить ($failed_count):"
    printf "%s" "$failed_tasks"
fi

remaining=$(grep -c '"status": "pending"' "$TASKS_FILE" 2>/dev/null) || remaining=0
done_total=$(grep -c '"status": "done"' "$TASKS_FILE" 2>/dev/null) || done_total=0
echo ""
echo "Итого: done=$done_total, pending=$remaining"
echo "==========================================="

if [ "$remaining" -eq 0 ]; then
    say -v Milena "Хозяин, я всё сделалъ!"
else
    say -v Milena "Хозяин, я закончилъ. Выполнено $completed_count, не удалось $failed_count, осталось $remaining."
fi