315 lines
11 KiB
C++
Executable File
315 lines
11 KiB
C++
Executable File
/*
|
|
* (C) Copyright 2014 Alex Spataru
|
|
*
|
|
* All rights reserved. This program and the accompanying materials
|
|
* are made available under the terms of the GNU Lesser General Public License
|
|
* (LGPL) version 2.1 which accompanies this distribution, and is available at
|
|
* http://www.gnu.org/licenses/lgpl-2.1.html
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
*/
|
|
|
|
//==============================================================================
|
|
// Class includes
|
|
//==============================================================================
|
|
|
|
#include "Downloader.h"
|
|
|
|
//==============================================================================
|
|
// System includes
|
|
//==============================================================================
|
|
|
|
#include <math.h>
|
|
|
|
#include <QDir>
|
|
#include <QFile>
|
|
#include <QProcess>
|
|
#include <QMessageBox>
|
|
#include <QNetworkReply>
|
|
#include <QDesktopServices>
|
|
#include <QNetworkAccessManager>
|
|
|
|
//==============================================================================
|
|
// Downloader::Downloader
|
|
//==============================================================================
|
|
|
|
Downloader::Downloader (QWidget* parent) : QWidget (parent) {
|
|
m_ui = new Ui::Downloader;
|
|
m_ui->setupUi (this);
|
|
|
|
/* Initialize private members */
|
|
m_manager = new QNetworkAccessManager();
|
|
|
|
/* Initialize internal values */
|
|
m_filePath = "";
|
|
m_startTime = 0;
|
|
m_useCustomProcedures = false;
|
|
|
|
/* Make the window look like a modal dialog */
|
|
setWindowIcon (QIcon ());
|
|
setWindowFlags (Qt::Dialog | Qt::CustomizeWindowHint | Qt::WindowTitleHint);
|
|
|
|
/* Configure the appearance and behavior of the buttons */
|
|
m_ui->openButton->setEnabled (false);
|
|
m_ui->openButton->setVisible (false);
|
|
connect (m_ui->stopButton, SIGNAL (clicked()),
|
|
this, SLOT (cancelDownload()));
|
|
connect (m_ui->openButton, SIGNAL (clicked()),
|
|
this, SLOT (installUpdate()));
|
|
|
|
/* Resize to fit */
|
|
setFixedSize (minimumSizeHint());
|
|
}
|
|
|
|
//==============================================================================
|
|
// Downloader::~Downloader
|
|
//==============================================================================
|
|
|
|
Downloader::~Downloader() {
|
|
delete m_ui;
|
|
delete m_reply;
|
|
delete m_manager;
|
|
}
|
|
|
|
//==============================================================================
|
|
// Downloader::useCustomInstallProcedures
|
|
//==============================================================================
|
|
|
|
bool Downloader::useCustomInstallProcedures() const {
|
|
return m_useCustomProcedures;
|
|
}
|
|
|
|
//==============================================================================
|
|
// Downloader::startDownload
|
|
//==============================================================================
|
|
|
|
void Downloader::startDownload (const QUrl& url) {
|
|
/* Reset UI */
|
|
m_ui->progressBar->setValue (0);
|
|
m_ui->stopButton->setText (tr ("Stop"));
|
|
m_ui->downloadLabel->setText (tr ("Downloading updates"));
|
|
m_ui->timeLabel->setText (tr ("Time remaining") + ": " + tr ("unknown"));
|
|
|
|
/* Start download */
|
|
m_startTime = QDateTime::currentDateTime().toTime_t();
|
|
m_reply = m_manager->get (QNetworkRequest (url));
|
|
|
|
/* Update UI when download progress changes or download finishes */
|
|
connect (m_reply, SIGNAL (downloadProgress (qint64, qint64)),
|
|
this, SLOT (updateProgress (qint64, qint64)));
|
|
connect (m_reply, SIGNAL (redirected (QUrl)),
|
|
this, SLOT (startDownload (QUrl)));
|
|
connect (m_reply, SIGNAL (finished()),
|
|
this, SLOT (onDownloadFinished()));
|
|
|
|
showNormal();
|
|
}
|
|
|
|
//==============================================================================
|
|
// Downloader::openDownload
|
|
//==============================================================================
|
|
|
|
void Downloader::openDownload() {
|
|
if (!m_filePath.isEmpty())
|
|
QDesktopServices::openUrl (QUrl::fromLocalFile (m_filePath));
|
|
|
|
else {
|
|
QMessageBox::critical (this,
|
|
tr ("Error"),
|
|
tr ("Cannot find downloaded update!"),
|
|
QMessageBox::Close);
|
|
}
|
|
}
|
|
|
|
//==============================================================================
|
|
// Downloader::installUpdate
|
|
//==============================================================================
|
|
|
|
void Downloader::installUpdate() {
|
|
if (useCustomInstallProcedures())
|
|
return;
|
|
|
|
QMessageBox box;
|
|
box.setIcon (QMessageBox::Question);
|
|
box.setDefaultButton (QMessageBox::Ok);
|
|
box.setStandardButtons (QMessageBox::Ok | QMessageBox::Cancel);
|
|
box.setInformativeText (tr ("Click \"OK\" to begin installing the update"));
|
|
box.setText ("<h3>" +
|
|
tr ("In order to install the update, you may need to "
|
|
"quit the application.")
|
|
+ "</h3>");
|
|
|
|
if (box.exec() == QMessageBox::Ok) {
|
|
if (!useCustomInstallProcedures())
|
|
openDownload();
|
|
}
|
|
|
|
else {
|
|
m_ui->openButton->setEnabled (true);
|
|
m_ui->openButton->setVisible (true);
|
|
m_ui->timeLabel->setText (tr ("Click the \"Open\" button to "
|
|
"apply the update"));
|
|
}
|
|
}
|
|
|
|
//==============================================================================
|
|
// Downloader::cancelDownload
|
|
//==============================================================================
|
|
|
|
void Downloader::cancelDownload() {
|
|
if (!m_reply->isFinished()) {
|
|
QMessageBox box;
|
|
box.setWindowTitle (tr ("Updater"));
|
|
box.setIcon (QMessageBox::Question);
|
|
box.setStandardButtons (QMessageBox::Yes | QMessageBox::No);
|
|
box.setText (tr ("Are you sure you want to cancel the download?"));
|
|
|
|
if (box.exec() == QMessageBox::Yes) {
|
|
hide();
|
|
m_reply->abort();
|
|
}
|
|
}
|
|
|
|
else
|
|
hide();
|
|
}
|
|
|
|
//==============================================================================
|
|
// Downloader::onDownloadFinished
|
|
//==============================================================================
|
|
|
|
void Downloader::onDownloadFinished() {
|
|
m_ui->stopButton->setText (tr ("Close"));
|
|
m_ui->downloadLabel->setText (tr ("Download complete!"));
|
|
m_ui->timeLabel->setText (tr ("The installer will open separately")
|
|
+ "...");
|
|
|
|
QByteArray data = m_reply->readAll();
|
|
|
|
if (!data.isEmpty()) {
|
|
QString name = m_reply->url().toString().split ("/").last();
|
|
|
|
/* Handle HTML redirections automatically */
|
|
if (data.startsWith ("<html") || data.startsWith ("<!DOCTYPE html"))
|
|
name.append (".html");
|
|
|
|
/* Save downloaded data to disk */
|
|
QFile file (QDir::tempPath() + "/" + name);
|
|
if (file.open (QIODevice::WriteOnly)) {
|
|
file.write (data);
|
|
file.close();
|
|
|
|
m_filePath = file.fileName();
|
|
emit downloadFinished (m_reply->url().toString(), m_filePath);
|
|
}
|
|
|
|
/* Open downloaded update */
|
|
m_reply->close();
|
|
installUpdate();
|
|
}
|
|
}
|
|
|
|
//==============================================================================
|
|
// Downloader::calculateSizes
|
|
//==============================================================================
|
|
|
|
void Downloader::calculateSizes (qint64 received, qint64 total) {
|
|
QString totalSize;
|
|
QString receivedSize;
|
|
|
|
if (total < 1024)
|
|
totalSize = tr ("%1 bytes").arg (total);
|
|
|
|
else if (total < 1048576)
|
|
totalSize = tr ("%1 KB").arg (round (total / 1024));
|
|
|
|
else
|
|
totalSize = tr ("%1 MB").arg (round (total / 1048576));
|
|
|
|
if (received < 1024)
|
|
receivedSize = tr ("%1 bytes").arg (received);
|
|
|
|
else if (received < 1048576)
|
|
receivedSize = tr ("%1 KB").arg (received / 1024);
|
|
|
|
else
|
|
receivedSize = tr ("%1 MB").arg (received / 1048576);
|
|
|
|
m_ui->downloadLabel->setText (tr ("Downloading updates")
|
|
+ " (" + receivedSize + " " + tr ("of")
|
|
+ " " + totalSize + ")");
|
|
}
|
|
|
|
//==============================================================================
|
|
// Downloader::updateProgress
|
|
//==============================================================================
|
|
|
|
void Downloader::updateProgress (qint64 received, qint64 total) {
|
|
if (total > 0) {
|
|
m_ui->progressBar->setMinimum (0);
|
|
m_ui->progressBar->setMaximum (100);
|
|
m_ui->progressBar->setValue ((received * 100) / total);
|
|
|
|
calculateSizes (received, total);
|
|
calculateTimeRemaining (received, total);
|
|
}
|
|
|
|
else {
|
|
m_ui->progressBar->setMinimum (0);
|
|
m_ui->progressBar->setMaximum (0);
|
|
m_ui->progressBar->setValue (-1);
|
|
m_ui->downloadLabel->setText (tr ("Downloading Updates") + "...");
|
|
m_ui->timeLabel->setText (QString ("%1: %2")
|
|
.arg (tr ("Time Remaining"))
|
|
.arg (tr ("Unknown")));
|
|
}
|
|
}
|
|
|
|
//==============================================================================
|
|
// Downloader::calculateTimeRemaining
|
|
//==============================================================================
|
|
|
|
void Downloader::calculateTimeRemaining (qint64 received, qint64 total) {
|
|
uint difference = QDateTime::currentDateTime().toTime_t() - m_startTime;
|
|
|
|
if (difference > 0) {
|
|
QString timeString;
|
|
float timeRemaining = total / (received / difference);
|
|
|
|
if (timeRemaining > 7200) {
|
|
timeRemaining /= 3600;
|
|
timeString = tr ("About %1 hours").arg (int (timeRemaining + 0.5));
|
|
}
|
|
|
|
else if (timeRemaining > 60) {
|
|
timeRemaining /= 60;
|
|
timeString = tr ("About %1 minutes").arg (int (timeRemaining + 0.5));
|
|
}
|
|
|
|
else if (timeRemaining <= 60)
|
|
timeString = tr ("%1 seconds").arg (int (timeRemaining + 0.5));
|
|
|
|
m_ui->timeLabel->setText (tr ("Time remaining") + ": " + timeString);
|
|
}
|
|
}
|
|
|
|
//==============================================================================
|
|
// Downloader::round
|
|
//==============================================================================
|
|
|
|
float Downloader::round (const float& input) {
|
|
return roundf (input * 100) / 100;
|
|
}
|
|
|
|
//==============================================================================
|
|
// Downloader::setUseCustomInstallProcedures
|
|
//==============================================================================
|
|
|
|
void Downloader::setUseCustomInstallProcedures (const bool& custom) {
|
|
m_useCustomProcedures = custom;
|
|
}
|