#pragma once

#include <tuple>
#include <array>
#include <string_view>
#include <source_location>

#include "TypeTraits.h"

namespace clice::refl {

namespace impl {

struct Any {
    consteval Any(std::size_t);

    template <typename T>
    consteval operator T () const;
};

template <typename T, std::size_t N>
consteval auto test() {
    return []<std::size_t... I>(std::index_sequence<I...>) {
        return requires { T{Any(I)...}; };
    }(std::make_index_sequence<N>{});
}

template <typename T, std::size_t N = 0>
consteval auto member_count() {
    if constexpr(test<T, N>() && !test<T, N + 1>()) {
        return N;
    } else {
        return member_count<T, N + 1>();
    }
}

template <typename T>
struct wrapper {
    T value;

    constexpr wrapper(T value) : value(value) {}
};

template <typename T>
struct storage {
    inline static T value;
};

template <wrapper T>
consteval auto member_name() {
    std::string_view name = std::source_location::current().function_name();
#if __GNUC__ && (!__clang__) && (!_MSC_VER)
    std::size_t start = name.rfind("::") + 2;
    std::size_t end = name.rfind(')');
    name = name.substr(start, end - start);
#elif __clang__
    std::size_t start = name.rfind(".") + 1;
    std::size_t end = name.rfind('}');
    name = name.substr(start, end - start);
#elif _MSC_VER
    std::size_t start = name.rfind("->") + 2;
    std::size_t end = name.rfind('}');
    name = name.substr(start, end - start);
#else
    static_assert(false, "Not supported compiler");
#endif
    if(name.rfind("::") != std::string_view::npos) {
        name = name.substr(name.rfind("::") + 2);
    }
    return name;
}

}  // namespace impl

template <typename T>
struct Struct {
    constexpr inline static bool reflectable =
        std::is_aggregate_v<std::remove_cvref_t<T>> &&
        std::is_default_constructible_v<std::remove_cvref_t<T>>;

    constexpr static std::size_t member_count() {
        return impl::member_count<T>();
    }

    constexpr static T& instance() {
        return impl::storage<T>::value;
    }

    template <typename Object>
    constexpr static auto collcet_members(Object&& object) {
        // clang-format off
        constexpr std::size_t count = member_count();
        if constexpr(count == 0) {
            return std::tuple{};
        } else if constexpr(count == 1) {
            auto&& [a] = object;
            return std::tuple{&a};
        } else if constexpr(count == 2) {
            auto&& [a, b] = object;
            return std::tuple{&a, &b};
        } else if constexpr(count == 3) {
            auto&& [a, b, c] = object;
            return std::tuple{&a, &b, &c};
        } else if constexpr(count == 4) {
            auto&& [a, b, c, d] = object;
            return std::tuple{&a, &b, &c, &d};
        } else if constexpr(count == 5) {
            auto&& [a, b, c, d, e] = object;
            return std::tuple{&a, &b, &c, &d, &e};
        } else if constexpr(count == 6) {
            auto&& [a, b, c, d, e, f] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f};
        } else if constexpr(count == 7) {
            auto&& [a, b, c, d, e, f, g] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g};
        } else if constexpr(count == 8) {
            auto&& [a, b, c, d, e, f, g, h] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h};
        } else if constexpr(count == 9) {
            auto&& [a, b, c, d, e, f, g, h, i] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i};
        } else if constexpr(count == 10) {
            auto&& [a, b, c, d, e, f, g, h, i, j] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j};
        } else if constexpr(count == 11) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k};
        } else if constexpr(count == 12) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l};
        } else if constexpr(count == 13) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m};
        } else if constexpr(count == 14) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n};
        } else if constexpr(count == 15) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o};
        } else if constexpr(count == 16) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p};
        } else if constexpr(count == 17) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q};
        } else if constexpr(count == 18) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q, &r};
        } else if constexpr(count == 19) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q, &r, &s};
        } else if constexpr(count == 20) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q, &r, &s, &t};
        } else if constexpr(count == 21) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q, &r, &s, &t, &u};
        } else if constexpr(count == 22) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q, &r, &s, &t, &u, &v};
        } else if constexpr(count == 23) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v, w] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q, &r, &s, &t, &u, &v, &w};
        } else if constexpr(count == 24) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v, w, x] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q, &r, &s, &t, &u, &v, &w, &x};
        } else if constexpr(count == 25) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v, w, x, y] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q, &r, &s, &t, &u, &v, &w, &x, &y};
        } else if constexpr(count == 26) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v, w, x, y, z] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q, &r, &s, &t, &u, &v, &w, &x, &y, &z};
        } else if constexpr(count == 27) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v, w, x, y, z, _0] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q, &r, &s, &t, &u, &v, &w, &x, &y, &z, &_0};
        } else if constexpr(count == 28) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v, w, x, y, z, _0, _1] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q, &r, &s, &t, &u, &v, &w, &x, &y, &z, &_0, &_1};
        } else if constexpr(count == 29) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v, w, x, y, z, _0, _1, _2] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q, &r, &s, &t, &u, &v, &w, &x, &y, &z, &_0, &_1, &_2};
        } else if constexpr(count == 30) {
            auto&& [a, b, c, d, e, f, g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v, w, x, y, z, _0, _1, _2, _3] = object;
            return std::tuple{&a, &b, &c, &d, &e, &f, &g, &h, &i, &j, &k, &l, &m, &n, &o, &p, &q, &r, &s, &t, &u, &v, &w, &x, &y, &z, &_0, &_1, &_2, &_3};
        } else {
            static_assert(count <= 30, "Not supported member count");
        }
        // clang-format on
    }
};

/// To check if the type is reflectable.
template <typename T>
concept reflectable = Struct<T>::reflectable;

template <typename... Ts>
struct Inheritance : Ts... {};

/// Use to define a reflectable struct with inheritance.
#define inherited_struct(name, ...)                                                                \
    struct name##Body;                                                                             \
    using name = clice::refl::Inheritance<__VA_ARGS__, name##Body>;                             \
    struct name##Body

template <typename... Ts>
struct Struct<Inheritance<Ts...>> {
    constexpr inline static bool reflectable = (refl::reflectable<Ts> && ...);

    constexpr static std::size_t member_count() {
        return (Struct<Ts>::member_count() + ...);
    }

    constexpr static auto& instance() {
        return impl::storage<Inheritance<Ts...>>::value;
    }

    template <typename Object>
    constexpr static auto collcet_members(Object&& object) {
        return std::tuple_cat(Struct<Ts>::collcet_members(static_cast<Ts&>(object))...);
    }
};

template <typename T>
using member_types =
    tuple_to_list_t<decltype(Struct<T>::collcet_members(std::declval<T>())), std::remove_pointer_t>;

/// Turn the return value of the callable to bool.
template <typename Callable>
constexpr auto foldable(const Callable& callable) {
    return [&](auto&&... args) {
        using Ret = std::invoke_result_t<Callable, decltype(args)...>;
        if constexpr(std::is_void_v<Ret>) {
            callable(args...);
            return true;
        } else {
            return bool(callable(args...));
        }
    };
}

template <reflectable Object, typename Callback>
constexpr bool foreach(Object&& object, const Callback& callback) {
    using S = Struct<std::remove_cvref_t<Object>>;
    constexpr auto count = S::member_count();
    auto members = S::collcet_members(object);
    constexpr auto static_members = S::collcet_members(S::instance());
    return [&]<std::size_t... Is>(std::index_sequence<Is...>) {
        return (foldable(callback)(impl::member_name<std::get<Is>(static_members)>(),
                                   *std::get<Is>(members)) &&
                ...);
    }(std::make_index_sequence<count>{});
}

/// Invoke callback for each member of lhs and rhs, return false
/// in callback to abort the iteration. Return true if all members are visited.
template <reflectable LHS, reflectable RHS, typename Callback>
constexpr bool foreach(LHS&& lhs, RHS&& rhs, const Callback& callback) {
    using L = Struct<std::remove_cvref_t<LHS>>;
    using R = Struct<std::remove_cvref_t<RHS>>;
    static_assert(L::member_count() == R::member_count(), "Member count mismatch");
    return [&]<std::size_t... Is>(std::index_sequence<Is...>) {
        return (foldable(callback)(*std::get<Is>(L::collcet_members(lhs)),
                                   *std::get<Is>(R::collcet_members(rhs))) &&
                ...);
    }(std::make_index_sequence<L::member_count()>{});
}

}  // namespace clice::refl