Add a checkParamCount() for EventStringifier as well
[quassel.git] / src / core / eventstringifier.cpp
1 /***************************************************************************
2  *   Copyright (C) 2005-2010 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 "eventstringifier.h"
22
23 #include "coresession.h"
24 #include "messageevent.h"
25
26 EventStringifier::EventStringifier(CoreSession *parent) : QObject(parent),
27   _coreSession(parent),
28   _whois(false)
29 {
30
31 }
32
33 void EventStringifier::displayMsg(NetworkEvent *event, Message::Type msgType, const QString &msg, const QString &sender,
34                                   const QString &target, Message::Flags msgFlags) {
35   if(event->flags().testFlag(EventManager::Silent))
36     return;
37
38   MessageEvent *msgEvent = createMessageEvent(event, msgType, msg, sender, target, msgFlags);
39   sendMessageEvent(msgEvent);
40 }
41
42 MessageEvent *EventStringifier::createMessageEvent(NetworkEvent *event, Message::Type msgType, const QString &msg, const QString &sender,
43                         const QString &target, Message::Flags msgFlags) {
44   MessageEvent *msgEvent = new MessageEvent(msgType, event->network(), msg, sender, target, msgFlags);
45   msgEvent->setTimestamp(event->timestamp());
46   return msgEvent;
47 }
48
49 void EventStringifier::sendMessageEvent(MessageEvent *event) {
50   coreSession()->eventManager()->sendEvent(event);
51 }
52
53 bool EventStringifier::checkParamCount(IrcEvent *e, int minParams) {
54   if(e->params().count() < minParams) {
55     if(e->type() == EventManager::IrcEventNumeric) {
56       qWarning() << "Command " << static_cast<IrcEventNumeric *>(e)->number() << " requires " << minParams << "params, got: " << e->params();
57     } else {
58       QString name = coreSession()->eventManager()->enumName(e->type());
59       qWarning() << qPrintable(name) << "requires" << minParams << "params, got:" << e->params();
60     }
61     e->stop();
62     return false;
63   }
64   return true;
65 }
66
67 void EventStringifier::processIrcEventNumeric(IrcEventNumeric *e) {
68   //qDebug() << e->number();
69   switch(e->number()) {
70   // Welcome, status, info messages. Just display these.
71   case 1: case 2: case 3: case 4: case 5:
72   case 221: case 250: case 251: case 252: case 253: case 254: case 255: case 265: case 266:
73   case 372: case 375:
74     displayMsg(e, Message::Server, e->params().join(" "), e->prefix());
75     break;
76
77   // Server error messages without param, just display them
78   case 409: case 411: case 412: case 422: case 424: case 445: case 446: case 451: case 462:
79   case 463: case 464: case 465: case 466: case 472: case 481: case 483: case 485: case 491: case 501: case 502:
80   case 431: // ERR_NONICKNAMEGIVEN
81     displayMsg(e, Message::Error, e->params().join(" "), e->prefix());
82     break;
83
84   // Server error messages, display them in red. First param will be appended.
85   case 401: {
86     QString target = e->params().takeFirst();
87     displayMsg(e, Message::Error, e->params().join(" ") + " " + target, e->prefix(), target, Message::Redirected);
88     break;
89   }
90
91   case 402: case 403: case 404: case 406: case 408: case 415: case 421: case 442: {
92     QString channelName = e->params().takeFirst();
93     displayMsg(e, Message::Error, e->params().join(" ") + " " + channelName, e->prefix());
94     break;
95   }
96
97   // Server error messages which will be displayed with a colon between the first param and the rest
98   case 413: case 414: case 423: case 441: case 444: case 461:  // FIXME see below for the 47x codes
99   case 467: case 471: case 473: case 474: case 475: case 476: case 477: case 478: case 482:
100   case 436: // ERR_NICKCOLLISION
101   {
102     QString p = e->params().takeFirst();
103     displayMsg(e, Message::Error, p + ": " + e->params().join(" "));
104     break;
105   }
106
107   // Ignore these commands.
108   case 321: case 366: case 376:
109     break;
110
111   // CAP stuff
112   case 903: case 904: case 905: case 906: case 907:
113   {
114     displayMsg(e, Message::Info, "CAP: " + e->params().join(""));
115     break;
116   }
117
118   // Everything else will be marked in red, so we can add them somewhere.
119   default:
120     if(_whois) {
121       // many nets define their own WHOIS fields. we fetch those not in need of special attention here:
122       displayMsg(e, Message::Server, tr("[Whois] ") + e->params().join(" "), e->prefix());
123     } else {
124       // FIXME figure out how/where to do this in the future
125       //if(coreSession()->ircListHelper()->requestInProgress(network()->networkId()))
126       //  coreSession()->ircListHelper()->reportError(params.join(" "));
127       //else
128         displayMsg(e, Message::Error, QString("%1 %2").arg(e->number(), 3, 10, QLatin1Char('0')).arg(e->params().join(" ")), e->prefix());
129     }
130   }
131 }
132
133 void EventStringifier::processIrcEventInvite(IrcEvent *e) {
134   displayMsg(e, Message::Invite, tr("%1 invited you to channel %2").arg(e->nick(), e->params().at(1)));
135 }
136
137 void EventStringifier::earlyProcessIrcEventKick(IrcEvent *e) {
138   IrcUser *victim = e->network()->ircUser(e->params().at(1));
139   if(victim) {
140     QString channel = e->params().at(0);
141     QString msg = victim->nick();
142     if(e->params().count() > 2)
143       msg += " " + e->params().at(2);
144
145     displayMsg(e, Message::Kick, msg, e->prefix(), channel);
146   }
147 }
148
149 // this needs to be called before the ircuser is renamed!
150 void EventStringifier::earlyProcessIrcEventNick(IrcEvent *e) {
151   if(!checkParamCount(e, 1))
152     return;
153
154   IrcUser *ircuser = e->network()->updateNickFromMask(e->prefix());
155   if(!ircuser) {
156     qWarning() << Q_FUNC_INFO << "Unknown IrcUser!";
157     return;
158   }
159
160   QString newnick = e->params().at(0);
161   QString oldnick = ircuser->nick();
162
163   QString sender = e->network()->isMyNick(oldnick) ? newnick : e->prefix();
164   foreach(const QString &channel, ircuser->channels())
165     displayMsg(e, Message::Nick, newnick, sender, channel);
166 }
167
168 void EventStringifier::earlyProcessIrcEventPart(IrcEvent *e) {
169   if(!checkParamCount(e, 1))
170     return;
171
172   QString channel = e->params().at(0);
173   QString msg = e->params().count() > 1? e->params().at(1) : QString();
174
175   displayMsg(e, Message::Part, msg, e->prefix(), channel);
176 }
177
178 void EventStringifier::processIrcEventPong(IrcEvent *e) {
179   QString timestamp = e->params().at(1);
180   QTime sendTime = QTime::fromString(timestamp, "hh:mm:ss.zzz");
181   if(!sendTime.isValid())
182     displayMsg(e, Message::Server, "PONG " + e->params().join(" "), e->prefix());
183 }
184
185 void EventStringifier::processIrcEventTopic(IrcEvent *e) {
186   displayMsg(e, Message::Topic, tr("%1 has changed topic for %2 to: \"%3\"")
187              .arg(e->nick(), e->params().at(0), e->params().at(1)), QString(), e->params().at(0));
188 }
189
190 void EventStringifier::processIrcEvent301(IrcEvent *e) {
191   QString nick = e->params().at(0);
192   QString awayMsg = e->params().at(1);
193   QString msg, target;
194   bool send = true;
195
196   // FIXME: proper redirection needed
197   if(_whois) {
198     msg = tr("[Whois] ");
199   } else {
200     target = nick;
201     IrcUser *ircuser = e->network()->ircUser(nick);
202     if(ircuser) {
203       int now = QDateTime::currentDateTime().toTime_t();
204       const int silenceTime = 60;
205       if(ircuser->lastAwayMessage() + silenceTime >= now)
206         send = false;
207       ircuser->setLastAwayMessage(now);
208     }
209   }
210   if(send)
211     displayMsg(e, Message::Server, msg + tr("%1 is away: \"%2\"").arg(nick, awayMsg), QString(), target);
212 }
213
214 /* RPL_UNAWAY - ":You are no longer marked as being away" */
215 void EventStringifier::processIrcEvent305(IrcEvent *e) {
216   displayMsg(e, Message::Server, tr("You are no longer marked as being away"));
217 }
218
219 /* RPL_NOWAWAY - ":You have been marked as being away" */
220 void EventStringifier::processIrcEvent306(IrcEvent *e) {
221   if(!e->network()->autoAwayActive())
222     displayMsg(e, Message::Server, tr("You have been marked as being away"));
223 }
224
225 /*
226 WHOIS-Message:
227    Replies 311 - 313, 317 - 319 are all replies generated in response to a WHOIS message.
228   and 301 (RPL_AWAY)
229               "<nick> :<away message>"
230 WHO-Message:
231    Replies 352 and 315 paired are used to answer a WHO message.
232
233 WHOWAS-Message:
234    Replies 314 and 369 are responses to a WHOWAS message.
235
236 */
237
238 /*  RPL_WHOISUSER - "<nick> <user> <host> * :<real name>" */
239 void EventStringifier::processIrcEvent311(IrcEvent *e) {
240   _whois = true;
241
242   const QString whoisUserString = tr("[Whois] %1 is %2 (%3)");
243
244   IrcUser *ircuser = e->network()->ircUser(e->params().at(0));
245   if(ircuser)
246     displayMsg(e, Message::Server, whoisUserString.arg(ircuser->nick(), ircuser->hostmask(), ircuser->realName()));
247   else {
248     QString host = QString("%1!%2@%3").arg(e->params().at(0), e->params().at(1), e->params().at(2));
249     displayMsg(e, Message::Server, whoisUserString.arg(e->params().at(0), host, e->params().last()));
250   }
251 }
252
253 /*  RPL_WHOISSERVER -  "<nick> <server> :<server info>" */
254 void EventStringifier::processIrcEvent312(IrcEvent *e) {
255   if(_whois)
256     displayMsg(e, Message::Server, tr("[Whois] %1 is online via %2 (%3)").arg(e->params().at(0), e->params().at(1), e->params().last()));
257   else
258     displayMsg(e, Message::Server, tr("[Whowas] %1 was online via %2 (%3)").arg(e->params().at(0), e->params().at(1), e->params().last()));
259 }
260
261 /*  RPL_WHOWASUSER - "<nick> <user> <host> * :<real name>" */
262 void EventStringifier::processIrcEvent314(IrcEvent *e) {
263   if(!checkParamCount(e, 3))
264     return;
265
266   displayMsg(e, Message::Server, tr("[Whowas] %1 was %2@%3 (%4)").arg(e->params()[0], e->params()[1], e->params()[2], e->params().last()));
267 }
268
269 /*  RPL_ENDOFWHO: "<name> :End of WHO list" */
270 void EventStringifier::processIrcEvent315(IrcEvent *e) {
271   QStringList p = e->params();
272   p.takeLast(); // should be "End of WHO list"
273   displayMsg(e, Message::Server, tr("[Who] End of /WHO list for %1").arg(p.join(" ")));
274 }
275
276 /*  RPL_WHOISIDLE - "<nick> <integer> :seconds idle"
277    (real life: "<nick> <integer> <integer> :seconds idle, signon time) */
278 void EventStringifier::processIrcEvent317(IrcEvent *e) {
279   int idleSecs = e->params()[1].toInt();
280
281   if(e->params().count() > 3) { // if we have more then 3 params we have the above mentioned "real life" situation
282     QDateTime loginTime = QDateTime::fromTime_t(e->params()[2].toInt());
283     displayMsg(e, Message::Server, tr("[Whois] %1 is logged in since %2").arg(e->params()[0], loginTime.toString()));
284   }
285   displayMsg(e, Message::Server, tr("[Whois] %1 is idling for %2 (since %3)")
286              .arg(e->params()[0], secondsToString(idleSecs), e->timestamp().toLocalTime().addSecs(-idleSecs).toString()));
287 }
288
289 /*  RPL_ENDOFWHOIS - "<nick> :End of WHOIS list" */
290 void EventStringifier::processIrcEvent318(IrcEvent *e) {
291   _whois = false;
292   displayMsg(e, Message::Server, tr("[Whois] End of /WHOIS list"));
293 }
294
295 /*  RPL_WHOISCHANNELS - "<nick> :*( ( "@" / "+" ) <channel> " " )" */
296 void EventStringifier::processIrcEvent319(IrcEvent *e) {
297   if(!checkParamCount(e, 2))
298     return;
299
300   QString nick = e->params().first();
301   QStringList op;
302   QStringList voice;
303   QStringList user;
304   foreach(QString channel, e->params().last().split(" ")) {
305     if(channel.startsWith("@"))
306        op.append(channel.remove(0,1));
307     else if(channel.startsWith("+"))
308       voice.append(channel.remove(0,1));
309     else
310       user.append(channel);
311   }
312   if(!user.isEmpty())
313     displayMsg(e, Message::Server, tr("[Whois] %1 is a user on channels: %2").arg(nick, user.join(" ")));
314   if(!voice.isEmpty())
315     displayMsg(e, Message::Server, tr("[Whois] %1 has voice on channels: %2").arg(nick, voice.join(" ")));
316   if(!op.isEmpty())
317     displayMsg(e, Message::Server, tr("[Whois] %1 is an operator on channels: %2").arg(nick, op.join(" ")));
318 }
319
320 /*  RPL_WHOISACCOUNT: "<nick> <account> :is authed as */
321 void EventStringifier::processIrcEvent330(IrcEvent *e) {
322   if(e->params().count() < 3)
323     return;
324
325   displayMsg(e, Message::Server, tr("[Whois] %1 is authed as %2").arg(e->params()[0], e->params()[1]));
326 }
327
328 // template
329 /*
330
331 void EventStringifier::processIrcEvent(IrcEvent *e) {
332
333 }
334
335 */