const std = @import("std");

pub fn build(b: *std.Build) void {
    const target = b.standardTargetOptions(.{});
    const optimize = b.standardOptimizeOption(.{});

    const triple = target.result;
    if (triple.os.tag != .linux or triple.cpu.arch != .x86_64) {
        std.log.warn("The build file only supports Linux x86-64 right now.\n", .{});
        std.log.warn("You can extend the build.zig to call the appropriate platform scripts in 3rd/libomtnet/build/ and 3rd/libomt/build.\n", .{});
    }

    // Build libomtnet
    const libomtnet_build = b.addSystemCommand(&.{
        "bash",
        "buildall.sh",
    });
    libomtnet_build.cwd = b.path("3rd/libomtnet/build");

    // Build libomt
    const libomt_build = b.addSystemCommand(&.{
        "bash",
        "buildlinuxx64.sh",
    });
    libomt_build.cwd = b.path("3rd/libomt/build");
    libomt_build.step.dependOn(&libomtnet_build.step);

    // Define executable
    const sender_exe = b.addExecutable(.{
        .name = "omtoy-sender",
        .root_module = b.createModule(.{
            .root_source_file = b.path("src/sender.zig"),
            .target = target,
            .optimize = optimize,
        }),
    });

    const receiver_exe = b.addExecutable(.{
        .name = "omtoy-receiver",
        .root_module = b.createModule(.{
            .root_source_file = b.path("src/receiver.zig"),
            .target = target,
            .optimize = optimize,
        }),
    });

    // Link libomt with the sender and receiver
    {
        const omt_output_dir = b.path("3rd/libomt/bin/Release/net8.0/linux-x64/publish");
        sender_exe.addIncludePath(omt_output_dir);
        sender_exe.addLibraryPath(omt_output_dir);
        sender_exe.linkSystemLibrary("omt");
        sender_exe.linkLibC();
        receiver_exe.addIncludePath(omt_output_dir);
        receiver_exe.addLibraryPath(omt_output_dir);
        receiver_exe.linkSystemLibrary("omt");
        receiver_exe.linkLibC();
    }

    b.installArtifact(sender_exe);
    b.installArtifact(receiver_exe);

    // Link libomt with the receiver
    {
        const omt_output_dir = b.path("3rd/libomt/bin/Release/net8.0/linux-x64/publish");
        // exe.step.dependOn(&libomt_build.step);
        sender_exe.addIncludePath(omt_output_dir);
        sender_exe.addLibraryPath(omt_output_dir);
        sender_exe.linkSystemLibrary("omt");
        sender_exe.linkLibC();
    }

    // Manual build step for libomt dependencies
    const build_omt_step = b.step("build-omt", "Build libomt and libomtnet");
    build_omt_step.dependOn(&libomt_build.step);

    // Add run step for sender
    const run_sender_step = b.step("sender-run", "Run the sender client");
    const run_sender_cmd = b.addRunArtifact(sender_exe);
    run_sender_step.dependOn(&run_sender_cmd.step);

    const run_receiver_step = b.step("receiver-run", "Run the receiver client");
    const run_receiver_cmd = b.addRunArtifact(receiver_exe);
    run_receiver_step.dependOn(&run_receiver_cmd.step);

    // Add run step for receiver

    run_sender_cmd.step.dependOn(b.getInstallStep());
    run_receiver_cmd.step.dependOn(b.getInstallStep());

    if (b.args) |args| {
        run_sender_cmd.addArgs(args);
        run_receiver_cmd.addArgs(args);
    }

    // TODO: write tests and stuff
    // const exe_tests = b.addTest(.{
    //     .root_module = sender_exe.root_module,
    // });

    // const run_exe_tests = b.addRunArtifact(exe_tests);

    // const test_step = b.step("test", "Run tests");
    // test_step.dependOn(&run_exe_tests.step);
}