#include "ledger.hpp"
#include "../consensus.hpp"
#include "../crypto.hpp"
#include "../conf.hpp"
#include "../util/version.hpp"
#include "../util/util.hpp"
#include "../msg/fbuf/common_helpers.hpp"
#include "ledger_common.hpp"
#include "ledger_serve.hpp"

#define RAW_DATA_RETURN(ret)                  \
    {                                         \
        if (ret == -1)                        \
            sqlite::rollback_transaction(db); \
        else                                  \
            sqlite::commit_transaction(db);   \
        if (users_stmt != NULL)               \
            sqlite3_finalize(users_stmt);     \
        if (outputs_stmt != NULL)             \
            sqlite3_finalize(outputs_stmt);   \
        if (inputs_stmt != NULL)              \
            sqlite3_finalize(inputs_stmt);    \
        if (in_fd != -1)                      \
            close(in_fd);                     \
        if (out_fd != -1)                     \
            close(out_fd);                    \
        return ret;                           \
    }

namespace ledger
{
    ledger_context ctx;
    ledger_record genesis;
    ledger::ledger_mount ledger_fs;         // Global ledger file system instance.
    ledger::ledger_sync ledger_sync_worker; // Global ledger file system sync instance.
    ledger::ledger_serve ledger_server;     // Ledger file server instance.

    constexpr uint32_t LEDGER_FS_ID = 1;
    constexpr int FILE_PERMS = 0644;

    /**
     * Perform ledger related initializations.
    */
    int init()
    {
        // Setup the static genesis ledger fields.
        {
            const std::string empty_hash = std::string(util::h32_empty.to_string_view());
            genesis.seq_no = 0;
            genesis.timestamp = 0;
            genesis.ledger_hash = empty_hash;
            genesis.prev_ledger_hash = empty_hash;
            genesis.data_hash = empty_hash;
            genesis.state_hash = empty_hash;
            genesis.config_hash = empty_hash;
            genesis.user_hash = empty_hash;
            genesis.input_hash = empty_hash;
            genesis.output_hash = empty_hash;
        }

        // Full history status is always set to false since this is ledger fs. Historical checkpoints are not required in ledger fs even in full history mode.
        if (ledger_fs.init(LEDGER_FS_ID, conf::ctx.ledger_hpfs_dir, conf::ctx.ledger_hpfs_mount_dir, conf::ctx.ledger_hpfs_rw_dir, false) == -1)
        {
            LOG_ERROR << "Ledger file system initialization failed.";
            return -1;
        }

        if (ledger_server.init("ledger", &ledger_fs) == -1)
        {
            LOG_ERROR << "Ledger file system serve worker initialization failed.";
            return -1;
        }

        if (ledger_sync_worker.init("ledger", &ledger_fs) == -1)
        {
            LOG_ERROR << "Ledger file system sync worker initialization failed.";
            return -1;
        }

        // Remove old shards that exceeds max shard range.
        const p2p::sequence_hash lcl_id = ctx.get_lcl_id();
        remove_old_shards(lcl_id.seq_no, PRIMARY_SHARD_SIZE, conf::cfg.node.history_config.max_primary_shards, PRIMARY_DIR);
        remove_old_shards(lcl_id.seq_no, RAW_SHARD_SIZE, conf::cfg.node.history_config.max_raw_shards, RAW_DIR);

        if (conf::cfg.node.history_config.max_raw_shards == 0)
            ctx.raw_shards_persisted = true;

        return 0;
    }

    /**
     * Perform deinit tasks related to ledger.
    */
    void deinit()
    {
        ledger_sync_worker.deinit();
        ledger_server.deinit();
        ledger_fs.deinit();
    }

    /**
     * Updates the ledger with the given proposal message.
     * @param proposal Consensus-reached Stage 3 proposal.
     * @param consensed_users Users and their raw inputs/outputs received in this consensus round.
     * @return Returns 0 on success -1 on error.
     */
    int update_ledger(const p2p::proposal &proposal, const consensus::consensed_user_map &consensed_users)
    {
        // Aquire hpfs rw session before writing into shards.
        if (ledger_fs.acquire_rw_session() == -1)
            return -1;

        p2p::sequence_hash lcl_id;
        if (update_primary_ledger(proposal, consensed_users, lcl_id) == -1 ||
            update_ledger_raw_data(proposal, consensed_users, lcl_id) == -1)
        {
            ledger_fs.release_rw_session();
            return -1;
        }

        return ledger_fs.release_rw_session();
    }

    /**
     * Updates the primary ledger with the given consensus information.
     * @param proposal Consensus-reached Stage 3 proposal.
     * @param consensed_users Users and their raw inputs/outputs received in this consensus round.
     * @param new_lcl_id The new ledger seq no. and hash.
     * @return 0 on success. -1 on failure.
     */
    int update_primary_ledger(const p2p::proposal &proposal, const consensus::consensed_user_map &consensed_users, p2p::sequence_hash &new_lcl_id)
    {
        const p2p::sequence_hash lcl_id = ctx.get_lcl_id();
        new_lcl_id.seq_no = lcl_id.seq_no + 1;

        sqlite3 *db = NULL;

        // Prepare shard folders and database and get the shard sequence number.
        uint64_t shard_seq_no;
        const int shard_res = prepare_shard(&db, shard_seq_no, new_lcl_id.seq_no, PRIMARY_SHARD_SIZE, PRIMARY_DIR, PRIMARY_DB, true);

        // Insert primary ledger record.
        if (shard_res >= 0 && insert_ledger_record(db, lcl_id, shard_seq_no, proposal, new_lcl_id) != -1)
        {
            sqlite::close_db(&db);
            ctx.set_lcl_id(new_lcl_id);

            const std::string shard_vpath = std::string(ledger::PRIMARY_DIR).append("/").append(std::to_string(shard_seq_no));
            util::h32 last_primary_shard_hash;
            if (ledger_fs.get_hash(last_primary_shard_hash, hpfs::RW_SESSION_NAME, shard_vpath) == -1)
            {
                LOG_ERROR << errno << ": Error reading shard hash: " << shard_seq_no;
                return -1;
            }

            // Update the last shard hash and shard seqence number tracker when a new ledger is created.
            ctx.set_last_primary_shard_id(p2p::sequence_hash{shard_seq_no, last_primary_shard_hash});

            // Update the hpfs log index file only in full history mode.
            if (conf::cfg.node.history == conf::HISTORY::FULL && sc::contract_fs.update_hpfs_log_index(new_lcl_id.seq_no) == -1)
            {
                LOG_ERROR << errno << ": Error updating the hpfs log index file.";
                return -1;
            }

            // Remove old shards if new one got created.
            if (shard_res == 1)
                remove_old_shards(new_lcl_id.seq_no, PRIMARY_SHARD_SIZE, conf::cfg.node.history_config.max_primary_shards, PRIMARY_DIR);

            return 0;
        }

        sqlite::close_db(&db);
        return -1;
    }

    int update_ledger_raw_data(const p2p::proposal &proposal, const consensus::consensed_user_map &consensed_users, const p2p::sequence_hash &lcl_id)
    {
        if ((conf::cfg.node.history != conf::HISTORY::FULL && conf::cfg.node.history_config.max_raw_shards == 0))
            return 0;

        const bool has_updates = !consensed_users.empty();

        // Prepare shard folders and database and get the shard sequence number.
        sqlite3 *db = NULL;
        uint64_t shard_seq_no;
        const int shard_res = prepare_shard(&db, shard_seq_no, lcl_id.seq_no, RAW_SHARD_SIZE, RAW_DIR, RAW_DB, has_updates);

        if (shard_res >= 0 && insert_raw_data_records(db, shard_seq_no, proposal, consensed_users, lcl_id) != -1)
        {
            sqlite::close_db(&db);

            // Update in-memory context raw shard hash after inserting new record.
            util::h32 last_raw_shard_hash;
            if (ledger_fs.get_hash(last_raw_shard_hash, hpfs::RW_SESSION_NAME, std::string(RAW_DIR).append("/").append(std::to_string(shard_seq_no))) != -1)
                ctx.set_last_raw_shard_id(p2p::sequence_hash{shard_seq_no, last_raw_shard_hash});

            // Remove old shards if new one got created.
            if (shard_res == 1)
                remove_old_shards(lcl_id.seq_no, RAW_SHARD_SIZE, conf::cfg.node.history_config.max_raw_shards, RAW_DIR);

            return 0;
        }

        sqlite::close_db(&db);
        return -1;
    }

    /**
     * Inserts new ledger record to the sqlite database.
     * @param db The sqlite db connection for primary ledger db.
     * @param current_lcl_id Current lcl id.
     * @param shard_seq_no Current primary shard seq no.
     * @param proposal The consensus proposal.
     * @param new_lcl_id Newly created ledger id.
     * @return 0 on success. -1 on failure.
     */
    int insert_ledger_record(sqlite3 *db, const p2p::sequence_hash &current_lcl_id, const uint64_t shard_seq_no,
                             const p2p::proposal &proposal, p2p::sequence_hash &new_lcl_id)
    {
        // Combined binary hash of consensus user binary pub keys.
        const std::string user_hash = crypto::get_list_hash(proposal.users);

        // Combined binary hash of consensus input hashes.
        std::vector<std::string_view> inp_hashes;
        for (const std::string &o_hash : proposal.input_ordered_hashes)
        {
            // We need to consider the last 32 bytes of each ordered hash to get input hash without the nonce prefix.
            inp_hashes.push_back(util::get_string_suffix(o_hash, BLAKE3_OUT_LEN));
        }
        const std::string input_hash = crypto::get_list_hash(inp_hashes);

        uint8_t seq_no_bytes[8], time_bytes[8];
        util::uint64_to_bytes(seq_no_bytes, current_lcl_id.seq_no);
        util::uint64_to_bytes(time_bytes, proposal.time);

        // Contruct binary string for data hash.
        std::vector<std::string_view> data;
        data.emplace_back((char *)seq_no_bytes, sizeof(seq_no_bytes));
        data.emplace_back((char *)time_bytes, sizeof(time_bytes));
        data.push_back(proposal.state_hash.to_string_view());
        data.push_back(proposal.patch_hash.to_string_view());
        data.push_back(user_hash);
        data.push_back(input_hash);
        data.push_back(proposal.output_hash);

        // Combined binary hash of data fields. blake3(seq_no + time + state_hash + patch_hash + user_hash + input_hash + output_hash)
        const std::string data_hash = crypto::get_list_hash(data);

        const std::string prev_ledger_hash(current_lcl_id.hash.to_string_view());

        // Ledger hash is the combined hash of previous ledger hash and the new data hash.
        new_lcl_id.hash = crypto::get_hash(prev_ledger_hash, data_hash);

        // Construct ledger struct with binary hashes.
        const ledger_record ledger{
            current_lcl_id.seq_no + 1,
            proposal.time,
            std::string(new_lcl_id.hash.to_string_view()),
            prev_ledger_hash,
            data_hash,
            std::string(proposal.state_hash.to_string_view()),
            std::string(proposal.patch_hash.to_string_view()),
            user_hash,
            input_hash,
            proposal.output_hash}; // Merkle root output hash.

        if (sqlite::insert_ledger_row(db, ledger) == -1)
        {
            LOG_ERROR << errno << ": Error creating the ledger, shard: " << shard_seq_no;
            return -1;
        }

        return 0;
    }

    /**
     * Populates the raw data db and blob files with consensed users, inputs and outputs records.
     * @param db The sqlite db connection for raw data db.
     * @param shard_seq_no Raw shard seq no.
     * @param proposal The consensus proposal.
     * @param consensed_users Consensed users and their inputs and outputs.
     * @param lcl_id Current ledger id.
     * @return 0 on success. -1 on failure.
     */
    int insert_raw_data_records(sqlite3 *db, const uint64_t shard_seq_no, const p2p::proposal &proposal,
                                const consensus::consensed_user_map &consensed_users, const p2p::sequence_hash &lcl_id)
    {
        // We keep sqlite records about users, inputs and outputs. To store raw input and output content, we use the corresponding blob file
        // within the shard. Each shard has a sqlite db, raw inputs blob file and raw outputs blob file.

        if (consensed_users.empty())
            return 0;

        const std::string shard_path = ledger_fs.physical_path(hpfs::RW_SESSION_NAME, std::string(RAW_DIR).append("/").append(std::to_string(shard_seq_no)).append("/"));

        // We reuse sqlite prepared statements to improve looping performance.

        sqlite3_stmt *users_stmt = sqlite::prepare_user_insert(db);
        sqlite3_stmt *outputs_stmt = NULL;
        sqlite3_stmt *inputs_stmt = NULL;

        int in_fd = -1;     // Raw inputs storage file for the shard. Only created and opened if there are any inputs.
        int out_fd = -1;    // Raw outputs storage file for the shard. Only created and opened if there are any outputs.
        size_t in_pos = 0;  // Current writing position offset of the inputs file.
        size_t out_pos = 0; // Current writing position offset of the outputs file.

        // Group all row insertions within a transaction for consistency.
        if (sqlite::begin_transaction(db) == -1)
            RAW_DATA_RETURN(-1);

        for (const auto &[pubkey, cu] : consensed_users)
        {
            if (sqlite::insert_user_record(users_stmt, lcl_id.seq_no, pubkey) == -1)
                RAW_DATA_RETURN(-1);

            if (!cu.consensed_inputs.empty())
            {
                if (inputs_stmt == NULL)
                    inputs_stmt = sqlite::prepare_user_input_insert(db);

                for (const consensus::consensed_user_input &cui : cu.consensed_inputs)
                {
                    // Create and open the raw inputs file for the shard if needed.
                    if (in_fd == -1 && (in_fd = create_raw_data_blob_file(shard_path, RAW_INPUTS_FILE, in_pos)) == -1)
                        RAW_DATA_RETURN(-1);

                    // Write the input to the blob file. Then we save the written offset and blob size in sqlite record.
                    std::string buf;
                    usr::input_store.read_buf(cui.input, buf);
                    if (write(in_fd, buf.data(), buf.size()) == -1)
                    {
                        LOG_ERROR << errno << ": Error when writing input blob.";
                        RAW_DATA_RETURN(-1);
                    }

                    // Insert sqlite record.
                    std::string_view hash = util::get_string_suffix(cui.ordered_hash, BLAKE3_OUT_LEN);
                    const uint64_t nonce = util::uint64_from_bytes((uint8_t *)cui.ordered_hash.data());

                    if (sqlite::insert_user_input_record(inputs_stmt, lcl_id.seq_no, pubkey, hash, nonce, in_pos, buf.size()) == -1)
                        RAW_DATA_RETURN(-1);

                    in_pos += buf.size(); // Increament the blob file write offset so next write will happen correctly.
                }
            }

            if (!cu.consensed_outputs.outputs.empty())
            {
                // Create and open the raw outputs file for the shard if needed.
                if (out_fd == -1 && (out_fd = create_raw_data_blob_file(shard_path, RAW_OUTPUTS_FILE, out_pos)) == -1)
                    RAW_DATA_RETURN(-1);

                // Write all the outputs of this user to the blob file. Then we save the written offset and output count in sqlite record.
                // First we write the list of offsets and sizes of each output. Then the outputs themselves.
                // [offset1][size1][offset2][size2]....[output1][output2]...

                // Prepare write header.
                const uint64_t output_count = cu.consensed_outputs.outputs.size();
                std::vector<uint8_t> header(output_count * (sizeof(off_t) + sizeof(size_t))); // Header containing list of [offset+size].
                off_t out_buf_offset = out_pos + header.size();                               // Output buffers will be written after the header.
                for (size_t i = 0; i < output_count; i++)
                {
                    const size_t output_size = cu.consensed_outputs.outputs[i].size();
                    uint8_t *header_pos = header.data() + (i * (sizeof(off_t) + sizeof(size_t)));
                    // Write the pair of offset+size of the individual output into the header.
                    util::uint64_to_bytes(header_pos, out_buf_offset);
                    util::uint64_to_bytes(header_pos + sizeof(size_t), output_size);
                    out_buf_offset += output_size;
                }

                // Write the header and output buffers.
                iovec memsegs[1 + output_count];
                memsegs[0] = iovec{header.data(), header.size()};
                uint64_t total_write_size = header.size();
                for (size_t i = 0; i < output_count; i++)
                {
                    const std::string &output = cu.consensed_outputs.outputs[i];
                    memsegs[i + 1] = iovec{(void *)output.data(), output.size()};
                    total_write_size += output.size();
                }
                if (writev(out_fd, memsegs, 1 + output_count) == -1)
                {
                    LOG_ERROR << errno << ": Error when writing outputs blobs.";
                    RAW_DATA_RETURN(-1);
                }

                // Insert sqlite record.
                // Prepare the output insertion stamement only once.
                if (outputs_stmt == NULL)
                    outputs_stmt = sqlite::prepare_user_output_insert(db);

                if (sqlite::insert_user_output_record(outputs_stmt, lcl_id.seq_no, pubkey, cu.consensed_outputs.hash, out_pos, output_count) == -1)
                    RAW_DATA_RETURN(-1);

                out_pos += total_write_size; // Increament the blob file write offset so next write will happen correctly.
            }
        }

        RAW_DATA_RETURN(0);
    }

    /**
     * Open or create the specified file name for appending raw blob data.
     * @param shard_path Parent shard directory.
     * @param file_name Name of the blob file.
     * @param file_size Current file size.
     * @return 0 on success. -1 on failure.
     */
    int create_raw_data_blob_file(const std::string &shard_path, const char *file_name, size_t &file_size)
    {
        const std::string file_path = shard_path + file_name;
        int fd = open(file_path.data(), O_WRONLY | O_APPEND | O_CREAT, FILE_PERMS);
        if (fd == -1)
            LOG_ERROR << errno << ": Error when creating file " << file_path;

        struct stat st;
        if (fstat(fd, &st) == -1)
            LOG_ERROR << errno << ": Error when stat of file " << file_path;

        file_size = st.st_size;
        return fd;
    }

    /**
     * Creates or open a db connection to the shard based on the params. This is used to create primary and raw shards.
     * @param db Database connection to be opened.
     * @param ledger_seq_no Ledger sequence number.
     * @param keep_db_connection Whether the sqlite db connection must be kept open or not.
     * @return 0 if shard already exists. 1 if new shard got created. -1 on failure.
     */
    int prepare_shard(sqlite3 **db, uint64_t &shard_seq_no, const uint64_t ledger_seq_no, const uint64_t shard_size,
                      const char *shard_dir, const char *db_name, const bool keep_db_connection)
    {
        // Construct shard path.
        shard_seq_no = (ledger_seq_no - 1) / shard_size;
        const std::string shard_path = ledger_fs.physical_path(hpfs::RW_SESSION_NAME, std::string(shard_dir).append("/").append(std::to_string(shard_seq_no)));
        const std::string db_path = shard_path + "/" + db_name;

        // This means this is the first ledger of the shard.
        // So create the shard folder and other required files.
        if ((ledger_seq_no - 1) % shard_size == 0)
        {
            // Creating the directory.
            if (util::create_dir_tree_recursive(shard_path) == -1)
            {
                LOG_ERROR << errno << ": Error creating the shard " << shard_path;
                return -1;
            }

            // Creating ledger database and open a database connection.
            if (sqlite::open_db(db_path, db, true) == -1)
            {
                LOG_ERROR << errno << ": Error creating the database " << db_name;
                return -1;
            }

            if ((shard_dir == PRIMARY_DIR && sqlite::initialize_ledger_db(*db) == -1) ||
                (shard_dir == RAW_DIR && sqlite::initialize_ledger_raw_db(*db) == -1))
            {
                LOG_ERROR << errno << ": Error initilizing the database " << db_name;
                return -1;
            }

            // Create and update the hp table with current ledger version.
            if (sqlite::create_hp_table(*db, version::LEDGER_VERSION) == -1)
            {
                LOG_ERROR << errno << ": Error creating hp table in " << db_name;
                return -1;
            }

            // Close the connection if it doesn't need to be retained.
            if (!keep_db_connection)
                sqlite::close_db(db);

            util::h32 prev_shard_hash;
            if (shard_seq_no > 0)
            {
                const std::string prev_shard_vpath = std::string(shard_dir) + "/" + std::to_string(shard_seq_no - 1);
                if (ledger_fs.get_hash(prev_shard_hash, hpfs::RW_SESSION_NAME, prev_shard_vpath) < 1)
                {
                    LOG_ERROR << errno << ": Error getting shard hash in vpath: " << prev_shard_vpath << " for previous shard hash.";
                    return -1;
                }
            }

            // Write the prev_shard.hash to the new folder.
            {
                const std::string shard_hash_file_path = shard_path + PREV_SHARD_HASH_FILENAME;
                const int fd = open(shard_hash_file_path.data(), O_CREAT | O_RDWR, FILE_PERMS);
                if (fd == -1)
                {
                    LOG_ERROR << errno << ": Error creating prev_shard.hash file in " << shard_path;
                    return -1;
                }

                struct iovec iov_vec[2];
                iov_vec[0].iov_base = version::LEDGER_VERSION_BYTES;
                iov_vec[0].iov_len = version::VERSION_BYTES_LEN;

                iov_vec[1].iov_base = &prev_shard_hash;
                iov_vec[1].iov_len = sizeof(util::h32);

                if (writev(fd, iov_vec, 2) == -1)
                {
                    LOG_ERROR << errno << ": Error writing to " << shard_hash_file_path << ".";
                    close(fd);
                    return -1;
                }
                close(fd);
            }

            // Persist newly created shard seq number as the max shard seq number.
            if (persist_max_shard_seq_no(shard_dir, shard_seq_no) == -1)
            {
                LOG_ERROR << "Error persisting maximum raw shard sequnce number.";
                return -1;
            }

            return 1;
        }
        else
        {
            if (keep_db_connection && sqlite::open_db(db_path, db, true) == -1)
            {
                LOG_ERROR << errno << ": Error openning the shard database " << db_path;
                return -1;
            }
            return 0;
        }
    }

    /**
     * Remove old shards that exceeds max shard range from file system.
     * @param lcl_seq_no Current ledger seq no.
     * @param shard_size Shard size to use.
     * @param max_shards Maximum shards to keep.
     * @param shard_parent_dir Shard parent directory.
     */
    void remove_old_shards(const uint64_t lcl_seq_no, const uint64_t shard_size, const uint64_t max_shards, std::string_view shard_parent_dir)
    {
        const uint64_t shard_seq_no = (lcl_seq_no - 1) / shard_size;

        // No removals if this is a full history node or we haven't yet reached the shard limit.
        if (conf::cfg.node.history == conf::HISTORY::FULL || max_shards > shard_seq_no)
            return;

        const uint64_t delete_from = shard_seq_no - max_shards;

        for (int i = delete_from; i >= 0; i--)
        {
            const std::string shard_path = std::string(ledger_fs.physical_path(hpfs::RW_SESSION_NAME, shard_parent_dir)).append("/").append(std::to_string(i));
            // Break the loop if there's no corresponding shard.
            // There cannot be shards which is less than this shard no. since shards are continous.
            if (!util::is_dir_exists(shard_path))
                break;

            if (util::remove_directory_recursively(shard_path) == -1)
            {
                LOG_ERROR << errno << ": Error deleting shard: " << shard_path;
                break;
            }
        }
    }

    /**
     * Cleanup and request historical shards according to the max we can keep.
     * @param shard_seq_no Latest shard sequence number.
     * @param shard_parent_dir Shard parent directory.
     */
    void persist_shard_history(const uint64_t shard_seq_no, std::string_view shard_parent_dir)
    {
        // Skip if shard cleanup and requesting has been already done.
        if ((shard_parent_dir == PRIMARY_DIR && ctx.primary_shards_persisted) || (shard_parent_dir == RAW_DIR && ctx.raw_shards_persisted))
            return;

        // Set persisted flag to true. So this cleanup won't get executed again.
        shard_parent_dir == PRIMARY_DIR ? ctx.primary_shards_persisted = true : ctx.raw_shards_persisted = true;
        const uint64_t max_shard_count = (shard_parent_dir == PRIMARY_DIR ? conf::cfg.node.history_config.max_primary_shards : conf::cfg.node.history_config.max_raw_shards);

        const std::string shard_dir_path = ledger_fs.physical_path(hpfs::RW_SESSION_NAME, shard_parent_dir);
        const std::list<std::string> shard_list = util::fetch_dir_entries(shard_dir_path);

        // Skip the sequence no file from the count.
        uint64_t shard_count = shard_list.size() - 1;

        // First, In history custom mode remove all the historical shards which are older than the min we can keep.
        if (conf::cfg.node.history == conf::HISTORY::CUSTOM && shard_seq_no >= max_shard_count)
        {
            for (const std::string &shard : shard_list)
            {
                // Skip the sequence no file.
                if (("/" + shard) == SHARD_SEQ_NO_FILENAME)
                    continue;

                uint64_t seq_no;
                if (util::stoull(shard, seq_no) != -1 && seq_no <= (shard_seq_no - max_shard_count))
                {
                    const std::string shard_path = std::string(shard_dir_path).append("/").append(shard);
                    if (util::is_dir_exists(shard_path) && util::remove_directory_recursively(shard_path) == -1)
                        LOG_ERROR << errno << ": Error deleting shard: " << shard;
                    else
                        shard_count--;
                }
            }
        }

        // In full history mode request for all the historical nodes if not exists, Otherwise request if max count haven't reached
        if (shard_seq_no >= shard_count && (conf::cfg.node.history == conf::HISTORY::FULL || shard_count < max_shard_count))
        {
            const uint64_t seq_no = shard_seq_no - shard_count;

            const std::string prev_shard_hash_file_path = shard_dir_path + "/" + std::to_string(seq_no + 1) + PREV_SHARD_HASH_FILENAME;
            const int fd = open(prev_shard_hash_file_path.c_str(), O_RDONLY | O_CLOEXEC);
            if (fd == -1)
            {
                LOG_DEBUG << "Cannot read " << prev_shard_hash_file_path;
                return;
            }

            util::h32 prev_shard_hash_from_file;
            // Start reading hash excluding version bytes.
            const int res = pread(fd, &prev_shard_hash_from_file, sizeof(util::h32), version::VERSION_BYTES_LEN);
            close(fd);
            if (res == -1)
            {
                LOG_ERROR << errno << ": Error reading hash file. " << prev_shard_hash_file_path;
                return;
            }

            const std::string shard_path = std::string(shard_parent_dir).append("/").append(std::to_string(seq_no));
            ledger_sync_worker.set_target_push_back(hpfs::sync_target{shard_path, prev_shard_hash_from_file, shard_path, hpfs::BACKLOG_ITEM_TYPE::DIR});
        }
    }

    /**
     * Get last ledger and update the context.
     * @param session_name Hpfs session name.
     * @param last_primary_shard_id Last primary shard id.
     * @return Returns 0 on success -1 on error.
     */
    int get_last_ledger_and_update_context(std::string_view session_name, const p2p::sequence_hash &last_primary_shard_id)
    {
        sqlite3 *db = NULL;
        const std::string shard_path = ledger_fs.physical_path(session_name, ledger::PRIMARY_DIR) + "/" + std::to_string(last_primary_shard_id.seq_no);

        if (last_primary_shard_id.empty())
        {
            // This is the genesis ledger.
            ctx.set_lcl_id(p2p::sequence_hash{0, util::h32_empty});
            return 0;
        }

        if (sqlite::open_db(shard_path + "/" + PRIMARY_DB, &db) == -1)
        {
            LOG_ERROR << errno << ": Error openning the shard database, shard: " << std::to_string(last_primary_shard_id.seq_no);
            return -1;
        }

        ledger_record last_ledger;
        if (sqlite::get_last_ledger(db, last_ledger) == -1)
        {
            sqlite::close_db(&db);
            return -1;
        }

        sqlite::close_db(&db);

        // Update new lcl information.
        p2p::sequence_hash lcl_id;
        lcl_id.seq_no = last_ledger.seq_no;
        lcl_id.hash = last_ledger.ledger_hash;
        ctx.set_lcl_id(lcl_id);

        return 0;
    }

    /**
     * Get the hash and shard sequence number of the last shard in the given parent directory.
     * @param session_name Hpfs session name.
     * @param last_shard_id Struct which holds last shard data. (sequence number and hash).
     * @param shard_parent_dir Parent director vpath of the shards.
     * @return 0 on success. -1 on error.
    */
    int get_last_shard_info(std::string_view session_name, p2p::sequence_hash &last_shard_id, const std::string &shard_parent_dir)
    {
        const std::string last_shard_seq_no_vpath = shard_parent_dir + SHARD_SEQ_NO_FILENAME;
        const std::string last_shard_seq_no_path = ledger_fs.physical_path(session_name, last_shard_seq_no_vpath);

        const int fd = open(last_shard_seq_no_path.data(), O_RDONLY, FILE_PERMS);
        if (fd == -1)
        {
            if (errno == ENOENT)
            {
                LOG_DEBUG << "Max shard sequence meta file not found. Starting from zero. " << last_shard_seq_no_path;
                // Return defaults of sequence hash(0 for shard_seq_no and empty hash for shard hash).
                last_shard_id = {};
                return 0;
            }
            else
            {
                LOG_ERROR << errno << ": Error opening meta " << last_shard_seq_no_path;
                return -1;
            }
        }
        uint8_t last_shard_seq_no_buf[8];
        if (pread(fd, last_shard_seq_no_buf, sizeof(last_shard_seq_no_buf), version::VERSION_BYTES_LEN) == -1)
        {
            LOG_ERROR << errno << ": Error reading " << last_shard_seq_no_path;
            close(fd);
            return -1;
        }
        close(fd);

        last_shard_id.seq_no = util::uint64_from_bytes(last_shard_seq_no_buf);
        const std::string shard_path = std::string(shard_parent_dir).append("/").append(std::to_string(last_shard_id.seq_no));
        if (ledger_fs.get_hash(last_shard_id.hash, session_name, shard_path) == -1)
        {
            LOG_ERROR << "Error reading last shard hash in " << shard_path;
            return -1;
        }

        return 0;
    }

    /**
     * Update max_shard.seq_no meta file with the given latest shard sequence number which can be used to identify last shard 
     * sequence number in startup.
     * @param shard_parent_dir Shard's parent directory. (primary or raw).
     * @param last_shard_seq_no Last shard sequence number of the given parent.
     * @return Return -1 on error and 0 on success.
    */
    int persist_max_shard_seq_no(const std::string &shard_parent_dir, const uint64_t last_shard_seq_no)
    {
        const std::string last_shard_seq_no_vpath = shard_parent_dir + SHARD_SEQ_NO_FILENAME;
        const std::string last_shard_seq_no_path = ledger_fs.physical_path(hpfs::RW_SESSION_NAME, last_shard_seq_no_vpath);

        // Open max_shard.seq_no in given parent directory.
        const int fd = open(last_shard_seq_no_path.data(), O_CREAT | O_RDWR, FILE_PERMS);
        if (fd == -1)
        {
            LOG_ERROR << errno << ": Error opening  " << last_shard_seq_no_path;
            return -1;
        }
        uint8_t seq_no_byte_str[8];
        util::uint64_to_bytes(seq_no_byte_str, last_shard_seq_no);

        struct iovec iov_vec[2];
        iov_vec[0].iov_base = version::LEDGER_VERSION_BYTES;
        iov_vec[0].iov_len = version::VERSION_BYTES_LEN;

        iov_vec[1].iov_base = seq_no_byte_str;
        iov_vec[1].iov_len = sizeof(seq_no_byte_str);

        if (writev(fd, iov_vec, 2) == -1)
        {
            LOG_ERROR << errno << ": Error updating the max_shard.seq_no file for shard " << last_shard_seq_no;
            close(fd);
            return -1;
        }
        close(fd);
        return 0;
    }

    /**
     * Calculate root hash of contract_fs from the ledger record of given seq_no.
     * @param root_hash The calculated root hash as of the given seq_no.
     * @param seq_no Ledger's sequence number.
     * @return Returns -1 on error and 0 on success.
    */
    int get_root_hash_from_ledger(util::h32 &root_hash, const uint64_t seq_no)
    {
        sqlite3 *db = NULL;
        const char *session_name = "root_hash_from_ledger";
        if (ledger_fs.start_ro_session(session_name, false) == -1)
            return -1;

        const uint64_t shard_seq_no = (seq_no - 1) / PRIMARY_SHARD_SIZE;

        const std::string shard_path = ledger_fs.physical_path(session_name, ledger::PRIMARY_DIR) + "/" + std::to_string(shard_seq_no);

        if (sqlite::open_db(shard_path + "/" + PRIMARY_DB, &db) == -1)
        {
            LOG_ERROR << errno << ": Error openning the shard database, shard: " << shard_seq_no;
            ledger_fs.stop_ro_session(session_name);
            return -1;
        }

        ledger::ledger_record ledger;
        if (sqlite::get_ledger_by_seq_no(db, seq_no, ledger) == -1)
        {
            LOG_ERROR << "Error getting ledger by sequence number: " << seq_no;
            sqlite::close_db(&db);
            ledger_fs.stop_ro_session(session_name);
            return -1;
        }
        sqlite::close_db(&db);
        ledger_fs.stop_ro_session(session_name);

        root_hash = hpfs::get_root_hash(ledger.config_hash, ledger.state_hash);
        return 0;
    }
} // namespace ledger