/* -------------------------------------------

  Copyright (C) 2025, Amlal El Mahrouss, all rights reserved.

------------------------------------------- */

#include <tools/libmkfs/hefs.h>
#include <tools/libmkfs/mkfs.h>
#include <cstdlib>
#include <fstream>

static uint16_t kNumericalBase = 10;

int main(int argc, char** argv) {
  if (argc < 2) {
    mkfs::console_out() << "fsck: hefs: usage: fsck.hefs -i=<input_device> -o=<origin>" << "\n";
    return EXIT_FAILURE;
  }

  auto args = mkfs::detail::build_args(argc, argv);

  auto opt_disk = mkfs::get_option<char>(args, "-i");

  auto origin = mkfs::get_option<char>(args, "-o");

  if (opt_disk.empty()) {
    mkfs::console_out() << "fsck: hefs: error: HeFS is empty! Exiting..." << "\n";
    return EXIT_FAILURE;
  }

  auto out_origin = 0L;

  if (!mkfs::detail::parse_signed(origin, out_origin, kNumericalBase)) {
    mkfs::console_out() << "hefs: error: Invalid -o <dec> argument.\n";
    return EXIT_FAILURE;
  }

  std::ifstream output_device(opt_disk, std::ios::binary);

  if (!output_device.good()) {
    mkfs::console_out() << "hefs: error: Unable to open output device: " << opt_disk << "\n";
    return EXIT_FAILURE;
  }

  output_device.seekg(out_origin);

  if (!output_device.good()) {
    mkfs::console_out() << "hefs: error: Failed seek to origin.\n";
    return EXIT_FAILURE;
  }

  mkfs::hefs::BootNode boot_node;

  std::memset(&boot_node, 0, sizeof(boot_node));

  output_device.read(reinterpret_cast<char*>(&boot_node), sizeof(boot_node));

  if (strncmp(boot_node.magic, kHeFSMagic, kHeFSMagicLen) != 0 || boot_node.sectorCount < 1 ||
      boot_node.sectorSize < kMkFsSectorSz) {
    mkfs::console_out() << "hefs: error: Device is not an HeFS disk: " << opt_disk << "\n";
    return EXIT_FAILURE;
  }

  if (boot_node.badSectors > kMkFsMaxBadSectors) {
    mkfs::console_out() << "hefs: error: HeFS disk has too much bad sectors: " << opt_disk << "\n";
    return EXIT_FAILURE;
  }

  mkfs::console_out() << "hefs: HeFS partition is is healthy, exiting...\n";

  output_device.close();

  return EXIT_SUCCESS;
}