#!/usr/bin/env python3
"""Health check for custom-html."""
import argparse
import os
import sys

sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..', '..'))
from utils.tests.helpers import http_get, resolve_domain


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--domain', default=os.environ.get('TEST_DOMAIN'))
    args = parser.parse_args()
    domain = args.domain or resolve_domain('custom-html')
    url = f"https://{domain}"

    print(f"Checking custom-html at {url} ...")
    status, _ = http_get(url)
    if status == 200:
        print(f"PASS: custom-html returned HTTP {status}")
    else:
        print(f"FAIL: custom-html returned HTTP {status} (expected 200)")
        sys.exit(1)


if __name__ == '__main__':
    main()