ExecWrapper tweaks
[quassel.git] / src / client / execwrapper.cpp
1 /***************************************************************************
2 *   Copyright (C) 2005-09 by the Quassel Project                          *
3 *   devel@quassel-irc.org                                                 *
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 2 of the License, or     *
8 *   (at your option) version 3.                                           *
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, write to the                         *
17 *   Free Software Foundation, Inc.,                                       *
18 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
19 ***************************************************************************/
20
21 #include <QFile>
22
23 #include "execwrapper.h"
24
25 #include "client.h"
26 #include "messagemodel.h"
27 #include "quassel.h"
28
29 ExecWrapper::ExecWrapper(QObject* parent) : QObject(parent) {
30   connect(&_process, SIGNAL(readyReadStandardOutput()), SLOT(processReadStdout()));
31   connect(&_process, SIGNAL(readyReadStandardError()), SLOT(processReadStderr()));
32   connect(&_process, SIGNAL(finished(int, QProcess::ExitStatus)), SLOT(processFinished(int, QProcess::ExitStatus)));
33   connect(&_process, SIGNAL(error(QProcess::ProcessError)), SLOT(processError(QProcess::ProcessError)));
34
35   connect(this, SIGNAL(stdout(QString)), SLOT(postStdout(QString)));
36   connect(this, SIGNAL(stderr(QString)), SLOT(postStderr(QString)));
37 }
38
39 void ExecWrapper::start(const BufferInfo &info, const QString &command) {
40   _bufferInfo = info;
41   QString params;
42
43   QRegExp rx("^\\s*(\\S+)(\\s+(.*))?$");
44   if(!rx.exactMatch(command)) {
45     emit stderr(tr("Invalid command string for /exec: %1").arg(command));
46   } else {
47     _scriptName = rx.cap(1);
48     params = rx.cap(3);
49   }
50
51   // Make sure we don't execute something outside a script dir
52   if(_scriptName.startsWith('/') || _scriptName.contains("../"))
53     emit stderr(tr("Name \"%1\" is invalid: / or ../ are not allowed!").arg(_scriptName));
54
55   else {
56     foreach(QString scriptDir, Quassel::scriptDirPaths()) {
57       QString fileName = scriptDir + _scriptName;
58       if(!QFile::exists(fileName))
59         continue;
60       _process.start(fileName + ' ' + params);
61       return;
62     }
63     emit stderr(tr("Could not find script \"%1\"").arg(_scriptName));
64   }
65
66   deleteLater(); // self-destruct
67 }
68
69 void ExecWrapper::postStdout(const QString &msg) {
70   if(_bufferInfo.isValid())
71     Client::userInput(_bufferInfo, msg);
72 }
73
74 void ExecWrapper::postStderr(const QString &msg) {
75   if(_bufferInfo.isValid())
76     Client::messageModel()->insertErrorMessage(_bufferInfo, msg);
77 }
78
79 void ExecWrapper::processFinished(int exitCode, QProcess::ExitStatus status) {
80   if(status == QProcess::CrashExit) {
81     emit stderr(tr("Script \"%1\" crashed with exit code %2.").arg(_scriptName).arg(exitCode));
82   }
83
84   // empty buffers
85   if(!_stdoutBuffer.isEmpty())
86     foreach(QString msg, _stdoutBuffer.split('\n'))
87       emit stdout(msg);
88   if(!_stderrBuffer.isEmpty())
89     foreach(QString msg, _stderrBuffer.split('\n'))
90     emit stderr(msg);
91
92   deleteLater();
93 }
94
95 void ExecWrapper::processError(QProcess::ProcessError error) {
96   emit stderr(tr("Script \"%1\" caused error %2.").arg(_scriptName).arg(error));
97   if(_process.state() != QProcess::Running)
98     deleteLater();
99 }
100
101 void ExecWrapper::processReadStdout() {
102   _stdoutBuffer.append(_process.readAllStandardOutput());
103   int idx;
104   while((idx = _stdoutBuffer.indexOf('\n')) >= 0) {
105     emit stdout(_stdoutBuffer.left(idx));
106     _stdoutBuffer = _stdoutBuffer.mid(idx + 1);
107   }
108 }
109
110 void ExecWrapper::processReadStderr() {
111   _stderrBuffer.append(_process.readAllStandardError());
112   int idx;
113   while((idx = _stderrBuffer.indexOf('\n')) >= 0) {
114     emit stderr(_stderrBuffer.left(idx));
115     _stderrBuffer = _stderrBuffer.mid(idx + 1);
116   }
117 }