e09457ad6948a904d577a1ed5d152bca7b2fa29d
[quassel.git] / src / common / message.cpp
1 /***************************************************************************
2  *   Copyright (C) 2005-08 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 "message.h"
22
23 #include "util.h"
24
25 #include <QDataStream>
26
27 Message::Message(BufferInfo bufferInfo, Type type, QString text, QString sender, quint8 flags)
28   : _timestamp(QDateTime::currentDateTime().toUTC()),
29     _bufferInfo(bufferInfo),
30     _text(text),
31     _sender(sender),
32     _type(type),
33     _flags(flags)
34 {
35 }
36
37 Message::Message(QDateTime ts,BufferInfo bufferInfo, Type type, QString text, QString sender, quint8 flags)
38   : _timestamp(ts),
39     _bufferInfo(bufferInfo),
40     _text(text),
41     _sender(sender),
42     _type(type),
43     _flags(flags)
44 {
45 }
46
47 QString Message::mircToInternal(QString mirc) {
48   mirc.replace('%', "%%");      // escape % just to be sure
49   mirc.replace('\x02', "%B");
50   mirc.replace('\x0f', "%O");
51   mirc.replace('\x12', "%R");
52   mirc.replace('\x16', "%R");
53   mirc.replace('\x1d', "%S");
54   mirc.replace('\x1f', "%U");
55
56   // Now we bring the color codes (\x03) in a sane format that can be parsed more easily later.
57   // %Dcfxx is foreground, %Dcbxx is background color, where xx is a 2 digit dec number denoting the color code.
58   // %Dc- turns color off.
59   // Note: We use the "mirc standard" as described in <http://www.mirc.co.uk/help/color.txt>.
60   //       This means that we don't accept something like \x03,5 (even though others, like WeeChat, do).
61   int pos = 0;
62   for(;;) {
63     pos = mirc.indexOf('\x03', pos);
64     if(pos < 0) break; // no more mirc color codes
65     QString ins, num;
66     int l = mirc.length();
67     int i = pos + 1;
68     // check for fg color
69     if(i < l && mirc[i].isDigit()) {
70       num = mirc[i++];
71       if(i < l && mirc[i].isDigit()) num.append(mirc[i++]);
72       else num.prepend('0');
73       ins = QString("%Dcf%1").arg(num);
74
75       if(i+1 < l && mirc[i] == ',' && mirc[i+1].isDigit()) {
76         i++;
77         num = mirc[i++];
78         if(i < l && mirc[i].isDigit()) num.append(mirc[i++]);
79         else num.prepend('0');
80         ins += QString("%Dcb%1").arg(num);
81       }
82     } else {
83       ins = "%Dc-";
84     }
85     mirc.replace(pos, i-pos, ins);
86   }
87   return mirc;
88 }
89
90 void Message::format() {
91   if(!_formattedText.isNull())
92     return; // already done
93   
94   QString user = userFromMask(sender());
95   QString host = hostFromMask(sender());
96   QString nick = nickFromMask(sender());
97   QString txt = mircToInternal(text());
98   QString bufferName = bufferInfo().bufferName();
99
100   _formattedTimestamp = tr("%DT[%1]").arg(timestamp().toLocalTime().toString("hh:mm:ss"));
101
102   QString s, t;
103   switch(type()) {
104     case Message::Plain:
105       s = tr("%DS<%1>").arg(nick); t = tr("%D0%1").arg(txt); break;
106     case Message::Server:
107       s = tr("%Ds*"); t = tr("%Ds%1").arg(txt); break;
108     case Message::Error:
109       s = tr("%De*"); t = tr("%De%1").arg(txt); break;
110     case Message::Join:
111       s = tr("%Dj-->"); t = tr("%Dj%DN%DU%1%DU%DN %DH(%2@%3)%DH has joined %DC%DU%4%DU%DC").arg(nick, user, host, bufferName); break;
112     case Message::Part:
113       s = tr("%Dp<--"); t = tr("%Dp%DN%DU%1%DU%DN %DH(%2@%3)%DH has left %DC%DU%4%DU%DC").arg(nick, user, host, bufferName);
114       if(!txt.isEmpty()) t = QString("%1 (%2)").arg(t).arg(txt);
115       break;
116     case Message::Quit:
117       s = tr("%Dq<--"); t = tr("%Dq%DN%DU%1%DU%DN %DH(%2@%3)%DH has quit").arg(nick, user, host);
118       if(!txt.isEmpty()) t = QString("%1 (%2)").arg(t).arg(txt);
119       break;
120     case Message::Kick:
121     { s = tr("%Dk<-*");
122     QString victim = txt.section(" ", 0, 0);
123         //if(victim == ui.ownNick->currentText()) victim = tr("you");
124     QString kickmsg = txt.section(" ", 1);
125     t = tr("%Dk%DN%DU%1%DU%DN has kicked %DN%DU%2%DU%DN from %DC%DU%3%DU%DC").arg(nick).arg(victim).arg(bufferName);
126     if(!kickmsg.isEmpty()) t = QString("%1 (%2)").arg(t).arg(kickmsg);
127     }
128     break;
129     case Message::Nick:
130       s = tr("%Dr<->");
131       if(nick == text()) t = tr("%DrYou are now known as %DN%1%DN").arg(txt);
132       else t = tr("%Dr%DN%1%DN is now known as %DN%DU%2%DU%DN").arg(nick, txt);
133       break;
134     case Message::Mode:
135       s = tr("%Dm***");
136       if(nick.isEmpty()) t = tr("%DmUser mode: %DM%1%DM").arg(text());
137       else t = tr("%DmMode %DM%1%DM by %DN%DU%2%DU%DN").arg(txt, nick);
138       break;
139     case Message::Action:
140       s = tr("%Da-*-");
141       t = tr("%Da%DN%DU%1%DU%DN %2").arg(nick).arg(txt);
142       break;
143     default:
144       s = tr("%De%1").arg(sender());
145       t = tr("%De[%1]").arg(txt);
146   }
147   _formattedSender = s;
148   _formattedText = t;
149 }
150
151 QString Message::formattedTimestamp() {
152   format();
153   return _formattedTimestamp;
154 }
155
156 QString Message::formattedSender() {
157   format();
158   return _formattedSender;
159 }
160
161 QString Message::formattedText() {
162   format();
163   return _formattedText;
164 }
165
166 /*
167 QString Message::formattedToHtml(const QString &f) {
168
169
170   return f;
171 }
172 */
173
174 QDataStream &operator<<(QDataStream &out, const Message &msg) {
175   out << (quint32)msg.timestamp().toTime_t() << (quint8)msg.type() << (quint8)msg.flags()
176       << msg.bufferInfo() << msg.sender().toUtf8() << msg.text().toUtf8();
177   return out;
178 }
179
180 QDataStream &operator>>(QDataStream &in, Message &msg) {
181   quint8 t, f;
182   quint32 ts;
183   QByteArray s, m;
184   BufferInfo buf;
185   in >> ts >> t >> f >> buf >> s >> m;
186   msg._type = (Message::Type)t;
187   msg._flags = (quint8)f;
188   msg._bufferInfo = buf;
189   msg._timestamp = QDateTime::fromTime_t(ts);
190   msg._sender = QString::fromUtf8(s);
191   msg._text = QString::fromUtf8(m);
192   return in;
193 }
194