1 /***************************************************************************
2 * Copyright (C) 2005-07 by The Quassel Team *
3 * devel@quassel-irc.org *
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) any later version. *
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. *
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 ***************************************************************************/
22 #include "clientproxy.h"
24 #include "buffertreemodel.h"
27 Client * Client::instanceptr = 0;
29 Client::ClientMode Client::clientMode;
30 QHash<BufferId, Buffer *> Client::buffers;
31 QHash<uint, BufferId> Client::bufferIds;
32 QHash<QString, QHash<QString, VarMap> > Client::nicks;
33 QHash<QString, bool> Client::connected;
34 QHash<QString, QString> Client::ownNick;
35 QList<BufferId> Client::coreBuffers;
38 Client *Client::instance() {
39 if(instanceptr) return instanceptr;
40 instanceptr = new Client();
44 void Client::destroy() {
50 clientProxy = ClientProxy::instance();
52 _bufferModel = new BufferTreeModel(this);
54 connect(this, SIGNAL(bufferSelected(Buffer *)), _bufferModel, SLOT(selectBuffer(Buffer *)));
55 connect(this, SIGNAL(bufferUpdated(Buffer *)), _bufferModel, SLOT(bufferUpdated(Buffer *)));
56 connect(this, SIGNAL(bufferActivity(Buffer::ActivityLevel, Buffer *)), _bufferModel, SLOT(bufferActivity(Buffer::ActivityLevel, Buffer *)));
58 // TODO: make this configurable (allow monolithic client to connect to remote cores)
59 if(Global::runMode == Global::Monolithic) clientMode = LocalCore;
60 else clientMode = RemoteCore;
63 void Client::init(AbstractUi *ui) {
64 instance()->mainUi = ui;
71 connect(&socket, SIGNAL(readyRead()), this, SLOT(serverHasData()));
72 connect(&socket, SIGNAL(connected()), this, SLOT(coreConnected()));
73 connect(&socket, SIGNAL(disconnected()), this, SLOT(coreDisconnected()));
74 connect(&socket, SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(serverError(QAbstractSocket::SocketError)));
76 connect(Global::instance(), SIGNAL(dataPutLocally(UserId, QString)), this, SLOT(updateCoreData(UserId, QString)));
77 connect(clientProxy, SIGNAL(csUpdateGlobalData(QString, QVariant)), this, SLOT(updateLocalData(QString, QVariant)));
79 connect(clientProxy, SIGNAL(send(ClientSignal, QVariant, QVariant, QVariant)), this, SLOT(recvProxySignal(ClientSignal, QVariant, QVariant, QVariant)));
80 connect(clientProxy, SIGNAL(csServerState(QString, QVariant)), this, SLOT(recvNetworkState(QString, QVariant)));
81 connect(clientProxy, SIGNAL(csServerConnected(QString)), this, SLOT(networkConnected(QString)));
82 connect(clientProxy, SIGNAL(csServerDisconnected(QString)), this, SLOT(networkDisconnected(QString)));
83 connect(clientProxy, SIGNAL(csDisplayMsg(Message)), this, SLOT(recvMessage(const Message &)));
84 connect(clientProxy, SIGNAL(csDisplayStatusMsg(QString, QString)), this, SLOT(recvStatusMsg(QString, QString)));
85 connect(clientProxy, SIGNAL(csTopicSet(QString, QString, QString)), this, SLOT(setTopic(QString, QString, QString)));
86 connect(clientProxy, SIGNAL(csNickAdded(QString, QString, VarMap)), this, SLOT(addNick(QString, QString, VarMap)));
87 connect(clientProxy, SIGNAL(csNickRemoved(QString, QString)), this, SLOT(removeNick(QString, QString)));
88 connect(clientProxy, SIGNAL(csNickRenamed(QString, QString, QString)), this, SLOT(renameNick(QString, QString, QString)));
89 connect(clientProxy, SIGNAL(csNickUpdated(QString, QString, VarMap)), this, SLOT(updateNick(QString, QString, VarMap)));
90 connect(clientProxy, SIGNAL(csOwnNickSet(QString, QString)), this, SLOT(setOwnNick(QString, QString)));
91 connect(clientProxy, SIGNAL(csBacklogData(BufferId, const QList<QVariant> &, bool)), this, SLOT(recvBacklogData(BufferId, QList<QVariant>, bool)));
92 connect(clientProxy, SIGNAL(csUpdateBufferId(BufferId)), this, SLOT(updateBufferId(BufferId)));
93 connect(this, SIGNAL(sendInput(BufferId, QString)), clientProxy, SLOT(gsUserInput(BufferId, QString)));
94 connect(this, SIGNAL(requestBacklog(BufferId, QVariant, QVariant)), clientProxy, SLOT(gsRequestBacklog(BufferId, QVariant, QVariant)));
98 layoutTimer = new QTimer(this);
99 layoutTimer->setInterval(0);
100 layoutTimer->setSingleShot(false);
101 connect(layoutTimer, SIGNAL(timeout()), this, SLOT(layoutMsg()));
103 /* make lookups by id faster */
104 foreach(BufferId id, coreBuffers) {
105 bufferIds[id.uid()] = id; // make lookups by id faster
106 buffer(id); // create all buffers, so we see them in the network views
107 emit requestBacklog(id, -1, -1); // TODO: use custom settings for backlog request
113 //delete _bufferModel;
114 foreach(Buffer *buf, buffers.values()) delete buf;
115 ClientProxy::destroy();
119 BufferTreeModel *Client::bufferModel() {
120 return instance()->_bufferModel;
123 void Client::coreConnected() {
127 void Client::coreDisconnected() {
131 void Client::updateCoreData(UserId, QString key) {
132 if(clientMode == LocalCore) return;
133 QVariant data = Global::data(key);
134 recvProxySignal(GS_UPDATE_GLOBAL_DATA, key, data, QVariant());
137 void Client::updateLocalData(QString key, QVariant data) {
138 Global::updateData(key, data);
141 void Client::recvProxySignal(ClientSignal sig, QVariant arg1, QVariant arg2, QVariant arg3) {
142 if(clientMode == LocalCore) return;
143 QList<QVariant> sigdata;
144 sigdata.append(sig); sigdata.append(arg1); sigdata.append(arg2); sigdata.append(arg3);
145 //qDebug() << "Sending signal: " << sigdata;
146 writeDataToDevice(&socket, QVariant(sigdata));
149 void Client::connectToCore(QString host, quint16 port) {
150 // TODO implement SSL
151 socket.connectToHost(host, port);
154 void Client::disconnectFromCore() {
158 void Client::serverError(QAbstractSocket::SocketError) {
159 emit coreConnectionError(socket.errorString());
162 void Client::serverHasData() {
164 while(readDataFromDevice(&socket, blockSize, item)) {
165 emit recvPartialItem(1,1);
166 QList<QVariant> sigdata = item.toList();
167 Q_ASSERT(sigdata.size() == 4);
168 ClientProxy::instance()->recv((CoreSignal)sigdata[0].toInt(), sigdata[1], sigdata[2], sigdata[3]);
172 emit recvPartialItem(socket.bytesAvailable(), blockSize);
176 void Client::networkConnected(QString net) {
177 connected[net] = true;
178 BufferId id = statusBufferId(net);
179 Buffer *b = buffer(id);
181 //b->displayMsg(Message(id, Message::Server, tr("Connected.")));
182 // TODO buffersUpdated();
185 void Client::networkDisconnected(QString net) {
186 foreach(BufferId id, buffers.keys()) {
187 if(id.network() != net) continue;
188 Buffer *b = buffer(id);
189 //b->displayMsg(Message(id, Message::Server, tr("Server disconnected."))); FIXME
192 connected[net] = false;
195 void Client::updateBufferId(BufferId id) {
196 bufferIds[id.uid()] = id; // make lookups by id faster
200 BufferId Client::bufferId(QString net, QString buf) {
201 foreach(BufferId id, buffers.keys()) {
202 if(id.network() == net && id.buffer() == buf) return id;
204 Q_ASSERT(false); // should never happen!
208 BufferId Client::statusBufferId(QString net) {
209 return bufferId(net, "");
213 Buffer * Client::buffer(BufferId id) {
214 Client *client = Client::instance();
215 if(!buffers.contains(id)) {
216 Buffer *b = new Buffer(id);
217 b->setOwnNick(ownNick[id.network()]);
218 connect(b, SIGNAL(userInput(BufferId, QString)), client, SLOT(userInput(BufferId, QString)));
219 connect(b, SIGNAL(bufferUpdated(Buffer *)), client, SIGNAL(bufferUpdated(Buffer *)));
220 connect(b, SIGNAL(bufferDestroyed(Buffer *)), client, SIGNAL(bufferDestroyed(Buffer *)));
222 emit client->bufferUpdated(b);
227 void Client::recvNetworkState(QString net, QVariant state) {
228 connected[net] = true;
229 setOwnNick(net, state.toMap()["OwnNick"].toString());
230 buffer(statusBufferId(net))->setActive(true);
231 VarMap t = state.toMap()["Topics"].toMap();
232 VarMap n = state.toMap()["Nicks"].toMap();
233 foreach(QVariant v, t.keys()) {
234 QString buf = v.toString();
235 BufferId id = bufferId(net, buf);
236 buffer(id)->setActive(true);
237 setTopic(net, buf, t[buf].toString());
239 foreach(QString nick, n.keys()) {
240 addNick(net, nick, n[nick].toMap());
244 void Client::recvMessage(const Message &msg) {
245 Buffer *b = buffer(msg.buffer);
247 Buffer::ActivityLevel level = Buffer::OtherActivity;
248 if(msg.type == Message::Plain || msg.type == Message::Notice){
249 level |= Buffer::NewMessage;
251 if(msg.flags & Message::Highlight){
252 level |= Buffer::Highlight;
254 emit bufferActivity(level, b);
259 void Client::recvStatusMsg(QString /*net*/, QString /*msg*/) {
260 //recvMessage(net, Message::server("", QString("[STATUS] %1").arg(msg)));
264 void Client::recvBacklogData(BufferId id, const QList<QVariant> &msgs, bool /*done*/) {
265 Buffer *b = buffer(id);
266 foreach(QVariant v, msgs) {
267 Message msg = v.value<Message>();
269 if(!layoutQueue.contains(b)) layoutQueue.append(b);
271 if(layoutQueue.count() && !layoutTimer->isActive()) layoutTimer->start();
274 void Client::layoutMsg() {
275 if(layoutQueue.count()) {
276 Buffer *b = layoutQueue.takeFirst(); // TODO make this the current buffer
277 if(b->layoutMsg()) layoutQueue.append(b); // Buffer has more messages in its queue --> Round Robin
279 if(!layoutQueue.count()) layoutTimer->stop();
282 AbstractUiMsg *Client::layoutMsg(const Message &msg) {
283 return instance()->mainUi->layoutMsg(msg);
286 void Client::userInput(BufferId id, QString msg) {
287 emit sendInput(id, msg);
290 void Client::setTopic(QString net, QString buf, QString topic) {
291 BufferId id = bufferId(net, buf);
292 if(!connected[id.network()]) return;
293 Buffer *b = buffer(id);
295 //if(!b->isActive()) {
296 // b->setActive(true);
301 void Client::addNick(QString net, QString nick, VarMap props) {
302 if(!connected[net]) return;
303 nicks[net][nick] = props;
304 VarMap chans = props["Channels"].toMap();
305 QStringList c = chans.keys();
306 foreach(QString bufname, c) {
307 buffer(bufferId(net, bufname))->addNick(nick, props);
311 void Client::renameNick(QString net, QString oldnick, QString newnick) {
312 if(!connected[net]) return;
313 QStringList chans = nicks[net][oldnick]["Channels"].toMap().keys();
314 foreach(QString c, chans) {
315 buffer(bufferId(net, c))->renameNick(oldnick, newnick);
317 nicks[net][newnick] = nicks[net].take(oldnick);
320 void Client::updateNick(QString net, QString nick, VarMap props) {
321 if(!connected[net]) return;
322 QStringList oldchans = nicks[net][nick]["Channels"].toMap().keys();
323 QStringList newchans = props["Channels"].toMap().keys();
324 foreach(QString c, newchans) {
325 if(oldchans.contains(c)) buffer(bufferId(net, c))->updateNick(nick, props);
326 else buffer(bufferId(net, c))->addNick(nick, props);
328 foreach(QString c, oldchans) {
329 if(!newchans.contains(c)) buffer(bufferId(net, c))->removeNick(nick);
331 nicks[net][nick] = props;
334 void Client::removeNick(QString net, QString nick) {
335 if(!connected[net]) return;
336 VarMap chans = nicks[net][nick]["Channels"].toMap();
337 foreach(QString bufname, chans.keys()) {
338 buffer(bufferId(net, bufname))->removeNick(nick);
340 nicks[net].remove(nick);
343 void Client::setOwnNick(QString net, QString nick) {
344 if(!connected[net]) return;
346 foreach(BufferId id, buffers.keys()) {
347 if(id.network() == net) {
348 buffers[id]->setOwnNick(nick);