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)
  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
  52 void
  53 Download::start (const QString& cmd, const QString& fpath, Video *video) {
  54
  55 #ifdef _0
  56     qDebug () << __PRETTY_FUNCTION__;
  57 #endif
  58
  59     Q_ASSERT (!cmd.isEmpty ());
  60     Q_ASSERT (!fpath.isEmpty ());
  61     Q_ASSERT (video != NULL);
  62
  63     _lastError.clear ();
  64
  65     QStringList args = cmd.split (" ");
  66
  67     args.replaceInStrings ("%u", video->get (Video::Link).toString ());
  68     args.replaceInStrings ("%f", fpath);
  69
  70     log << args.join (" ") + "\n";
  71
  72     _proc.setProcessChannelMode (QProcess::MergedChannels);
  73
  74     show ();
  75     _proc.start (args.takeFirst (), args);
  76     exec ();
  77 }
  78
  79 void
  80 Download::onCurlStarted ()
  81     { setLabelText (tr ("Starting download ...")); }
  82
  83 void
  84 Download::onCurlError (QProcess::ProcessError n) {
  85 #ifdef _0
  86     qDebug () << __PRETTY_FUNCTION__;
  87 #endif
  88     emit error (NomNom::to_process_errmsg (n));
  89     cancel ();
  90 }
  91
  92 static const QRegExp rx_prog ("^(\\d+).*(\\d+:\\d+:\\d+|\\d+d \\d+h)\\s+(\\w+)$");
  93 static const QRegExp rx_err  ("curl:\\s+(.*)$");
  94 static const QRegExp rx_rate ("(\\D+)");   // rate unit
  95
  96 void
  97 Download::onCurlReadyRead () {
  98
  99 #ifdef _0
 100     qDebug () << __PRETTY_FUNCTION__;
 101 #endif
 102
 103     static char data[1024];
 104
 105     while (_proc.readLine (data, sizeof (data))) {
 106
 107         QString ln = QString::fromLocal8Bit (data).simplified ();
 108
 109         if (rx_err.indexIn (ln) != -1) {
 110             _lastError = "curl: " +rx_err.cap (1);
 111             continue;
 112         }
 113
 114         if (ln.split (" ").count () < 12)
 115             continue; // Full line updates only, PLZKTHXBYE.
 116
 117 #ifdef _0
 118         qDebug () << ln;
 119         qDebug () << "--";
 120 #endif
 121
 122         if (rx_prog.indexIn (ln) != -1) {
 123
 124             enum {
 125                 PERCENT = 1,
 126                 ETA,
 127                 RATE
 128             };
 129
 130 #ifdef _0
 131             qDebug ()
 132                 << rx_prog.cap (PERCENT)
 133                 << rx_prog.cap (ETA)
 134                 << rx_prog.cap (RATE);
 135             qDebug ()
 136                 << ln;
 137 #endif
 138
 139             setValue (rx_prog.cap (PERCENT).toInt ());
 140
 141             QString rate = rx_prog.cap (RATE);
 142
 143             if (rx_rate.indexIn (rate) == -1) {
 144                 rate = QString ("%1k").arg (rate.toLongLong ()/1024.0,2,'f',1);
 145             }
 146
 147             const QString s = tr("Copying at %1, %2")
 148                 .arg (rate)
 149                 .arg (rx_prog.cap (ETA))
 150                 ;
 151
 152             setLabelText (s);
 153         }
 154
 155         else
 156             log << ln;
 157
 158     }
 159
 160 }
 161
 162 void
 163 Download::onCurlFinished (int exitCode, QProcess::ExitStatus es) {
 164
 165 #ifdef _0
 166     qDebug () << __PRETTY_FUNCTION__;
 167 #endif
 168
 169     switch (es) {
 170
 171     case QProcess::NormalExit:
 172
 173         switch (exitCode) {
 174
 175         default: emit error (_lastError); break;
 176
 177         case  0: break;
 178
 179         } // Switch exitCode.
 180
 181     break; // NormalExit.
 182
 183     default: break;
 184
 185     } // Switch ExitStatus.
 186
 187     cancel ();
 188 }
 189
 190 void
 191 Download::onCanceled () {
 192
 193 #ifdef _0
 194     qDebug () << __PRETTY_FUNCTION__;
 195 #endif
 196
 197     if (_proc.state () == QProcess::Running)
 198         _proc.kill ();
 199 }
 200
 201