#include <QHBoxLayout>
#include <QVBoxLayout>
#include <QPushButton>
#include "posemanagewidget.h"
#include "theme.h"
#include "poseeditwidget.h"
#include "infolabel.h"

PoseManageWidget::PoseManageWidget(const Document *document, QWidget *parent) :
    QWidget(parent),
    m_document(document)
{
    QPushButton *addPoseButton = new QPushButton(Theme::awesome()->icon(fa::plus), tr("Add Pose..."));
    addPoseButton->hide();
    
    connect(addPoseButton, &QPushButton::clicked, this, &PoseManageWidget::showAddPoseDialog);
    
    QHBoxLayout *toolsLayout = new QHBoxLayout;
    toolsLayout->addWidget(addPoseButton);
    
    m_poseListWidget = new PoseListWidget(document);
    connect(m_poseListWidget, &PoseListWidget::modifyPose, this, &PoseManageWidget::showPoseDialog);
    
    InfoLabel *infoLabel = new InfoLabel;
    infoLabel->hide();
    
    auto refreshInfoLabel = [=]() {
        if (m_document->currentRigSucceed()) {
            if (m_document->rigType == RigType::Animal) {
                infoLabel->setText("");
                infoLabel->hide();
                addPoseButton->show();
            } else {
                infoLabel->setText(tr("Pose editor doesn't support this rig type yet: ") + RigTypeToDispName(m_document->rigType));
                infoLabel->show();
                addPoseButton->hide();
            }
        } else {
            infoLabel->setText(tr("Missing Rig"));
            infoLabel->show();
            addPoseButton->hide();
        }
    };
    
    connect(m_document, &Document::resultRigChanged, this, refreshInfoLabel);
    connect(m_document, &Document::cleanup, this, refreshInfoLabel);
    
    QVBoxLayout *mainLayout = new QVBoxLayout;
    mainLayout->addWidget(infoLabel);
    mainLayout->addLayout(toolsLayout);
    mainLayout->addWidget(m_poseListWidget);
    
    setLayout(mainLayout);
}

PoseListWidget *PoseManageWidget::poseListWidget()
{
    return m_poseListWidget;
}

QSize PoseManageWidget::sizeHint() const
{
    return QSize(Theme::sidebarPreferredWidth, 0);
}

void PoseManageWidget::showAddPoseDialog()
{
    showPoseDialog(QUuid());
}

void PoseManageWidget::showPoseDialog(QUuid poseId)
{
    PoseEditWidget *poseEditWidget = new PoseEditWidget(m_document);
    poseEditWidget->setAttribute(Qt::WA_DeleteOnClose);
    if (!poseId.isNull()) {
        const Pose *pose = m_document->findPose(poseId);
        if (nullptr != pose) {
            poseEditWidget->setEditPoseId(poseId);
            poseEditWidget->setEditPoseName(pose->name);
            poseEditWidget->setEditPoseFrames(pose->frames);
            poseEditWidget->setEditPoseTurnaroundImageId(pose->turnaroundImageId);
            poseEditWidget->clearUnsaveState();
        }
    }
    poseEditWidget->show();
    connect(poseEditWidget, &QDialog::destroyed, [=]() {
        emit unregisterDialog((QWidget *)poseEditWidget);
    });
    emit registerDialog((QWidget *)poseEditWidget);
}