Update upstream source from tag 'upstream/3.6'

[minitube] / src / downloadmanager.cpp

diff --git a/src/downloadmanager.cpp b/src/downloadmanager.cpp
index 36ede9bf384930b10bf2aa82affc8229b5ddc7ed..4ea5d8c7b0a0f789eb137dd738286f1bdac46c7c 100644 (file)
--- a/src/downloadmanager.cpp
+++ b/src/downloadmanager.cpp
@@ -1,14 +1,43 @@
+/* $BEGIN_LICENSE
+
+This file is part of Minitube.
+Copyright 2009, Flavio Tordini <flavio.tordini@gmail.com>
+
+Minitube is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+Minitube 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 General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with Minitube.  If not, see <http://www.gnu.org/licenses/>.
+
+$END_LICENSE */
+
 #include "downloadmanager.h"
 #include "downloaditem.h"
 #include "downloadmodel.h"
 #include "video.h"
 #include "constants.h"
+#include "mainwindow.h"
+#ifdef APP_ACTIVATION
+#include "activation.h"
+#endif
+#ifdef APP_EXTRA
+#include "extra.h"
+#endif
+#include "datautils.h"
+#include "iconutils.h"
 
 static DownloadManager *downloadManagerInstance = 0;
 
-DownloadManager::DownloadManager(QObject *parent) :
-        QObject(parent),
-        downloadModel(new DownloadModel(this, this))
+DownloadManager::DownloadManager(QWidget *parent) :
+    QObject(parent),
+    downloadModel(new DownloadModel(this, this))
 { }
 
 DownloadManager* DownloadManager::instance() {
@@ -24,53 +53,29 @@ void DownloadManager::clear() {
 
 int DownloadManager::activeItems() {
     int num = 0;
-    foreach (DownloadItem *item, items) {
+    for (DownloadItem *item : items) {
         if (item->status() == Downloading || item->status() == Starting) num++;
     }
     return num;
 }
 
 DownloadItem* DownloadManager::itemForVideo(Video* video) {
-    foreach (DownloadItem *item, items) {
-        if (item->getVideo()->id() == video->id()) return item;
+    for (DownloadItem *item : items) {
+        if (item->getVideo()->getId() == video->getId()) return item;
     }
     return 0;
 }
 
 void DownloadManager::addItem(Video *video) {
     // qDebug() << __FUNCTION__ << video->title();
-    
-#ifdef APP_DEMO
-    if (video->duration() >= 60*4) {
-        QMessageBox msgBox;
-        msgBox.setIconPixmap(QPixmap(":/images/app.png").scaled(64, 64, Qt::KeepAspectRatio, Qt::SmoothTransformation));
-        msgBox.setText(tr("This is just the demo version of %1.").arg(Constants::APP_NAME));
-        msgBox.setInformativeText(
-                tr("It can only download videos shorter than %1 minutes so you can test the download functionality.")
-                .arg(4));
-        msgBox.setModal(true);
-
-        QPushButton *quitButton = msgBox.addButton(tr("Continue"), QMessageBox::RejectRole);
-        QPushButton *buyButton = msgBox.addButton(tr("Get the full version"), QMessageBox::ActionRole);
-
-        msgBox.exec();
-
-        if (msgBox.clickedButton() == buyButton) {
-            QDesktopServices::openUrl(QString(Constants::WEBSITE) + "#download");
-        }
-
-        return;
-    }
-#endif
-    
 
     DownloadItem *item = itemForVideo(video);
     if (item != 0) {
         if (item->status() == Failed || item->status() == Idle) {
-            qDebug() << "Restarting download" << video->title();
+            qDebug() << "Restarting download" << video->getTitle();
             item->tryAgain();
         } else {
-            qDebug() << "Already downloading video" << video->title();
+            qDebug() << "Already downloading video" << video->getTitle();
         }
         return;
     }
@@ -93,22 +98,16 @@ void DownloadManager::gotStreamUrl(QUrl url) {
 
     video->disconnect(this);
 
-    QString path = currentDownloadFolder();
+    QString basename = DataUtils::stringToFilename(video->getTitle());
+    if (basename.isEmpty()) basename = video->getId();
 
-    // TODO ensure all chars are filename compatible
-    QString basename = video->title().simplified();
-    basename.replace('(', '[');
-    basename.replace(')', ']');
-    basename.replace('/', '-');
-    basename.replace('\\', '-');
-    QString filename = path + "/" + basename + ".mp4";
+    QString filename = currentDownloadFolder() + "/" + basename + ".mp4";
 
     Video *videoCopy = video->clone();
     DownloadItem *item = new DownloadItem(videoCopy, url, filename, this);
 
-    int row = items.count();
-    downloadModel->beginInsertRows(QModelIndex(), row, row);
-    items.append(item);
+    downloadModel->beginInsertRows(QModelIndex(), 0, 0);
+    items.prepend(item);
     downloadModel->endInsertRows();
 
     // connect(item, SIGNAL(statusChanged()), SLOT(updateStatusMessage()));
@@ -120,20 +119,40 @@ void DownloadManager::gotStreamUrl(QUrl url) {
 
 void DownloadManager::itemFinished() {
     if (activeItems() == 0) emit finished();
+#ifdef APP_EXTRA
+    DownloadItem *item = static_cast<DownloadItem*>(sender());
+    if (!item) {
+        qDebug() << "Cannot get item in" << __FUNCTION__;
+        return;
+    }
+    Video *video = item->getVideo();
+    if (!video) return;
+    QString stats = tr("%1 downloaded in %2").arg(
+                DownloadItem::formattedFilesize(item->bytesTotal()),
+                DownloadItem::formattedTime(item->totalTime(), false));
+    Extra::notify(tr("Download finished"), video->getTitle(), stats);
+#endif
 }
 
 void DownloadManager::updateStatusMessage() {
-    QString message = tr("%n Download(s)", "", items.size());
+    QString message = tr("%n Download(s)", "", activeItems());
     emit statusMessageChanged(message);
 }
 
 QString DownloadManager::defaultDownloadFolder() {
     // download in the Movies system folder
-    QString path = QDesktopServices::storageLocation(QDesktopServices::MoviesLocation);
-    QDir moviesDir(path);
+    QString path = QStandardPaths::writableLocation(QStandardPaths::MoviesLocation);
+
+    const QDir moviesDir(path);
     if (!moviesDir.exists()) {
         // fallback to Desktop
-        path = QDesktopServices::storageLocation(QDesktopServices::DesktopLocation);
+        path = QStandardPaths::writableLocation(QStandardPaths::DesktopLocation);
+
+        const QDir desktopDir(path);
+        if (!desktopDir.exists()) {
+            // fallback to Home
+            path = QStandardPaths::writableLocation(QStandardPaths::HomeLocation);
+        }
     }
     return path;
 }