src/DownloadDiag.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 "Preferences.h"
  23 #include "DownloadDiag.h"
  24
  25 // main.cpp
  26 extern SharedPreferences shPrefs;
  27 extern Log log;
  28
  29 DownloadDialog::DownloadDialog (QWidget *parent/*=NULL*/)
  30   : QProgressDialog (parent), _canceled (false)
  31 {
  32 #define _wrap(s,sl) \
  33     do { connect (&_proc, SIGNAL(s), this, SLOT(sl)); } while (0)
  34
  35   _wrap (started (), onCurlStarted ());
  36
  37   _wrap (error (QProcess::ProcessError),
  38          onCurlError (QProcess::ProcessError));
  39
  40   _wrap (readyRead (), onCurlReadyRead ());
  41
  42   _wrap (finished (int, QProcess::ExitStatus),
  43          onCurlFinished (int, QProcess::ExitStatus));
  44
  45 #undef _wrap
  46
  47   connect (this, SIGNAL(canceled()), this, SLOT (onCanceled()));
  48
  49   setWindowModality (Qt::WindowModal);
  50   setAutoClose (false);
  51
  52   _proc.setProcessChannelMode (QProcess::MergedChannels);
  53 }
  54
  55 void
  56 DownloadDialog::start (const QString& cmd, const QString& fpath, Video *video)
  57 {
  58   Q_ASSERT (!cmd.isEmpty ());
  59   Q_ASSERT (!fpath.isEmpty ());
  60   Q_ASSERT (video != NULL);
  61
  62   _lastError.clear ();
  63
  64   QStringList args = cmd.split (" ");
  65
  66   args.replaceInStrings ("%u", video->get (Video::Link).toString ());
  67   args.replaceInStrings ("%f", fpath);
  68
  69   log << args.join (" ") + "\n";
  70
  71   _canceled = false;
  72
  73   show ();
  74   _proc.start (args.takeFirst (), args);
  75   exec ();
  76 }
  77
  78 void
  79 DownloadDialog::onCurlStarted ()
  80 {
  81   setLabelText (tr ("Starting download ..."));
  82   if (shPrefs.get (SharedPreferences::MinWhenStarts).toBool ())
  83     parentWidget ()->showMinimized ();
  84 }
  85
  86 void
  87 DownloadDialog::onCurlError (QProcess::ProcessError n)
  88 {
  89   if (!_canceled)
  90     {
  91       hide ();
  92       NomNom::crit (parentWidget (), NomNom::to_process_errmsg (n));
  93       emit error ();
  94       cancel ();
  95     }
  96 }
  97
  98 static const QRegExp rx_err  ("curl:\\s+(.*)$");
  99 static const QRegExp rx_rate ("(\\D+)");   // rate unit
 100
 101 void
 102 DownloadDialog::onCurlReadyRead ()
 103 {
 104
 105   static char data[1024];
 106
 107   while (_proc.readLine (data, sizeof (data)))
 108     {
 109       QString ln = QString::fromLocal8Bit (data).simplified ();
 110
 111       if (rx_err.indexIn (ln) != -1)
 112         {
 113           _lastError = "curl: " +rx_err.cap (1);
 114           continue;
 115         }
 116
 117       QStringList lst = ln.split (" ");
 118
 119       if (lst.count () < 12)
 120         continue; // Full line updates only.
 121
 122 #ifdef _0
 123       qDebug () << lst;
 124 #endif
 125
 126       enum
 127       {
 128         PERCENT = 0,
 129         ETA     = 10,
 130         RATE    = 11
 131       };
 132
 133       setValue (lst[PERCENT].toInt ());
 134
 135       QString rate = lst[RATE];
 136
 137       if (rx_rate.indexIn (rate) == -1)
 138         rate = QString ("%1k").arg (rate.toLongLong ()/1024.0,2,'f',1);
 139
 140       const QString s = tr("Copying at %1, %2")
 141                         .arg (rate)
 142                         .arg (lst[ETA]);
 143
 144       setLabelText (s);
 145     }
 146
 147 }
 148
 149 void
 150 DownloadDialog::onCurlFinished (int exitCode, QProcess::ExitStatus exitStatus)
 151 {
 152   if (exitStatus == QProcess::NormalExit && exitCode == 0)
 153     ;
 154   else
 155     {
 156       if (!_canceled)
 157         {
 158           hide ();
 159           NomNom::crit (parentWidget (), _lastError);
 160           emit error ();
 161         }
 162     }
 163
 164   cancel ();
 165 }
 166
 167 void
 168 DownloadDialog::onCanceled ()
 169 {
 170   _canceled = true;
 171
 172   if (_proc.state () == QProcess::Running)
 173     _proc.kill ();
 174 }
 175
 176 // vim: set ts=2 sw=2 tw=72 expandtab: