geode/loader/include/Geode/modify/IDManager.hpp

#pragma once

#include "../DefaultInclude.hpp"

namespace cocos2d {
    class CCNode;
}

namespace geode {
    template<class T>
    concept IDProvidable = requires {
        { T::CLASS_NAME } -> std::convertible_to<const char*>;
    };

    class GEODE_DLL NodeStringIDManager {
    public:
        template<class T>
        using Provider = void(GEODE_CALL*)(T*);

    protected:
        std::unordered_map<std::string, Provider<cocos2d::CCNode>> m_providers;

    public:
        static NodeStringIDManager* get();

        template<IDProvidable T>
        void registerProvider(void(GEODE_CALL* fun)(T*)) {
            m_providers.insert({
                T::CLASS_NAME,
                reinterpret_cast<Provider<cocos2d::CCNode>>(fun)
            });
        }

        template<IDProvidable T>
        bool provide(T* layer) const {
            if (m_providers.count(T::CLASS_NAME)) {
                m_providers.at(T::CLASS_NAME)(layer);
                return true;
            }
            return false;
        }
    };
}

namespace geode {
    template<class For>
    void GEODE_CALL geodeInternalProvideIDsFor(For* cls) {
        cls->provide();
    }
}

#define $register_ids(Layer_, ...) \
	struct GEODE_CONCAT(ProvideIDsFor, Layer_);\
	$execute {\
		NodeStringIDManager::get()->registerProvider(\
            &geodeInternalProvideIDsFor<GEODE_CONCAT(ProvideIDsFor, Layer_)>\
        );\
	};\
    struct GEODE_CONCAT(ProvideIDsFor, Layer_) : public Layer_ {\
        void provide() __VA_ARGS__\
    };\