#include "rippleeffectbutton.hpp"

#include <QDebug>
#include <QPainter>

#include "colorscheme.hpp"

namespace chatterino {
namespace widgets {

RippleEffectButton::RippleEffectButton(BaseWidget *parent)
    : BaseWidget(parent)

{
    connect(&effectTimer, &QTimer::timeout, this, &RippleEffectButton::onMouseEffectTimeout);

    this->effectTimer.setInterval(20);
    this->effectTimer.start();
}

void RippleEffectButton::setMouseEffectColor(boost::optional<QColor> color)
{
    this->mouseEffectColor = color;
}

void RippleEffectButton::paintEvent(QPaintEvent *)
{
    QPainter painter(this);

    this->fancyPaint(painter);
}

void RippleEffectButton::fancyPaint(QPainter &painter)
{
    QColor c;

    if (this->mouseEffectColor) {
        c = this->mouseEffectColor.get();
    } else {
        c = this->colorScheme.isLightTheme() ? QColor(0, 0, 0) : QColor(255, 255, 255);
    }

    if (this->hoverMultiplier > 0) {
        QRadialGradient gradient(mousePos.x(), mousePos.y(), 50, mousePos.x(), mousePos.y());

        gradient.setColorAt(
            0, QColor(c.red(), c.green(), c.blue(), (int)(24 * this->hoverMultiplier)));
        gradient.setColorAt(
            1, QColor(c.red(), c.green(), c.blue(), (int)(12 * this->hoverMultiplier)));

        painter.fillRect(this->rect(), gradient);
    }

    for (auto effect : this->clickEffects) {
        QRadialGradient gradient(effect.position.x(), effect.position.y(),
                                 effect.progress * (float)width() * 2, effect.position.x(),
                                 effect.position.y());

        gradient.setColorAt(
            0, QColor(c.red(), c.green(), c.blue(), (int)((1 - effect.progress) * 95)));
        gradient.setColorAt(
            0.9999, QColor(c.red(), c.green(), c.blue(), (int)((1 - effect.progress) * 95)));
        gradient.setColorAt(1, QColor(c.red(), c.green(), c.blue(), (int)(0)));

        painter.fillRect(this->rect(), gradient);
    }
}

void RippleEffectButton::enterEvent(QEvent *)
{
    this->mouseOver = true;
}

void RippleEffectButton::leaveEvent(QEvent *)
{
    this->mouseOver = false;
}

void RippleEffectButton::mousePressEvent(QMouseEvent *event)
{
    if (event->button() != Qt::LeftButton) {
        return;
    }

    this->clickEffects.push_back(ClickEffect(event->pos()));

    this->mouseDown = true;
}

void RippleEffectButton::mouseReleaseEvent(QMouseEvent *event)
{
    if (event->button() != Qt::LeftButton) {
        return;
    }

    this->mouseDown = false;

    if (this->rect().contains(event->pos())) {
        emit clicked();
    }
}

void RippleEffectButton::mouseMoveEvent(QMouseEvent *event)
{
    this->mousePos = event->pos();
}

void RippleEffectButton::onMouseEffectTimeout()
{
    bool performUpdate = false;

    if (selected) {
        if (this->hoverMultiplier != 0) {
            this->hoverMultiplier = std::max(0.0, this->hoverMultiplier - 0.1);
            performUpdate = true;
        }
    } else if (mouseOver) {
        if (this->hoverMultiplier != 1) {
            this->hoverMultiplier = std::min(1.0, this->hoverMultiplier + 0.5);
            performUpdate = true;
        }
    } else {
        if (this->hoverMultiplier != 0) {
            this->hoverMultiplier = std::max(0.0, this->hoverMultiplier - 0.3);
            performUpdate = true;
        }
    }

    if (this->clickEffects.size() != 0) {
        performUpdate = true;

        for (auto it = this->clickEffects.begin(); it != this->clickEffects.end();) {
            (*it).progress += mouseDown ? 0.02 : 0.07;

            if ((*it).progress >= 1.0) {
                it = this->clickEffects.erase(it);
            } else {
                it++;
            }
        }
    }

    if (performUpdate) {
        update();
    }
}

}  // namespace widgets
}  // namespace chatterino