# |$|*file*packdisk.py*| |$|*noclean*|
# |$|*artifact*disk.bin*packdisk.py&kernel.bin*"python3 {artifact0} {file} {artifact1}"*|
from sys import argv
import struct, math
# USAGE: python3 packdisk.py <disk-location> <kernel.bin>
assert len(argv) == 3
print(argv)

f_disk = open(argv[1], "wb")

f_kernel = open(argv[2], "rb")

f_kernel.seek(0, 2)
kernel_size = f_kernel.tell()
kernel_sector_count = math.ceil(kernel_size / 512)
print(kernel_size, kernel_sector_count)
f_kernel.seek(0, 0)

# write the first sector
"""
struct sec_1 {
    uint32_t kern_lba;
    uint32_t kern_block_count;
    void*(64) kern_addr;
    padding to 512;
}
"""
#                                    HARDCODED
#                                    VV    VV
sec_1 = struct.pack("<IIQ"+("x"*(512-16)), 8, kernel_sector_count, 0x7B00)

f_disk.write(sec_1)
f_disk.write(f_kernel.read(kernel_size))
padding_count = 512 - (kernel_size%512)
f_disk.write(struct.pack("x"*padding_count))
f_kernel.close()
f_disk.close()