const std = @import("std");
const Io = std.Io;
const Md5 = std.crypto.hash.Md5;

const trace = @import("../../tracy.zig").trace;
const Compilation = @import("../../Compilation.zig");
const ParallelHasher = @import("hasher.zig").ParallelHasher;

/// Calculates Md5 hash of each chunk in parallel and then hashes all Md5 hashes to produce
/// the final digest.
/// While this is NOT a correct MD5 hash of the contents, this methodology is used by LLVM/LLD
/// and we will use it too as it seems accepted by Apple OSes.
/// TODO LLD also hashes the output filename to disambiguate between same builds with different
/// output files. Should we also do that?
pub fn calcUuid(comp: *const Compilation, file: Io.File, file_size: u64, out: *[Md5.digest_length]u8) !void {
    const tracy = trace(@src());
    defer tracy.end();

    const gpa = comp.gpa;
    const io = comp.io;

    const chunk_size: usize = 1024 * 1024;
    const num_chunks: usize = std.math.cast(usize, @divTrunc(file_size, chunk_size)) orelse return error.Overflow;
    const actual_num_chunks = if (@rem(file_size, chunk_size) > 0) num_chunks + 1 else num_chunks;

    const hashes = try gpa.alloc([Md5.digest_length]u8, actual_num_chunks);
    defer gpa.free(hashes);

    try ParallelHasher(Md5).hash(gpa, io, file, hashes, .{
        .chunk_size = chunk_size,
        .max_file_size = file_size,
    });

    const final_buffer = try gpa.alloc(u8, actual_num_chunks * Md5.digest_length);
    defer gpa.free(final_buffer);

    for (hashes, 0..) |hash, i| {
        @memcpy(final_buffer[i * Md5.digest_length ..][0..Md5.digest_length], &hash);
    }

    Md5.hash(final_buffer, out, .{});
    conform(out);
}

inline fn conform(out: *[Md5.digest_length]u8) void {
    // LC_UUID uuids should conform to RFC 4122 UUID version 4 & UUID version 5 formats
    out[6] = (out[6] & 0x0F) | (3 << 4);
    out[8] = (out[8] & 0x3F) | 0x80;
}