src/ProcProgDiag.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 #ifdef ENABLE_VERBOSE
  22 #include <QDebug>
  23 #endif
  24
  25 #include "ProcProgDiag.h"
  26
  27 static QHash<QString,QRegExp> rx_labels;
  28
  29 ProcessProgressDialog::ProcessProgressDialog(QWidget *parent/*=NULL*/)
  30   : QProgressDialog(parent), _canceled(false), _errcode(-1)
  31 {
  32   _proc.setProcessChannelMode(QProcess::MergedChannels);
  33
  34 #define _conn(s,sl) \
  35     do { connect(&_proc, SIGNAL(s), this, SLOT(sl)); } while (0)
  36   _conn(started(), onProcStarted());
  37   _conn(error(QProcess::ProcessError), onProcError(QProcess::ProcessError));
  38   _conn(readyRead(), onProcReadyRead());
  39   _conn(finished(int, QProcess::ExitStatus),
  40         onProcFinished(int, QProcess::ExitStatus));
  41 #undef _conn
  42
  43   connect(this, SIGNAL(canceled()), this, SLOT(onCanceled()));
  44
  45   setWindowModality(Qt::WindowModal);
  46   setAutoClose(false);
  47
  48 #define _wrap(s,r) do { rx_labels[s] = QRegExp(r); } while (0)
  49   _wrap(tr("Checking..."),  "^:: Check");
  50   _wrap(tr("Fetching..."),  "^:: Fetch");
  51   _wrap(tr("Verifying..."), "^:: Verify");
  52 #undef _wrap
  53 }
  54
  55 bool ProcessProgressDialog::canceled() const
  56 {
  57   return _canceled;
  58 }
  59
  60 bool ProcessProgressDialog::failed() const
  61 {
  62   return !_errmsg.isEmpty();
  63 }
  64
  65 QString ProcessProgressDialog::errmsg() const
  66 {
  67   return _errmsg;
  68 }
  69
  70 int ProcessProgressDialog::errcode() const
  71 {
  72   return _errcode;
  73 }
  74
  75 void ProcessProgressDialog::start(QStringList& args)
  76 {
  77 #ifdef ENABLE_VERBOSE
  78   qDebug() << __PRETTY_FUNCTION__ << __LINE__ << "args=" << args;
  79 #endif
  80   _canceled = false;
  81
  82   _buffer.clear();
  83   _errmsg.clear();
  84   _args = args;
  85
  86   show();
  87 #ifdef _1
  88   _proc.setProcessEnvironment(NomNom::proc_env());
  89 #endif
  90   _proc.start(args.takeFirst(), args);
  91   exec();
  92 }
  93
  94 void ProcessProgressDialog::onProcStarted()
  95 {
  96 #ifdef ENABLE_VERBOSE
  97   qDebug() << __PRETTY_FUNCTION__ << __LINE__;
  98 #endif
  99 }
 100
 101 void ProcessProgressDialog::onProcError(QProcess::ProcessError n)
 102 {
 103 #ifdef ENABLE_VERBOSE
 104   qDebug() << __PRETTY_FUNCTION__ << __LINE__ << "code=" << n;
 105 #endif
 106   if (!_canceled)
 107     {
 108       hide();
 109       _errcode = static_cast<int>(n);
 110       _errmsg  = _proc.errorString();
 111       emit error();
 112     }
 113   cancel();
 114 }
 115
 116 static void update_label(QProgressDialog *d,
 117                          const QString& ln)
 118 {
 119   QHashIterator<QString, QRegExp> i(rx_labels);
 120   while (i.hasNext())
 121     {
 122       i.next();
 123       if (i.value().indexIn(ln) != -1)
 124         d->setLabelText(i.key());
 125     }
 126 }
 127
 128 void ProcessProgressDialog::onProcReadyRead()
 129 {
 130   static char data[1024];
 131   while (_proc.readLine(data, sizeof(data)))
 132     {
 133       const QString ln = QString::fromLocal8Bit(data);
 134       update_label(this, ln);
 135       _buffer += ln;
 136 #ifdef ENABLE_VERBOSE
 137       qDebug() << __PRETTY_FUNCTION__ << __LINE__ << "ln=" << ln;
 138 #endif
 139     }
 140 }
 141
 142 static QRegExp rx_error("error:\\s+(.*)$");
 143
 144 void ProcessProgressDialog::onProcFinished(int ec, QProcess::ExitStatus es)
 145 {
 146   if (es == QProcess::NormalExit && ec == 0)
 147     emit finished(_buffer);
 148   else
 149     {
 150       if (!_canceled)
 151         {
 152           hide();
 153           _errmsg = _buffer;
 154           if (rx_error.indexIn(_buffer) != -1)
 155             {
 156               _errmsg  = rx_error.cap(1).simplified();
 157               _errcode = static_cast<int>(ec);
 158             }
 159 #ifdef ENABLE_VERBOSE
 160           qDebug() << __PRETTY_FUNCTION__ << __LINE__ << "errmsg=" << _errmsg;
 161 #endif
 162           emit error();
 163         }
 164     }
 165   cancel();
 166 }
 167
 168 void ProcessProgressDialog::onCanceled()
 169 {
 170   _canceled = true;
 171   if (_proc.state() == QProcess::Running)
 172     _proc.kill();
 173 }
 174
 175 // vim: set ts=2 sw=2 tw=72 expandtab: