"""Проверки состояния для Docker HEALTHCHECK: `healthcheck.py api` или `healthcheck.py collector`."""
import datetime
import json
import sys
import urllib.request

import cidr_collector as cc


def check_api(url="http://127.0.0.1:8000/health"):
    with urllib.request.urlopen(url, timeout=5) as response:
        return response.status == 200


def check_collector():
    """Демон жив, пока heartbeat в status.json не старше STATUS_STALE_AFTER."""
    with open(cc.STATUS_FILE) as f:
        updated_at = datetime.datetime.fromisoformat(json.load(f)["updated_at"])
    return (datetime.datetime.now() - updated_at).total_seconds() < cc.STATUS_STALE_AFTER


CHECKS = {"api": check_api, "collector": check_collector}


if __name__ == "__main__":
    target = sys.argv[1] if len(sys.argv) > 1 else ""
    try:
        ok = CHECKS[target]()
    except Exception as e:
        print(f"{target} unhealthy: {e}")
        ok = False
    sys.exit(0 if ok else 1)