ComponentBindingHelper.hpp

Go to the documentation of this file.

/*
** EPITECH PROJECT, 2026
** rtype
** File description:
** ComponentBindingHelper - Simplifies component registration
*/
 
#pragma once
 
#include <functional>
#include <sol/sol.hpp>
#include <string>
#include <unordered_map>
#include <vector>
#include "common/ECSWrapper/ECSWorld.hpp"
 
namespace scripting::bindings {
 
    struct ComponentBinding {
        std::string name;
        std::function<void(sol::state &)> bindFunc;
    };
 
    class ComponentBindingHelper {
       public:
        template <typename T>
        void add(const std::string &name, std::function<void(sol::state &)> bindFunc);
 
        template <typename T>
        void registerComponent(const std::string &name);
 
        void applyComponentBindings(sol::state &lua);
 
        void applyEntityMethods(sol::usertype<ecs::wrapper::Entity> &entityType);
 
        void applyRemoveFunction(sol::state &lua, ecs::wrapper::ECSWorld *world);
 
        const std::vector<ComponentBinding> &getBindings() const;
 
        void clear();
 
       private:
        std::vector<ComponentBinding> _bindings;
        std::unordered_map<std::string, std::function<sol::object(ecs::wrapper::Entity &, sol::this_state)>>
            _getters;
        std::unordered_map<std::string, std::function<bool(ecs::wrapper::Entity &)>> _hasCheckers;
        std::unordered_map<std::string, std::function<void(ecs::wrapper::Entity &)>> _removers;
    };
 
    // Template implementation
    template <typename T>
    void ComponentBindingHelper::add(const std::string &name, std::function<void(sol::state &)> bindFunc) {
        _bindings.push_back({name, bindFunc});
 
        _getters[name] = [](ecs::wrapper::Entity &e, sol::this_state L) -> sol::object {
            sol::state_view lua(L);
            if (!e.has<T>()) {
                return sol::lua_nil;
            }
            return sol::make_object(lua, &e.get<T>());
        };
 
        _hasCheckers[name] = [](ecs::wrapper::Entity &e) -> bool {
            return e.has<T>();
        };
 
        _removers[name] = [](ecs::wrapper::Entity &e) {
            e.remove<T>();
        };
    }
 
    template <typename T>
    void ComponentBindingHelper::registerComponent(const std::string &name) {
        // Store direct lambdas that sol2 can understand
        _getters[name] = [](ecs::wrapper::Entity &e, sol::this_state L) -> sol::object {
            sol::state_view lua(L);
            if (!e.has<T>()) {
                return sol::lua_nil;
            }
            return sol::make_object(lua, &e.get<T>());
        };
 
        // CRITICAL: Use a direct lambda that sol2 can properly bind
        // The key is to not wrap it in std::function before assigning to usertype
        _hasCheckers[name] = [](ecs::wrapper::Entity &e) -> bool {
            return e.has<T>();
        };
 
        _removers[name] = [](ecs::wrapper::Entity &e) {
            e.remove<T>();
        };
    }
 
}  // namespace scripting::bindings