#include "flatbutton_internal.h"
#include <QAbstractTransition>
#include <QEventTransition>
#include <QSignalTransition>
#include <QPropertyAnimation>
#include <QSequentialAnimationGroup>
#include "flatbutton.h"
#include "lib/style.h"

FlatButtonDelegate::FlatButtonDelegate(FlatButton *parent)
    : QStateMachine(parent),
      button(parent),
      _normalState(new QState(this)),
      _normalFocusedState(new QState(this)),
      _hoveredState(new QState(this)),
      _hoveredFocusedState(new QState(this)),
      _pressedState(new QState(this)),
      _focusHaloSize(85)
{
    setInitialState(_normalState);

    assignProperties();

    addTransition(QEvent::Enter, _normalFocusedState, _hoveredFocusedState);
    addTransition(QEvent::FocusIn, _normalState, _normalFocusedState);
    addTransition(QEvent::FocusOut, _normalFocusedState, _normalState);
    addTransition(QEvent::Enter, _normalState, _hoveredState);
    addTransition(QEvent::Leave, _hoveredFocusedState, _normalFocusedState);
    addTransition(QEvent::Leave, _hoveredState, _normalState);
    addTransition(QEvent::MouseButtonPress, _hoveredState, _pressedState);
    addTransition(QEvent::MouseButtonRelease, _pressedState, _hoveredFocusedState);

    QSequentialAnimationGroup *group = new QSequentialAnimationGroup(this);

    QPropertyAnimation *grow = new QPropertyAnimation;
    QPropertyAnimation *shrink = new QPropertyAnimation;

    grow->setTargetObject(this);
    grow->setPropertyName("focusHaloSize");
    grow->setStartValue(85);
    grow->setEndValue(100);
    grow->setEasingCurve(QEasingCurve::InOutSine);
    grow->setDuration(840);

    shrink->setTargetObject(this);
    shrink->setPropertyName("focusHaloSize");
    shrink->setStartValue(100);
    shrink->setEndValue(85);
    shrink->setEasingCurve(QEasingCurve::InOutSine);
    shrink->setDuration(840);

    group->addAnimation(grow);
    group->addAnimation(shrink);
    group->setLoopCount(-1);
    group->start();

    start();
}

FlatButtonDelegate::~FlatButtonDelegate()
{
}

void FlatButtonDelegate::setBackgroundOpacity(qreal opacity)
{
    _backgroundOpacity = opacity;
    button->update();
}

qreal FlatButtonDelegate::backgroundOpacity() const
{
    return _backgroundOpacity;
}

void FlatButtonDelegate::setBackgroundColor(const QColor &color)
{
    _backgroundColor = color;
    button->update();
}

QColor FlatButtonDelegate::backgroundColor() const
{
    return _backgroundColor;
}

void FlatButtonDelegate::setFocusHaloOpacity(qreal opacity)
{
    _focusHaloOpacity = opacity;
    button->update();
}

qreal FlatButtonDelegate::focusHaloOpacity() const
{
    return _focusHaloOpacity;
}

void FlatButtonDelegate::setFocusHaloSize(int size)
{
    _focusHaloSize = size;
    button->update();
}

int FlatButtonDelegate::focusHaloSize() const
{
    return _focusHaloSize;
}

void FlatButtonDelegate::assignProperties()
{
    QColor textColor = button->palette().color(QPalette::Active,
                                               QPalette::ButtonText);

    _normalState->assignProperty(this, "backgroundOpacity", 0);
    _normalState->assignProperty(this, "backgroundColor", textColor);
    _normalState->assignProperty(this, "focusHaloOpacity", 0);

    _normalFocusedState->assignProperty(this, "backgroundOpacity", 0);
    _normalFocusedState->assignProperty(this, "backgroundColor", textColor);
    _normalFocusedState->assignProperty(this, "focusHaloOpacity", 0.12);

    _hoveredState->assignProperty(this, "backgroundOpacity", 0.15);
    _hoveredState->assignProperty(this, "backgroundColor", textColor);
    _hoveredState->assignProperty(this, "focusHaloOpacity", 0);

    _hoveredFocusedState->assignProperty(this, "backgroundOpacity", 0.15);
    _hoveredFocusedState->assignProperty(this, "backgroundColor", textColor);
    _normalFocusedState->assignProperty(this, "focusHaloOpacity", 0.12);

    _pressedState->assignProperty(this, "backgroundOpacity", 0.15);
    _pressedState->assignProperty(this, "backgroundColor", textColor);
    _pressedState->assignProperty(this, "focusHaloOpacity", 0);
}

void FlatButtonDelegate::addTransition(QEvent::Type eventType,
                                       QState *fromState,
                                       QState *toState)
{
    QAbstractTransition *transition = new QEventTransition(button, eventType);
    transition->setTargetState(toState);

    QPropertyAnimation *animation;

    animation = new QPropertyAnimation(this, "backgroundOpacity");
    animation->setDuration(200);
    transition->addAnimation(animation);

    animation = new QPropertyAnimation(this, "backgroundColor");
    animation->setDuration(200);
    transition->addAnimation(animation);

    animation = new QPropertyAnimation(this, "focusHaloOpacity");
    animation->setDuration(350);
    transition->addAnimation(animation);

    fromState->addTransition(transition);
}