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