import os
import time

import requests

domain_map = {"OLD": os.getenv("OLD_DOMAIN"), "NEW": os.getenv("NEW_DOMAIN")}


def do_check(domain):
    url = "http://{}/status/200".format(domain)
    try:
        resp = requests.get(url, timeout=3, headers={"kong-debug": "1"})
    except requests.RequestException as e:
        return False, "HTTP_CALL_EXCEPTION"
    if resp.status_code != 200:
        return False, "UNEXPECTED_STATUS_CODE"
    if resp.headers.get("Kong-Route-Name") != "kong.go-httpbin-ingress.00":
        return False, "UNEXPECTED_HEADER"
    return True, ""


while True:
    for key, domain in domain_map.items():
        success, code = do_check(domain)
        if success:
            print("\033[0;36;40m{} {} SUCCESS\033[0m".format(key, domain))
        else:
            print("\033[0;32;40m{} {} {}\033[0m".format(key, domain, code))
    time.sleep(1)
    print()