#include <Helena/Helena.hpp>
//#include <Test/Systems/ECSManager.hpp>

// Test System
class TestSystem
{
public:
    TestSystem() {
        // Start listen Engine events
        Helena::Engine::SubscribeEvent<Helena::Events::Engine::Init>    (&TestSystem::OnInit);
        Helena::Engine::SubscribeEvent<Helena::Events::Engine::Config>  (&TestSystem::OnConfig);
        Helena::Engine::SubscribeEvent<Helena::Events::Engine::Execute> (&TestSystem::OnExecute);
        Helena::Engine::SubscribeEvent<Helena::Events::Engine::Tick>    (&TestSystem::OnTick);
        Helena::Engine::SubscribeEvent<Helena::Events::Engine::Update>  (&TestSystem::OnUpdate);
        Helena::Engine::SubscribeEvent<Helena::Events::Engine::Render>  (&TestSystem::OnRender);
        Helena::Engine::SubscribeEvent<Helena::Events::Engine::Finalize>(&TestSystem::OnFinalize);
        Helena::Engine::SubscribeEvent<Helena::Events::Engine::Shutdown>(&TestSystem::OnShutdown);

        // Start listen events from system ECSManager
        Helena::Engine::SubscribeEvent<Helena::Events::ECSManager::CreateEntity>(&TestSystem::OnCreateEntity);
        Helena::Engine::SubscribeEvent<Helena::Events::ECSManager::RemoveEntity>(&TestSystem::OnRemoveEntity);
    }
    ~TestSystem() = default;

    // if event type is empty we can ignore argument in callback
    void OnInit() {
        HELENA_MSG_DEBUG("EventInit");

        auto& ecs = Helena::Engine::GetSystem<Helena::Systems::ECSManager>();
        ecs.CreateEntity(); // Create entity and trigger event CreateEntity
    }

    void OnConfig() {
        HELENA_MSG_DEBUG("EngineConfig");
    }

    void OnExecute() {
        HELENA_MSG_DEBUG("EngineExecute");
    }

    void OnTick(const Helena::Events::Engine::Tick event) {
        HELENA_MSG_DEBUG("EngineTick: {:.4f}", event.deltaTime);
    }

    void OnUpdate(const Helena::Events::Engine::Update event) {
        HELENA_MSG_DEBUG("EngineUpdate: {:.4f}", event.fixedTime);
    }

    void OnRender(const Helena::Events::Engine::Render event) {
        HELENA_MSG_DEBUG("EngineRender: {:.4f}", event.deltaTime);
    }

    void OnFinalize() {
        HELENA_MSG_DEBUG("EngineFinalize");
    }

    void OnShutdown() {
        HELENA_MSG_DEBUG("EngineShutdown");
    }

    void OnCreateEntity(const Helena::Events::ECSManager::CreateEntity& event) {
        auto& ecs = Helena::Engine::GetSystem<Helena::Systems::ECSManager>();
        auto& userInfo = ecs.AddComponent<UserInfo>(event.Entity, "Helena", 30u);

        HELENA_MSG_DEBUG("Entity created, user name: {}, age: {}", userInfo.name, userInfo.age);
        ecs.RemoveEntity(event.Entity); // Removed entity after trigger RemoveEntity event
    }

    void OnRemoveEntity(const Helena::Events::ECSManager::RemoveEntity& event) {
        auto& ecs = Helena::Engine::GetSystem<Helena::Systems::ECSManager>();
        auto& userInfo = ecs.GetComponent<UserInfo>(event.Entity);

        HELENA_MSG_DEBUG("Entity: {} removed!", userInfo.name);
    }
};

int main(int argc, char** argv)
{
    // Engine started from Initialize method
    Helena::Engine::Context::Initialize();          // Initialize Context
    Helena::Engine::Context::SetAppName("Helena");  // Set application name
    Helena::Engine::Context::SetTickrate(60.f);     // Set fixed update frequency
    Helena::Engine::Context::SetMain([]()           // Register systems happen in this callback
    {
        // Register all used systems
        Helena::Engine::RegisterSystem<Helena::Systems::ECSManager>();  // Entity Component System
        Helena::Engine::RegisterSystem<TestSystem>();                   // Test System
    });

    // Engine loop
    while(Helena::Engine::Heartbeat()) {}

    return 0;
}