#include "banner_widget.h"

#include "../../../pixel_map_generator.h"

#include <QLinearGradient>
#include <QMouseEvent>
#include <QPainter>
#include <QVBoxLayout>

BannerWidget::BannerWidget(QWidget *parent, const QString &text, Qt::Orientation orientation, int transparency_)
    : QWidget(parent), gradientOrientation(orientation), transparency(qBound(0, transparency_, 100))
{
    auto layout = new QHBoxLayout(this);

    iconLabel = new QLabel(this);
    iconLabel->setAlignment(Qt::AlignLeft | Qt::AlignVCenter);

    // Create the banner label and set properties
    bannerLabel = new QLabel(text, this);
    bannerLabel->setAlignment(Qt::AlignCenter);

    QString textColor;
    if (transparency > 50) {
        textColor = " color: white;";
    }
    bannerLabel->setStyleSheet("font-size: 24px; font-weight: bold;" + textColor);

    layout->addWidget(iconLabel);
    layout->addWidget(bannerLabel);
    layout->addWidget(new QLabel(this)); // add dummy label to force text label to be centered
    setLayout(layout);

    // Set minimum height for the widget
    setMinimumHeight(50);
    connect(this, &BannerWidget::buddyVisibilityChanged, this, &BannerWidget::toggleBuddyVisibility);

    updateDropdownIconState();
}

void BannerWidget::mousePressEvent(QMouseEvent *event)
{
    QWidget::mousePressEvent(event);
    if (clickable) {
        emit buddyVisibilityChanged();
    }
}

void BannerWidget::setText(const QString &text) const
{
    bannerLabel->setText(text);
}

void BannerWidget::setClickable(bool _clickable)
{
    clickable = _clickable;
    updateDropdownIconState();
}

void BannerWidget::setBuddy(QWidget *_buddy)
{
    buddy = _buddy;
    updateDropdownIconState();
}

void BannerWidget::toggleBuddyVisibility() const
{
    if (buddy) {
        buddy->setVisible(!buddy->isVisible());
        updateDropdownIconState();
    }
}

void BannerWidget::updateDropdownIconState() const
{
    if (clickable && buddy) {
        iconLabel->setPixmap(DropdownIconPixmapGenerator::generatePixmap(24, !buddy->isHidden()));
    } else {
        // we cannot directly hide the iconLabel, since it's needed to center the text; set an empty image instead
        iconLabel->setPixmap(QPixmap());
    }
}

void BannerWidget::paintEvent(QPaintEvent *event)
{
    Q_UNUSED(event);

    QPainter painter(this);

    // Calculate alpha based on transparency percentage
    int alpha = (255 * transparency) / 100;

    // Determine gradient direction
    QLinearGradient gradient;
    if (gradientOrientation == Qt::Vertical) {
        gradient = QLinearGradient(rect().topLeft(), rect().bottomLeft());
    } else {
        gradient = QLinearGradient(rect().topLeft(), rect().topRight());
    }

    // Set neutral gradient colors with calculated transparency
    gradient.setColorAt(0, QColor(200, 200, 200, alpha));       // Light grey with alpha
    gradient.setColorAt(1, QColor(100, 100, 100, alpha / 1.5)); // Darker grey, slightly more transparent

    // Fill the widget background with the gradient
    painter.fillRect(rect(), gradient);
}