/******************************************************************************
*	File		-	entity.h
*	Author		-	Joey Pollack
*	Date		-	2022/05/23 (y/m/d)
*	Mod Date	-	2022/05/23 (y/m/d)
*	Description	-	Provides functionality to work with world entities
******************************************************************************/

#ifndef LUNARIUM_ENTITY_H_
#define LUNARIUM_ENTITY_H_

#include <utils/uuid.h>
#include <utils/logger.h>
#include <entt/entt.hpp>
#include <world/world.h>

namespace lunarium
{
    class Entity
    {
    public:
        // TODO: Maybe change this to take the World instead of the registry
        Entity(World& w);
        Entity(World& w, LUUID uuid, entt::entity handle = entt::null);

        // These component methods are based on methods from the Hazel engine, written by Cherno
        // https://github.com/TheCherno/Hazel/blob/dev/Hazel/src/Hazel/Scene/Entity.h
        template<typename T, typename... Args>
		T& AddComponent(Args&&... args)
		{
			if (HasComponent<T>())
            {
                Logger::Error(LogCategory::GAME_SYSTEM, "Entity already has component!");
                return GetComponent<T>();
            }
			T& component = mWorld.GetEntityRegistry()->emplace<T>(mHandle, std::forward<Args>(args)...);
			return component;
		}

        template<typename T>
		T& GetComponent()
		{
			if (!HasComponent<T>())
            {
                Logger::Error(LogCategory::GAME_SYSTEM, "Entity does not have component!");
            }
			return mWorld.GetEntityRegistry()->get<T>(mHandle);
		}

		template<typename T>
		bool HasComponent()
		{
			return mWorld.GetEntityRegistry()->all_of<T>(mHandle);
		}

		template<typename T>
		void RemoveComponent()
		{
			if (!HasComponent<T>())
            {
                Logger::Error(LogCategory::GAME_SYSTEM, "Entity does not have component!");
            }
			mWorld.GetEntityRegistry()->remove<T>(mHandle);
		}

        LUUID GetUUID() const;

    private:
        LUUID mUUID;
        entt::entity mHandle;
        World& mWorld; // The world the entity is contained within
        //entt::registry* mpRegistry;

    private:
        void Init();
    };
}

#endif  // LUNARIUM_ENTITY_H_