96 lines
3 KiB
Python
96 lines
3 KiB
Python
#!/usr/bin/env python3
|
|
import json
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import urllib.request
|
|
import urllib.error
|
|
from datetime import datetime, timezone
|
|
from pathlib import Path
|
|
|
|
USER_NAME = "Ronaldo Freitas Dias"
|
|
|
|
|
|
def load_config():
|
|
config_file = Path.home() / ".config" / "health-monitor"
|
|
config = {}
|
|
if config_file.exists():
|
|
for line in config_file.read_text().splitlines():
|
|
if "=" in line:
|
|
key, value = line.split("=", 1)
|
|
config[key.strip()] = value.strip()
|
|
return config
|
|
|
|
|
|
def ping(url: str, body: bytes, timeout: int) -> int:
|
|
req = urllib.request.Request(
|
|
url, data=body, method="POST",
|
|
headers={"User-Agent": "health-monitor/1.0", "Content-Type": "application/json"}
|
|
)
|
|
with urllib.request.urlopen(req, timeout=timeout) as response:
|
|
return response.getcode()
|
|
|
|
|
|
def ping_fail(url: str, body: bytes, timeout: int) -> None:
|
|
fail_url = url.rstrip("/") + "/fail"
|
|
try:
|
|
req = urllib.request.Request(
|
|
fail_url, data=body, method="POST",
|
|
headers={"User-Agent": "health-monitor/1.0", "Content-Type": "application/json"}
|
|
)
|
|
urllib.request.urlopen(req, timeout=timeout)
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
def main():
|
|
config = load_config()
|
|
url = sys.argv[1] if len(sys.argv) > 1 else os.environ.get("HEALTHCHECK_URL") or config.get("HEALTHCHECK_URL")
|
|
if not url:
|
|
print("ERROR: HEALTHCHECK_URL not set", file=sys.stderr)
|
|
return 1
|
|
|
|
device = os.environ.get("DEVICE_NAME") or config.get("DEVICE_NAME", "unknown")
|
|
timeout = int(os.environ.get("HEALTHCHECK_TIMEOUT", 10))
|
|
try:
|
|
ips = subprocess.check_output(["hostname", "-I"], timeout=5).decode().split()
|
|
except Exception:
|
|
ips = []
|
|
|
|
try:
|
|
seconds = int(float(Path("/proc/uptime").read_text().split()[0]))
|
|
d, rem = divmod(seconds, 86400)
|
|
h, rem = divmod(rem, 3600)
|
|
m = rem // 60
|
|
uptime = f"{d}d {h}h {m}m"
|
|
except Exception:
|
|
uptime = "unknown"
|
|
|
|
body = json.dumps({"user": USER_NAME, "device": device, "ips": ips, "uptime": uptime}).encode()
|
|
|
|
try:
|
|
status = ping(url, body, timeout)
|
|
if 200 <= status < 300:
|
|
print(f"{datetime.now(timezone.utc).isoformat()} status={status} device={device} url={url}")
|
|
return 0
|
|
|
|
print(f"ERROR: HTTP {status} url={url}", file=sys.stderr)
|
|
ping_fail(url, body, timeout)
|
|
return 2
|
|
|
|
except urllib.error.HTTPError as exc:
|
|
print(f"HTTPError {exc.code} {exc.reason} url={url}", file=sys.stderr)
|
|
ping_fail(url, body, timeout)
|
|
return 3
|
|
except urllib.error.URLError as exc:
|
|
print(f"URLError {exc.reason} url={url}", file=sys.stderr)
|
|
ping_fail(url, body, timeout)
|
|
return 4
|
|
except Exception as exc:
|
|
print(f"ERROR {exc} url={url}", file=sys.stderr)
|
|
ping_fail(url, body, timeout)
|
|
return 5
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|