src/Dialog_Update.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // LameXP - Audio Encoder Front-End
   3 // Copyright (C) 2004-2015 LoRd_MuldeR <MuldeR2@GMX.de>
   4 //
   5 // This program is free software; you can redistribute it and/or modify
   6 // it under the terms of the GNU General Public License as published by
   7 // the Free Software Foundation; either version 2 of the License, or
   8 // (at your option) any later version, but always including the *additional*
   9 // restrictions defined in the "License.txt" file.
  10 //
  11 // This program is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU General Public License for more details.
  15 //
  16 // You should have received a copy of the GNU General Public License along
  17 // with this program; if not, write to the Free Software Foundation, Inc.,
  18 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 //
  20 // http://www.gnu.org/licenses/gpl-2.0.txt
  21 ///////////////////////////////////////////////////////////////////////////////
  22
  23 #include "Dialog_Update.h"
  24
  25 //UIC includes
  26 #include "UIC_UpdateDialog.h"
  27
  28 //LameXP includes
  29 #include "Global.h"
  30 #include "Dialog_LogView.h"
  31 #include "Model_Settings.h"
  32
  33 //MUtils
  34 #include <MUtils/UpdateChecker.h>
  35 #include <MUtils/Version.h>
  36 #include <MUtils/Exception.h>
  37 #include <MUtils/Sound.h>
  38 #include <MUtils/GUI.h>
  39 #include <MUtils/OSSupport.h>
  40 #include <MUtils/Taskbar7.h>
  41
  42 //Qt includes
  43 #include <QClipboard>
  44 #include <QFileDialog>
  45 #include <QTimer>
  46 #include <QProcess>
  47 #include <QDesktopServices>
  48 #include <QUrl>
  49 #include <QCloseEvent>
  50 #include <QMovie>
  51 #include <QMessageBox>
  52
  53 ///////////////////////////////////////////////////////////////////////////////
  54
  55 #define SHOW_HINT(TEXT, ICON) do \
  56 { \
  57         ui->hintLabel->setText((TEXT)); \
  58         ui->hintIcon->setPixmap(QIcon((ICON)).pixmap(16,16)); \
  59         ui->hintIcon->show(); \
  60         ui->hintLabel->show(); \
  61 } \
  62 while(0)
  63
  64 #define UPDATE_TASKBAR(STATE, ICON) do \
  65 { \
  66         m_taskbar->setTaskbarState((STATE)); \
  67         m_taskbar->setOverlayIcon(&QIcon((ICON))); \
  68 } \
  69 while(0)
  70
  71 ///////////////////////////////////////////////////////////////////////////////
  72
  73 UpdateDialog::UpdateDialog(const SettingsModel *const settings, QWidget *parent)
  74 :
  75         QDialog(parent),
  76         ui(new Ui::UpdateDialog),
  77         m_taskbar(new MUtils::Taskbar7(parent)),
  78         m_settings(settings),
  79         m_logFile(new QStringList()),
  80         m_betaUpdates(settings ? (settings->autoUpdateCheckBeta() || lamexp_version_demo()) : lamexp_version_demo()),
  81         m_success(false),
  82         m_firstShow(true),
  83         m_updateReadyToInstall(false),
  84         m_updaterProcess(NULL),
  85         m_binaryUpdater(lamexp_tools_lookup("wupdate.exe")),
  86         m_binaryWGet(lamexp_tools_lookup("wget.exe")),
  87         m_binaryGnuPG(lamexp_tools_lookup("gpgv.exe")),
  88         m_binaryKeys(lamexp_tools_lookup("gpgv.gpg"))
  89 {
  90         if(m_binaryUpdater.isEmpty())
  91         {
  92                 MUTILS_THROW("Tools not initialized correctly!");
  93         }
  94
  95         //Init the dialog, from the .ui file
  96         ui->setupUi(this);
  97         setWindowFlags(windowFlags() ^ Qt::WindowContextHelpButtonHint);
  98
  99         //Disable "X" button
 100         MUtils::GUI::enable_close_button(this, false);
 101
 102         //Init animation
 103         m_animator.reset(new QMovie(":/images/Loading3.gif"));
 104         ui->labelAnimationCenter->setMovie(m_animator.data());
 105         m_animator->start();
 106
 107         //Indicate beta updates
 108         if(m_betaUpdates)
 109         {
 110                 setWindowTitle(windowTitle().append(" [Beta]"));
 111         }
 112
 113         //Enable button
 114         connect(ui->retryButton, SIGNAL(clicked()), this, SLOT(checkForUpdates()));
 115         connect(ui->installButton, SIGNAL(clicked()), this, SLOT(applyUpdate()));
 116         connect(ui->infoLabel, SIGNAL(linkActivated(QString)), this, SLOT(linkActivated(QString)));
 117         connect(ui->logButton, SIGNAL(clicked()), this, SLOT(logButtonClicked()));
 118
 119         //Enable progress bar
 120         connect(ui->progressBar, SIGNAL(valueChanged(int)), this, SLOT(progressBarValueChanged(int)));
 121 }
 122
 123 UpdateDialog::~UpdateDialog(void)
 124 {
 125         if(m_animator)
 126         {
 127                 m_animator->stop();
 128         }
 129
 130         if(!m_thread.isNull())
 131         {
 132                 if(!m_thread->wait(1000))
 133                 {
 134                         m_thread->terminate();
 135                         m_thread->wait();
 136                 }
 137         }
 138
 139         m_taskbar->setTaskbarState(MUtils::Taskbar7::TASKBAR_STATE_NONE);
 140         m_taskbar->setOverlayIcon(NULL);
 141
 142         delete ui;
 143 }
 144
 145 void UpdateDialog::showEvent(QShowEvent *event)
 146 {
 147         QDialog::showEvent(event);
 148
 149         if(m_firstShow)
 150         {
 151                 if(m_thread.isNull())
 152                 {
 153                         m_thread.reset(new MUtils::UpdateChecker(m_binaryWGet, m_binaryGnuPG, m_binaryKeys, QLatin1String("LameXP"), lamexp_version_build(), m_betaUpdates));
 154                         connect(m_thread.data(), SIGNAL(statusChanged(int)), this, SLOT(threadStatusChanged(int)));
 155                         connect(m_thread.data(), SIGNAL(progressChanged(int)), this, SLOT(threadProgressChanged(int)));
 156                         connect(m_thread.data(), SIGNAL(messageLogged(QString)), this, SLOT(threadMessageLogged(QString)));
 157                         connect(m_thread.data(), SIGNAL(finished()), this, SLOT(threadFinished()));
 158                         connect(m_thread.data(), SIGNAL(terminated()), this, SLOT(threadFinished()));
 159                 }
 160
 161                 threadStatusChanged(m_thread->getUpdateStatus());
 162                 ui->labelVersionInstalled->setText(QString("%1 %2 (%3)").arg(tr("Build"), QString::number(lamexp_version_build()), MUtils::Version::app_build_date().toString(Qt::ISODate)));
 163                 ui->labelVersionLatest->setText(QString("(%1)").arg(tr("Unknown")));
 164
 165                 ui->installButton->setEnabled(false);
 166                 ui->closeButton->setEnabled(false);
 167                 ui->retryButton->setEnabled(false);
 168                 ui->logButton->setEnabled(false);
 169                 ui->retryButton->hide();
 170                 ui->logButton->hide();
 171                 ui->infoLabel->hide();
 172                 ui->hintLabel->hide();
 173                 ui->hintIcon->hide();
 174                 ui->frameAnimation->hide();
 175
 176                 ui->progressBar->setMaximum(m_thread->getMaximumProgress());
 177                 ui->progressBar->setValue(0);
 178
 179                 m_updaterProcess = NULL;
 180
 181                 QTimer::singleShot(0, this, SLOT(updateInit()));
 182                 m_firstShow = false;
 183         }
 184 }
 185
 186 void UpdateDialog::closeEvent(QCloseEvent *event)
 187 {
 188         if(!ui->closeButton->isEnabled())
 189         {
 190                 event->ignore();
 191         }
 192         else
 193         {
 194                 m_taskbar->setTaskbarState(MUtils::Taskbar7::TASKBAR_STATE_NONE);
 195                 m_taskbar->setOverlayIcon(NULL);
 196         }
 197 }
 198
 199 void UpdateDialog::keyPressEvent(QKeyEvent *e)
 200 {
 201         if(e->key() == Qt::Key_F11)
 202         {
 203                 if(ui->closeButton->isEnabled()) logButtonClicked();
 204         }
 205         else if((e->key() == Qt::Key_F12) && e->modifiers().testFlag(Qt::ControlModifier))
 206         {
 207                 if(ui->closeButton->isEnabled()) testKnownHosts();
 208         }
 209         else
 210         {
 211                 QDialog::keyPressEvent(e);
 212         }
 213 }
 214
 215 bool UpdateDialog::event(QEvent *e)
 216 {
 217         if((e->type() == QEvent::ActivationChange) && (m_updaterProcess != NULL))
 218         {
 219                 MUtils::GUI::bring_to_front(m_updaterProcess);
 220         }
 221         return QDialog::event(e);
 222 }
 223
 224 void UpdateDialog::updateInit(void)
 225 {
 226         setMinimumSize(size());
 227         setMaximumHeight(height());
 228         QTimer::singleShot(0, this, SLOT(checkForUpdates()));
 229 }
 230
 231 void UpdateDialog::checkForUpdates(void)
 232 {
 233         if(m_thread->isRunning())
 234         {
 235                 qWarning("Update in progress, cannot check for updates now!");
 236         }
 237
 238         if(!MUtils::OS::user_is_admin())
 239         {
 240                 qWarning("User is not in the \"admin\" group, cannot update!");
 241                 QString message;
 242                 message += QString("<nobr>%1</nobr><br>").arg(tr("Sorry, but only users in the \"Administrators\" group can install updates."));
 243                 message += QString("<nobr>%1</nobr>").arg(tr("Please start application from an administrator account and try again!"));
 244                 if(QMessageBox::critical(this, this->windowTitle(), message, tr("Discard"), tr("Ignore")) != 1)
 245                 {
 246                         ui->closeButton->setEnabled(true);
 247                         close(); return;
 248                 }
 249         }
 250
 251         m_taskbar->setTaskbarState(MUtils::Taskbar7::TASKBAR_STATE_NORMAL);
 252         m_taskbar->setOverlayIcon(&QIcon(":/icons/transmit_blue.png"));
 253
 254         ui->progressBar->setValue(0);
 255         ui->installButton->setEnabled(false);
 256         ui->closeButton->setEnabled(false);
 257         ui->retryButton->setEnabled(false);
 258         ui->logButton->setEnabled(false);
 259         if(ui->infoLabel->isVisible()) ui->infoLabel->hide();
 260         if(ui->hintLabel->isVisible()) ui->hintLabel->hide();
 261         if(ui->hintIcon->isVisible()) ui->hintIcon->hide();
 262         ui->frameAnimation->show();
 263
 264         QApplication::processEvents(QEventLoop::ExcludeUserInputEvents);
 265         QApplication::setOverrideCursor(Qt::WaitCursor);
 266
 267         m_logFile->clear();
 268         m_thread->start();
 269 }
 270
 271 void UpdateDialog::threadStatusChanged(const int status)
 272 {
 273         switch(status)
 274         {
 275         case MUtils::UpdateChecker::UpdateStatus_NotStartedYet:
 276                 ui->statusLabel->setText(tr("Initializing, please wait..."));
 277                 break;
 278         case MUtils::UpdateChecker::UpdateStatus_CheckingConnection:
 279                 ui->statusLabel->setText(tr("Testing your internet connection, please wait..."));
 280                 break;
 281         case MUtils::UpdateChecker::UpdateStatus_FetchingUpdates:
 282                 ui->statusLabel->setText(tr("Checking for new updates online, please wait..."));
 283                 break;
 284         case MUtils::UpdateChecker::UpdateStatus_CompletedUpdateAvailable:
 285                 ui->statusLabel->setText(tr("A new version of LameXP is available!"));
 286                 SHOW_HINT(tr("We highly recommend all users to install this update as soon as possible."), ":/icons/shield_exclamation.png");
 287                 UPDATE_TASKBAR(MUtils::Taskbar7::TASKBAR_STATE_NORMAL, ":/icons/shield_exclamation.png");
 288                 break;
 289         case MUtils::UpdateChecker::UpdateStatus_CompletedNoUpdates:
 290                 ui->statusLabel->setText(tr("No new updates available at this time."));
 291                 SHOW_HINT(tr("Your version of LameXP is still up-to-date. Please check for updates regularly!"), ":/icons/shield_green.png");
 292                 UPDATE_TASKBAR(MUtils::Taskbar7::TASKBAR_STATE_NORMAL, ":/icons/shield_green.png");
 293                 break;
 294         case MUtils::UpdateChecker::UpdateStatus_CompletedNewVersionOlder:
 295                 ui->statusLabel->setText(tr("Your version appears to be newer than the latest release."));
 296                 SHOW_HINT(tr("This usually indicates your are currently using a pre-release version of LameXP."), ":/icons/shield_blue.png");
 297                 UPDATE_TASKBAR(MUtils::Taskbar7::TASKBAR_STATE_NORMAL, ":/icons/shield_error.png");
 298                 break;
 299         case MUtils::UpdateChecker::UpdateStatus_ErrorNoConnection:
 300                 ui->statusLabel->setText(tr("It appears that the computer currently is offline!"));
 301                 SHOW_HINT(tr("Please make sure your computer is connected to the internet and try again."), ":/icons/network_error.png");
 302                 UPDATE_TASKBAR(MUtils::Taskbar7::TASKBAR_STATE_NORMAL, ":/icons/exclamation.png");
 303                 break;
 304         case MUtils::UpdateChecker::UpdateStatus_ErrorConnectionTestFailed:
 305                 ui->statusLabel->setText(tr("Network connectivity test has failed!"));
 306                 SHOW_HINT(tr("Please make sure your computer is connected to the internet and try again."), ":/icons/network_error.png");
 307                 UPDATE_TASKBAR(MUtils::Taskbar7::TASKBAR_STATE_NORMAL, ":/icons/exclamation.png");
 308                 break;
 309         case MUtils::UpdateChecker::UpdateStatus_ErrorFetchUpdateInfo:
 310                 ui->statusLabel->setText(tr("Failed to fetch update information from server!"));
 311                 SHOW_HINT(tr("Sorry, the update server might be busy at this time. Plase try again later."), ":/icons/server_error.png");
 312                 UPDATE_TASKBAR(MUtils::Taskbar7::TASKBAR_STATE_NORMAL, ":/icons/exclamation.png");
 313                 break;
 314         default:
 315                 qWarning("Unknown status %d !!!", int(status));
 316         }
 317 }
 318
 319 void UpdateDialog::threadProgressChanged(const int progress)
 320 {
 321         ui->progressBar->setValue(progress);
 322 }
 323
 324 void UpdateDialog::threadMessageLogged(const QString &message)
 325 {
 326         (*m_logFile) << message;
 327 }
 328
 329 void UpdateDialog::threadFinished(void)
 330 {
 331         const bool bSuccess = m_thread->getSuccess();
 332
 333         ui->closeButton->setEnabled(true);
 334         if(ui->frameAnimation->isVisible()) ui->frameAnimation->hide();
 335         ui->progressBar->setValue(ui->progressBar->maximum());
 336
 337         if(!bSuccess)
 338         {
 339                 if(m_settings->soundsEnabled()) MUtils::Sound::play_sound("error", true);
 340         }
 341         else
 342         {
 343                 const bool bHaveUpdate = (m_thread->getUpdateStatus() == MUtils::UpdateChecker::UpdateStatus_CompletedUpdateAvailable);
 344                 ui->installButton->setEnabled(bHaveUpdate);
 345                 MUtils::Sound::beep(bHaveUpdate ? MUtils::Sound::BEEP_NFO : MUtils::Sound::BEEP_WRN);
 346
 347                 if(const MUtils::UpdateCheckerInfo *const updateInfo = m_thread->getUpdateInfo())
 348                 {
 349                         ui->infoLabel->setText(QString("%1<br><a href=\"%2\">%2</a>").arg(tr("More information available at:"), updateInfo->getDownloadSite()));
 350                         ui->labelVersionLatest->setText(QString("%1 %2 (%3)").arg(tr("Build"), QString::number(updateInfo->getBuildNo()), updateInfo->getBuildDate().toString(Qt::ISODate)));
 351                         ui->infoLabel->show();
 352                 }
 353
 354                 m_success = true;
 355         }
 356
 357         ui->retryButton->setVisible(!bSuccess);
 358         ui->logButton->setVisible(!bSuccess);
 359         ui->retryButton->setEnabled(!bSuccess);
 360         ui->logButton->setEnabled(!bSuccess);
 361
 362         QApplication::restoreOverrideCursor();
 363 }
 364
 365 void UpdateDialog::linkActivated(const QString &link)
 366 {
 367         QDesktopServices::openUrl(QUrl(link));
 368 }
 369
 370 void UpdateDialog::applyUpdate(void)
 371 {
 372         ui->installButton->setEnabled(false);
 373         ui->closeButton->setEnabled(false);
 374         ui->retryButton->setEnabled(false);
 375
 376         if(const MUtils::UpdateCheckerInfo *updateInfo = m_thread->getUpdateInfo())
 377         {
 378                 ui->statusLabel->setText(tr("Update is being downloaded, please be patient..."));
 379                 ui->frameAnimation->show();
 380                 if(ui->hintLabel->isVisible()) ui->hintLabel->hide();
 381                 if(ui->hintIcon->isVisible()) ui->hintIcon->hide();
 382                 int oldMax = ui->progressBar->maximum();
 383                 int oldMin = ui->progressBar->minimum();
 384                 ui->progressBar->setRange(0, 0);
 385                 QApplication::processEvents(QEventLoop::ExcludeUserInputEvents);
 386
 387                 QProcess process;
 388                 QStringList args;
 389                 QEventLoop loop;
 390
 391                 MUtils::init_process(process, QFileInfo(m_binaryUpdater).absolutePath(), false);
 392
 393                 connect(&process, SIGNAL(error(QProcess::ProcessError)), &loop, SLOT(quit()));
 394                 connect(&process, SIGNAL(finished(int,QProcess::ExitStatus)), &loop, SLOT(quit()));
 395
 396                 args << QString("/Location=%1").arg(updateInfo->getDownloadAddress());
 397                 args << QString("/Filename=%1").arg(updateInfo->getDownloadFilename());
 398                 args << QString("/TicketID=%1").arg(updateInfo->getDownloadFilecode());
 399                 args << QString("/CheckSum=%1").arg(updateInfo->getDownloadChecksum());
 400                 args << QString("/ToFolder=%1").arg(QDir::toNativeSeparators(QDir(QApplication::applicationDirPath()).canonicalPath()));
 401                 args << QString("/ToExFile=%1.exe").arg(QFileInfo(QFileInfo(QApplication::applicationFilePath()).canonicalFilePath()).completeBaseName());
 402                 args << QString("/AppTitle=LameXP (Build #%1)").arg(QString::number(updateInfo->getBuildNo()));
 403
 404                 QApplication::setOverrideCursor(Qt::WaitCursor);
 405                 UPDATE_TASKBAR(MUtils::Taskbar7::TASKBAR_STATE_INTERMEDIATE, ":/icons/transmit_blue.png");
 406
 407                 process.start(m_binaryUpdater, args);
 408                 bool updateStarted = process.waitForStarted();
 409                 if(updateStarted)
 410                 {
 411                         m_updaterProcess = MUtils::OS::process_id(&process);
 412                         loop.exec(QEventLoop::ExcludeUserInputEvents);
 413                 }
 414
 415                 m_updaterProcess = NULL;
 416                 QApplication::restoreOverrideCursor();
 417
 418                 ui->hintLabel->show();
 419                 ui->hintIcon->show();
 420                 ui->progressBar->setRange(oldMin, oldMax);
 421                 ui->progressBar->setValue(oldMax);
 422                 ui->frameAnimation->hide();
 423
 424                 if(updateStarted && (process.exitCode() == 0))
 425                 {
 426                         ui->statusLabel->setText(tr("Update ready to install. Applicaion will quit..."));
 427                         m_updateReadyToInstall = true;
 428                         m_taskbar->setTaskbarState(MUtils::Taskbar7::TASKBAR_STATE_NONE);
 429                         m_taskbar->setOverlayIcon(NULL);
 430                         accept();
 431                 }
 432                 else
 433                 {
 434                         ui->statusLabel->setText(tr("Update failed. Please try again or download manually!"));
 435                         m_taskbar->setTaskbarState(MUtils::Taskbar7::TASKBAR_STATE_ERROR);
 436                         m_taskbar->setOverlayIcon(&QIcon(":/icons/exclamation.png"));
 437                         m_taskbar->setTaskbarProgress(100, 100);
 438                 }
 439         }
 440
 441         ui->installButton->setEnabled(true);
 442         ui->closeButton->setEnabled(true);
 443 }
 444
 445 void UpdateDialog::logButtonClicked(void)
 446 {
 447         LogViewDialog *logView = new LogViewDialog(this);
 448         logView->exec(*m_logFile);
 449         MUTILS_DELETE(logView);
 450 }
 451
 452 void UpdateDialog::progressBarValueChanged(int value)
 453 {
 454         m_taskbar->setTaskbarProgress(value, ui->progressBar->maximum());
 455 }
 456
 457 void UpdateDialog::testKnownHosts(void)
 458 {
 459         ui->statusLabel->setText("Testing all known hosts, this may take a few minutes...");
 460
 461         if(MUtils::UpdateChecker *testThread = new MUtils::UpdateChecker(m_binaryWGet, m_binaryGnuPG, m_binaryKeys, QLatin1String("LameXP"), lamexp_version_build(), m_betaUpdates, true))
 462         {
 463                 QEventLoop loop;
 464                 m_logFile->clear();
 465
 466                 connect(testThread, SIGNAL(messageLogged(QString)), this, SLOT(threadMessageLogged(QString)));
 467                 connect(testThread, SIGNAL(finished()), &loop, SLOT(quit()));
 468                 connect(testThread, SIGNAL(terminated()), &loop, SLOT(quit()));
 469
 470                 testThread->start();
 471                 while(testThread->isRunning())
 472                 {
 473                         QTimer::singleShot(5000, &loop, SLOT(quit()));
 474                         loop.exec(QEventLoop::ExcludeUserInputEvents);
 475                 }
 476
 477                 MUTILS_DELETE(testThread);
 478                 logButtonClicked();
 479         }
 480
 481         ui->statusLabel->setText("Test completed.");
 482         MUtils::Sound::beep(MUtils::Sound::BEEP_NFO);
 483 }