// SPDX-License-Identifier: GPL-3.0-or-later
// SPDX-FileCopyrightText: 2019-2023 Ivan Baidakou

#include "test-utils.h"
#include "model/cluster.h"
#include "model/misc/version_utils.h"
#include "diff-builder.h"

using namespace syncspirit;
using namespace syncspirit::test;
using namespace syncspirit::model;

TEST_CASE("version_utils", "[model]") {
    proto::Vector lhs, rhs;

    SECTION("empty versions") { CHECK(compare(lhs, rhs) == version_relation_t::identity); }

    SECTION("non-empty, identical") {
        auto lc = lhs.add_counters();
        lc->set_id(1);
        lc->set_value(1);

        auto rc = rhs.add_counters();
        rc->set_id(1);
        rc->set_value(1);
        CHECK(compare(lhs, rhs) == version_relation_t::identity);
    }

    SECTION("non-empty, conflict") {
        auto lc = lhs.add_counters();
        lc->set_id(5);
        lc->set_value(1);

        auto rc = rhs.add_counters();
        rc->set_id(6);
        rc->set_value(1);
        CHECK(compare(lhs, rhs) == version_relation_t::conflict);
    }

    SECTION("trivial, newer") {
        auto lc = lhs.add_counters();
        lc->set_id(1);
        lc->set_value(2);

        auto rc = rhs.add_counters();
        rc->set_id(1);
        rc->set_value(1);
        CHECK(compare(lhs, rhs) == version_relation_t::newer);
    }

    SECTION("trivial, older") {
        auto lc = lhs.add_counters();
        lc->set_id(1);
        lc->set_value(1);

        auto rc = rhs.add_counters();
        rc->set_id(1);
        rc->set_value(2);
        CHECK(compare(lhs, rhs) == version_relation_t::older);
    }

    SECTION("newer") {
        auto lc_1 = lhs.add_counters();
        lc_1->set_id(1);
        lc_1->set_value(1);

        auto lc_2 = lhs.add_counters();
        lc_2->set_id(3);
        lc_2->set_value(1);

        auto rc = rhs.add_counters();
        rc->set_id(1);
        rc->set_value(1);
        CHECK(compare(lhs, rhs) == version_relation_t::newer);
    }
}

TEST_CASE("file_info_t::need_download", "[model]") {
    auto my_id = device_id_t::from_string("KHQNO2S-5QSILRK-YX4JZZ4-7L77APM-QNVGZJT-EKU7IFI-PNEPBMY-4MXFMQD").value();
    auto peer_id = device_id_t::from_string("VUV42CZ-IQD5A37-RPEBPM4-VVQK6E4-6WSKC7B-PVJQHHD-4PZD44V-ENC6WAZ").value();
    auto my_device = device_t::create(my_id, "my-device").value();
    auto peer_device = device_t::create(peer_id, "peer-device").value();

    auto cluster = cluster_ptr_t(new cluster_t(my_device, 1, 1));
    cluster->get_devices().put(my_device);
    cluster->get_devices().put(peer_device);

    auto &folders = cluster->get_folders();
    auto builder = diff_builder_t(*cluster);
    builder.create_folder("1234-5678", "some/path", "my-label").share_folder(peer_id.get_sha256(), "1234-5678");
    REQUIRE(builder.apply());

    auto folder = folders.by_id("1234-5678");
    auto &folder_infos = folder->get_folder_infos();
    auto folder_my = folder_infos.by_device(*my_device);
    auto folder_peer = folder_infos.by_device(*peer_device);

    auto pr_file = proto::FileInfo();
    pr_file.set_name("a.txt");

    SECTION("file is empty => no download") {
        auto file_my = file_info_t::create(cluster->next_uuid(), pr_file, folder_my).value();
        auto file_peer = file_info_t::create(cluster->next_uuid(), pr_file, folder_peer).value();
        CHECK(!file_my->need_download(*file_peer));
    }

    SECTION("locked file => no download") {
        auto file_my = file_info_t::create(cluster->next_uuid(), pr_file, folder_my).value();
        auto file_peer = file_info_t::create(cluster->next_uuid(), pr_file, folder_peer).value();
        file_my->lock();
        CHECK(!file_my->need_download(*file_peer));
    }

    pr_file.set_block_size(5);
    pr_file.set_size(5);
    auto *peer_v = pr_file.mutable_version();
    auto *peer_c1 = peer_v->add_counters();
    peer_c1->set_id(1);
    peer_c1->set_value(1);

    auto b = pr_file.add_blocks();
    b->set_hash(utils::sha256_digest("12345").value());
    b->set_weak_hash(555);
    auto bi = block_info_t::create(*b).value();
    auto &blocks_map = cluster->get_blocks();
    blocks_map.put(bi);
    auto bbb = blocks_map.get(b->hash());
    REQUIRE(bbb);

    auto file_my = file_info_t::create(cluster->next_uuid(), pr_file, folder_my).value();
    file_my->remove_blocks();
    file_my->assign_block(bbb, 0);

    SECTION("versions are identical, no local file => download") {
        auto file_peer = file_info_t::create(cluster->next_uuid(), pr_file, folder_peer).value();
        CHECK(file_my->need_download(*file_peer));
    }

    SECTION("versions are identical, already available => no download") {
        auto file_peer = file_info_t::create(cluster->next_uuid(), pr_file, folder_peer).value();
        file_my->mark_local_available(0);
        REQUIRE(file_my->is_locally_available());
        CHECK(!file_my->need_download(*file_peer));
    }

    SECTION("peer's version is newer => download") {
        auto *peer_c2 = peer_v->add_counters();
        peer_c2->set_id(1);
        peer_c2->set_value(2);
        auto file_peer = file_info_t::create(cluster->next_uuid(), pr_file, folder_peer).value();

        file_my->mark_local_available(0);
        CHECK(file_my->need_download(*file_peer));
    }

    SECTION("peer's counter is newer => download") {
        peer_c1->set_value(2);
        auto file_peer = file_info_t::create(cluster->next_uuid(), pr_file, folder_peer).value();

        file_my->mark_local_available(0);
        CHECK(file_my->need_download(*file_peer));
    }
}

TEST_CASE("file_info_t::local_file", "[model]") {
    auto my_id = device_id_t::from_string("KHQNO2S-5QSILRK-YX4JZZ4-7L77APM-QNVGZJT-EKU7IFI-PNEPBMY-4MXFMQD").value();
    auto peer_id = device_id_t::from_string("VUV42CZ-IQD5A37-RPEBPM4-VVQK6E4-6WSKC7B-PVJQHHD-4PZD44V-ENC6WAZ").value();
    auto my_device = device_t::create(my_id, "my-device").value();
    auto peer_device = device_t::create(peer_id, "peer-device").value();

    auto cluster = cluster_ptr_t(new cluster_t(my_device, 1, 1));
    cluster->get_devices().put(my_device);
    cluster->get_devices().put(peer_device);

    auto &folders = cluster->get_folders();
    auto builder = diff_builder_t(*cluster);
    builder.create_folder("1234-5678", "some/path", "my-label").share_folder(peer_id.get_sha256(), "1234-5678");
    REQUIRE(builder.apply());

    auto folder = folders.by_id("1234-5678");
    auto &folder_infos = folder->get_folder_infos();
    auto folder_my = folder_infos.by_device(*my_device);
    auto folder_peer = folder_infos.by_device(*peer_device);

    auto pr_file = proto::FileInfo();
    pr_file.set_name("a.txt");
    auto version = pr_file.mutable_version();
    auto c1 = version->add_counters();
    c1->set_id(1);
    c1->set_value(1);

    SECTION("no local file") {
        auto file_peer = file_info_t::create(cluster->next_uuid(), pr_file, folder_peer).value();
        CHECK(!file_peer->local_file());
    }

    SECTION("there is identical local file") {
        auto file_my = file_info_t::create(cluster->next_uuid(), pr_file, folder_my).value();
        folder_my->add(file_my, false);
        auto file_peer = file_info_t::create(cluster->next_uuid(), pr_file, folder_peer).value();
        folder_peer->add(file_peer, false);
        file_my->set_source(file_peer);

        auto lf = file_peer->local_file();
        REQUIRE(lf);
        REQUIRE(lf == file_my);
    }

    SECTION("peer version is newer (1)") {
        auto file_my = file_info_t::create(cluster->next_uuid(), pr_file, folder_my).value();
        folder_my->add(file_my, false);

        auto c2 = version->add_counters();
        c2->set_id(2);
        c2->set_value(1);
        auto file_peer = file_info_t::create(cluster->next_uuid(), pr_file, folder_peer).value();

        REQUIRE(!file_peer->local_file());
    }
}

TEST_CASE("source file", "[model]") {
    auto my_id = device_id_t::from_string("KHQNO2S-5QSILRK-YX4JZZ4-7L77APM-QNVGZJT-EKU7IFI-PNEPBMY-4MXFMQD").value();
    auto peer_id = device_id_t::from_string("VUV42CZ-IQD5A37-RPEBPM4-VVQK6E4-6WSKC7B-PVJQHHD-4PZD44V-ENC6WAZ").value();
    auto my_device = device_t::create(my_id, "my-device").value();
    auto peer_device = device_t::create(peer_id, "peer-device").value();

    auto cluster = cluster_ptr_t(new cluster_t(my_device, 1, 1));
    cluster->get_devices().put(my_device);
    cluster->get_devices().put(peer_device);

    auto &folders = cluster->get_folders();
    auto builder = diff_builder_t(*cluster);
    builder.create_folder("1234-5678", "some/path", "my-label").share_folder(peer_id.get_sha256(), "1234-5678");
    REQUIRE(builder.apply());

    auto folder = folders.by_id("1234-5678");
    auto &folder_infos = folder->get_folder_infos();
    auto folder_my = folder_infos.by_device(*my_device);
    auto folder_peer = folder_infos.by_device(*peer_device);

    auto pr_file = proto::FileInfo();
    pr_file.set_name("a.txt");
    auto version = pr_file.mutable_version();
    auto c1 = version->add_counters();
    c1->set_id(1);
    c1->set_value(peer_device->as_uint());

    auto my_file = file_info_t::create(cluster->next_uuid(), pr_file, folder_my).value();
    SECTION("no peer file exists") { CHECK(!my_file->get_source()); }

    auto peer_file = file_info_t::create(cluster->next_uuid(), pr_file, folder_peer).value();
    my_file->set_source(peer_file);
    folder_peer->add(peer_file, false);
    CHECK(my_file->get_source());

    SECTION("reset") {
        my_file->set_source({});
        CHECK(!my_file->get_source());
    }

    SECTION("peer has different version") {
        c1->set_id(2);
        peer_file = file_info_t::create(cluster->next_uuid(), pr_file, folder_peer).value();
        folder_peer->add(peer_file, false);
        CHECK(!my_file->get_source());
    }
}

TEST_CASE("file_info_t::check_consistency", "[model]") {
    auto my_id = device_id_t::from_string("KHQNO2S-5QSILRK-YX4JZZ4-7L77APM-QNVGZJT-EKU7IFI-PNEPBMY-4MXFMQD").value();
    auto my_device = device_t::create(my_id, "my-device").value();

    auto cluster = cluster_ptr_t(new cluster_t(my_device, 1, 1));
    cluster->get_devices().put(my_device);

    auto &folders = cluster->get_folders();
    auto builder = diff_builder_t(*cluster);
    builder.create_folder("1234-5678", "some/path", "my-label");
    REQUIRE(builder.apply());

    auto folder = folders.by_id("1234-5678");
    auto &folder_infos = folder->get_folder_infos();
    auto folder_my = folder_infos.by_device(*my_device);

    auto pr_file = proto::FileInfo();
    pr_file.set_name("a.txt");
    pr_file.set_block_size(5);
    pr_file.set_size(5);

    auto b = pr_file.add_blocks();
    b->set_hash(utils::sha256_digest("12345").value());
    b->set_weak_hash(555);
    b->set_size(5ul);
    auto bi = block_info_t::create(*b).value();
    auto &blocks_map = cluster->get_blocks();
    blocks_map.put(bi);
    auto bbb = blocks_map.get(b->hash());
    REQUIRE(bbb);

    REQUIRE(builder.local_update(folder->get_id(), pr_file).apply());
    auto file_my = folder_my->get_file_infos().by_name(pr_file.name());
    CHECK(file_my->get_size() == 5);
}

TEST_CASE("file_info_t::create, inconsistent source") {
    auto my_id = device_id_t::from_string("KHQNO2S-5QSILRK-YX4JZZ4-7L77APM-QNVGZJT-EKU7IFI-PNEPBMY-4MXFMQD").value();
    auto my_device = device_t::create(my_id, "my-device").value();

    auto cluster = cluster_ptr_t(new cluster_t(my_device, 1, 1));
    cluster->get_devices().put(my_device);

    auto &folders = cluster->get_folders();
    auto builder = diff_builder_t(*cluster);
    builder.create_folder("1234-5678", "some/path", "my-label");
    REQUIRE(builder.apply());

    auto folder = folders.by_id("1234-5678");
    auto &folder_infos = folder->get_folder_infos();
    auto folder_my = folder_infos.by_device(*my_device);

    auto pr_block = proto::BlockInfo();
    pr_block.set_size(131072);

    auto pr_file = proto::FileInfo();
    pr_file.set_name("a.txt");
    *pr_file.add_blocks() = pr_block;
    pr_file.set_block_size(131072);
    pr_file.set_size(0);

    auto my_file = file_info_t::create(cluster->next_uuid(), pr_file, folder_my).value();
    CHECK(my_file->get_block_size() == 0);
    CHECK(my_file->get_blocks().size() == 0);
}