├── .claude/
│   ├── settings.local.json
│   └── logs/
├── ...
└── checkpoint.sh

{
    "permissions": {
      "deny": [
      ]
    },
    "hooks": {
      "Stop": [
        {
          "hooks": [
            {
              "type": "command",
              "command": "YOUR_PROJECT_DIR/checkpoint.sh",
              "timeout": 30000
            }
          ]
        }
      ]
    }
  }

#!/bin/bash
# Claude Code stop hook: checkpoint manager

# Thư mục và tập tin log
LOG_DIR=".claude/logs"
LOG_FILE="$LOG_DIR/checkpoint.log"
mkdir -p "$LOG_DIR"

# Mức độ ghi log qua biến môi trường
CHECKPOINT_DEBUG=${CHECKPOINT_DEBUG:-false}
CHECKPOINT_VERBOSE=${CHECKPOINT_VERBOSE:-true}

# Xoay vòng log: >50KB, giữ lại hiện tại + .old
rotate_logs() {
    if [ -f "$LOG_FILE" ] && [ $(wc -c < "$LOG_FILE" 2>/dev/null || echo 0) -gt 51200 ]; then
        # Xóa tệp cũ, di chuyển tệp hiện tại đến .old
        [ -f "$LOG_FILE.old" ] && rm -f "$LOG_FILE.old"
        mv "$LOG_FILE" "$LOG_FILE.old"
    fi
}

# Hàm hỗ trợ ghi log (có điều kiện)
log_debug() {
    [ "$CHECKPOINT_DEBUG" = "true" ] && echo "[$(date '+%H:%M:%S')] DEBUG: $1" >> "$LOG_FILE"
}

log_info() {
    [ "$CHECKPOINT_VERBOSE" = "true" ] && echo "[$(date '+%H:%M:%S')] INFO: $1" >> "$LOG_FILE"
}

log_error() {
    echo "[$(date '+%H:%M:%S')] ERROR: $1" >> "$LOG_FILE"
}

# Xoay vòng log
rotate_logs

# Bắt đầu ghi debug
log_debug "Stop hook started with args: $*"

# Đọc dữ liệu JSON đầu vào của Claude Code
HOOK_DATA=""
if [ -t 0 ]; then
    log_debug "No stdin data detected (terminal mode)"
else
    log_debug "Reading JSON data from stdin"
    HOOK_DATA=$(cat)
    log_debug "Received JSON: $HOOK_DATA"
fi

# Guard: stop_hook_active true => exit (ngăn chặn vòng lặp)
if echo "$HOOK_DATA" | grep -q '"stop_hook_active":\s*true'; then
    log_info "Stop hook already active, exiting to prevent loop"
    exit 0
fi

# Xác minh repo git
if ! git rev-parse --git-dir >/dev/null 2>&1; then
    log_debug "Not a git repository, exiting"
    exit 1
fi

# Kiểm tra thay đổi trong cây làm việc
GIT_STATUS=$(git status --porcelain 2>/dev/null)
if [ -z "$GIT_STATUS" ]; then
    log_debug "No changes detected, exiting"
    exit 0
fi

log_info "Changes detected: $(echo "$GIT_STATUS" | wc -l) files"

# Xây dựng thông điệp stash (arg hoặc trạng thái git)
if [ -n "$1" ]; then
    MESSAGE="$1"
    log_debug "Using command line message: $MESSAGE"
else
    # Từ trạng thái, lấy tên tệp được sắp xếp theo thời gian sửa đổi (tối đa 3)
    FILES_WITH_TIME=""
    while IFS= read -r line; do
        if [ -n "$line" ]; then
            FILE_PATH=$(echo "$line" | cut -c4-)
            if [ -f "$FILE_PATH" ]; then
                # Lưu với thời gian sửa đổi tính bằng giây
                MOD_TIME=$(stat -f "%m" "$FILE_PATH" 2>/dev/null || echo "0")
                FILES_WITH_TIME="$FILES_WITH_TIME$MOD_TIME:$FILE_PATH\n"
            fi
        fi
    done <<< "$GIT_STATUS"

    # Sắp xếp theo thời gian sửa đổi desc, giữ lại tên
    SORTED_FILES=$(echo -e "$FILES_WITH_TIME" | grep -v "^$" | sort -rn -t: -k1 | cut -d: -f2- | head -3)

    # Soạn thảo thông điệp
    CHANGED_FILES=$(echo "$SORTED_FILES" | tr '\n' ' ' | sed 's/ $//')
    TOTAL_COUNT=$(echo "$GIT_STATUS" | wc -l)
    if [ $TOTAL_COUNT -gt 3 ]; then
        MESSAGE="$CHANGED_FILES (and $(($TOTAL_COUNT - 3)) more)"
    else
        MESSAGE="$CHANGED_FILES"
    fi
    log_debug "Generated message from git status (sorted by modification time): $MESSAGE"
fi

# Tạo nhãn checkpoint
TIMESTAMP=$(date '+%y%m%d:%H:%M:%S')
STASH_MESSAGE="agent: $TIMESTAMP $MESSAGE"

log_info "Creating checkpoint: $STASH_MESSAGE"

# Thực hiện git stash
if git stash push --include-untracked -m "$STASH_MESSAGE" >/dev/null 2>&1; then
    log_info "Stash created successfully"

    # Áp dụng lại stash
    if git stash apply stash@{0} >/dev/null 2>&1; then
        log_info "Stash reapplied successfully"
    else
        log_debug "Warning: Failed to reapply stash"
    fi
else
    log_debug "Error: Failed to create stash"
    exit 1
fi

# Vòng đời stash của agent (tối đa 10)
AGENT_STASH_COUNT=$(git stash list 2>/dev/null | grep "agent:" | wc -l)
log_debug "Current agent stash count: $AGENT_STASH_COUNT"

while [ $AGENT_STASH_COUNT -gt 10 ]; do
    OLDEST_AGENT_STASH=$(git stash list 2>/dev/null | grep "agent:" | tail -1 | cut -d: -f1)
    if [ -n "$OLDEST_AGENT_STASH" ]; then
        log_info "Removing oldest agent stash: $OLDEST_AGENT_STASH"
        git stash drop "$OLDEST_AGENT_STASH" >/dev/null 2>&1
        AGENT_STASH_COUNT=$((AGENT_STASH_COUNT - 1))
    else
        break
    fi
done

# Thông điệp thành công
FINAL_MESSAGE="Checkpoint saved: stash@{0} - $STASH_MESSAGE"
echo "$FINAL_MESSAGE"
log_info "$FINAL_MESSAGE"

git stash list

git stash apply stash@{0}