import contextlib
import logging
import os
import subprocess
import sys
import tempfile
import time

EXPECTED_TIME_TO_BOOT = 60  # seconds
RESET = "\033[0m"
BOLD = "\033[1m"
RED = "\033[31m"
OBJECTS = os.environ.get("OBJECTS", tempfile.mkdtemp(prefix="osbuild-"))
OUTPUT_DIR = os.environ.get("OUTPUT_DIR", tempfile.mkdtemp(prefix="osbuild-"))
OSBUILD = os.environ.get("OSBUILD", "osbuild")
IMAGE_PATH = os.environ.get("IMAGE_PATH", OUTPUT_DIR + "/base.qcow2")


logging.basicConfig(level=logging.getLevelName(os.environ.get("TESTS_LOGLEVEL", "INFO")))


def run_osbuild(pipeline: str, check=True):
    cmd = [OSBUILD, "--objects", OBJECTS, "-o", OUTPUT_DIR, pipeline]
    logging.info(f"Running osbuild: {cmd}")
    osbuild = subprocess.run(cmd, capture_output=True)
    if osbuild.returncode != 0:
        logging.error(f"{RED}osbuild failed!{RESET}")
        print(f"{BOLD}STDERR{RESET}")
        print(osbuild.stderr.decode())
        print(f"{BOLD}STDOUT{RESET}")
        print(osbuild.stdout.decode())
        if check:
            sys.exit(1)

    return osbuild.returncode


def rel_path(fname: str) -> str:
    return os.path.join(os.path.dirname(__file__), fname)


def build_web_server_image():
    run_osbuild(rel_path("4-all.json"))
    # run_osbuild(rel_path("1-create-base.json"))
    # r = run_osbuild(rel_path("2-configure-web-server.json"), check=False)
    # # TODO: remove this workaround once the grub2 stage works on the first try :-)
    # if r != 0:
    #     run_osbuild(rel_path("2-configure-web-server.json"))
    # run_osbuild(rel_path("3-compose-qcow2.json"))


def uname() -> str:
    uname = subprocess.run(["uname"], capture_output=True)
    return uname.stdout.decode("utf-8").strip()


@contextlib.contextmanager
def boot_image(path: str):
    acceleration = ["-accel", "kvm:hvf:tcg"]
    network = ["-net", "nic,model=rtl8139", "-net", "user,hostfwd=tcp::8888-:8888"]
    cmd = ["qemu-system-x86_64", "-nographic", "-m", "1024", "-snapshot"] + acceleration + [path] + network
    logging.info(f"Booting image: {cmd}")
    vm = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    try:
        time.sleep(EXPECTED_TIME_TO_BOOT)
        yield None
    finally:
        vm.kill()


def test_web_server():
    cmd = ["curl", "-s", "http://127.0.0.1:8888/index"]
    logging.info(f"Running curl: {cmd}")
    curl = subprocess.run(cmd, capture_output=True)
    logging.info(f"Curl returned: code={curl.returncode}, stdout={curl.stdout.decode()}, stderr={curl.stderr.decode()}")
    assert curl.returncode == 0
    assert curl.stdout.decode("utf-8").strip() == "hello, world!"


if __name__ == '__main__':
    if uname() == "Darwin":
        logging.info("Running on macOS. Skipping image build.")
    else:
        logging.info("Building image")
        build_web_server_image()

    logging.info("Running tests")
    tests = [test_web_server]
    with boot_image(IMAGE_PATH):
        for test in tests:
            try:
                test()
                print(f"{RESET}{BOLD}{test.__name__}: Success{RESET}")
            except AssertionError as e:
                print(f"{RESET}{BOLD}{test.__name__}: {RESET}{RED}Fail{RESET}")
                print(e)