CompoundTagVariant.h

#pragma once
 
#include "mc/_HeaderOutputPredefine.h"
#include "mc/nbt/ByteArrayTag.h"
#include "mc/nbt/ByteTag.h"
#include "mc/nbt/CompoundTag.h"
#include "mc/nbt/DoubleTag.h"
#include "mc/nbt/EndTag.h"
#include "mc/nbt/FloatTag.h"
#include "mc/nbt/Int64Tag.h"
#include "mc/nbt/IntArrayTag.h"
#include "mc/nbt/IntTag.h"
#include "mc/nbt/ListTag.h"
#include "mc/nbt/ShortTag.h"
#include "mc/nbt/StringTag.h"
 
#include "ll/api/Expected.h"
#include "ll/api/base/Concepts.h"
#include "ll/api/base/Meta.h"
 
// auto generated forward declare list
// clang-format off
class ByteArrayTag;
class ByteTag;
class CompoundTag;
class DoubleTag;
class EndTag;
class FloatTag;
class Int64Tag;
class IntArrayTag;
class IntTag;
class ListTag;
class ShortTag;
class StringTag;
class Tag;
// clang-format on
 
class CompoundTagVariant {
public:
    // CompoundTagVariant inner types define
    using Types = ::ll::meta::TypeList<
        ::EndTag,
        ::ByteTag,
        ::ShortTag,
        ::IntTag,
        ::Int64Tag,
        ::FloatTag,
        ::DoubleTag,
        ::ByteArrayTag,
        ::StringTag,
        ::ListTag,
        ::CompoundTag,
        ::IntArrayTag>;
 
    using Variant = Types::to<::std::variant>;
 
    template <bool Const>
    class Iterator {
    public:
        using iterator_category = std::bidirectional_iterator_tag;
        using difference_type   = ptrdiff_t;
        using value_type        = Tag;
        using reference         = std::conditional_t<Const, value_type const, value_type>&;
        using pointer           = std::add_pointer_t<reference>;
 
        std::conditional_t<
            Const,
            std::variant<CompoundTagVariant const*, CompoundTag::const_iterator, ListTag::const_iterator>,
            std::variant<CompoundTagVariant*, CompoundTag::iterator, ListTag::iterator>>
            iter;
 
        static Iterator makeBegin(auto& var) noexcept {
            Iterator res;
            switch (var.index()) {
            case Tag::List:
                res.iter.emplace<2>(var.get<ListTag>().begin());
                break;
            case Tag::Compound:
                res.iter.emplace<1>(var.get<CompoundTag>().begin());
                break;
            case Tag::End:
                res.iter.emplace<0>(std::addressof(var) + 1);
                break;
            default:
                res.iter.emplace<0>(std::addressof(var));
            }
            return res;
        }
 
        static Iterator makeEnd(auto& var) noexcept {
            Iterator res;
            switch (var.index()) {
            case Tag::List:
                res.iter.emplace<2>(var.get<ListTag>().end());
                break;
            case Tag::Compound:
                res.iter.emplace<1>(var.get<CompoundTag>().end());
                break;
            default:
                res.iter.emplace<0>(std::addressof(var) + 1);
            }
            return res;
        }
 
        [[nodiscard]] reference operator*() const noexcept {
            switch (iter.index()) {
            case 0:
                return std::get<0>(iter)->get();
            case 1:
                return std::get<1>(iter)->second.get();
            case 2:
                return *std::get<2>(iter)->get();
            default:
                LL_UNREACHABLE;
            }
            LL_UNREACHABLE;
        }
 
        [[nodiscard]] pointer operator->() const noexcept { return std::addressof(**this); }
 
        Iterator& operator++() noexcept {
            std::visit([](auto& val) { ++val; }, iter);
            return *this;
        }
 
        Iterator operator++(int) noexcept {
            Iterator tmp = *this;
            ++*this;
            return tmp;
        }
 
        Iterator& operator--() noexcept {
            std::visit([](auto& val) { --val; }, iter);
            return *this;
        }
 
        Iterator operator--(int) noexcept {
            Iterator tmp = *this;
            --*this;
            return tmp;
        }
 
        [[nodiscard]] bool operator==(Iterator const& r) const noexcept { return this->iter == r.iter; }
    };
 
    using iterator       = Iterator<false>;
    using const_iterator = Iterator<true>;
 
public:
    // member variables
    Variant mTagStorage;
 
public:
    // destructor thunk
    // NOLINTBEGIN
    MCFOLD void $dtor();
    // NOLINTEND
 
public:
    LLNDAPI static ll::Expected<CompoundTagVariant>
    parse(std::string_view snbt, optional_ref<size_t> parsedLength = std::nullopt) noexcept;
 
    [[nodiscard]] constexpr CompoundTagVariant() = default;
 
    [[nodiscard]] constexpr CompoundTagVariant(CompoundTagVariant&&) = default;
 
    LL_MAY_CONSTEXPR CompoundTagVariant& operator=(CompoundTagVariant&&) = default;
 
    [[nodiscard]] constexpr CompoundTagVariant(CompoundTagVariant const&) = default;
 
    LL_MAY_CONSTEXPR CompoundTagVariant& operator=(CompoundTagVariant const&) = default;
 
    template <class T>
        requires(requires(T o) { mTagStorage = std::move(o); })
    constexpr CompoundTagVariant& operator=(T other) {
        mTagStorage = std::move(other);
        return *this;
    }
 
    [[nodiscard]] constexpr CompoundTagVariant(std::nullptr_t) {}
 
    [[nodiscard]] bool operator==(CompoundTagVariant const& other) const { return get() == other.get(); }
 
    [[nodiscard]] CompoundTagVariant(Variant tag) : mTagStorage(std::move(tag)) {}
 
    template <class T, class... Args>
    [[nodiscard]] constexpr CompoundTagVariant(std::in_place_type_t<T>, Args&&... args)
    : mTagStorage(std::in_place_type<T>, std::forward<Args>(args)...) {}
 
    [[nodiscard]] constexpr CompoundTagVariant(std::initializer_list<CompoundTag::TagMap::value_type> tagPairs)
    : mTagStorage(std::in_place_type<CompoundTag>, tagPairs) {}
 
    [[nodiscard]] CompoundTagVariant(UniqueTagPtr&& tag) {
        if (!tag) {
            return;
        }
        ::ll::meta::visitIndex<Types::size>(static_cast<size_t>(tag->getId()), [&]<size_t I> {
            mTagStorage = std::move((Types::get<I>&)*tag);
        });
    }
    [[nodiscard]] CompoundTagVariant(UniqueTagPtr const& tag) : CompoundTagVariant(tag ? tag->copy() : nullptr) {}
    template <std::derived_from<Tag> T>
    [[nodiscard]] constexpr CompoundTagVariant(T tag) : mTagStorage(std::move(tag)) {}
    template <std::integral T>
    [[nodiscard]] constexpr CompoundTagVariant(T integer) {
        constexpr size_t size = sizeof(T);
        if constexpr (size == 1) {
            mTagStorage = ByteTag{integer};
        } else if constexpr (size == 2) {
            mTagStorage = ShortTag{integer};
        } else if constexpr (size == 4) {
            mTagStorage = IntTag{integer};
        } else {
            mTagStorage = Int64Tag{integer};
        }
    }
    [[nodiscard]] constexpr CompoundTagVariant(std::byte b) : mTagStorage(ByteTag{b}) {}
 
    [[nodiscard]] constexpr CompoundTagVariant(float f) : mTagStorage(FloatTag{f}) {}
 
    [[nodiscard]] constexpr CompoundTagVariant(double d) : mTagStorage(DoubleTag{d}) {}
 
    [[nodiscard]] constexpr CompoundTagVariant(std::string s)
    : mTagStorage(std::in_place_type<StringTag>, std::move(s)) {}
 
    [[nodiscard]] constexpr CompoundTagVariant(std::string_view s) : mTagStorage(std::in_place_type<StringTag>, s) {}
 
    template <size_t N>
    [[nodiscard]] constexpr CompoundTagVariant(char const (&str)[N])
    : CompoundTagVariant(std::string_view{str, N - 1}) {}
 
    [[nodiscard]] iterator       begin() noexcept { return iterator::makeBegin(*this); }
    [[nodiscard]] const_iterator begin() const noexcept { return cbegin(); }
    [[nodiscard]] const_iterator cbegin() const noexcept { return const_iterator::makeBegin(*this); }
 
    [[nodiscard]] iterator       end() noexcept { return iterator::makeEnd(*this); }
    [[nodiscard]] const_iterator end() const noexcept { return cend(); }
    [[nodiscard]] const_iterator cend() const noexcept { return const_iterator::makeEnd(*this); }
 
    [[nodiscard]] constexpr Tag::Type index() const noexcept { return Tag::Type(mTagStorage.index()); }
    [[nodiscard]] constexpr Tag::Type getId() const noexcept { return index(); }
 
    template <std::derived_from<Tag> T>
    [[nodiscard]] constexpr bool hold() const noexcept {
        return std::holds_alternative<T>(mTagStorage);
    }
    [[nodiscard]] constexpr bool hold(::Tag::Type type) const noexcept { return getId() == type; }
 
    // consistency with json
    [[nodiscard]] constexpr bool is_array() const noexcept { return hold(Tag::List); }
    [[nodiscard]] constexpr bool is_binary() const noexcept { return hold(Tag::ByteArray) || hold(Tag::IntArray); }
    [[nodiscard]] constexpr bool is_boolean() const noexcept { return hold(Tag::Byte); }
    [[nodiscard]] constexpr bool is_null() const noexcept { return hold(Tag::End); }
    [[nodiscard]] constexpr bool is_number_float() const noexcept { return hold(Tag::Float) || hold(Tag::Double); }
    [[nodiscard]] constexpr bool is_number_integer() const noexcept {
        return hold(Tag::Byte) || hold(Tag::Short) || hold(Tag::Int) || hold(Tag::Int64);
    }
    [[nodiscard]] constexpr bool is_object() const noexcept { return hold(Tag::Compound); }
    [[nodiscard]] constexpr bool is_string() const noexcept { return hold(Tag::String); }
    [[nodiscard]] constexpr bool is_number() const noexcept { return is_number_float() || is_number_integer(); }
    [[nodiscard]] constexpr bool is_primitive() const noexcept {
        return is_null() || is_string() || is_number() || is_binary();
    }
    [[nodiscard]] constexpr bool is_structured() const noexcept { return is_array() || is_object(); }
 
    [[nodiscard]] constexpr CompoundTag::TagMap const& items() const { return get<CompoundTag>().mTags; }
    [[nodiscard]] constexpr CompoundTag::TagMap&       items() { return get<CompoundTag>().mTags; }
 
    [[nodiscard]] bool contains(std::string_view key) const noexcept {
        if (is_object()) {
            return get<CompoundTag>().contains(key);
        }
        return false;
    }
 
    [[nodiscard]] bool contains(std::string_view key, Tag::Type type) const noexcept {
        if (is_object()) {
            return get<CompoundTag>().contains(key, type);
        }
        return false;
    }
 
    template <std::derived_from<Tag> T>
    [[nodiscard]] bool contains(std::string_view key) const noexcept {
        constexpr size_t idx = Types::index<T>;
        return contains(key, (Tag::Type)idx);
    }
 
    [[nodiscard]] constexpr size_t size() const noexcept {
        switch (index()) {
        case Tag::Byte:
        case Tag::Short:
        case Tag::Int:
        case Tag::Int64:
        case Tag::Float:
        case Tag::Double:
        case Tag::String:
            return 1;
        case Tag::List:
            return get<ListTag>().size();
        case Tag::Compound:
            return get<CompoundTag>().size();
        case Tag::IntArray:
            return get<IntArrayTag>().size();
        case Tag::ByteArray:
            return get<ByteArrayTag>().size();
        case Tag::End:
            return 0;
        default:
            LL_UNREACHABLE;
        }
    }
 
    template <std::derived_from<Tag> T>
    [[nodiscard]] constexpr T& get() {
        return std::get<T>(mTagStorage);
    }
 
    template <std::derived_from<Tag> T>
    [[nodiscard]] constexpr T const& get() const {
        return std::get<T>(mTagStorage);
    }
 
    [[nodiscard]] Tag& get() { return reinterpret_cast<Tag&>(mTagStorage); }
 
    [[nodiscard]] Tag const& get() const { return reinterpret_cast<Tag const&>(mTagStorage); }
 
    template <std::derived_from<Tag> T>
    constexpr T& emplace() {
        return mTagStorage.emplace<T>();
    }
 
    [[nodiscard]] UniqueTagPtr& operator[](size_t index) {
        if (hold(Tag::List)) {
            return get<ListTag>()[index];
        } else {
            throw std::runtime_error("tag not hold an array");
        }
    }
 
    [[nodiscard]] UniqueTagPtr const& operator[](size_t index) const {
        if (hold(Tag::List)) {
            return get<ListTag>()[index];
        } else {
            throw std::runtime_error("tag not hold an array");
        }
    }
 
    [[nodiscard]] CompoundTagVariant& operator[](std::string_view index) {
        if (is_null()) {
            mTagStorage = CompoundTag{};
        }
        if (!hold(Tag::Compound)) {
            throw std::runtime_error("tag not hold an object");
        }
        return get<CompoundTag>()[index];
    }
 
    [[nodiscard]] CompoundTagVariant const& operator[](std::string_view index) const {
        if (!hold(Tag::Compound)) {
            throw std::runtime_error("tag not hold an object");
        }
        return get<CompoundTag>()[index];
    }
 
    template <size_t N>
    [[nodiscard]] CompoundTagVariant& operator[](char const (&index)[N]) { // make EDG happy
        return operator[](std::string_view{index, N - 1});
    }
 
    template <size_t N>
    [[nodiscard]] CompoundTagVariant const& operator[](char const (&index)[N]) const { // make EDG happy
        return operator[](std::string_view{index, N - 1});
    }
 
    [[nodiscard]] UniqueTagPtr toUniqueCopy() const& {
        return std::visit(
            [](auto& val) -> std::unique_ptr<Tag> { return std::make_unique<std::decay_t<decltype(val)>>(val); },
            mTagStorage
        );
    }
 
    [[nodiscard]] UniqueTagPtr toUnique() && {
        return std::visit(
            [](auto&& val) -> std::unique_ptr<Tag> {
                return std::make_unique<std::decay_t<decltype(val)>>(std::move(val));
            },
            mTagStorage
        );
    }
 
    LLNDAPI std::string toSnbt(SnbtFormat snbtFormat = SnbtFormat::PrettyFilePrint, uchar indent = 4) const noexcept;
 
    std::string dump(uchar indent = 4, SnbtFormat snbtFormat = SnbtFormat::PrettyFilePrint) const noexcept {
        return toSnbt(snbtFormat, indent);
    }
 
    void push_back(CompoundTagVariant val) {
        if (is_null()) {
            mTagStorage = ListTag{};
        }
        if (!hold(Tag::List)) {
            throw std::runtime_error("tag not hold an array");
        }
        get<ListTag>().add(std::move(val).toUnique());
    }
 
    template <class T>
        requires(std::is_arithmetic_v<T> && !ll::traits::is_char_v<T>)
    [[nodiscard]] constexpr operator T() const {
        if (is_number()) {
            return std::visit(
                [](auto& val) -> T {
                    if constexpr (std::is_convertible_v<std::decay_t<decltype(val)>, T>) {
                        return (T)val;
                    } else {
                        return {};
                    }
                },
                mTagStorage
            );
        } else {
            throw std::runtime_error("tag not hold an number");
        }
    }
    [[nodiscard]] operator std::string const&() const { return get<StringTag>(); }
    [[nodiscard]] operator std::string&() { return get<StringTag>(); }
    [[nodiscard]] operator std::string&&() && { return std::move(get<StringTag>()); }
    [[nodiscard]] operator std::string_view() const { return get<StringTag>(); }
    static CompoundTagVariant object(std::initializer_list<CompoundTag::TagMap::value_type> init = {}) {
        return CompoundTagVariant{std::in_place_type<CompoundTag>, init};
    }
    static CompoundTagVariant array(std::initializer_list<CompoundTagVariant> init = {}) {
        return CompoundTagVariant{std::in_place_type<ListTag>, init};
    }
};
 
[[nodiscard]] inline auto CompoundTag::begin() noexcept { return mTags.begin(); }
[[nodiscard]] inline auto CompoundTag::begin() const noexcept { return mTags.begin(); }
[[nodiscard]] inline auto CompoundTag::end() noexcept { return mTags.end(); }
[[nodiscard]] inline auto CompoundTag::end() const noexcept { return mTags.end(); }
[[nodiscard]] inline auto CompoundTag::rbegin() noexcept { return mTags.rbegin(); }
[[nodiscard]] inline auto CompoundTag::rbegin() const noexcept { return mTags.rbegin(); }
[[nodiscard]] inline auto CompoundTag::rend() noexcept { return mTags.rend(); }
[[nodiscard]] inline auto CompoundTag::rend() const noexcept { return mTags.rend(); }
[[nodiscard]] inline auto CompoundTag::cbegin() const noexcept { return mTags.cbegin(); }
[[nodiscard]] inline auto CompoundTag::cend() const noexcept { return mTags.cend(); }
[[nodiscard]] inline auto CompoundTag::crbegin() const noexcept { return mTags.crbegin(); }
[[nodiscard]] inline auto CompoundTag::crend() const noexcept { return mTags.crend(); }
inline bool               CompoundTag::erase(std::string_view name) {
    if (auto it = mTags.find(name); it != mTags.end()) {
        mTags.erase(it);
        return true;
    }
    return false;
}
[[nodiscard]] inline CompoundTagVariant& CompoundTag::operator[](std::string_view index) {
    if (auto it = mTags.find(index); it != mTags.end()) {
        return it->second;
    }
    return mTags[std::string{index}];
}
[[nodiscard]] inline CompoundTagVariant const& CompoundTag::operator[](std::string_view index) const {
    if (auto it = mTags.find(index); it != mTags.end()) {
        return it->second;
    }
    throw std::out_of_range("invalid nbt key");
}
[[nodiscard]] inline void CompoundTag::rename(std::string_view name, std::string_view newName) {
    if (auto it = mTags.find(name); it != mTags.end()) {
        CompoundTagVariant tmp{std::move(it->second)};
        mTags.erase(it);
        mTags.try_emplace(std::string{newName}, std::move(tmp));
    }
}
 
[[nodiscard]] constexpr ListTag::ListTag(std::vector<CompoundTagVariant> tags) {
    if (tags.empty()) {
        mType = Tag::End;
    } else {
        mType = tags.front().index();
        reserve(tags.size());
        for (auto& tag : tags) {
            emplace_back(std::move(tag).toUnique());
        }
    }
}
[[nodiscard]] constexpr ListTag::ListTag(std::initializer_list<CompoundTagVariant> tags) {
    if (tags.size() == 0) {
        mType = Tag::End;
    } else {
        mType = tags.begin()->index();
        reserve(tags.size());
        for (auto& tag : tags) {
            emplace_back(tag.toUniqueCopy());
        }
    }
}
[[nodiscard]] inline bool operator==(UniqueTagPtr const& l, CompoundTagVariant const& r) {
    return l ? (*l == r.get()) : false;
}
[[nodiscard]] inline bool operator==(CompoundTagVariant const& l, UniqueTagPtr const& r) {
    return r ? (l.get() == *r) : false;
}
 
[[nodiscard]] inline UniqueTagPtr::UniqueTagPtr(CompoundTagVariant&& r) : ptr(std::move(r).toUnique().release()) {}
 
[[nodiscard]] inline UniqueTagPtr::UniqueTagPtr(CompoundTagVariant const& r) : ptr(r.toUniqueCopy().release()) {}
 
inline UniqueTagPtr& UniqueTagPtr::operator=(CompoundTagVariant&& r) {
    reset(std::move(r).toUnique().release());
    return *this;
}
inline UniqueTagPtr& UniqueTagPtr::operator=(CompoundTagVariant const& r) {
    reset(r.toUniqueCopy().release());
    return *this;
}
 
template <std::derived_from<Tag> T>
[[nodiscard]] inline T& UniqueTagPtr::get() const {
    if (hold<T>()) {
        return *static_cast<T*>(ptr);
    }
    throw std::runtime_error("not the expected type");
}
[[nodiscard]] inline bool UniqueTagPtr::contains(std::string_view key) const noexcept {
    if (is_object()) {
        return get<CompoundTag>().contains(key);
    }
    return false;
}
[[nodiscard]] inline bool UniqueTagPtr::contains(std::string_view key, Tag::Type type) const noexcept {
    if (is_object()) {
        return get<CompoundTag>().contains(key, type);
    }
    return false;
}
[[nodiscard]] inline size_t UniqueTagPtr::size() const noexcept {
    switch (index()) {
    case Tag::Byte:
    case Tag::Short:
    case Tag::Int:
    case Tag::Int64:
    case Tag::Float:
    case Tag::Double:
    case Tag::String:
        return 1;
    case Tag::List:
        return get<ListTag>().size();
    case Tag::Compound:
        return get<CompoundTag>().size();
    case Tag::IntArray:
        return get<IntArrayTag>().size();
    case Tag::ByteArray:
        return get<ByteArrayTag>().size();
    case Tag::End:
        return 0;
    default:
        LL_UNREACHABLE;
    }
}
[[nodiscard]] inline UniqueTagPtr::operator std::string const&() const { return get<StringTag>(); }
[[nodiscard]] inline UniqueTagPtr::operator std::string&() & { return get<StringTag>(); }
[[nodiscard]] inline UniqueTagPtr::operator std::string&&() && { return std::move(get<StringTag>()); }
[[nodiscard]] inline UniqueTagPtr::operator std::string_view() const { return get<StringTag>(); }
 
[[nodiscard]] inline UniqueTagPtr& UniqueTagPtr::operator[](size_t index) {
    if (hold(Tag::List)) {
        return get<ListTag>()[index];
    } else {
        throw std::runtime_error("tag not hold an array");
    }
}
[[nodiscard]] inline UniqueTagPtr const& UniqueTagPtr::operator[](size_t index) const {
    if (hold(Tag::List)) {
        return get<ListTag>()[index];
    } else {
        throw std::runtime_error("tag not hold an array");
    }
}
[[nodiscard]] inline CompoundTagVariant& UniqueTagPtr::operator[](std::string_view index) {
    if (is_null()) {
        emplace<CompoundTag>();
    }
    if (!hold(Tag::Compound)) {
        throw std::runtime_error("tag not hold an object");
    }
    return get<CompoundTag>()[index];
}
[[nodiscard]] inline CompoundTagVariant const& UniqueTagPtr::operator[](std::string_view index) const {
    if (!hold(Tag::Compound)) {
        throw std::runtime_error("tag not hold an object");
    }
    return get<CompoundTag>()[index];
}
template <class T>
    requires(std::is_arithmetic_v<T> && !ll::traits::is_char_v<T>)
[[nodiscard]] inline UniqueTagPtr::operator T() const {
    if (is_number()) {
        return ll::meta::visitIndex<CompoundTagVariant::Types::size>((size_t)getId(), [&]<size_t I>() -> T {
            auto& val = *static_cast<CompoundTagVariant::Types::get<I>*>(ptr);
            if constexpr (std::is_convertible_v<std::decay_t<decltype(val)>, T>) {
                return (T)val;
            } else {
                return {};
            }
        });
    } else {
        throw std::runtime_error("tag not hold an number");
    }
}