src/DownloadDiag.cpp

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