#include "splitoverlay.hpp"

#include <QEvent>
#include <QGraphicsOpacityEffect>
#include <QGridLayout>
#include <QPainter>
#include <QPushButton>

#include "application.hpp"
#include "singletons/resourcemanager.hpp"
#include "widgets/split.hpp"

namespace chatterino {
namespace widgets {

SplitOverlay::SplitOverlay(Split *parent)
    : BaseWidget(parent)
    , split(parent)
{
    QGridLayout *layout = new QGridLayout(this);
    layout->setMargin(1);
    layout->setSpacing(1);

    layout->setRowStretch(1, 1);
    layout->setRowStretch(3, 1);
    layout->setColumnStretch(1, 1);
    layout->setColumnStretch(3, 1);

    QPushButton *move = new QPushButton(getApp()->resources->split.move, QString());
    QPushButton *left = new QPushButton(getApp()->resources->split.left, QString());
    QPushButton *right = new QPushButton(getApp()->resources->split.right, QString());
    QPushButton *up = new QPushButton(getApp()->resources->split.up, QString());
    QPushButton *down = new QPushButton(getApp()->resources->split.down, QString());

    move->setGraphicsEffect(new QGraphicsOpacityEffect(this));
    left->setGraphicsEffect(new QGraphicsOpacityEffect(this));
    right->setGraphicsEffect(new QGraphicsOpacityEffect(this));
    up->setGraphicsEffect(new QGraphicsOpacityEffect(this));
    down->setGraphicsEffect(new QGraphicsOpacityEffect(this));

    move->setFlat(true);
    left->setFlat(true);
    right->setFlat(true);
    up->setFlat(true);
    down->setFlat(true);

    layout->addWidget(move, 2, 2);
    layout->addWidget(left, 2, 0);
    layout->addWidget(right, 2, 4);
    layout->addWidget(up, 0, 2);
    layout->addWidget(down, 4, 2);

    move->installEventFilter(new ButtonEventFilter(this, SplitMove));
    left->installEventFilter(new ButtonEventFilter(this, SplitLeft));
    right->installEventFilter(new ButtonEventFilter(this, SplitRight));
    up->installEventFilter(new ButtonEventFilter(this, SplitUp));
    down->installEventFilter(new ButtonEventFilter(this, SplitDown));

    move->setFocusPolicy(Qt::NoFocus);
    left->setFocusPolicy(Qt::NoFocus);
    right->setFocusPolicy(Qt::NoFocus);
    up->setFocusPolicy(Qt::NoFocus);
    down->setFocusPolicy(Qt::NoFocus);

    move->setCursor(Qt::SizeAllCursor);

    this->managedConnect(this->scaleChanged, [=](float scale) {
        int a = scale * 40;
        QSize size(a, a);

        move->setIconSize(size);
        left->setIconSize(size);
        right->setIconSize(size);
        up->setIconSize(size);
        down->setIconSize(size);
    });
}

void SplitOverlay::paintEvent(QPaintEvent *event)
{
    QPainter painter(this);
    painter.fillRect(this->rect(), QColor(0, 0, 0, 90));

    QRect rect;
    switch (this->hoveredElement) {
        case SplitLeft: {
            rect = QRect(0, 0, this->width() / 2, this->height());
        } break;
        case SplitRight: {
            rect = QRect(this->width() / 2, 0, this->width() / 2, this->height());
        } break;
        case SplitUp: {
            rect = QRect(0, 0, this->width(), this->height() / 2);
        } break;
        case SplitDown: {
            rect = QRect(0, this->height() / 2, this->width(), this->height() / 2);
        } break;
    }
    if (!rect.isNull()) {
        painter.setPen(QColor(0, 148, 255, 0x70));
        painter.setBrush(QColor(0, 148, 255, 0x30));
        painter.drawRect(rect);
    }
}

SplitOverlay::ButtonEventFilter::ButtonEventFilter(SplitOverlay *_parent, HoveredElement _element)
    : QObject(_parent)
    , parent(_parent)
    , hoveredElement(_element)
{
}

bool SplitOverlay::ButtonEventFilter::eventFilter(QObject *watched, QEvent *event)
{
    switch (event->type()) {
        case QEvent::Enter: {
            QGraphicsOpacityEffect *effect =
                dynamic_cast<QGraphicsOpacityEffect *>(((QWidget *)watched)->graphicsEffect());

            if (effect != nullptr) {
                effect->setOpacity(1);
            }

            this->parent->hoveredElement = this->hoveredElement;
            this->parent->update();
        } break;
        case QEvent::Leave: {
            QGraphicsOpacityEffect *effect =
                dynamic_cast<QGraphicsOpacityEffect *>(((QWidget *)watched)->graphicsEffect());

            if (effect != nullptr) {
                effect->setOpacity(0.7);
            }

            this->parent->hoveredElement = HoveredElement::None;
            this->parent->update();
        } break;
        case QEvent::MouseButtonPress: {
            if (this->hoveredElement == HoveredElement::SplitMove) {
                QMouseEvent *mouseEvent = static_cast<QMouseEvent *>(event);
                if (mouseEvent->button() == Qt::LeftButton) {
                    this->parent->split->drag();
                }
                return true;
            }
        } break;
    }
    return QObject::eventFilter(watched, event);
}

}  // namespace widgets
}  // namespace chatterino