// Licensed under the Apache-2.0 license. See README.md for details.

#include "ProgressWidget.h"
#include <QProgressBar>
#include <QLabel>
#include <QVBoxLayout>
#include <QEventLoop>

#include "tasks/Task.h"

ProgressWidget::ProgressWidget(QWidget *parent)
    : QWidget(parent)
{
    m_label = new QLabel(this);
    m_label->setWordWrap(true);
    m_bar = new QProgressBar(this);
    m_bar->setMinimum(0);
    m_bar->setMaximum(100);
    QVBoxLayout *layout = new QVBoxLayout(this);
    layout->addWidget(m_label);
    layout->addWidget(m_bar);
    layout->addStretch();
    setLayout(layout);
}

void ProgressWidget::start(std::shared_ptr<Task> task)
{
    if (m_task)
    {
        disconnect(m_task.get(), 0, this, 0);
    }
    m_task = task;
    connect(m_task.get(), &Task::finished, this, &ProgressWidget::handleTaskFinish);
    connect(m_task.get(), &Task::status, this, &ProgressWidget::handleTaskStatus);
    connect(m_task.get(), &Task::progress, this, &ProgressWidget::handleTaskProgress);
    connect(m_task.get(), &Task::destroyed, this, &ProgressWidget::taskDestroyed);
    if (!m_task->isRunning())
    {
        QMetaObject::invokeMethod(m_task.get(), "start", Qt::QueuedConnection);
    }
}
bool ProgressWidget::exec(std::shared_ptr<Task> task)
{
    QEventLoop loop;
    connect(task.get(), &Task::finished, &loop, &QEventLoop::quit);
    start(task);
    if (task->isRunning())
    {
        loop.exec();
    }
    return task->wasSuccessful();
}

void ProgressWidget::handleTaskFinish()
{
    if (!m_task->wasSuccessful())
    {
        m_label->setText(m_task->failReason());
    }
}
void ProgressWidget::handleTaskStatus(const QString &status)
{
    m_label->setText(status);
}
void ProgressWidget::handleTaskProgress(qint64 current, qint64 total)
{
    m_bar->setMaximum(total);
    m_bar->setValue(current);
}
void ProgressWidget::taskDestroyed()
{
    m_task = nullptr;
}
