src/Download.cpp

   1 /*
   2 * Copyright (C) 2010 Toni Gundogdu.
   3 *
   4 * This program is free software: you can redistribute it and/or modify
   5 * it under the terms of the GNU General Public License as published by
   6 * the Free Software Foundation, either version 3 of the License, or
   7 * (at your option) any later version.
   8 *
   9 * This program is distributed in the hope that it will be useful,
  10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 * GNU General Public License for more details.
  13 *
  14 * You should have received a copy of the GNU General Public License
  15 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16 */
  17
  18 #include <QDebug>
  19
  20 #include "util.h"
  21 #include "Log.h"
  22 #include "Download.h"
  23
  24 // main.cpp
  25 extern Log log;
  26
  27 Download::Download (QWidget *parent/*=NULL*/)
  28     : QProgressDialog (parent), _canceled (false)
  29 {
  30
  31 #define _wrap(s,sl) \
  32     do { connect (&_proc, SIGNAL(s), this, SLOT(sl)); } while (0)
  33
  34     _wrap (started (), onCurlStarted ());
  35
  36     _wrap (error (QProcess::ProcessError),
  37         onCurlError (QProcess::ProcessError));
  38
  39     _wrap (readyRead (), onCurlReadyRead ());
  40
  41     _wrap (finished (int, QProcess::ExitStatus),
  42         onCurlFinished (int, QProcess::ExitStatus));
  43
  44 #undef _wrap
  45
  46     connect (this, SIGNAL(canceled()), this, SLOT (onCanceled()));
  47
  48     setWindowModality (Qt::WindowModal);
  49     setAutoClose (false);
  50
  51     _proc.setProcessChannelMode (QProcess::MergedChannels);
  52 }
  53
  54 void
  55 Download::start (const QString& cmd, const QString& fpath, Video *video) {
  56
  57     Q_ASSERT (!cmd.isEmpty ());
  58     Q_ASSERT (!fpath.isEmpty ());
  59     Q_ASSERT (video != NULL);
  60
  61     _lastError.clear ();
  62
  63     QStringList args = cmd.split (" ");
  64
  65     args.replaceInStrings ("%u", video->get (Video::Link).toString ());
  66     args.replaceInStrings ("%f", fpath);
  67
  68     log << args.join (" ") + "\n";
  69
  70     _canceled = false;
  71
  72     show ();
  73     _proc.start (args.takeFirst (), args);
  74     exec ();
  75 }
  76
  77 void
  78 Download::onCurlStarted ()
  79     { setLabelText (tr ("Starting download ...")); }
  80
  81 void
  82 Download::onCurlError (QProcess::ProcessError n) {
  83
  84     if (!_canceled) {
  85         emit error (NomNom::to_process_errmsg (n));
  86         cancel ();
  87     }
  88 }
  89
  90 static const QRegExp rx_prog ("^(\\d+).*(\\d+:\\d+:\\d+|\\d+d \\d+h)\\s+(\\w+)$");
  91 static const QRegExp rx_err  ("curl:\\s+(.*)$");
  92 static const QRegExp rx_rate ("(\\D+)");   // rate unit
  93
  94 void
  95 Download::onCurlReadyRead () {
  96
  97     static char data[1024];
  98
  99     while (_proc.readLine (data, sizeof (data))) {
 100
 101         QString ln = QString::fromLocal8Bit (data).simplified ();
 102
 103         if (rx_err.indexIn (ln) != -1) {
 104             _lastError = "curl: " +rx_err.cap (1);
 105             continue;
 106         }
 107
 108         if (ln.split (" ").count () < 12)
 109             continue; // Full line updates only, PLZKTHXBYE.
 110
 111 #ifdef _0
 112         qDebug () << ln;
 113         qDebug () << "--";
 114 #endif
 115
 116         if (rx_prog.indexIn (ln) != -1) {
 117
 118             enum {
 119                 PERCENT = 1,
 120                 ETA,
 121                 RATE
 122             };
 123
 124 #ifdef _0
 125             qDebug ()
 126                 << rx_prog.cap (PERCENT)
 127                 << rx_prog.cap (ETA)
 128                 << rx_prog.cap (RATE);
 129             qDebug ()
 130                 << ln;
 131 #endif
 132
 133             setValue (rx_prog.cap (PERCENT).toInt ());
 134
 135             QString rate = rx_prog.cap (RATE);
 136
 137             if (rx_rate.indexIn (rate) == -1) {
 138                 rate = QString ("%1k").arg (rate.toLongLong ()/1024.0,2,'f',1);
 139             }
 140
 141             const QString s = tr("Copying at %1, %2")
 142                 .arg (rate)
 143                 .arg (rx_prog.cap (ETA))
 144                 ;
 145
 146             setLabelText (s);
 147         }
 148
 149         else
 150             log << ln;
 151
 152     }
 153
 154 }
 155
 156 void
 157 Download::onCurlFinished (int exitCode, QProcess::ExitStatus exitStatus) {
 158
 159     if (exitStatus == QProcess::NormalExit && exitCode == 0)
 160         ;
 161     else {
 162         if (!_canceled)
 163             emit error (_lastError);
 164     }
 165
 166     cancel ();
 167 }
 168
 169 void
 170 Download::onCanceled () {
 171
 172     _canceled = true;
 173
 174     if (_proc.state () == QProcess::Running)
 175         _proc.kill ();
 176 }
 177
 178