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