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