import contextlib
import logging
import subprocess
import time

from .config import *


@contextlib.contextmanager
def boot_image(file_name: 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 + [f"{OUTPUT_DIR}/{file_name}"] + 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()


@contextlib.contextmanager
def extract_image(file_name: str):
    extract_dir = tempfile.mkdtemp(prefix="osbuild-")
    subprocess.run(["tar", "xf", f"{OUTPUT_DIR}/{file_name}"], cwd=extract_dir, check=True)
    try:
        yield extract_dir
    finally:
        # Clean up?
        pass