#pragma once

#include <cstring>
#include <mach-o/dyld.h>
#include <type_traits>
#include <typeinfo>

namespace geode {
    using dylib_t = void*;

    struct PlatformInfo {
        dylib_t m_dylib;
    };
}

namespace geode::base {
    GEODE_NOINLINE inline uintptr_t get() {
        static uintptr_t base = _dyld_get_image_vmaddr_slide(0) + 0x100000000;
        return base;
    }
}

extern "C" inline uintptr_t _geode_ios_base() {
    return geode::base::get();
}

namespace geode::cast {
    using uinthalf_t = uint32_t;
    using inthalf_t = int32_t;

    struct DummyClass {
        virtual ~DummyClass() {}
    };

    struct DummySingleClass : DummyClass {};

    struct DummyClass2 {};

    struct DummyMultipleClass : DummySingleClass, DummyClass2 {};

    struct ClassTypeinfoType {
        void** m_typeinfoVtable;
        char const* m_typeinfoName;
    };

    struct SingleClassTypeinfoType : ClassTypeinfoType {
        ClassTypeinfoType* m_baseClassTypeinfo;
    };

#pragma pack(push, 1)

    struct MultipleClassSingleEntryType {
        ClassTypeinfoType* m_baseClassTypeinfo;
        uint8_t m_visibilityFlag;
        inthalf_t m_offset;
        uint8_t m_padding[sizeof(inthalf_t) - 1];
    };

#pragma pack(pop)

    struct MultipleClassTypeinfoType : ClassTypeinfoType {
        uint32_t m_flags;
        uint32_t m_numBaseClass;
        MultipleClassSingleEntryType m_baseClasses[0x100];
    };

    struct VtableTypeinfoType {
        inthalf_t m_offset;
        ClassTypeinfoType* m_typeinfo;
    };

    struct VtableType {
        void* m_vtable[0x100];
    };

    struct CompleteVtableType : VtableTypeinfoType, VtableType {};

    inline void** typeinfoVtableOf(void* ptr) {
        auto vftable = *reinterpret_cast<VtableType**>(ptr);

        auto typeinfoPtr =
            static_cast<VtableTypeinfoType*>(static_cast<CompleteVtableType*>(vftable));

        return typeinfoPtr->m_typeinfo->m_typeinfoVtable;
    }

    inline void* traverseTypeinfoFor(
        void* ptr, ClassTypeinfoType* typeinfo, char const* afterIdent
    ) {
        DummySingleClass dummySingleClass;
        DummyMultipleClass dummyMultipleClass;

        {
            auto optionIdent = typeinfo->m_typeinfoName;
            if (std::strcmp(optionIdent, afterIdent) == 0) {
                return ptr;
            }
        }
        if (typeinfo->m_typeinfoVtable == typeinfoVtableOf(&dummySingleClass)) {
            auto siTypeinfo = static_cast<SingleClassTypeinfoType*>(typeinfo);
            return traverseTypeinfoFor(ptr, siTypeinfo->m_baseClassTypeinfo, afterIdent);
        }
        else if (typeinfo->m_typeinfoVtable == typeinfoVtableOf(&dummyMultipleClass)) {
            auto vmiTypeinfo = static_cast<MultipleClassTypeinfoType*>(typeinfo);
            for (int i = 0; i < vmiTypeinfo->m_numBaseClass; ++i) {
                auto& entry = vmiTypeinfo->m_baseClasses[i];
                auto optionPtr = reinterpret_cast<std::byte*>(ptr) + entry.m_offset;
                auto ret = traverseTypeinfoFor(optionPtr, entry.m_baseClassTypeinfo, afterIdent);
                if (ret != nullptr) return ret;
            }
        }

        return nullptr;
    }

    template <class After, class Before>
    inline After typeinfo_cast(Before ptr) {
        static_assert(
            std::is_polymorphic_v<std::remove_pointer_t<Before>>, "Input is not a polymorphic type"
        );
        auto basePtr = dynamic_cast<void*>(ptr);
        auto vftable = *reinterpret_cast<VtableType**>(basePtr);
        auto typeinfo =
            static_cast<VtableTypeinfoType*>(static_cast<CompleteVtableType*>(vftable))->m_typeinfo;

        auto afterTypeinfo =
            reinterpret_cast<ClassTypeinfoType const*>(&typeid(std::remove_pointer_t<After>));
        auto afterIdent = afterTypeinfo->m_typeinfoName;

        return static_cast<After>(traverseTypeinfoFor(basePtr, typeinfo, afterIdent));
    }
}