/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * Copyright 2019, Danny Robson <danny@nerdcruft.net>
 */

#pragma once

#include "fwd.hpp"

#include "../platform.hpp"
#include "../debug/panic.hpp"
#include "../introspection/name.hpp"
#include "../log.hpp"
#include "../typeidx.hpp"
#include "../view.hpp"

#include <map>

namespace cruft::parse::enumeration {
    namespace detail {
        /// Map a type to the underlying enum type if it is an enum, else
        /// provided the identity mapping. Useful so simplify template logic
        /// when dealing with types that aren't quite enums proper.
        template <typename EnumT, typename = void>
        struct underlying_else_identity { using type = EnumT; };


        // Specialise for the enum case.
        template <typename EnumT>
        struct underlying_else_identity<
            EnumT,
            std::enable_if_t<std::is_enum_v<EnumT>>
        > { using type = std::underlying_type_t<EnumT>; };


        template <typename EnumT>
        struct underlying_else_identity<
            EnumT,
            std::void_t<typename EnumT::underlying_type>
        > { using type = typename underlying_else_identity<typename EnumT::underlying_type>::type; };


        template <typename EnumT>
        using underlying_else_identity_t = typename underlying_else_identity<EnumT>::type;


        /// Abstract base class for lookups from string names to typed
        /// integers.
        ///
        /// An exception should be thrown if the type is incorrect. This
        /// shouldn't be the case because the caller is expected to know the
        /// type beforehand.
        ///
        /// Do not perform type casting - lossless or otherwise.
        struct lookup_base {
            virtual ~lookup_base () = default;

            virtual i08 as_i08 (cruft::view<char const*> &) const& = 0;
            virtual i16 as_i16 (cruft::view<char const*> &) const& = 0;
            virtual i32 as_i32 (cruft::view<char const*> &) const& = 0;
            virtual i64 as_i64 (cruft::view<char const*> &) const& = 0;

            virtual u08 as_u08 (cruft::view<char const*> &) const& = 0;
            virtual u16 as_u16 (cruft::view<char const*> &) const& = 0;
            virtual u32 as_u32 (cruft::view<char const*> &) const& = 0;
            virtual u64 as_u64 (cruft::view<char const*> &) const& = 0;
        };


        template <typename EnumT>
        struct lookup_callback : public lookup_base {
            using function_t =  std::function<EnumT(cruft::view<char const*>&)>;

            lookup_callback (
                 function_t &&_callback
            ) : m_callback (std::move (_callback))
            { ; }

            i08 as_i08 (cruft::view<char const*> &str) const& override { return get<i08> (str); }
            i16 as_i16 (cruft::view<char const*> &str) const& override { return get<i16> (str); }
            i32 as_i32 (cruft::view<char const*> &str) const& override { return get<i32> (str); }
            i64 as_i64 (cruft::view<char const*> &str) const& override { return get<i64> (str); }

            u08 as_u08 (cruft::view<char const*> &str) const& override { return get<u08> (str); }
            u16 as_u16 (cruft::view<char const*> &str) const& override { return get<u16> (str); }
            u32 as_u32 (cruft::view<char const*> &str) const& override { return get<u32> (str); }
            u64 as_u64 (cruft::view<char const*> &str) const& override { return get<u64> (str); }

        private:
            template <typename ValueT>
            ValueT
            get (cruft::view<char const*> &str) const&
            {
                if constexpr (std::is_same_v<ValueT, underlying_else_identity_t<EnumT>>) {
                    // Double check the callee changes the remaining string size.
                    auto const len = str.size ();
                    auto const res = m_callback (str);
                    if (len == str.size ())
                        throw std::invalid_argument ("Unknown enum value");
                    return static_cast<ValueT> (res);
                } else {
                    throw std::runtime_error ("Invalid underlying type");
                }
            }


            function_t m_callback;
        };


        /// A simple lookup from a supplied mapping of strings-to-underlying,
        /// specialised on the EnumT that the lookup is targetting.
        template <typename EnumT>
        struct lookup_concrete : public lookup_base {
            using underlying_type = underlying_else_identity_t<EnumT>;
            using cache_type = std::map<std::string_view, EnumT>;

            lookup_concrete (cache_type _cache)
                : m_cache (std::move (_cache))
            { ; }

            i08 as_i08 (cruft::view<char const*> &str) const& override { return get<i08> (str); }
            i16 as_i16 (cruft::view<char const*> &str) const& override { return get<i16> (str); }
            i32 as_i32 (cruft::view<char const*> &str) const& override { return get<i32> (str); }
            i64 as_i64 (cruft::view<char const*> &str) const& override { return get<i64> (str); }

            u08 as_u08 (cruft::view<char const*> &str) const& override { return get<u08> (str); }
            u16 as_u16 (cruft::view<char const*> &str) const& override { return get<u16> (str); }
            u32 as_u32 (cruft::view<char const*> &str) const& override { return get<u32> (str); }
            u64 as_u64 (cruft::view<char const*> &str) const& override { return get<u64> (str); }

        private:
            /// A common lookup helper for strings-to-some_value_type.
            /// Provided so that we don't need to replicate the lookup and
            /// error logic for each integer case.
            template <typename ValueT>
            ValueT
            get (cruft::view<char const*> &str) const&
            {
                static_assert (std::is_integral_v<ValueT>);

                if constexpr (std::is_same_v<ValueT, underlying_type>) {
                    for (auto const &[key,val]: m_cache) {
                        if (equal (key, str)) {
                            str = str.consume (key.size ());
                            // We must cast the value here so that enum
                            // classes are supported.
                            return static_cast<ValueT> (val);
                        }
                    }

                    throw std::invalid_argument ("Unknown enum value");
                } else {
                    throw std::runtime_error ("Invalid underlying type");
                }
            }

            cache_type m_cache;
        };


        /// Resolves a global map from typeidx to lookup providers for the
        /// enumeration type.
        std::map<
            int,
            std::unique_ptr<lookup_base>
        >&
        cache (void);


        template <typename EnumT>
        cookie
        setup [[nodiscard]] (std::unique_ptr<lookup_base> &&lookup)
        {
            auto &cache = detail::cache ();
            auto const index = cruft::typeidx<EnumT> ();

            auto [pos, success] = cache.insert ({
                index,
                std::move (lookup)
            });

            if (!success)
                // clang#XXX
                // fmtlib#1824
                //     generates "unused type alias 'char_type'" under clang
                #if defined(COMPILER_CLANG)
                #pragma GCC diagnostic push
                #pragma GCC diagnostic ignored "-Wunused-local-typedef"
                #endif
                LOG_WARN (
                    "duplicate parse setup for {} was ignored",
                    cruft::introspection::name::bare<EnumT> ()
                );
                #if defined(COMPILER_CLANG)
                #pragma GCC diagnostic push -Wno-unused-local-typedef
                #endif

            return cookie {};
        }
    };


    /// Establish a mapping from strings-to-enums and return a cookie that
    /// will deregister the mapping at destruction time.
    ///
    /// Results when calling multiple times for the same type are undefined
    /// though some form of runtime warning will likely be emitted, and it's
    /// unlikely to crash your application).
    template <typename EnumT>
    cookie
    setup [[nodiscard]] (std::map<std::string_view, EnumT> mapping)
    {
        return detail::setup<EnumT> (
            std::make_unique<
                detail::lookup_concrete<EnumT>
            > (std::move (mapping))
        );
    };


    template <typename EnumT>
    cookie
    setup [[nodiscard]] (std::function<EnumT(cruft::view<char const*>&)> &&func)
    {
        return detail::setup<EnumT> (
            std::make_unique<
                detail::lookup_callback<EnumT>
            > (std::move (func))
        );
    }


    /// Returns true if a typeidx has been registered for enumeration
    /// reflection.
    inline
    bool is_registered (int idx)
    {
        auto const &dat = detail::cache ();
        auto const  pos = dat.find (idx);
        return pos != dat.cend ();
    }


    /// Returns true if a typename has been registered for enumation
    /// reflection.
    template <typename ValueT>
    bool is_registered (void)
    {
        return is_registered (cruft::typeidx<ValueT> ());
    }


    /// Lookup an enumeration value given a string representation.
    ///
    /// The string view is taken by reference and will be modified so that it
    /// points to the remainder of the string after the parsed value.
    template <typename EnumT>
    EnumT
    value (int const idx, cruft::view<char const*> &str)
    {
        auto const &reg = detail::cache ();
        auto const  pos = reg.find (idx);

        if (pos == reg.cend ())
            throw std::invalid_argument ("Unknown enumeration");

        auto const &obj = *pos->second;

        static_assert (
            sizeof (EnumT) == 1 ||
            sizeof (EnumT) == 2 ||
            sizeof (EnumT) == 4 ||
            sizeof (EnumT) == 8
        );

        using underlying_type = detail::underlying_else_identity_t<EnumT>;
        static_assert (std::is_integral_v<underlying_type>);

        if constexpr (std::is_signed_v<underlying_type>) {
            if constexpr (sizeof (EnumT) == 1) return static_cast<EnumT> (obj.as_i08 (str));
            if constexpr (sizeof (EnumT) == 2) return static_cast<EnumT> (obj.as_i16 (str));
            if constexpr (sizeof (EnumT) == 4) return static_cast<EnumT> (obj.as_i32 (str));
            if constexpr (sizeof (EnumT) == 8) return static_cast<EnumT> (obj.as_i64 (str));

            unreachable ();
        } else {
            if constexpr (sizeof (EnumT) == 1) return static_cast<EnumT> (obj.as_u08 (str));
            if constexpr (sizeof (EnumT) == 2) return static_cast<EnumT> (obj.as_u16 (str));
            if constexpr (sizeof (EnumT) == 4) return static_cast<EnumT> (obj.as_u32 (str));
            if constexpr (sizeof (EnumT) == 8) return static_cast<EnumT> (obj.as_u64 (str));

            unreachable ();
        }
    }


    template <typename EnumT>
    EnumT
    value (cruft::view<char const*> &str)
    {
        return value<EnumT> (typeidx<EnumT> (), str);
    }


    /// Lookup an enumeration value from the string representation given a
    /// typeidx.
    ///
    /// If the entire string isn't consumed an exception will be thrown.
    template <typename EnumT>
    EnumT
    from_string (int const idx, cruft::view<char const*> src)
    {
        auto const res = value<EnumT> (idx, src);
        if (!src.empty ())
            throw std::runtime_error ("Invalid conversion");
        return res;
    }


    /// Lookup an enumeration value from the string representation
    ///
    /// If the entire string isn't consumed an exception will be thrown.
    template <typename EnumT>
    EnumT
    from_string (cruft::view<char const*> src)
    {
        return from_string<EnumT> (typeidx<EnumT> (), src);
    }
}