// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

// NOLINTBEGIN
#undef NDEBUG

#include <array>
#include <cassert>
#include <filesystem>
#include <iostream>

#include <aws/store/filesystem/posixFileSystem.hpp>
#include <aws/store/stream/fileStream.hpp>
#include <aws/store/stream/memoryStream.hpp>
#include <sys/resource.h>

void do_memory() {
    rusage rusage{};
    getrusage(RUSAGE_SELF, &rusage);

    printf("resident size max: %lu KB\n", (rusage.ru_maxrss) / 1024);
}

class MyLogger final : public aws::store::logging::Logger {
    void log(const aws::store::logging::LogLevel l, const std::string &msg) const override {
        if (l < level) {
            return;
        }
        if (l >= aws::store::logging::LogLevel::Warning) {
            std::cerr << l << " " << msg << std::endl;
        } else {
            std::cout << l << " " << msg << std::endl;
        }
    }
};

int main() {
    srand(static_cast<uint32_t>(time(nullptr)));
    auto data = std::array<char, 128>{};
    for (char &i : data) {
        i = static_cast<char>((rand() % 64) + 64);
    }

    constexpr int NUM_RECORDS = 100000;
    constexpr bool use_kv = false;

    std::filesystem::remove_all(std::filesystem::current_path() / "stream1");

    auto start = std::chrono::high_resolution_clock::now();
    {
        auto fs =
            std::make_shared<aws::store::filesystem::PosixFileSystem>(std::filesystem::current_path() / "stream1");
        auto logger = std::make_shared<MyLogger>();

        if (use_kv) {
            auto kv_or = aws::store::kv::KV::openOrCreate(aws::store::kv::KVOptions{
                false,
                fs,
                logger,
                "m",
                16 * 1024 * 1024,
            });
            if (!kv_or.ok()) {
                std::cerr << kv_or.err().msg << std::endl;
                std::terminate();
            }
            auto kv = kv_or.val();
            for (int i = 0; i < NUM_RECORDS; i++) {
                auto x =
                    kv->put("key" + std::to_string(i), aws::store::common::BorrowedSlice{data.data(), data.size()});
            }
        } else {
            // auto s = MemoryStream::openOrCreate(StreamOptions{.maximum_size_bytes = 500 * 1024 * 1024});
            auto s_or = aws::store::stream::FileStream::openOrCreate(aws::store::stream::StreamOptions{
                1024 * 1024,
                10 * 1024 * 1024,
                false,
                fs,
                logger,
                aws::store::kv::KVOptions{
                    false,
                    fs,
                    logger,
                    "m",
                    512 * 1024,
                },
            });
            if (!s_or.ok()) {
                std::cerr << s_or.err().msg << std::endl;
                std::terminate();
            }
            auto s = s_or.val();

            std::cout << "loaded checkpoint: "
                      << s->openOrCreateIterator("a", aws::store::stream::IteratorOptions{}).sequence_number
                      << std::endl;

            aws::store::common::Expected<uint64_t, aws::store::stream::StreamError> last_sequence_number{0};
            for (int i = 0; i < NUM_RECORDS; i++) {
                last_sequence_number = s->append(aws::store::common::BorrowedSlice{data.data(), data.size()},
                                                 aws::store::stream::AppendOptions{});
            }

            auto last_record_or = s->read(last_sequence_number.val(), aws::store::stream::ReadOptions{});
            if (last_record_or.ok()) {
                std::cout << last_record_or.val().data.string() << std::endl;
            } else {
                std::cerr << last_record_or.err().msg << std::endl;
            }

            for (auto r : s->openOrCreateIterator("a", aws::store::stream::IteratorOptions{})) {
                if (r.ok()) {
                    // Do something with the record....
                    // std::cout << r.val().sequence_number << std::endl;
                    r.val().checkpoint();
                } else {
                    std::cout << r.err().msg << std::endl;
                    break;
                }
            }

            std::cout << "last checkpoint: "
                      << s->openOrCreateIterator("a", aws::store::stream::IteratorOptions{}).sequence_number
                      << std::endl;
            (void)s->deleteIterator("a");
            std::cout << "after deleting iterator: "
                      << s->openOrCreateIterator("a", aws::store::stream::IteratorOptions{}).sequence_number
                      << std::endl;
        }
    }

    auto end = std::chrono::high_resolution_clock::now();
    std::cout << std::chrono::duration_cast<std::chrono::milliseconds>(end - start).count() << "ms" << std::endl;
    do_memory();
}
// NOLINTEND