#include "InitUpdateButton.hpp"

#include "widgets/dialogs/UpdateDialog.hpp"
#include "widgets/helper/RippleEffectButton.hpp"

namespace chatterino {

void initUpdateButton(RippleEffectButton &button,
                      std::unique_ptr<UpdateDialog> &handle,
                      pajlada::Signals::SignalHolder &signalHolder)
{
    button.hide();

    // show update prompt when clicking the button
    QObject::connect(&button, &RippleEffectButton::clicked, [&button, &handle] {
        (void)(handle);

        auto dialog = new UpdateDialog();
        dialog->setActionOnFocusLoss(BaseWindow::Delete);
        dialog->move(button.mapToGlobal(
            QPoint(int(-100 * button.getScale()), button.height())));
        dialog->show();
        dialog->raise();

        dialog->buttonClicked.connect([&button](auto buttonType) {
            switch (buttonType) {
                case UpdateDialog::Dismiss: {
                    button.hide();
                } break;
                case UpdateDialog::Install: {
                    Updates::getInstance().installUpdates();
                } break;
            }
        });

        //        handle.reset(dialog);
        //        dialog->closing.connect([&handle] { handle.release(); });
    });

    // update image when state changes
    auto updateChange = [&button](auto) {
        button.setVisible(Updates::getInstance().shouldShowUpdateButton());

        auto imageUrl = Updates::getInstance().isError()
                            ? ":/images/download_update_error.png"
                            : ":/images/download_update.png";
        button.setPixmap(QPixmap(imageUrl));
    };

    updateChange(Updates::getInstance().getStatus());

    signalHolder.managedConnect(
        Updates::getInstance().statusUpdated,
        [updateChange](auto status) { updateChange(status); });
}

}  // namespace chatterino