#pragma once

#include <cstdlib>
#include <string>
#include <string_view>
#include <fstream>
#include <filesystem>
#include <chrono>
#include <vector>
#include <functional>
#include <thread>
#include <mutex>
#include <condition_variable>

namespace common_helpers {

class KillableWorker
{
private:
    std::thread thread_obj{};

    // don't run immediately, wait some time
    std::chrono::milliseconds initial_delay{};
    // time between each invokation
    std::chrono::milliseconds polling_time{};

    std::function<bool()> should_kill{};

    std::function<bool(void *)> thread_job;

    std::mutex kill_thread_mutex{};
    std::condition_variable kill_thread_cv{};
    bool kill_thread{};
    
    void thread_proc(void *data);

public:
    KillableWorker(
        std::function<bool(void *)> thread_proc = {},
        std::chrono::milliseconds initial_delay = {},
        std::chrono::milliseconds polling_time = {},
        std::function<bool()> should_kill = {});
    ~KillableWorker();

    KillableWorker& operator=(const KillableWorker &other);

    // spawn the thread if necessary
    bool start(void *data = nullptr);
    // kill the thread if necessary
    void kill();
};

bool create_dir(std::string_view dir);
bool create_dir(std::wstring_view dir);

void write(std::ofstream &file, std::string_view data);

std::wstring str_to_w(std::string_view str);
std::string wstr_to_a(std::wstring_view wstr);

bool starts_with_i(std::string_view target, std::string_view query);
bool starts_with_i(std::wstring_view target, std::wstring_view query);

bool ends_with_i(std::string_view target, std::string_view query);
bool ends_with_i(std::wstring_view target, std::wstring_view query);

std::string string_strip(std::string_view str);

std::string uint8_vector_to_hex_string(const std::vector<uint8_t> &v);

bool str_cmp_insensitive(std::string_view str1, std::string_view str2);
bool str_cmp_insensitive(std::wstring_view str1, std::wstring_view str2);

std::string ascii_to_lowercase(std::string data);

void thisThreadYieldFor(std::chrono::microseconds u);

void consume_bom(std::ifstream &input);

std::string to_lower(std::string_view str);
std::wstring to_lower(std::wstring_view wstr);

std::string to_upper(std::string_view str);
std::wstring to_upper(std::wstring_view wstr);

std::string to_absolute(std::string_view path, std::string_view base = std::string_view());
std::wstring to_absolute(std::wstring_view path, std::wstring_view base = std::wstring_view());

bool file_exist(const std::filesystem::path &filepath);
bool file_exist(const std::string &filepath);
bool file_exist(const std::wstring &filepath);

bool file_size(const std::filesystem::path &filepath, size_t &size);
bool file_size(const std::string &filepath, size_t &size);
bool file_size(const std::wstring &filepath, size_t &size);

bool dir_exist(const std::filesystem::path &dirpath);
bool dir_exist(const std::string &dirpath);
bool dir_exist(const std::wstring &dirpath);

bool remove_file(const std::filesystem::path &filepath);
bool remove_file(const std::string &filepath);
bool remove_file(const std::wstring &filepath);

// between 0 and max, 0 and max are included
size_t rand_number(size_t max);

std::string get_utc_time();

std::wstring to_wstr(std::string_view str);
std::string to_str(std::wstring_view wstr);

std::string str_replace_all(std::string_view source, std::string_view substr, std::string_view replace);

}