/*
 * This file is part of smolbote. It's copyrighted by the contributors recorded
 * in the version control history of the file, available from its original
 * location: https://neueland.iserlohn-fortress.net/gitea/aqua/smolbote
 *
 * SPDX-License-Identifier: GPL-3.0
 */

#ifndef SMOLBOTE_CONFIGURATION_H
#define SMOLBOTE_CONFIGURATION_H

#include <initializer_list>
#include <memory>
#include <optional>
#include <string>
#include <type_traits>
#include <unordered_map>
#include <variant>

#if defined(__clang__)
#define consumable(X) [[clang::consumable(X)]]
#define return_typestate(X) [[clang::return_typestate(X)]]
#define callable_when(X) [[clang::callable_when(X)]]
#define param_typestate(X) [[clang::param_typestate(X)]]
#else
#define consumable(X)
#define return_typestate(X)
#define callable_when(X)
#define param_typestate(X)
#endif

typedef std::variant<std::string, int, bool> conf_value_t;

template <typename T>
concept concept_value_t = std::is_arithmetic<T>::value || std::is_same<T, bool>::value || std::is_constructible<T, std::string>::value;

class consumable(unconsumed) Configuration : private std::unordered_map<std::string, conf_value_t>
{
    friend std::ostream &operator<<(std::ostream &out, const Configuration &obj);

public:
    return_typestate(unconsumed) Configuration();
    return_typestate(unconsumed) Configuration(std::initializer_list<std::pair<std::string, conf_value_t>> l) noexcept;

    Configuration(const Configuration &) = delete;
    Configuration &operator=(const Configuration &) = delete;

    return_typestate(unconsumed) Configuration(Configuration && other param_typestate(unconsumed)) = default;
    Configuration &operator=(Configuration &&) = delete;

    ~Configuration() = default;

    callable_when(unconsumed) void read_file(const std::string &location);
    callable_when(unconsumed) void read(std::basic_istream<char> & input);

    template <typename T>
    callable_when(unconsumed) [[nodiscard]] std::optional<T> value(const char *path) const
    {
        if(use_global) {
            return instance()->value<T>(path);
        }

        if(count(path) == 0) {
            return std::nullopt;
        }

        return std::get<T>(at(path));
    }

    template <concept_value_t T>
    callable_when(unconsumed) [[nodiscard]] std::optional<T> value(const char *path) const
    {
        if(use_global) {
            return instance()->value<T>(path);
        }

        if(this->count(path) == 0) {
            return std::nullopt;
        }

        // path is guaranteed to exist
        const auto value = at(path);

        if(std::holds_alternative<int>(value)) {
            if constexpr(std::is_arithmetic<T>::value) {
                return static_cast<T>(std::get<int>(value));
            } else if constexpr(std::is_constructible<T, std::string>::value) {
                return T{ std::to_string(std::get<int>(value)) };
            }

        } else if(std::holds_alternative<bool>(value)) {
            if constexpr(std::is_constructible<T, bool>::value) {
                return std::get<bool>(value);
            } else if constexpr(std::is_constructible<T, const char *>::value) {
                return std::get<bool>(value) ? T{ "true" } : T{ "false" };
            }

        } else if(std::holds_alternative<std::string>(value)) {
            auto str = std::get<std::string>(value);

            try {
                if constexpr(std::is_floating_point<T>::value) {
                    return static_cast<T>(std::stod(str));
                } else if constexpr(std::is_arithmetic<T>::value) {
                    return static_cast<T>(std::stol(str));
                }
            } catch(std::invalid_argument &) {
                return std::nullopt;
            }

            if(str.front() == '~') {
                str.replace(0, 1, m_homePath);
            }

            if constexpr(std::is_constructible<T, std::string>::value) {
                return T{ str };
            }
        }

        return std::nullopt;
    }

    template <typename T>
    callable_when(unconsumed) T &shortcut(T & /* unused */, const char * /* unused */) const
    {
        return T{};
    }

    static void move_global(std::unique_ptr<Configuration> && conf);

private:
    static Configuration *instance();

    const std::string m_homePath;
    const bool use_global = false;
};

#ifndef NO_QT_SPEC
#include "qt_specialization.h"
#endif

std::ostream &operator<<(std::ostream &out, const Configuration &obj);

#endif // SMOLBOTE_CONFIGURATION_H