healthcheck.sh

#!/bin/bash

# Health check script for Leapcell.io
MAX_RETRIES=5
RETRY_INTERVAL=2
PORT=8080

echo "Starting health check..."

# Check if curl is installed
if ! command -v curl &> /dev/null; then
    echo "curl is not installed. Installing..."
    apt-get update && apt-get install -y curl
fi

# Check if the process is running
check_process() {
    if pgrep -f "python.*quickinfo.py" > /dev/null; then
        echo "Bot process is running"
        return 0
    else
        echo "Bot process is not running"
        return 1
    fi
}

# Check the health endpoint
check_endpoint() {
    for i in $(seq 1 $MAX_RETRIES); do
        echo "Checking health endpoint (Attempt $i of $MAX_RETRIES)..."
        response=$(curl -s -o /dev/null -w "%{http_code}" http://127.0.0.1:$PORT/kaithheathcheck)
        if [ "$response" = "200" ]; then
            echo "Health endpoint check passed!"
            return 0
        fi
        echo "Attempt $i failed. Retrying in $RETRY_INTERVAL seconds..."
        sleep $RETRY_INTERVAL
    done
    echo "Health endpoint check failed after $MAX_RETRIES attempts"
    return 1
}

# Check MongoDB connection
check_mongodb() {
    if nc -z -w5 $(echo $MONGODB_URL | cut -d'@' -f2 | cut -d'/' -f1 | cut -d':' -f1) \
                  $(echo $MONGODB_URL | cut -d'@' -f2 | cut -d'/' -f1 | cut -d':' -f2); then
        echo "MongoDB connection check passed!"
        return 0
    else
        echo "MongoDB connection check failed!"
        return 1
    fi
}

# Main health check function
main_health_check() {
    # Check process
    if ! check_process; then
        exit 1
    fi

    # Check endpoint
    if ! check_endpoint; then
        exit 1
    fi

    # Check MongoDB
    if ! check_mongodb; then
        exit 1
    fi

    echo "All health checks passed successfully!"
    exit 0
}

# Execute the main health check
main_health_check