99 lines
2.7 KiB
Bash
99 lines
2.7 KiB
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
# Health check script for the Home Server Agent
|
||
|
|
# This script can be used with monitoring tools like Nagios, Zabbix, etc.
|
||
|
|
|
||
|
|
AGENT_URL="${AGENT_URL:-http://localhost:3000}"
|
||
|
|
API_TOKEN="${API_TOKEN:-}"
|
||
|
|
TIMEOUT="${TIMEOUT:-10}"
|
||
|
|
|
||
|
|
# Function to check if agent is healthy
|
||
|
|
check_health() {
|
||
|
|
local response
|
||
|
|
local status_code
|
||
|
|
|
||
|
|
if [ -n "$API_TOKEN" ]; then
|
||
|
|
response=$(curl -s -w "%{http_code}" -m "$TIMEOUT" \
|
||
|
|
-H "Authorization: Bearer $API_TOKEN" \
|
||
|
|
"$AGENT_URL/health" 2>/dev/null)
|
||
|
|
else
|
||
|
|
response=$(curl -s -w "%{http_code}" -m "$TIMEOUT" \
|
||
|
|
"$AGENT_URL/health" 2>/dev/null)
|
||
|
|
fi
|
||
|
|
|
||
|
|
status_code="${response: -3}"
|
||
|
|
|
||
|
|
if [ "$status_code" = "200" ]; then
|
||
|
|
echo "OK: Home Server Agent is healthy"
|
||
|
|
return 0
|
||
|
|
else
|
||
|
|
echo "CRITICAL: Home Server Agent is not responding (HTTP $status_code)"
|
||
|
|
return 2
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Function to check game server status
|
||
|
|
check_gameserver() {
|
||
|
|
local server_name="$1"
|
||
|
|
local response
|
||
|
|
local status_code
|
||
|
|
|
||
|
|
if [ -z "$server_name" ]; then
|
||
|
|
echo "UNKNOWN: Server name not provided"
|
||
|
|
return 3
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ -z "$API_TOKEN" ]; then
|
||
|
|
echo "UNKNOWN: API_TOKEN not provided"
|
||
|
|
return 3
|
||
|
|
fi
|
||
|
|
|
||
|
|
response=$(curl -s -w "%{http_code}" -m "$TIMEOUT" \
|
||
|
|
-H "Authorization: Bearer $API_TOKEN" \
|
||
|
|
"$AGENT_URL/api/gameserver/$server_name/status" 2>/dev/null)
|
||
|
|
|
||
|
|
status_code="${response: -3}"
|
||
|
|
response_body="${response%???}"
|
||
|
|
|
||
|
|
if [ "$status_code" = "200" ]; then
|
||
|
|
# Parse JSON response to get status
|
||
|
|
status=$(echo "$response_body" | grep -o '"status":"[^"]*"' | cut -d'"' -f4)
|
||
|
|
|
||
|
|
case "$status" in
|
||
|
|
"running")
|
||
|
|
echo "OK: $server_name is running"
|
||
|
|
return 0
|
||
|
|
;;
|
||
|
|
"stopped")
|
||
|
|
echo "WARNING: $server_name is stopped"
|
||
|
|
return 1
|
||
|
|
;;
|
||
|
|
*)
|
||
|
|
echo "CRITICAL: $server_name status unknown ($status)"
|
||
|
|
return 2
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
else
|
||
|
|
echo "CRITICAL: Cannot check $server_name status (HTTP $status_code)"
|
||
|
|
return 2
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# Main execution
|
||
|
|
case "$1" in
|
||
|
|
"health")
|
||
|
|
check_health
|
||
|
|
;;
|
||
|
|
"gameserver")
|
||
|
|
check_gameserver "$2"
|
||
|
|
;;
|
||
|
|
*)
|
||
|
|
echo "Usage: $0 {health|gameserver <server_name>}"
|
||
|
|
echo "Environment variables:"
|
||
|
|
echo " AGENT_URL - URL of the Home Server Agent (default: http://localhost:3000)"
|
||
|
|
echo " API_TOKEN - Authentication token for API access"
|
||
|
|
echo " TIMEOUT - Request timeout in seconds (default: 10)"
|
||
|
|
exit 3
|
||
|
|
;;
|
||
|
|
esac
|