1 /***************************************************************************
2 * Copyright (C) 2005-08 by the Quassel Project *
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) version 3. *
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 ***************************************************************************/
23 #include "bufferinfo.h"
24 #include "buffermodel.h"
25 #include "buffersettings.h"
26 #include "buffersyncer.h"
27 #include "bufferviewmanager.h"
28 #include "clientbacklogmanager.h"
29 #include "clientirclisthelper.h"
32 #include "ircchannel.h"
35 #include "messagemodel.h"
37 #include "networkmodel.h"
38 #include "quasselui.h"
39 #include "signalproxy.h"
42 QPointer<Client> Client::instanceptr = 0;
43 AccountId Client::_currentCoreAccount = 0;
45 /*** Initialization/destruction ***/
47 Client *Client::instance() {
49 instanceptr = new Client();
53 void Client::destroy() {
55 instanceptr->deleteLater();
58 void Client::init(AbstractUi *ui) {
59 instance()->mainUi = ui;
63 Client::Client(QObject *parent)
66 _signalProxy(new SignalProxy(SignalProxy::Client, this)),
71 _backlogManager(new ClientBacklogManager(this)),
72 _bufferViewManager(0),
73 _ircListHelper(new ClientIrcListHelper(this)),
75 _connectedToCore(false),
78 _monitorBuffer = new Buffer(BufferInfo(), this);
79 _signalProxy->synchronize(_ircListHelper);
81 connect(_backlogManager, SIGNAL(backlog(BufferId, const QVariantList &)),
82 this, SLOT(receiveBacklog(BufferId, const QVariantList &)));
90 _currentCoreAccount = 0;
91 _networkModel = new NetworkModel(this);
93 connect(this, SIGNAL(bufferUpdated(BufferInfo)),
94 _networkModel, SLOT(bufferUpdated(BufferInfo)));
95 connect(this, SIGNAL(networkRemoved(NetworkId)),
96 _networkModel, SLOT(networkRemoved(NetworkId)));
98 _bufferModel = new BufferModel(_networkModel);
99 _messageModel = mainUi->createMessageModel(this);
101 SignalProxy *p = signalProxy();
103 p->attachSlot(SIGNAL(displayMsg(const Message &)), this, SLOT(recvMessage(const Message &)));
104 p->attachSlot(SIGNAL(displayStatusMsg(QString, QString)), this, SLOT(recvStatusMsg(QString, QString)));
106 p->attachSlot(SIGNAL(bufferInfoUpdated(BufferInfo)), this, SLOT(updateBufferInfo(BufferInfo)));
107 p->attachSignal(this, SIGNAL(sendInput(BufferInfo, QString)));
108 p->attachSignal(this, SIGNAL(requestNetworkStates()));
110 p->attachSignal(this, SIGNAL(requestCreateIdentity(const Identity &)), SIGNAL(createIdentity(const Identity &)));
111 p->attachSignal(this, SIGNAL(requestUpdateIdentity(const Identity &)), SIGNAL(updateIdentity(const Identity &)));
112 p->attachSignal(this, SIGNAL(requestRemoveIdentity(IdentityId)), SIGNAL(removeIdentity(IdentityId)));
113 p->attachSlot(SIGNAL(identityCreated(const Identity &)), this, SLOT(coreIdentityCreated(const Identity &)));
114 p->attachSlot(SIGNAL(identityRemoved(IdentityId)), this, SLOT(coreIdentityRemoved(IdentityId)));
116 p->attachSignal(this, SIGNAL(requestCreateNetwork(const NetworkInfo &)), SIGNAL(createNetwork(const NetworkInfo &)));
117 p->attachSignal(this, SIGNAL(requestUpdateNetwork(const NetworkInfo &)), SIGNAL(updateNetwork(const NetworkInfo &)));
118 p->attachSignal(this, SIGNAL(requestRemoveNetwork(NetworkId)), SIGNAL(removeNetwork(NetworkId)));
119 p->attachSlot(SIGNAL(networkCreated(NetworkId)), this, SLOT(coreNetworkCreated(NetworkId)));
120 p->attachSlot(SIGNAL(networkRemoved(NetworkId)), this, SLOT(coreNetworkRemoved(NetworkId)));
122 connect(p, SIGNAL(disconnected()), this, SLOT(disconnectFromCore()));
124 //connect(mainUi, SIGNAL(connectToCore(const QVariantMap &)), this, SLOT(connectToCore(const QVariantMap &)));
125 connect(mainUi, SIGNAL(disconnectFromCore()), this, SLOT(disconnectFromCore()));
126 connect(this, SIGNAL(connected()), mainUi, SLOT(connectedToCore()));
127 connect(this, SIGNAL(disconnected()), mainUi, SLOT(disconnectedFromCore()));
129 layoutTimer = new QTimer(this);
130 layoutTimer->setInterval(0);
131 layoutTimer->setSingleShot(false);
132 connect(layoutTimer, SIGNAL(timeout()), this, SLOT(layoutMsg()));
136 /*** public static methods ***/
138 AccountId Client::currentCoreAccount() {
139 return _currentCoreAccount;
142 void Client::setCurrentCoreAccount(AccountId id) {
143 _currentCoreAccount = id;
146 QList<BufferInfo> Client::allBufferInfos() {
147 QList<BufferInfo> bufferids;
148 foreach(Buffer *buffer, buffers()) {
149 bufferids << buffer->bufferInfo();
154 QList<Buffer *> Client::buffers() {
155 return instance()->_buffers.values();
159 Buffer *Client::statusBuffer(const NetworkId &networkId) const {
160 if(_statusBuffers.contains(networkId))
161 return _statusBuffers[networkId];
166 Buffer *Client::buffer(BufferId bufferId) {
167 if(instance()->_buffers.contains(bufferId))
168 return instance()->_buffers[bufferId];
173 Buffer *Client::buffer(BufferInfo bufferInfo) {
174 Buffer *buff = buffer(bufferInfo.bufferId());
177 Client *client = Client::instance();
178 buff = new Buffer(bufferInfo, client);
179 connect(buff, SIGNAL(destroyed()), client, SLOT(bufferDestroyed()));
180 client->_buffers[bufferInfo.bufferId()] = buff;
181 if(bufferInfo.type() == BufferInfo::StatusBuffer)
182 client->_statusBuffers[bufferInfo.networkId()] = buff;
184 emit client->bufferUpdated(bufferInfo);
186 // I don't like this: but currently there isn't really a prettier way:
187 if(isSynced()) { // this slows down syncing a lot, so disable it during sync
188 QModelIndex bufferIdx = networkModel()->bufferIndex(bufferInfo.bufferId());
189 bufferModel()->setCurrentIndex(bufferModel()->mapFromSource(bufferIdx));
196 bool Client::isConnected() {
197 return instance()->_connectedToCore;
200 bool Client::isSynced() {
201 return instance()->_syncedToCore;
204 /*** Network handling ***/
206 QList<NetworkId> Client::networkIds() {
207 return instance()->_networks.keys();
210 const Network * Client::network(NetworkId networkid) {
211 if(instance()->_networks.contains(networkid)) return instance()->_networks[networkid];
215 void Client::createNetwork(const NetworkInfo &info) {
216 emit instance()->requestCreateNetwork(info);
219 void Client::updateNetwork(const NetworkInfo &info) {
220 emit instance()->requestUpdateNetwork(info);
223 void Client::removeNetwork(NetworkId id) {
224 emit instance()->requestRemoveNetwork(id);
227 void Client::addNetwork(Network *net) {
228 net->setProxy(signalProxy());
229 signalProxy()->synchronize(net);
230 networkModel()->attachNetwork(net);
231 connect(net, SIGNAL(destroyed()), instance(), SLOT(networkDestroyed()));
232 instance()->_networks[net->networkId()] = net;
233 emit instance()->networkCreated(net->networkId());
236 void Client::coreNetworkCreated(NetworkId id) {
237 if(_networks.contains(id)) {
238 qWarning() << "Creation of already existing network requested!";
241 Network *net = new Network(id, this);
245 void Client::coreNetworkRemoved(NetworkId id) {
246 if(!_networks.contains(id))
248 Network *net = _networks.take(id);
249 emit networkRemoved(net->networkId());
253 /*** Identity handling ***/
255 QList<IdentityId> Client::identityIds() {
256 return instance()->_identities.keys();
259 const Identity * Client::identity(IdentityId id) {
260 if(instance()->_identities.contains(id)) return instance()->_identities[id];
264 void Client::createIdentity(const Identity &id) {
265 emit instance()->requestCreateIdentity(id);
268 void Client::updateIdentity(const Identity &id) {
269 emit instance()->requestUpdateIdentity(id);
272 void Client::removeIdentity(IdentityId id) {
273 emit instance()->requestRemoveIdentity(id);
276 void Client::coreIdentityCreated(const Identity &other) {
277 if(!_identities.contains(other.id())) {
278 Identity *identity = new Identity(other, this);
279 _identities[other.id()] = identity;
280 identity->setInitialized();
281 signalProxy()->synchronize(identity);
282 emit identityCreated(other.id());
284 qWarning() << tr("Identity already exists in client!");
288 void Client::coreIdentityRemoved(IdentityId id) {
289 if(_identities.contains(id)) {
290 emit identityRemoved(id);
291 Identity *i = _identities.take(id);
297 void Client::userInput(BufferInfo bufferInfo, QString message) {
298 emit instance()->sendInput(bufferInfo, message);
301 /*** core connection stuff ***/
303 void Client::setConnectedToCore(QIODevice *sock, AccountId id) {
305 signalProxy()->addPeer(socket);
306 _connectedToCore = true;
307 setCurrentCoreAccount(id);
310 void Client::setSyncedToCore() {
311 // create buffersyncer
312 Q_ASSERT(!_bufferSyncer);
313 _bufferSyncer = new BufferSyncer(this);
314 connect(bufferSyncer(), SIGNAL(lastSeenMsgSet(BufferId, MsgId)), this, SLOT(updateLastSeenMsg(BufferId, MsgId)));
315 connect(bufferSyncer(), SIGNAL(bufferRemoved(BufferId)), this, SLOT(bufferRemoved(BufferId)));
316 connect(bufferSyncer(), SIGNAL(bufferRenamed(BufferId, QString)), this, SLOT(bufferRenamed(BufferId, QString)));
317 signalProxy()->synchronize(bufferSyncer());
319 // attach backlog manager
320 signalProxy()->synchronize(backlogManager());
322 // create a new BufferViewManager
323 _bufferViewManager = new BufferViewManager(signalProxy(), this);
325 _syncedToCore = true;
327 emit coreConnectionStateChanged(true);
330 void Client::setSecuredConnection() {
331 emit securedConnection();
334 void Client::disconnectFromCore() {
337 _connectedToCore = false;
341 socket->deleteLater();
343 _syncedToCore = false;
345 emit coreConnectionStateChanged(false);
347 // Clear internal data. Hopefully nothing relies on it at this point.
348 setCurrentCoreAccount(0);
351 _bufferSyncer->deleteLater();
355 if(_bufferViewManager) {
356 _bufferViewManager->deleteLater();
357 _bufferViewManager = 0;
360 _networkModel->clear();
362 QHash<BufferId, Buffer *>::iterator bufferIter = _buffers.begin();
363 while(bufferIter != _buffers.end()) {
364 Buffer *buffer = bufferIter.value();
365 disconnect(buffer, SIGNAL(destroyed()), this, 0);
366 bufferIter = _buffers.erase(bufferIter);
367 buffer->deleteLater();
369 Q_ASSERT(_buffers.isEmpty());
371 _statusBuffers.clear();
373 QHash<NetworkId, Network*>::iterator netIter = _networks.begin();
374 while(netIter != _networks.end()) {
375 Network *net = netIter.value();
376 emit networkRemoved(net->networkId());
377 disconnect(net, SIGNAL(destroyed()), this, 0);
378 netIter = _networks.erase(netIter);
381 Q_ASSERT(_networks.isEmpty());
383 QHash<IdentityId, Identity*>::iterator idIter = _identities.begin();
384 while(idIter != _identities.end()) {
385 Identity *id = idIter.value();
386 emit identityRemoved(id->id());
387 idIter = _identities.erase(idIter);
390 Q_ASSERT(_identities.isEmpty());
396 void Client::setCoreConfiguration(const QVariantMap &settings) {
397 SignalProxy::writeDataToDevice(socket, settings);
402 void Client::updateBufferInfo(BufferInfo id) {
403 emit bufferUpdated(id);
406 void Client::bufferDestroyed() {
407 Buffer *buffer = static_cast<Buffer *>(sender());
408 QHash<BufferId, Buffer *>::iterator iter = _buffers.begin();
409 while(iter != _buffers.end()) {
410 if(iter.value() == buffer) {
411 iter = _buffers.erase(iter);
417 QHash<NetworkId, Buffer *>::iterator statusIter = _statusBuffers.begin();
418 while(statusIter != _statusBuffers.end()) {
419 if(statusIter.value() == buffer) {
420 statusIter = _statusBuffers.erase(statusIter);
427 void Client::networkDestroyed() {
428 Network *net = static_cast<Network *>(sender());
429 QHash<NetworkId, Network *>::iterator netIter = _networks.begin();
430 while(netIter != _networks.end()) {
431 if(*netIter == net) {
432 netIter = _networks.erase(netIter);
440 void Client::recvMessage(const Message &msg) {
441 //checkForHighlight(msg);
442 _messageModel->insertMessage(msg);
445 void Client::recvStatusMsg(QString /*net*/, QString /*msg*/) {
446 //recvMessage(net, Message::server("", QString("[STATUS] %1").arg(msg)));
449 void Client::receiveBacklog(BufferId bufferId, const QVariantList &msgs) {
450 //checkForHighlight(msg);
451 //QTime start = QTime::currentTime();
452 foreach(QVariant v, msgs) {
453 _messageModel->insertMessage(v.value<Message>());
455 //qDebug() << "processed" << msgs.count() << "backlog lines in" << start.msecsTo(QTime::currentTime());
458 void Client::layoutMsg() {
459 if(layoutQueue.isEmpty()) {
464 Buffer *buffer = layoutQueue.takeFirst();
465 if(buffer->layoutMsg()) {
466 layoutQueue.append(buffer); // Buffer has more messages in its queue --> Round Robin
470 if(layoutQueue.isEmpty())
474 AbstractUiMsg *Client::layoutMsg(const Message &msg) {
475 return instance()->mainUi->layoutMsg(msg);
478 void Client::checkForHighlight(Message &msg) {
479 NotificationSettings notificationSettings;
480 const Network *net = network(msg.bufferInfo().networkId());
481 if(net && !net->myNick().isEmpty()) {
482 QStringList nickList;
483 if(notificationSettings.highlightNick() == NotificationSettings::CurrentNick) {
484 nickList << net->myNick();
485 } else if(notificationSettings.highlightNick() == NotificationSettings::AllNicks) {
486 const Identity *myIdentity = identity(net->identity());
488 nickList = myIdentity->nicks();
490 foreach(QString nickname, nickList) {
491 QRegExp nickRegExp("^(.*\\W)?" + QRegExp::escape(nickname) + "(\\W.*)?$");
492 if((msg.type() & (Message::Plain | Message::Notice | Message::Action))
493 && !(msg.flags() & Message::Self)
494 && nickRegExp.exactMatch(msg.contents())) {
495 msg.setFlags(msg.flags() | Message::Highlight);
500 foreach(QVariant highlight, notificationSettings.highlightList()) {
501 QVariantMap highlightRule = highlight.toMap();
502 if(!highlightRule["enable"].toBool())
504 Qt::CaseSensitivity caseSensitivity = highlightRule["cs"].toBool() ? Qt::CaseSensitive : Qt::CaseInsensitive;
505 QString name = highlightRule["name"].toString();
507 if(highlightRule["regex"].toBool()) {
508 userRegExp = QRegExp(name, caseSensitivity);
510 userRegExp = QRegExp("^(.*\\W)?" + QRegExp::escape(name) + "(\\W.*)?$", caseSensitivity);
512 if((msg.type() & (Message::Plain | Message::Notice | Message::Action))
513 && !(msg.flags() & Message::Self)
514 && userRegExp.exactMatch(msg.contents())) {
515 msg.setFlags(msg.flags() | Message::Highlight);
522 void Client::updateLastSeenMsg(BufferId id, const MsgId &msgId) {
523 Buffer *b = buffer(id);
525 qWarning() << "Client::updateLastSeen(): Unknown buffer" << id;
528 b->setLastSeenMsg(msgId);
531 void Client::setBufferLastSeenMsg(BufferId id, const MsgId &msgId) {
534 bufferSyncer()->requestSetLastSeenMsg(id, msgId);
537 void Client::removeBuffer(BufferId id) {
538 if(!bufferSyncer()) return;
539 bufferSyncer()->requestRemoveBuffer(id);
542 void Client::bufferRemoved(BufferId bufferId) {
543 // first remove the buffer from hash. this prohibits further lastSeenUpdates
545 if(_buffers.contains(bufferId)) {
546 buff = _buffers.take(bufferId);
547 layoutQueue.removeAll(buff);
548 disconnect(buff, 0, this, 0);
551 // then we select a sane buffer (status buffer)
552 /* we have to manually select a buffer because otherwise inconsitent changes
553 * to the model might occur:
554 * the result of a buffer removal triggers a change in the selection model.
555 * the newly selected buffer might be a channel that hasn't been selected yet
556 * and a new nickview would be created (which never heard of the "rowsAboutToBeRemoved").
557 * this new view (and/or) its sort filter will then only receive a "rowsRemoved" signal.
559 QModelIndex current = bufferModel()->currentIndex();
560 if(current.data(NetworkModel::BufferIdRole).value<BufferId>() == bufferId) {
561 bufferModel()->setCurrentIndex(current.sibling(0,0));
564 // and remove it from the model
565 networkModel()->removeBuffer(bufferId);
571 void Client::bufferRenamed(BufferId bufferId, const QString &newName) {
572 QModelIndex bufferIndex = networkModel()->bufferIndex(bufferId);
573 if(bufferIndex.isValid()) {
574 networkModel()->setData(bufferIndex, newName, Qt::DisplayRole);