#include "block_reader.hpp"

#include "crypto/block/block-auto.h"
#include "block_manipulator/factory.hpp"
#include "utils.hpp"

using namespace test::fisherman;

auto main(int argc, char **argv) -> int {
  if (argc < 3) {
    std::cerr << "Usage: " << argv[0] << " /path/to/rootdb config.json\n";
    return 1;
  }

  SET_VERBOSITY_LEVEL(VERBOSITY_NAME(INFO));  // TODO: add to config

  std::string db_path = argv[1];
  std::string json_file_path = argv[2];

  auto content_res = read_file_to_buffer(json_file_path);
  if (content_res.is_error()) {
    std::cerr << "Error reading JSON file: " << content_res.error().message().str() << std::endl;
    return 1;
  }
  td::BufferSlice content = content_res.move_as_ok();

  td::Parser parser(content.as_slice());
  auto decode_result = do_json_decode(parser, 100);
  if (decode_result.is_error()) {
    std::cerr << "JSON parse error: " << decode_result.error().message().str() << std::endl;
    return 1;
  }

  auto js = decode_result.move_as_ok();
  auto &js_obj = js.get_object();
  auto blk_id_obj_res = td::get_json_object_field(js_obj, "block_id", td::JsonValue::Type::Object, false);
  CHECK(blk_id_obj_res.is_ok());
  auto blk_id_res = parse_block_id_from_json(blk_id_obj_res.move_as_ok());
  if (blk_id_res.is_error()) {
    std::cerr << "Error extracting BlockIdExt: " << blk_id_res.error().message().str() << std::endl;
    return 1;
  }
  ton::BlockIdExt blk_id = blk_id_res.move_as_ok();

  BlockDataLoader loader(db_path);

  auto blk_data_result = loader.load_block_data(blk_id);
  if (blk_data_result.is_error()) {
    std::cerr << "Error loading block data: " << blk_data_result.error().message().str() << std::endl;
    return 1;
  }

  auto blk_data = blk_data_result.move_as_ok();
  LOG(INFO) << "BlockId: " << blk_data->block_id().to_str();
  LOG(INFO) << "Block data size: " << blk_data->data().size() << " bytes";

  LOG(INFO) << "Cell has block record = " << block::gen::Block().validate_ref(10000000, blk_data->root_cell()) << "\n";

  std::ostringstream os;
  block::gen::Block().print_ref(os, blk_data->root_cell());
  LOG(INFO) << "Block = " << os.str();

  block::gen::Block::Record block_rec;
  bool ok = block::gen::Block().cell_unpack(blk_data->root_cell(), block_rec);
  CHECK(ok);

  block::gen::BlockInfo::Record info_rec;
  block::gen::BlockInfo().cell_unpack(block_rec.info, info_rec);
  LOG(INFO) << "Block.info after_merge=" << info_rec.after_merge << ", after_split=" << info_rec.after_split;

  auto manipulation_config = td::get_json_object_field(js_obj, "manipulation", td::JsonValue::Type::Object, false);
  CHECK(manipulation_config.is_ok());
  ManipulatorFactory().create(manipulation_config.move_as_ok())->modify(block_rec);

  LOG(INFO) << "Block after manipulation:";
  block::gen::BlockInfo().cell_unpack(block_rec.info, info_rec);
  LOG(INFO) << "Block.info after_merge=" << info_rec.after_merge << ", after_split=" << info_rec.after_split;

  return 0;
}