#include "fader.h"
#include "imgui_utils.h"
#include <user/config.h>

static bool g_isFading;
static bool g_isFadeIn;

static float g_startTime;

static float g_duration;
static ImU32 g_colour = IM_COL32_BLACK;
static std::function<void()> g_endCallback;
static float g_endCallbackDelay;

void Fader::Draw()
{
    if (!s_isVisible)
        return;

    auto time = (ImGui::GetTime() - g_startTime) / g_duration;
    auto alpha = 1.0f;

    if (time >= g_duration)
    {
        if (time >= g_duration + g_endCallbackDelay)
        {
            if (g_endCallback)
            {
                g_endCallback();
                g_endCallback = nullptr;
            }

            g_isFading = false;
        }
    }
    else
    {
        alpha = g_isFadeIn
            ? Lerp(1, 0, time)
            : Lerp(0, 1, time);
    }

    if (g_isFadeIn && !g_isFading)
        return;

    auto colour = IM_COL32(g_colour & 0xFF, (g_colour >> 8) & 0xFF, (g_colour >> 16) & 0xFF, 255 * alpha);

    ImGui::GetForegroundDrawList()->AddRectFilled({ 0, 0 }, ImGui::GetIO().DisplaySize, colour);
}

static void DoFade(bool isFadeIn, float duration, std::function<void()> endCallback, float endCallbackDelay)
{
    if (g_isFading)
        return;

    g_isFading = true;
    g_isFadeIn = isFadeIn;
    g_startTime = ImGui::GetTime();
    g_duration = duration;
    g_endCallback = endCallback;
    g_endCallbackDelay = endCallbackDelay;

    Fader::s_isVisible = true;
}

void Fader::SetFadeColour(ImU32 colour)
{
    g_colour = colour;
}

void Fader::FadeIn(float duration, std::function<void()> endCallback, float endCallbackDelay)
{
    DoFade(true, duration, endCallback, endCallbackDelay);
}

void Fader::FadeOut(float duration, std::function<void()> endCallback, float endCallbackDelay)
{
    DoFade(false, duration, endCallback, endCallbackDelay);
}