LandPPT/docker-healthcheck.sh

#!/bin/bash

# LandPPT Docker Health Check Script
# This script performs comprehensive health checks for the LandPPT application

set -e

# Configuration
HOST=${HOST:-localhost}
PORT=${PORT:-8000}
TIMEOUT=${HEALTH_CHECK_TIMEOUT:-30}

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

# Logging function
log() {
    echo -e "${GREEN}[$(date +'%Y-%m-%d %H:%M:%S')] $1${NC}"
}

warn() {
    echo -e "${YELLOW}[$(date +'%Y-%m-%d %H:%M:%S')] WARNING: $1${NC}"
}

error() {
    echo -e "${RED}[$(date +'%Y-%m-%d %H:%M:%S')] ERROR: $1${NC}"
}

# Check if service is responding
check_http_health() {
    log "Checking HTTP health endpoint..."
    
    if curl -f -s --max-time $TIMEOUT "http://${HOST}:${PORT}/health" > /dev/null; then
        log "✅ HTTP health check passed"
        return 0
    else
        error "❌ HTTP health check failed"
        return 1
    fi
}

# Check if API documentation is accessible
check_api_docs() {
    log "Checking API documentation..."
    
    if curl -f -s --max-time $TIMEOUT "http://${HOST}:${PORT}/docs" > /dev/null; then
        log "✅ API documentation accessible"
        return 0
    else
        warn "⚠️ API documentation not accessible"
        return 1
    fi
}

# Check configured database mode (informational only)
check_database_mode() {
    log "Checking database mode..."

    local database_url="${DATABASE_URL:-}"
    if [ -z "$database_url" ]; then
        warn "⚠️ DATABASE_URL not set; application will use its internal default"
        return 0
    fi

    if [[ "$database_url" == sqlite:* ]]; then
        log "✅ SQLite mode configured"
        return 0
    fi

    log "✅ External database configured via DATABASE_URL"
    return 0
}

# Check required directories
check_directories() {
    log "Checking required directories..."
    
    local dirs=("/app/uploads" "/app/temp" "/app/data")
    local all_good=true
    
    for dir in "${dirs[@]}"; do
        if [ -d "$dir" ]; then
            log "✅ Directory $dir exists"
        else
            warn "⚠️ Directory $dir missing"
            all_good=false
        fi
    done
    
    if $all_good; then
        return 0
    else
        return 1
    fi
}

# Check Python process
check_python_process() {
    log "Checking Python process..."

    # `pgrep` is provided by `procps` on Debian. Some slim images may not have it.
    # If it's missing, rely on the HTTP health check as the critical signal.
    if ! command -v pgrep >/dev/null 2>&1; then
        warn "pgrep not found; skipping process check (install procps to enable it)"
        return 0
    fi

    if pgrep -f "python.*run.py" > /dev/null; then
        log "✅ Python process running"
        return 0
    fi

    # Fallback for uvicorn multi-worker where the parent/worker cmdline may not match `run.py` exactly.
    if pgrep -f "uvicorn" > /dev/null; then
        log "✅ Uvicorn process running"
        return 0
    fi

    error "❌ Python/Uvicorn process not found"
    return 1
}

# Main health check function
main() {
    log "Starting LandPPT health check..."
    
    local exit_code=0
    
    # Critical checks (must pass)
    if ! check_http_health; then
        exit_code=1
    fi
    
    if ! check_python_process; then
        exit_code=1
    fi
    
    # Non-critical checks (warnings only)
    check_api_docs || true
    check_database_mode || true
    check_directories || true
    
    if [ $exit_code -eq 0 ]; then
        log "🎉 All critical health checks passed!"
    else
        error "💥 Some critical health checks failed!"
    fi
    
    exit $exit_code
}

# Run health check
main "$@"