106 lines
3.2 KiB
Bash
Executable File
106 lines
3.2 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# hybrid-health.sh
|
|
# Checks the health of a running opendataloader-pdf hybrid server.
|
|
# Works on Windows (Git Bash), macOS, and Linux.
|
|
# Outputs key=value pairs for machine readability.
|
|
#
|
|
# Vendored verbatim from the opendataloader-pdf upstream agent skill
|
|
# (skills/odl-pdf/scripts/hybrid-health.sh), Apache-2.0.
|
|
|
|
set -euo pipefail
|
|
|
|
DEFAULT_URL="http://localhost:5002"
|
|
HYBRID_URL="${DEFAULT_URL}"
|
|
|
|
# Parse arguments
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
--url)
|
|
if [[ $# -lt 2 ]]; then
|
|
echo "Error: --url requires a value" >&2
|
|
exit 1
|
|
fi
|
|
HYBRID_URL="$2"
|
|
shift 2
|
|
;;
|
|
--url=*)
|
|
HYBRID_URL="${1#--url=}"
|
|
shift
|
|
;;
|
|
*)
|
|
echo "Unknown argument: $1" >&2
|
|
echo "Usage: $0 [--url <url>]" >&2
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
# Validate the URL before use: reject empty or malformed values.
|
|
# Require the form http(s)://host[:port] (optional trailing slash; no path).
|
|
if [[ -z "${HYBRID_URL}" ]]; then
|
|
echo "Error: --url must not be empty" >&2
|
|
echo "Usage: $0 [--url <http(s)://host[:port]>]" >&2
|
|
exit 1
|
|
fi
|
|
# Note the excluded '@': a URL with userinfo (https://user:pass@host) is rejected
|
|
# so credentials are never echoed back to stdout.
|
|
if [[ ! "${HYBRID_URL}" =~ ^https?://[^[:space:]/@]+(:[0-9]+)?/?$ ]]; then
|
|
if [[ "${HYBRID_URL}" == *@* ]]; then
|
|
echo "Error: --url must not contain embedded credentials (userinfo '@'); pass a plain host[:port]" >&2
|
|
else
|
|
echo "Error: --url must be of the form http(s)://host[:port] (got: '${HYBRID_URL}')" >&2
|
|
fi
|
|
echo "Usage: $0 [--url <http(s)://host[:port]>]" >&2
|
|
exit 1
|
|
fi
|
|
|
|
HEALTH_ENDPOINT="${HYBRID_URL%/}/health"
|
|
|
|
# Detect available HTTP client
|
|
_http_get_status() {
|
|
local url="$1"
|
|
if command -v curl &>/dev/null; then
|
|
curl --silent --output /dev/null --write-out "%{http_code}" \
|
|
--max-time 5 --connect-timeout 3 "$url" 2>/dev/null
|
|
elif command -v wget &>/dev/null; then
|
|
wget --quiet --server-response --spider --timeout=5 "$url" 2>&1 \
|
|
| awk '/HTTP\//{print $2}' | tail -1
|
|
else
|
|
echo "none"
|
|
fi
|
|
}
|
|
|
|
HTTP_STATUS=$(_http_get_status "${HEALTH_ENDPOINT}" || true)
|
|
|
|
# No HTTP client available to probe — this is NOT "server stopped"; the check
|
|
# could not run at all. Report a distinct state so callers don't misread it.
|
|
if [[ "${HTTP_STATUS}" == "none" ]]; then
|
|
echo "HYBRID_SERVER=error"
|
|
echo "HYBRID_URL=${HYBRID_URL}"
|
|
echo "HYBRID_STATUS=client-missing"
|
|
echo ""
|
|
echo "Cannot probe the hybrid server: no HTTP client (curl or wget) is available. Install one, or check the server manually."
|
|
exit 0
|
|
fi
|
|
|
|
# Interpret result
|
|
if [[ -z "${HTTP_STATUS}" || "${HTTP_STATUS}" == "000" ]]; then
|
|
echo "HYBRID_SERVER=stopped"
|
|
echo "HYBRID_URL=${HYBRID_URL}"
|
|
echo "HYBRID_STATUS=none"
|
|
echo ""
|
|
echo "Hybrid server is not running at ${HYBRID_URL}. Start it with: opendataloader-pdf-hybrid"
|
|
exit 0
|
|
fi
|
|
|
|
# The script always exits 0 (a completed health probe is not itself a failure).
|
|
# The result is on stdout: HYBRID_SERVER=running means reachable; stopped/error
|
|
# mean not usable. Callers must branch on that value, NOT on the exit code.
|
|
if [[ "${HTTP_STATUS}" =~ ^2 ]]; then
|
|
echo "HYBRID_SERVER=running"
|
|
else
|
|
echo "HYBRID_SERVER=error"
|
|
fi
|
|
|
|
echo "HYBRID_URL=${HYBRID_URL}"
|
|
echo "HYBRID_STATUS=${HTTP_STATUS}" |