import time, uos


class RAMBlockDevice:
    def __init__(self, block_size, num_blocks):
        self.block_size = block_size
        self.data = bytearray(num_blocks * self.block_size)

    def readblocks(self, block, buf, off=0):
        addr = block * self.block_size + off
        for i in range(len(buf)):
            buf[i] = self.data[addr + i]

    def writeblocks(self, block, buf, off=None):
        if off is None:
            # do erase, then write
            for i in range(len(buf) // self.block_size):
                self.ioctl(6, block + i)
            off = 0
        addr = block * self.block_size + off
        self.data[addr:addr + len(buf)] = buf

    def ioctl(self, op, arg):
        if op == 4:  # block count
            return len(self.data) // self.block_size
        if op == 5:  # block size
            return self.block_size
        if op == 6:  # erase block
            return 0

print(time.time())
print(time.localtime(time.time()))

bdev1 = RAMBlockDevice(512, 64)
uos.VfsFat.mkfs(bdev1)
uos.mount(uos.VfsFat(bdev1), "/fat")

bdev2 = RAMBlockDevice(256, 16)
uos.VfsLfs2.mkfs(bdev2)
uos.mount(uos.VfsLfs2(bdev2), "/lfs")

with open("/fat/test", "w") as f:
    f.write("hello")

with open("/lfs/test", "w") as f:
    f.write("hello")

s = uos.stat("/fat/test")
print(s)
print(time.localtime(s[-1]))

s = uos.stat("/lfs/test")
print(s)
print(time.localtime(s[-1]))