SConstruct

import os

dirs = [
    "boot",
    "kernel",
]

args = Variables()
args.Add(EnumVariable("build", "Set build type", "debug", allowed_values=["debug", "release"]))
args.Add(BoolVariable("framepointers", "Force frame pointers for stack unwinding in panics", 1))
args.Add(BoolVariable("headless", "Run headless in qemu", 0))

env = Environment(ENV=os.environ, variables=args)
Export("env")

for d in dirs:
    build_dir = os.path.join("build", d)
    SConscript(os.path.join(d, "SConscript"), variant_dir=build_dir, duplicate=0)

Import("kernel_lib", "boot_objs")

env["LINKFLAGS"] = ["-T", "linker.ld", "-n", "-g", "--Map", "./build/symbols.map"]
env["CC"] = "ld"  # awful but this has taken long enough

build = env.Program(["build/iso/boot/DomeOS", "build/symbols.map"], boot_objs, LIBS=[kernel_lib])
env.Depends(build, ["linker.ld", "SConstruct"])
domeos = env.Command("build/symbols.bin", ["build/symbols.map", build], [
    # build packed link map
    "cargo run --manifest-path kernel/helpers/ld-link-map/Cargo.toml --release --bin create-packed build/symbols.map build/symbols.bin",

    # patch in packed link map
    "cargo run --manifest-path kernel/helpers/patcher/Cargo.toml --release build/iso/boot/DomeOS build/symbols.bin",
])


# create grub structure and make iso
def mk_grub(env, target, source):
    os.makedirs("build/iso/boot/grub", exist_ok=True)
    grub = \
        """set timeout=0
        set default=0
        menuentry "domeos" {
            multiboot /boot/DomeOS some args in grub.cfg
            boot
        }"""

    with open("build/iso/boot/grub/grub.cfg", "w") as f:
        f.write(grub)

    env.Execute("grub-mkrescue -o build/DomeOS.iso build/iso")


iso = env.Command("build/DomeOS.iso", [domeos, build, "build/boot"], action=mk_grub)
Default(iso)  # only build this by default
Clean(iso, "build")  # delete whole build dir on clean

# run command
qemu_cmd = "qemu-system-x86_64 -cdrom build/DomeOS.iso -monitor stdio -serial file:serial.log -d cpu_reset,int -D qemu-logfile -no-reboot"
if "debug" in COMMAND_LINE_TARGETS:
    qemu_cmd += " -s -S"
if env["headless"]:
    qemu_cmd += " -nographic"


def PhonyTarget(target, action):
    env = Environment(ENV=os.environ, BUILDERS={"phony": Builder(action=action)})
    phony = env.phony(target=target, source="SConstruct")
    AlwaysBuild(phony)
    Requires(phony, iso)


run_qemu = PhonyTarget(["run", "debug"], qemu_cmd)
env.Depends(run_qemu, iso)


# really crappy host testing
def host_tests(target, source, env):
    modules = [
        "kernel/kernel",
        ("kernel/memory", ["--features", "std"]),
        ("kernel/common", ["--features", "std"]),
        "kernel/userspace/syscall",
        "kernel/helpers/ld-link-map",
        "kernel/helpers/patcher"]

    # im so fed up of trying to fit test discovery and building of tests into scons' sick model that im doing it
    # all right here right now, fk it

    import subprocess
    for module in modules:
        if isinstance(module, tuple):
            module, extra_args = module
        else:
            extra_args = []
        subprocess.run(
            ["cargo", "test", "-Zbuild-std", "--manifest-path", "Cargo.toml", "--target", "x86_64-unknown-linux-gnu", *extra_args],
            cwd=module, check=True)

    return []


tests_host = env.Command("test", [], action=host_tests)
env.AlwaysBuild(tests_host)