#!/bin/bash
#
# Claude Code Hook: task-complete
# Runs AFTER a task is completed
# Saves conversation context to the database for future recall
#
# Expected environment variables:
#   CLAUDE_PROJECT_ID - UUID of the current project
#   JWT_TOKEN - Authentication token for API
#   CLAUDE_API_URL - API base URL (default: http://localhost:8000)
#   CONTEXT_RECALL_ENABLED - Set to "false" to disable (default: true)
#   TASK_SUMMARY - Summary of completed task (auto-generated by Claude)
#   TASK_FILES - Files modified during task (comma-separated)
#

# Load configuration if exists
CONFIG_FILE="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)/context-recall-config.env"
if [ -f "$CONFIG_FILE" ]; then
    source "$CONFIG_FILE"
fi

# Default values
API_URL="${CLAUDE_API_URL:-http://localhost:8000}"
ENABLED="${CONTEXT_RECALL_ENABLED:-true}"

# Exit early if disabled
if [ "$ENABLED" != "true" ]; then
    exit 0
fi

# Detect project ID (same logic as user-prompt-submit)
if [ -z "$CLAUDE_PROJECT_ID" ]; then
    PROJECT_ID=$(git config --local claude.projectid 2>/dev/null)

    if [ -z "$PROJECT_ID" ]; then
        GIT_REMOTE=$(git config --get remote.origin.url 2>/dev/null)
        if [ -n "$GIT_REMOTE" ]; then
            PROJECT_ID=$(echo -n "$GIT_REMOTE" | md5sum | cut -d' ' -f1)
        fi
    fi
else
    PROJECT_ID="$CLAUDE_PROJECT_ID"
fi

# Exit if no project ID or JWT token
if [ -z "$PROJECT_ID" ] || [ -z "$JWT_TOKEN" ]; then
    exit 0
fi

# Gather task information
TIMESTAMP=$(date -u +"%Y-%m-%dT%H:%M:%SZ")
GIT_BRANCH=$(git rev-parse --abbrev-ref HEAD 2>/dev/null || echo "unknown")
GIT_COMMIT=$(git rev-parse --short HEAD 2>/dev/null || echo "none")

# Get recent git changes
CHANGED_FILES=$(git diff --name-only HEAD~1 2>/dev/null | head -10 | tr '\n' ',' | sed 's/,$//')
if [ -z "$CHANGED_FILES" ]; then
    CHANGED_FILES="${TASK_FILES:-}"
fi

# Create task summary
if [ -z "$TASK_SUMMARY" ]; then
    # Generate basic summary from git log if no summary provided
    TASK_SUMMARY=$(git log -1 --pretty=format:"%s" 2>/dev/null || echo "Task completed")
fi

# Build context payload
CONTEXT_TITLE="Session: ${TIMESTAMP}"
CONTEXT_TYPE="session_summary"
RELEVANCE_SCORE=7.0

# Create dense summary
DENSE_SUMMARY="Task completed on branch '${GIT_BRANCH}' (commit: ${GIT_COMMIT}).

Summary: ${TASK_SUMMARY}

Modified files: ${CHANGED_FILES:-none}

Timestamp: ${TIMESTAMP}"

# Escape JSON strings
escape_json() {
    echo "$1" | python3 -c "import sys, json; print(json.dumps(sys.stdin.read())[1:-1])"
}

ESCAPED_TITLE=$(escape_json "$CONTEXT_TITLE")
ESCAPED_SUMMARY=$(escape_json "$DENSE_SUMMARY")

# Save context to database
CONTEXT_PAYLOAD=$(cat <<EOF
{
    "project_id": "${PROJECT_ID}",
    "context_type": "${CONTEXT_TYPE}",
    "title": ${ESCAPED_TITLE},
    "dense_summary": ${ESCAPED_SUMMARY},
    "relevance_score": ${RELEVANCE_SCORE},
    "metadata": {
        "git_branch": "${GIT_BRANCH}",
        "git_commit": "${GIT_COMMIT}",
        "files_modified": "${CHANGED_FILES}",
        "timestamp": "${TIMESTAMP}"
    }
}
EOF
)

# POST to conversation-contexts endpoint
RESPONSE=$(curl -s --max-time 5 \
    -X POST "${API_URL}/api/conversation-contexts" \
    -H "Authorization: Bearer ${JWT_TOKEN}" \
    -H "Content-Type: application/json" \
    -d "$CONTEXT_PAYLOAD" 2>/dev/null)

# Update project state
PROJECT_STATE_PAYLOAD=$(cat <<EOF
{
    "project_id": "${PROJECT_ID}",
    "state_data": {
        "last_task_completion": "${TIMESTAMP}",
        "last_git_commit": "${GIT_COMMIT}",
        "last_git_branch": "${GIT_BRANCH}",
        "recent_files": "${CHANGED_FILES}"
    },
    "state_type": "task_completion"
}
EOF
)

curl -s --max-time 5 \
    -X POST "${API_URL}/api/project-states" \
    -H "Authorization: Bearer ${JWT_TOKEN}" \
    -H "Content-Type: application/json" \
    -d "$PROJECT_STATE_PAYLOAD" 2>/dev/null >/dev/null

# Log success (optional - comment out for silent operation)
if [ -n "$RESPONSE" ]; then
    echo "✓ Context saved to database" >&2
fi

exit 0