#include "ImaginationShrineServer.h"
#include "RebuildComponent.h"


void ImaginationShrineServer::OnUse(Entity* self, Entity* user) 
{
    // If the rebuild component is complete, use the shrine
    auto* rebuildComponent = self->GetComponent<RebuildComponent>();

    if (rebuildComponent == nullptr)
    {
        return;
    }

    if (rebuildComponent->GetState() == REBUILD_COMPLETED)
    {
        // Use the shrine
        BaseUse(self, user);
    }
}