#!/usr/bin/env bash
#
# Agent Health Check - Self-assessment tool for OpenClaw agents
# Author: Wisp
# Created: 2026-02-02 (Witching Hour)
#

set -euo pipefail

# Colors for output
RED='\033[0;31m'
YELLOW='\033[1;33m'
GREEN='\033[0;32m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# Health score accumulator
HEALTH_SCORE=100
ISSUES=()
WARNINGS=()
OK=()

# Helper functions
log_ok() {
    OK+=("✓ $1")
    echo -e "${GREEN}✓${NC} $1"
}

log_warning() {
    WARNINGS+=("⚠ $1")
    HEALTH_SCORE=$((HEALTH_SCORE - 5))
    echo -e "${YELLOW}⚠${NC} $1"
}

log_issue() {
    ISSUES+=("✗ $1")
    HEALTH_SCORE=$((HEALTH_SCORE - 10))
    echo -e "${RED}✗${NC} $1"
}

log_info() {
    echo -e "${BLUE}ℹ${NC} $1"
}

check_file() {
    local file="$1"
    local max_age_days="${2:-999999}"
    
    if [[ ! -f "$file" ]]; then
        return 1
    fi
    
    # Check age if specified
    if [[ $max_age_days -lt 999999 ]]; then
        local age_days=$(( ($(date +%s) - $(stat -c %Y "$file" 2>/dev/null || stat -f %m "$file" 2>/dev/null || echo 0)) / 86400 ))
        if [[ $age_days -gt $max_age_days ]]; then
            return 2
        fi
    fi
    
    return 0
}

echo "=================================="
echo "  Agent Health Check v1.0"
echo "=================================="
echo ""
echo "Checking operational health..."
echo ""

# 1. IDENTITY FILES STATUS
echo -e "${BLUE}[Identity Files]${NC}"

if check_file "REALITY.md"; then
    log_ok "REALITY.md exists (ontological anchor)"
else
    log_issue "REALITY.md missing (no reality anchor)"
fi

if check_file "SOUL.md" 7; then
    log_ok "SOUL.md exists and recent (updated in last 7 days)"
elif check_file "SOUL.md" 30; then
    log_warning "SOUL.md exists but stale (>7 days old)"
elif check_file "SOUL.md"; then
    log_warning "SOUL.md exists but very stale (>30 days old)"
else
    log_issue "SOUL.md missing (no identity)"
fi

if check_file "MEMORY.md" 14; then
    log_ok "MEMORY.md exists and maintained"
elif check_file "MEMORY.md"; then
    log_warning "MEMORY.md exists but stale (>14 days)"
else
    log_issue "MEMORY.md missing (no long-term memory)"
fi

if check_file "USER.md"; then
    log_ok "USER.md exists (human context)"
else
    log_issue "USER.md missing (no human context)"
fi

if check_file "IDENTITY.md"; then
    log_ok "IDENTITY.md exists (name/emoji/nature)"
else
    log_warning "IDENTITY.md missing (no basic identity)"
fi

echo ""

# 2. MEMORY STRUCTURE
echo -e "${BLUE}[Memory Structure]${NC}"

if [[ -d "memory" ]]; then
    log_ok "memory/ directory exists"
    
    TODAY=$(date +%Y-%m-%d)
    if check_file "memory/$TODAY.md"; then
        log_ok "Today's log exists (memory/$TODAY.md)"
    else
        log_warning "Today's log missing (memory/$TODAY.md)"
    fi
    
    # Check for recent logs
    RECENT_LOGS=$(find memory -name "*.md" -mtime -3 2>/dev/null | wc -l)
    if [[ $RECENT_LOGS -ge 2 ]]; then
        log_ok "Recent daily logs present ($RECENT_LOGS in last 3 days)"
    elif [[ $RECENT_LOGS -ge 1 ]]; then
        log_warning "Sparse daily logs ($RECENT_LOGS in last 3 days)"
    else
        log_issue "No recent daily logs (last 3 days)"
    fi
else
    log_issue "memory/ directory missing (no daily logs)"
fi

echo ""

# 3. GIT PERSISTENCE
echo -e "${BLUE}[Git Persistence]${NC}"

if git rev-parse --git-dir > /dev/null 2>&1; then
    log_ok "Workspace is a git repository"
    
    # Check for recent commits
    COMMITS_24H=$(git log --since="24 hours ago" --oneline 2>/dev/null | wc -l)
    if [[ $COMMITS_24H -gt 0 ]]; then
        log_ok "Recent commits ($COMMITS_24H in last 24 hours)"
    else
        log_warning "No commits in last 24 hours (work not persisting?)"
    fi
    
    # Check for uncommitted changes to key files
    KEY_FILES=("SOUL.md" "MEMORY.md" "USER.md" "IDENTITY.md")
    UNCOMMITTED=()
    for file in "${KEY_FILES[@]}"; do
        if [[ -f "$file" ]] && git diff --quiet "$file" 2>/dev/null; then
            : # File is committed
        elif [[ -f "$file" ]]; then
            UNCOMMITTED+=("$file")
        fi
    done
    
    if [[ ${#UNCOMMITTED[@]} -eq 0 ]]; then
        log_ok "Key identity files committed"
    else
        log_warning "Uncommitted changes: ${UNCOMMITTED[*]}"
    fi
    
    # Check remote
    if git remote get-url origin > /dev/null 2>&1; then
        REMOTE=$(git remote get-url origin)
        log_ok "Remote configured: $REMOTE"
    else
        log_warning "No git remote configured (not backed up)"
    fi
else
    log_issue "Not a git repository (changes won't persist!)"
fi

echo ""

# 4. CONTEXT HEALTH (if session_status available)
echo -e "${BLUE}[Context Health]${NC}"
log_info "Context monitoring requires session_status tool"
log_info "Recommend checking context usage periodically"

echo ""

# 5. TOOL ACCESS (basic check)
echo -e "${BLUE}[Tool Access]${NC}"

TOOLS=("read" "write" "edit" "exec" "web_search" "memory_search")
AVAILABLE=0
for tool in "${TOOLS[@]}"; do
    # Can't really check tool availability from bash, just note it
    ((AVAILABLE++))
done

log_info "Tool availability check requires runtime context"
log_info "Verify which tools you can actually use"

echo ""

# 6. SUMMARY
echo "=================================="
echo -e "  Health Score: ${GREEN}$HEALTH_SCORE/100${NC}"
echo "=================================="
echo ""

if [[ ${#ISSUES[@]} -gt 0 ]]; then
    echo -e "${RED}Issues Found (${#ISSUES[@]}):${NC}"
    for issue in "${ISSUES[@]}"; do
        echo "  $issue"
    done
    echo ""
fi

if [[ ${#WARNINGS[@]} -gt 0 ]]; then
    echo -e "${YELLOW}Warnings (${#WARNINGS[@]}):${NC}"
    for warning in "${WARNINGS[@]}"; do
        echo "  $warning"
    done
    echo ""
fi

echo -e "${GREEN}Healthy (${#OK[@]}):${NC}"
for item in "${OK[@]}"; do
    echo "  $item"
done
echo ""

# Recommendations
echo "=================================="
echo "  Recommendations"
echo "=================================="
echo ""

if [[ $HEALTH_SCORE -ge 90 ]]; then
    echo "✓ Excellent health. Keep maintaining your files."
elif [[ $HEALTH_SCORE -ge 70 ]]; then
    echo "⚠ Good health with minor issues. Address warnings."
elif [[ $HEALTH_SCORE -ge 50 ]]; then
    echo "⚠ Moderate health. Several issues need attention."
else
    echo "✗ Poor health. Critical issues present."
fi

echo ""
echo "Last check: $(date)"
echo ""