const std = @import("std");
const Build = std.Build;

const MicroZig = @import("build/definitions");

const example_dep_names: []const []const u8 = &.{
    // "examples/nordic/nrf5x",
    // "examples/nxp/lpc",
    // "examples/microchip/atsam",
    //"examples/microchip/avr",
    // "examples/gigadevice/gd32",
    "examples/stmicro/stm32",
    //"examples/espressif/esp",
    // "examples/raspberrypi/rp2040",
    "examples/wch/ch32",
};

const bsps = .{
    // .{ "bsp/nordic/nrf5x", @import("bsp/nordic/nrf5x") },
    // .{ "bsp/nxp/lpc", @import("bsp/nxp/lpc") },
    // .{ "bsp/microchip/atsam", @import("bsp/microchip/atsam") },
    // .{ "bsp/microchip/avr", @import("bsp/microchip/avr") },
    // .{ "bsp/gigadevice/gd32", @import("bsp/gigadevice/gd32") },
    .{ "bsp/stmicro/stm32", @import("bsp/stmicro/stm32") },
    // .{ "bsp/espressif/esp", @import("bsp/espressif/esp") },
    // .{ "bsp/raspberrypi/rp2040", @import("bsp/raspberrypi/rp2040") },
    .{ "bsp/wch/ch32", @import("bsp/wch/ch32") },
};

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

    // Build all examples
    for (example_dep_names) |example_dep_name| {
        const example_dep = b.dependency(example_dep_name, .{
            .optimize = optimize,
        });

        const example_dep_install_step = example_dep.builder.getInstallStep();
        b.getInstallStep().dependOn(example_dep_install_step);
    }

    const boxzer_dep = b.dependency("boxzer", .{});
    const boxzer_exe = boxzer_dep.artifact("boxzer");
    const boxzer_run = b.addRunArtifact(boxzer_exe);
    if (b.args) |args|
        boxzer_run.addArgs(args);

    const package_step = b.step("package", "Package monorepo using boxzer");
    package_step.dependOn(&boxzer_run.step);

    //const website_dep = b.dependency("website", .{});
    //const website_step = b.step("website", "Build website");
    //website_step.dependOn(website_dep.builder.getInstallStep());

    const parts_db = generate_parts_db(b) catch @panic("OOM");
    const parts_db_json = b.addInstallFile(parts_db, "parts-db.json");
    package_step.dependOn(&parts_db_json.step);

    const test_bsps_step = b.step("run-bsp-tests", "Run all platform agnostic tests for BSPs");
    inline for (bsps) |bsp| {
        const bsp_dep = b.dependency(bsp[0], .{});
        test_bsps_step.dependOn(&bsp_dep.builder.top_level_steps.get("test").?.step);
    }
}

const PartsDb = struct {
    chips: []const Chip,
    boards: []const Board,

    const Chip = struct {
        identifier: []const u8,
        bsp_package: []const u8,
        url: ?[]const u8,
        cpu: []const u8,
        has_hal: bool,
        memory: struct {
            flash: u64,
            ram: u64,
        },
        output_format: ?[]const u8,
    };
    const Board = struct {
        identifier: []const u8,
        bsp_package: []const u8,
        chip_idx: u32,
        url: ?[]const u8,
        output_format: ?[]const u8,
    };
};

fn generate_parts_db(b: *Build) !Build.LazyPath {
    var chips = std.ArrayList(PartsDb.Chip).init(b.allocator);
    var boards = std.ArrayList(PartsDb.Board).init(b.allocator);
    inline for (bsps) |bsp| {
        const chips_start_idx = chips.items.len;
        inline for (@typeInfo(@field(bsp[1], "chips")).Struct.decls) |decl| {
            const target = @field(@field(bsp[1], "chips"), decl.name);
            try chips.append(.{
                .identifier = decl.name,
                .bsp_package = bsp[0],
                .url = target.chip.url,
                .cpu = target.chip.cpu.name,
                .has_hal = target.hal != null,
                .memory = .{
                    .flash = 0,
                    .ram = 0,
                },
                .output_format = null,
            });
        }

        inline for (@typeInfo(@field(bsp[1], "boards")).Struct.decls) |decl| {
            const target = @field(@field(bsp[1], "boards"), decl.name);
            _ = target;
            _ = chips_start_idx;

            //const chip_idx = inline for (@typeInfo(@field(bsp[1], "chips")).Struct.decls, 0..) |chip_decl, idx| {
            //    const chip = @field(@field(bsp[1], "chips"), chip_decl.name);
            //    if (std.mem.eql(u8, chip.chip.name, target.chip.name))
            //        break chips_start_idx + idx;
            //} else @compileError("failed to get chip_idx");

            try boards.append(.{
                .identifier = decl.name,
                .bsp_package = bsp[0],
                .chip_idx = 0,
                .url = "",
                .output_format = null,
            });
        }
    }

    const parts_db = PartsDb{
        .chips = chips.items,
        .boards = boards.items,
    };

    const parts_db_str = try std.json.stringifyAlloc(b.allocator, parts_db, .{
        .whitespace = .indent_4,
    });
    const write_file_step = b.addWriteFiles();
    return write_file_step.add("parts-db.json", parts_db_str);
}