clone-conversation.sh

#!/usr/bin/env bash
#
# clone-conversation.sh - Claude Code 대화 복제
#
# 순수 bash 구현 - Python/Node 의존성 없음.
# macOS (bash 3.2+) 및 Linux에서 동작합니다.
#
# 사용법:
#   clone-conversation.sh <session-id> [project-path]
#
# 인수:
#   session-id    복제할 대화의 UUID (필수)
#   project-path  프로젝트 경로 (기본값: 현재 디렉토리)
#
# 예시:
#   clone-conversation.sh d96c899d-7501-4e81-a31b-e0095bb3b501
#   clone-conversation.sh d96c899d-7501-4e81-a31b-e0095bb3b501 /home/user/myproject
#
# 복제 후 'claude -r'을 사용하면 원본과 복제된 대화를 모두 볼 수 있습니다.
#

set -euo pipefail

CLAUDE_DIR="${HOME}/.claude"
PROJECTS_DIR="${CLAUDE_DIR}/projects"
HISTORY_FILE="${CLAUDE_DIR}/history.jsonl"
TODOS_DIR="${CLAUDE_DIR}/todos"

# 출력용 색상
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

log_info() { echo -e "${BLUE}[INFO]${NC} $1"; }
log_success() { echo -e "${GREEN}[SUCCESS]${NC} $1"; }
log_warning() { echo -e "${YELLOW}[WARNING]${NC} $1"; }
log_error() { echo -e "${RED}[ERROR]${NC} $1" >&2; }

usage() {
    echo "사용법: $0 <session-id> [project-path]"
    echo ""
    echo "인수:"
    echo "  session-id    복제할 대화의 UUID (필수)"
    echo "  project-path  프로젝트 경로 (기본값: 현재 디렉토리)"
    exit 1
}

# UUID 생성 - Mac과 Linux 모두에서 동작
generate_uuid() {
    if command -v uuidgen &> /dev/null; then
        uuidgen | tr '[:upper:]' '[:lower:]'
    elif [ -f /proc/sys/kernel/random/uuid ]; then
        cat /proc/sys/kernel/random/uuid
    else
        # $RANDOM을 사용한 대체 방법
        printf '%04x%04x-%04x-%04x-%04x-%04x%04x%04x\n' \
            $((RANDOM)) $((RANDOM)) $((RANDOM)) \
            $((RANDOM & 0x0fff | 0x4000)) \
            $((RANDOM & 0x3fff | 0x8000)) \
            $((RANDOM)) $((RANDOM)) $((RANDOM))
    fi
}

convert_path_to_dirname() {
    echo "$1" | sed 's|^/||' | sed 's|/|-|g' | sed 's|^|-|'
}

find_conversation_file() {
    local session_id="$1"
    local project_path="$2"
    local project_dirname
    project_dirname=$(convert_path_to_dirname "$project_path")
    local project_dir="${PROJECTS_DIR}/${project_dirname}"
    local conv_file="${project_dir}/${session_id}.jsonl"

    if [ -f "$conv_file" ]; then
        echo "$conv_file"
        return 0
    fi

    # 모든 프로젝트 디렉토리에서 찾기 시도
    local found_file
    found_file=$(find "$PROJECTS_DIR" -name "${session_id}.jsonl" 2>/dev/null | head -1)

    if [ -n "$found_file" ]; then
        echo "$found_file"
        return 0
    fi

    return 1
}

get_project_from_conv_file() {
    local conv_file="$1"
    local project_dirname
    project_dirname=$(dirname "$conv_file" | xargs basename)
    echo "$project_dirname" | sed 's|^-|/|' | sed 's|-|/|g'
}

# awk 스크립트용 UUID 사전 생성
pre_generate_uuids() {
    local count="$1"
    local output_file="$2"

    if command -v hexdump &> /dev/null; then
        # 빠른 경로: hexdump
        hexdump -vn $((count * 16)) -e '4/1 "%02x" "-" 2/1 "%02x" "-" 2/1 "%02x" "-" 2/1 "%02x" "-" 6/1 "%02x" "\n"' /dev/urandom > "$output_file"
    elif [[ -r /proc/sys/kernel/random/uuid ]]; then
        # Linux 대체
        for ((i=0; i<count; i++)); do
            cat /proc/sys/kernel/random/uuid
        done > "$output_file"
    elif command -v uuidgen &> /dev/null; then
        # macOS/BSD 대체
        for ((i=0; i<count; i++)); do
            uuidgen | tr '[:upper:]' '[:lower:]'
        done > "$output_file"
    else
        log_error "UUID 생성 방법을 찾을 수 없음 (hexdump, /proc/sys/kernel/random/uuid, 또는 uuidgen 필요)"
        return 1
    fi
}

clone_conversation() {
    local source_session="$1"
    local project_path="$2"

    # 복제 태그용 타임스탬프 생성 (예: "Jan 7 14:30")
    local clone_timestamp
    clone_timestamp=$(date "+%b %-d %H:%M")
    local clone_tag="[CLONED ${clone_timestamp}]"

    # 소스 파일 찾기
    local source_file
    if ! source_file=$(find_conversation_file "$source_session" "$project_path"); then
        log_error "세션의 대화 파일을 찾을 수 없음: $source_session"
        log_info "검색 위치: ${PROJECTS_DIR}"
        log_info "사용 가능한 대화:"
        find "$PROJECTS_DIR" -name "*.jsonl" -type f 2>/dev/null | while read -r f; do
            local fname
            fname=$(basename "$f")
            if [[ ${#fname} -eq 42 && "$fname" =~ ^[a-f0-9-]+\.jsonl$ ]]; then
                echo "  - ${fname%.jsonl}"
            fi
        done
        exit 1
    fi

    log_info "소스 대화 발견: $source_file"

    if [ -z "$project_path" ]; then
        project_path=$(get_project_from_conv_file "$source_file")
    fi

    # 새 세션 ID 생성
    local new_session
    new_session=$(generate_uuid)
    log_info "새 세션 ID 생성됨: $new_session"

    # 대상 파일
    local project_dirname
    project_dirname=$(convert_path_to_dirname "$project_path")
    local project_dir="${PROJECTS_DIR}/${project_dirname}"
    local target_file="${project_dir}/${new_session}.jsonl"

    mkdir -p "$project_dir"
    log_info "대화 복제 중: $target_file"

    # 최적화된 첫 번째 패스: 마지막 깨끗한 사용자 메시지가 clone/half-clone 명령인지 확인
    # grep -n을 사용하여 한 번에 모든 사용자 메시지를 찾은 후 필터링
    local stop_at_line=0
    local last_clone_cmd_line=0
    local last_clean_user_line=0

    # 라인 번호와 함께 모든 사용자 메시지 라인 가져오기 (라인별 grep보다 훨씬 빠름)
    # 깨끗한 사용자 메시지만 필터링 (tool_result 아님, isMeta 아님)
    local clean_user_lines
    clean_user_lines=$(grep -n '"type":"user"' "$source_file" | grep -v '"type":"tool_result"' | grep -v '"isMeta":true' || true)

    if [ -n "$clean_user_lines" ]; then
        # 마지막 깨끗한 사용자 메시지 라인 가져오기
        local last_line_info
        last_line_info=$(echo "$clean_user_lines" | tail -1)
        last_clean_user_line=$(echo "$last_line_info" | cut -d: -f1)

        # clone 명령인지 확인
        if echo "$last_line_info" | grep -qE '<command-message>(dx:)?clone</command-message>|<command-message>(dx:)?half-clone</command-message>' 2>/dev/null; then
            last_clone_cmd_line=$last_clean_user_line
        fi
    fi

    # 마지막 깨끗한 사용자 메시지가 clone 명령이면 그 전에서 중지
    if [ "$last_clone_cmd_line" -gt 0 ] && [ "$last_clone_cmd_line" -eq "$last_clean_user_line" ]; then
        stop_at_line=$last_clone_cmd_line
        log_info "/clone 명령 및 이후 메시지 제외 (라인 $stop_at_line부터)"
    fi

    # awk용 UUID 사전 생성 (추정: uuid, parentUuid, messageId에 라인당 3개)
    local total_lines
    total_lines=$(wc -l < "$source_file")
    local uuid_count=$((total_lines * 3 + 100))  # 추가 버퍼
    local uuid_file
    uuid_file=$(mktemp)
    trap "rm -f '$uuid_file'" EXIT
    log_info "UUID 사전 생성 중..."
    pre_generate_uuids "$uuid_count" "$uuid_file"

    # awk로 처리 - 단일 패스, 라인당 외부 명령 없음
    log_info "awk로 처리 중..."
    awk -v stop_at_line="$stop_at_line" \
         -v new_session="$new_session" \
         -v clone_tag="$clone_tag" \
         -v uuid_file="$uuid_file" '
    BEGIN {
        first_user = 1
        uuid_idx = 0
        # 사전 생성된 UUID 로드
        while ((getline uuid < uuid_file) > 0) {
            uuids[uuid_idx++] = uuid
        }
        close(uuid_file)
        next_uuid = 0
    }

    function get_new_uuid(old_uuid) {
        if (old_uuid in uuid_map) {
            return uuid_map[old_uuid]
        }
        new_uuid = uuids[next_uuid++]
        uuid_map[old_uuid] = new_uuid
        return new_uuid
    }

    function extract_uuid(line, key,    pattern, match_str, uuid) {
        pattern = "\"" key "\":\"[a-f0-9][a-f0-9-]*[a-f0-9]\""
        if (match(line, pattern)) {
            match_str = substr(line, RSTART, RLENGTH)
            uuid = match_str
            gsub("\"" key "\":\"", "", uuid)
            gsub("\"", "", uuid)
            return uuid
        }
        return ""
    }

    stop_at_line > 0 && NR >= stop_at_line { exit }
    /^$/ { next }

    {
        line = $0

        # sessionId 교체
        old_session = extract_uuid(line, "sessionId")
        if (old_session != "") {
            gsub("\"sessionId\":\"" old_session "\"", "\"sessionId\":\"" new_session "\"", line)
        }

        # uuid 교체 (parentUuid, sessionId 제외)
        if (match(line, /"uuid":"[a-f0-9][a-f0-9-]*[a-f0-9]"/)) {
            match_str = substr(line, RSTART, RLENGTH)
            old_uuid = match_str
            gsub("\"uuid\":\"", "", old_uuid)
            gsub("\"", "", old_uuid)
            new_uuid = get_new_uuid(old_uuid)
            gsub("\"uuid\":\"" old_uuid "\"", "\"uuid\":\"" new_uuid "\"", line)
        }

        # parentUuid 교체
        old_parent = extract_uuid(line, "parentUuid")
        if (old_parent != "") {
            new_parent = get_new_uuid(old_parent)
            gsub("\"parentUuid\":\"" old_parent "\"", "\"parentUuid\":\"" new_parent "\"", line)
        }

        # messageId 교체
        old_msgid = extract_uuid(line, "messageId")
        if (old_msgid != "") {
            new_msgid = get_new_uuid(old_msgid)
            gsub("\"messageId\":\"" old_msgid "\"", "\"messageId\":\"" new_msgid "\"", line)
        }

        # 첫 번째 사용자 메시지에 태그 추가
        if (first_user && index(line, "\"type\":\"user\"") > 0) {
            gsub("\"content\":\"", "\"content\":\"" clone_tag " ", line)
            gsub("\"text\":\"", "\"text\":\"" clone_tag " ", line)
            first_user = 0
        }

        print line
    }
    ' "$source_file" > "$target_file"

    local line_count
    line_count=$(wc -l < "$target_file" | tr -d ' ')
    log_success "$line_count줄을 $target_file에 기록함"

    # claude -r 목록 상단에 나타나도록 파일 수정 시간 갱신
    touch "$target_file"

    # history.jsonl 업데이트
    log_info "히스토리 파일 업데이트 중..."

    # 첫 번째 사용자 메시지에서 표시 텍스트 가져오기
    local display_text
    display_text=$(grep '"type":"user"' "$source_file" | head -1 | \
        grep -oE '"content":"[^"]*"' | head -1 | \
        sed 's/"content":"//;s/"$//' | \
        head -c 200 || echo "[복제된 대화]")

    if [ -z "$display_text" ]; then
        # 배열 형식 시도
        display_text=$(grep '"type":"user"' "$source_file" | head -1 | \
            grep -oE '"text":"[^"]*"' | head -1 | \
            sed 's/"text":"//;s/"$//' | \
            head -c 200 || echo "[복제된 대화]")
    fi

    display_text="${clone_tag} ${display_text}"

    # 타임스탬프 (밀리초)
    local timestamp
    if [[ "$OSTYPE" == "darwin"* ]]; then
        timestamp=$(( $(date +%s) * 1000 + 1000 ))
    else
        timestamp=$(( $(date +%s%3N) + 1000 ))
    fi

    # JSON용 이스케이프
    display_text=$(echo "$display_text" | sed 's/\\/\\\\/g' | sed 's/"/\\"/g' | tr '\n' ' ')

    # 히스토리 항목 추가
    echo "{\"display\":\"${display_text}\",\"pastedContents\":{},\"timestamp\":${timestamp},\"project\":\"${project_path}\",\"sessionId\":\"${new_session}\"}" >> "$HISTORY_FILE"
    echo "히스토리 항목이 추가되었습니다"

    # 할 일 파일이 있으면 복사
    local old_todo_file="${TODOS_DIR}/${source_session}-agent-${source_session}.json"
    local new_todo_file="${TODOS_DIR}/${new_session}-agent-${new_session}.json"

    if [ -f "$old_todo_file" ]; then
        log_info "할 일 파일 복사 중..."
        cp "$old_todo_file" "$new_todo_file"
    fi

    log_success "대화가 성공적으로 복제되었습니다!"
    echo ""
    echo "원본 세션:   $source_session"
    echo "새 세션:     $new_session"
    echo "프로젝트:    $project_path"
    echo ""
    echo "복제된 대화를 이어가려면 다음을 사용하세요:"
    echo "  claude -r"
    echo ""
    echo "${clone_tag} 표시가 있는 대화를 선택하세요"
}

# 메인
if [ $# -lt 1 ]; then
    usage
fi

SESSION_ID="$1"
PROJECT_PATH="${2:-$(pwd)}"

# 세션 ID 유효성 검사
if ! [[ "$SESSION_ID" =~ ^[a-f0-9]{8}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{12}$ ]]; then
    log_error "잘못된 세션 ID 형식. UUID 형식 예시: d96c899d-7501-4e81-a31b-e0095bb3b501"
    exit 1
fi

if [ ! -d "$CLAUDE_DIR" ]; then
    log_error "Claude 디렉토리를 찾을 수 없음: $CLAUDE_DIR"
    exit 1
fi

clone_conversation "$SESSION_ID" "$PROJECT_PATH"