#ifndef VBUTTONWITHWIDGET_H
#define VBUTTONWITHWIDGET_H

#include <QPushButton>
#include <QString>
#include <QIcon>
#include <QWidgetAction>

class QDragEnterEvent;
class QDropEvent;
class QPaintEvent;
class VButtonWithWidget;

// Abstract class for the widget used by VButtonWithWidget.
// Widget need to inherit this class if drag/drop is needed.
class VButtonPopupWidget
{
public:
    VButtonPopupWidget() : m_btn(NULL)
    {
    }

    virtual bool isAcceptDrops() const = 0;
    virtual bool handleDragEnterEvent(QDragEnterEvent *p_event) = 0;
    virtual bool handleDropEvent(QDropEvent *p_event) = 0;
    virtual void handleAboutToShow() = 0;

    void setButton(VButtonWithWidget *p_btn)
    {
        m_btn = p_btn;
    }

    VButtonWithWidget *getButton() const
    {
        return m_btn;
    }

private:
    VButtonWithWidget *m_btn;
};

class VButtonWidgetAction : public QWidgetAction
{
    Q_OBJECT
public:
    VButtonWidgetAction(QWidget *p_widget, QWidget *p_parent)
        : QWidgetAction(p_parent), m_widget(p_widget)
    {
    }

    QWidget *createWidget(QWidget *p_parent)
    {
        m_widget->setParent(p_parent);
        return m_widget;
    }

private:
    QWidget *m_widget;
};

// A QPushButton with popup widget.
class VButtonWithWidget : public QPushButton
{
    Q_OBJECT
public:
    VButtonWithWidget(QWidget *p_widget,
                      QWidget *p_parent = Q_NULLPTR);

    VButtonWithWidget(const QString &p_text,
                      QWidget *p_widget,
                      QWidget *p_parent = Q_NULLPTR);

    VButtonWithWidget(const QIcon &p_icon,
                      const QString &p_text,
                      QWidget *p_widget,
                      QWidget *p_parent = Q_NULLPTR);

    QWidget *getPopupWidget() const;

    // Show the popup widget.
    void showPopupWidget();

    // Set the bubble to display a number @p_num.
    // @p_num: -1 to hide the bubble.
    void setBubbleNumber(int p_num);

signals:
    // Emit when popup widget is about to show.
    void popupWidgetAboutToShow(QWidget *p_widget);

protected:
    // To accept specific drop.
    void dragEnterEvent(QDragEnterEvent *p_event) Q_DECL_OVERRIDE;

    // Drop the data.
    void dropEvent(QDropEvent *p_event) Q_DECL_OVERRIDE;

    void paintEvent(QPaintEvent *p_event) Q_DECL_OVERRIDE;

private:
    void init();

    // Get VButtonWithWidget from m_popupWidget.
    VButtonPopupWidget *getButtonPopupWidget() const;

    QWidget *m_popupWidget;

    QColor m_bubbleBg;
    QColor m_bubbleFg;

    // String to display in the bubble.
    // Empty to hide bubble.
    QString m_bubbleStr;
};

inline VButtonPopupWidget *VButtonWithWidget::getButtonPopupWidget() const
{
    return dynamic_cast<VButtonPopupWidget *>(m_popupWidget);
}

#endif // VBUTTONWITHWIDGET_H