#include "UpdateDialog.hpp"

#include "singletons/Updates.hpp"
#include "util/LayoutCreator.hpp"
#include "widgets/Label.hpp"

#include <QDialogButtonBox>
#include <QPushButton>
#include <QVBoxLayout>

namespace chatterino {

UpdateDialog::UpdateDialog()
    : BaseWindow(nullptr,
                 BaseWindow::Flags(BaseWindow::Frameless | BaseWindow::TopMost |
                                   BaseWindow::EnableCustomFrame))
{
    auto layout =
        LayoutCreator<UpdateDialog>(this).setLayoutType<QVBoxLayout>();

    layout.emplace<Label>("You shouldn't be seeing this dialog.")
        .assign(&this->ui_.label);

    auto buttons = layout.emplace<QDialogButtonBox>();
    auto install = buttons->addButton("Install", QDialogButtonBox::AcceptRole);
    this->ui_.installButton = install;
    auto dismiss = buttons->addButton("Dismiss", QDialogButtonBox::RejectRole);

    QObject::connect(install, &QPushButton::clicked, this, [this] {
        Updates::getInstance().installUpdates();
        this->close();
    });
    QObject::connect(dismiss, &QPushButton::clicked, this, [this] {
        this->buttonClicked.invoke(Dismiss);
        this->close();
    });

    this->updateStatusChanged(Updates::getInstance().getStatus());
    this->connections_.managedConnect(
        Updates::getInstance().statusUpdated,
        [this](auto status) { this->updateStatusChanged(status); });

    this->setScaleIndependantHeight(150);
}

void UpdateDialog::updateStatusChanged(Updates::Status status)
{
    this->ui_.installButton->setVisible(status == Updates::UpdateAvailable);

    switch (status)
    {
        case Updates::UpdateAvailable:
        {
            this->ui_.label->setText(
                QString("An update (%1) is available.\n\nDo you want to "
                        "download and install it?")
                    .arg(Updates::getInstance().getOnlineVersion()));
            this->updateGeometry();
        }
        break;

        case Updates::SearchFailed:
        {
            this->ui_.label->setText("Failed to load version information.");
        }
        break;

        case Updates::Downloading:
        {
            this->ui_.label->setText(
                "Downloading updates.\n\nChatterino will restart "
                "automatically when the download is done.");
        }
        break;

        case Updates::DownloadFailed:
        {
            this->ui_.label->setText("Failed to download the update.");
        }
        break;

        case Updates::WriteFileFailed:
        {
            this->ui_.label->setText("Failed to save the update to disk.");
        }
        break;

        default:;
    }
}

}  // namespace chatterino