#include "fs/ext2/ext2_block_reader.h"

#include "mammoth/debug.h"

glcr::ErrorOr<glcr::SharedPtr<Ext2BlockReader>> Ext2BlockReader::Init(
    const DenaliInfo& denali_info) {
  // Read 1024 bytes from 1024 offset.
  // FIXME: Don't assume 512 byte sectors somehow.
  DenaliClient client(denali_info.denali_endpoint());
  ReadRequest req;
  req.set_device_id(denali_info.device_id());
  req.set_lba(denali_info.lba_offset() + 2);
  req.set_size(2);
  ReadResponse resp;
  RET_ERR(client.Read(req, resp));
  MappedMemoryRegion superblock =
      MappedMemoryRegion::FromCapability(resp.memory());

  return glcr::SharedPtr<Ext2BlockReader>(
      new Ext2BlockReader(glcr::Move(client), denali_info.device_id(),
                          denali_info.lba_offset(), superblock));
}

Superblock* Ext2BlockReader::GetSuperblock() {
  return reinterpret_cast<Superblock*>(super_block_region_.vaddr());
}

uint64_t Ext2BlockReader::SectorsPerBlock() {
  return 1 << (GetSuperblock()->log_block_size + 1);
}

uint64_t Ext2BlockReader::BlockSize() {
  return 1024 << (GetSuperblock()->log_block_size);
}

uint64_t Ext2BlockReader::NumberOfBlockGroups() {
  return ((GetSuperblock()->blocks_count - 1) /
          GetSuperblock()->blocks_per_group) +
         1;
}

uint64_t Ext2BlockReader::BgdtBlockNum() {
  return (BlockSize() == 1024) ? 2 : 1;
}

uint64_t Ext2BlockReader::BgdtBlockSize() {
  return ((NumberOfBlockGroups() * sizeof(BlockGroupDescriptor) - 1) /
          BlockSize()) +
         1;
}

uint64_t Ext2BlockReader::InodeSize() {
  constexpr uint64_t kDefaultInodeSize = 0x80;
  return GetSuperblock()->rev_level >= 1 ? GetSuperblock()->inode_size
                                         : kDefaultInodeSize;
}

uint64_t Ext2BlockReader::InodeTableBlockSize() {
  return (InodeSize() * GetSuperblock()->inodes_per_group) / BlockSize();
}

glcr::ErrorOr<MappedMemoryRegion> Ext2BlockReader::ReadBlock(
    uint64_t block_number) {
  return ReadBlocks(block_number, 1);
}
glcr::ErrorOr<MappedMemoryRegion> Ext2BlockReader::ReadBlocks(
    uint64_t block_number, uint64_t num_blocks) {
  ReadRequest req;
  req.set_device_id(device_id_);
  req.set_lba(lba_offset_ + block_number * SectorsPerBlock());
  req.set_size(num_blocks * SectorsPerBlock());
  ReadResponse resp;
  RET_ERR(denali_.Read(req, resp));
  return MappedMemoryRegion::FromCapability(resp.memory());
}

glcr::ErrorOr<MappedMemoryRegion> Ext2BlockReader::ReadBlocks(
    const glcr::Vector<uint64_t>& block_list) {
  ReadManyRequest req;
  req.set_device_id(device_id_);
  // FIXME: We should have better ergonomics for setting a repeated field in
  // Yunq.
  for (uint64_t i = 0; i < block_list.size(); i++) {
    uint64_t sector = lba_offset_ + block_list.at(i) * SectorsPerBlock();
    for (uint64_t j = 0; j < SectorsPerBlock(); j++) {
      req.add_lba(sector + j);
    }
  }
  ReadResponse resp;
  RET_ERR(denali_.ReadMany(req, resp));
  return MappedMemoryRegion::FromCapability(resp.memory());
}

Ext2BlockReader::Ext2BlockReader(DenaliClient&& denali, uint64_t device_id,
                                 uint64_t lba_offset,
                                 MappedMemoryRegion super_block)
    : denali_(glcr::Move(denali)),
      device_id_(device_id),
      lba_offset_(lba_offset),
      super_block_region_(super_block) {}