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"
36 # include "messagemodel.h"
39 #include "networkmodel.h"
40 #include "quasselui.h"
41 #include "signalproxy.h"
44 QPointer<Client> Client::instanceptr = 0;
45 AccountId Client::_currentCoreAccount = 0;
47 /*** Initialization/destruction ***/
49 Client *Client::instance() {
51 instanceptr = new Client();
55 void Client::destroy() {
57 instanceptr->deleteLater();
60 void Client::init(AbstractUi *ui) {
61 instance()->mainUi = ui;
65 Client::Client(QObject *parent)
68 _signalProxy(new SignalProxy(SignalProxy::Client, this)),
73 _backlogManager(new ClientBacklogManager(this)),
74 _bufferViewManager(0),
75 _ircListHelper(new ClientIrcListHelper(this)),
77 _connectedToCore(false),
80 _monitorBuffer = new Buffer(BufferInfo(), this);
81 _signalProxy->synchronize(_ircListHelper);
83 connect(_backlogManager, SIGNAL(backlog(BufferId, const QVariantList &)),
84 this, SLOT(receiveBacklog(BufferId, const QVariantList &)));
92 _currentCoreAccount = 0;
93 _networkModel = new NetworkModel(this);
95 connect(this, SIGNAL(bufferUpdated(BufferInfo)),
96 _networkModel, SLOT(bufferUpdated(BufferInfo)));
97 connect(this, SIGNAL(networkRemoved(NetworkId)),
98 _networkModel, SLOT(networkRemoved(NetworkId)));
100 _bufferModel = new BufferModel(_networkModel);
102 _messageModel = mainUi->createMessageModel(this);
104 SignalProxy *p = signalProxy();
106 p->attachSlot(SIGNAL(displayMsg(const Message &)), this, SLOT(recvMessage(const Message &)));
107 p->attachSlot(SIGNAL(displayStatusMsg(QString, QString)), this, SLOT(recvStatusMsg(QString, QString)));
109 p->attachSlot(SIGNAL(bufferInfoUpdated(BufferInfo)), this, SLOT(updateBufferInfo(BufferInfo)));
110 p->attachSignal(this, SIGNAL(sendInput(BufferInfo, QString)));
111 p->attachSignal(this, SIGNAL(requestNetworkStates()));
113 p->attachSignal(this, SIGNAL(requestCreateIdentity(const Identity &)), SIGNAL(createIdentity(const Identity &)));
114 p->attachSignal(this, SIGNAL(requestUpdateIdentity(const Identity &)), SIGNAL(updateIdentity(const Identity &)));
115 p->attachSignal(this, SIGNAL(requestRemoveIdentity(IdentityId)), SIGNAL(removeIdentity(IdentityId)));
116 p->attachSlot(SIGNAL(identityCreated(const Identity &)), this, SLOT(coreIdentityCreated(const Identity &)));
117 p->attachSlot(SIGNAL(identityRemoved(IdentityId)), this, SLOT(coreIdentityRemoved(IdentityId)));
119 p->attachSignal(this, SIGNAL(requestCreateNetwork(const NetworkInfo &)), SIGNAL(createNetwork(const NetworkInfo &)));
120 p->attachSignal(this, SIGNAL(requestUpdateNetwork(const NetworkInfo &)), SIGNAL(updateNetwork(const NetworkInfo &)));
121 p->attachSignal(this, SIGNAL(requestRemoveNetwork(NetworkId)), SIGNAL(removeNetwork(NetworkId)));
122 p->attachSlot(SIGNAL(networkCreated(NetworkId)), this, SLOT(coreNetworkCreated(NetworkId)));
123 p->attachSlot(SIGNAL(networkRemoved(NetworkId)), this, SLOT(coreNetworkRemoved(NetworkId)));
125 connect(p, SIGNAL(disconnected()), this, SLOT(disconnectFromCore()));
127 //connect(mainUi, SIGNAL(connectToCore(const QVariantMap &)), this, SLOT(connectToCore(const QVariantMap &)));
128 connect(mainUi, SIGNAL(disconnectFromCore()), this, SLOT(disconnectFromCore()));
129 connect(this, SIGNAL(connected()), mainUi, SLOT(connectedToCore()));
130 connect(this, SIGNAL(disconnected()), mainUi, SLOT(disconnectedFromCore()));
132 layoutTimer = new QTimer(this);
133 layoutTimer->setInterval(0);
134 layoutTimer->setSingleShot(false);
135 connect(layoutTimer, SIGNAL(timeout()), this, SLOT(layoutMsg()));
139 /*** public static methods ***/
141 AccountId Client::currentCoreAccount() {
142 return _currentCoreAccount;
145 void Client::setCurrentCoreAccount(AccountId id) {
146 _currentCoreAccount = id;
149 QList<BufferInfo> Client::allBufferInfos() {
150 QList<BufferInfo> bufferids;
151 foreach(Buffer *buffer, buffers()) {
152 bufferids << buffer->bufferInfo();
157 QList<Buffer *> Client::buffers() {
158 return instance()->_buffers.values();
162 Buffer *Client::statusBuffer(const NetworkId &networkId) const {
163 if(_statusBuffers.contains(networkId))
164 return _statusBuffers[networkId];
169 Buffer *Client::buffer(BufferId bufferId) {
170 if(instance()->_buffers.contains(bufferId))
171 return instance()->_buffers[bufferId];
176 Buffer *Client::buffer(BufferInfo bufferInfo) {
177 Buffer *buff = buffer(bufferInfo.bufferId());
180 Client *client = Client::instance();
181 buff = new Buffer(bufferInfo, client);
182 connect(buff, SIGNAL(destroyed()), client, SLOT(bufferDestroyed()));
183 client->_buffers[bufferInfo.bufferId()] = buff;
184 if(bufferInfo.type() == BufferInfo::StatusBuffer)
185 client->_statusBuffers[bufferInfo.networkId()] = buff;
187 emit client->bufferUpdated(bufferInfo);
189 // I don't like this: but currently there isn't really a prettier way:
190 if(isSynced()) { // this slows down syncing a lot, so disable it during sync
191 QModelIndex bufferIdx = networkModel()->bufferIndex(bufferInfo.bufferId());
192 bufferModel()->setCurrentIndex(bufferModel()->mapFromSource(bufferIdx));
199 bool Client::isConnected() {
200 return instance()->_connectedToCore;
203 bool Client::isSynced() {
204 return instance()->_syncedToCore;
207 /*** Network handling ***/
209 QList<NetworkId> Client::networkIds() {
210 return instance()->_networks.keys();
213 const Network * Client::network(NetworkId networkid) {
214 if(instance()->_networks.contains(networkid)) return instance()->_networks[networkid];
218 void Client::createNetwork(const NetworkInfo &info) {
219 emit instance()->requestCreateNetwork(info);
222 void Client::updateNetwork(const NetworkInfo &info) {
223 emit instance()->requestUpdateNetwork(info);
226 void Client::removeNetwork(NetworkId id) {
227 emit instance()->requestRemoveNetwork(id);
230 void Client::addNetwork(Network *net) {
231 net->setProxy(signalProxy());
232 signalProxy()->synchronize(net);
233 networkModel()->attachNetwork(net);
234 connect(net, SIGNAL(destroyed()), instance(), SLOT(networkDestroyed()));
235 instance()->_networks[net->networkId()] = net;
236 emit instance()->networkCreated(net->networkId());
239 void Client::coreNetworkCreated(NetworkId id) {
240 if(_networks.contains(id)) {
241 qWarning() << "Creation of already existing network requested!";
244 Network *net = new Network(id, this);
248 void Client::coreNetworkRemoved(NetworkId id) {
249 if(!_networks.contains(id))
251 Network *net = _networks.take(id);
252 emit networkRemoved(net->networkId());
256 /*** Identity handling ***/
258 QList<IdentityId> Client::identityIds() {
259 return instance()->_identities.keys();
262 const Identity * Client::identity(IdentityId id) {
263 if(instance()->_identities.contains(id)) return instance()->_identities[id];
267 void Client::createIdentity(const Identity &id) {
268 emit instance()->requestCreateIdentity(id);
271 void Client::updateIdentity(const Identity &id) {
272 emit instance()->requestUpdateIdentity(id);
275 void Client::removeIdentity(IdentityId id) {
276 emit instance()->requestRemoveIdentity(id);
279 void Client::coreIdentityCreated(const Identity &other) {
280 if(!_identities.contains(other.id())) {
281 Identity *identity = new Identity(other, this);
282 _identities[other.id()] = identity;
283 identity->setInitialized();
284 signalProxy()->synchronize(identity);
285 emit identityCreated(other.id());
287 qWarning() << tr("Identity already exists in client!");
291 void Client::coreIdentityRemoved(IdentityId id) {
292 if(_identities.contains(id)) {
293 emit identityRemoved(id);
294 Identity *i = _identities.take(id);
300 void Client::userInput(BufferInfo bufferInfo, QString message) {
301 emit instance()->sendInput(bufferInfo, message);
304 /*** core connection stuff ***/
306 void Client::setConnectedToCore(QIODevice *sock, AccountId id) {
308 signalProxy()->addPeer(socket);
309 _connectedToCore = true;
310 setCurrentCoreAccount(id);
313 void Client::setSyncedToCore() {
314 // create buffersyncer
315 Q_ASSERT(!_bufferSyncer);
316 _bufferSyncer = new BufferSyncer(this);
317 connect(bufferSyncer(), SIGNAL(lastSeenMsgSet(BufferId, MsgId)), this, SLOT(updateLastSeenMsg(BufferId, MsgId)));
318 connect(bufferSyncer(), SIGNAL(bufferRemoved(BufferId)), this, SLOT(bufferRemoved(BufferId)));
319 connect(bufferSyncer(), SIGNAL(bufferRenamed(BufferId, QString)), this, SLOT(bufferRenamed(BufferId, QString)));
320 signalProxy()->synchronize(bufferSyncer());
322 // attach backlog manager
323 signalProxy()->synchronize(backlogManager());
325 // create a new BufferViewManager
326 _bufferViewManager = new BufferViewManager(signalProxy(), this);
328 _syncedToCore = true;
330 emit coreConnectionStateChanged(true);
333 void Client::setSecuredConnection() {
334 emit securedConnection();
337 void Client::disconnectFromCore() {
340 _connectedToCore = false;
344 socket->deleteLater();
346 _syncedToCore = false;
348 emit coreConnectionStateChanged(false);
350 // Clear internal data. Hopefully nothing relies on it at this point.
351 setCurrentCoreAccount(0);
354 _bufferSyncer->deleteLater();
358 if(_bufferViewManager) {
359 _bufferViewManager->deleteLater();
360 _bufferViewManager = 0;
363 _networkModel->clear();
365 QHash<BufferId, Buffer *>::iterator bufferIter = _buffers.begin();
366 while(bufferIter != _buffers.end()) {
367 Buffer *buffer = bufferIter.value();
368 disconnect(buffer, SIGNAL(destroyed()), this, 0);
369 bufferIter = _buffers.erase(bufferIter);
370 buffer->deleteLater();
372 Q_ASSERT(_buffers.isEmpty());
374 _statusBuffers.clear();
376 QHash<NetworkId, Network*>::iterator netIter = _networks.begin();
377 while(netIter != _networks.end()) {
378 Network *net = netIter.value();
379 emit networkRemoved(net->networkId());
380 disconnect(net, SIGNAL(destroyed()), this, 0);
381 netIter = _networks.erase(netIter);
384 Q_ASSERT(_networks.isEmpty());
386 QHash<IdentityId, Identity*>::iterator idIter = _identities.begin();
387 while(idIter != _identities.end()) {
388 Identity *id = idIter.value();
389 emit identityRemoved(id->id());
390 idIter = _identities.erase(idIter);
393 Q_ASSERT(_identities.isEmpty());
399 void Client::setCoreConfiguration(const QVariantMap &settings) {
400 SignalProxy::writeDataToDevice(socket, settings);
405 void Client::updateBufferInfo(BufferInfo id) {
406 emit bufferUpdated(id);
409 void Client::bufferDestroyed() {
410 Buffer *buffer = static_cast<Buffer *>(sender());
411 QHash<BufferId, Buffer *>::iterator iter = _buffers.begin();
412 while(iter != _buffers.end()) {
413 if(iter.value() == buffer) {
414 iter = _buffers.erase(iter);
420 QHash<NetworkId, Buffer *>::iterator statusIter = _statusBuffers.begin();
421 while(statusIter != _statusBuffers.end()) {
422 if(statusIter.value() == buffer) {
423 statusIter = _statusBuffers.erase(statusIter);
430 void Client::networkDestroyed() {
431 Network *net = static_cast<Network *>(sender());
432 QHash<NetworkId, Network *>::iterator netIter = _networks.begin();
433 while(netIter != _networks.end()) {
434 if(*netIter == net) {
435 netIter = _networks.erase(netIter);
444 void Client::recvMessage(const Message &message) {
445 Message msg = message;
448 checkForHighlight(msg);
450 // FIXME clean up code! (dup)
452 // TODO: make redirected messages show up in the correct buffer!
454 if(msg.flags() & Message::Redirected) {
455 BufferSettings bufferSettings;
456 bool inStatus = bufferSettings.value("UserMessagesInStatusBuffer", QVariant(true)).toBool();
457 bool inQuery = bufferSettings.value("UserMessagesInQueryBuffer", QVariant(false)).toBool();
458 bool inCurrent = bufferSettings.value("UserMessagesInCurrentBuffer", QVariant(false)).toBool();
461 b = statusBuffer(msg.bufferInfo().networkId());
464 } else if(!inQuery && !inCurrent) { // make sure the message get's shown somewhere
465 b = buffer(msg.bufferInfo());
471 b = buffer(msg.bufferInfo().bufferId());
474 } else if(!inStatus && !inCurrent) { // make sure the message get's shown somewhere
475 b = statusBuffer(msg.bufferInfo().networkId());
476 if(!b) b = buffer(msg.bufferInfo()); // seems like we have to create the buffer anyways...
482 BufferId currentId = bufferModel()->currentIndex().data(NetworkModel::BufferIdRole).value<BufferId>();
483 b = buffer(currentId);
484 if(b && currentId != msg.bufferInfo().bufferId() && !inQuery) {
486 } else if(!inStatus && !inQuery) { // make sure the message get's shown somewhere
487 b = statusBuffer(msg.bufferInfo().networkId());
488 if(!b) b = buffer(msg.bufferInfo()); // seems like we have to create the buffer anyways...
493 // the regular case: we can deliver where it was supposed to go
494 b = buffer(msg.bufferInfo());
497 //bufferModel()->updateBufferActivity(msg);
499 if(msg.type() == Message::Plain || msg.type() == Message::Notice || msg.type() == Message::Action) {
500 const Network *net = network(msg.bufferInfo().networkId());
501 QString networkName = net != 0
502 ? net->networkName() + ":"
504 QString sender = networkName + msg.bufferInfo().bufferName() + ":" + msg.sender();
505 Message mmsg = Message(msg.timestamp(), msg.bufferInfo(), msg.type(), msg.contents(), sender, msg.flags());
506 monitorBuffer()->appendMsg(mmsg);
508 emit messageReceived(msg);
512 void Client::recvMessage(const Message &msg) {
513 //checkForHighlight(msg);
514 _messageModel->insertMessage(msg);
519 void Client::recvStatusMsg(QString /*net*/, QString /*msg*/) {
520 //recvMessage(net, Message::server("", QString("[STATUS] %1").arg(msg)));
524 void Client::receiveBacklog(BufferId bufferId, const QVariantList &msgs) {
525 //checkForHighlight(msg);
526 foreach(QVariant v, msgs) {
527 _messageModel->insertMessage(v.value<Message>());
533 void Client::receiveBacklog(BufferId bufferId, const QVariantList &msgs) {
534 Buffer *buffer_ = buffer(bufferId);
536 qWarning() << "Client::receiveBacklog(): received Backlog for unknown Buffer:" << bufferId;
541 return; // no work to be done...
543 QVariantList::const_iterator msgIter = msgs.constBegin();
544 QVariantList::const_iterator msgIterEnd = msgs.constEnd();
546 while(msgIter != msgIterEnd) {
547 msg = (*msgIter).value<Message>();
548 checkForHighlight(msg);
549 buffer_->prependMsg(msg);
553 if(!layoutQueue.contains(buffer_))
554 layoutQueue.append(buffer_);
556 if(!layoutTimer->isActive()) {
557 layoutTimer->start();
562 void Client::layoutMsg() {
563 if(layoutQueue.isEmpty()) {
568 Buffer *buffer = layoutQueue.takeFirst();
569 if(buffer->layoutMsg()) {
570 layoutQueue.append(buffer); // Buffer has more messages in its queue --> Round Robin
574 if(layoutQueue.isEmpty())
578 AbstractUiMsg *Client::layoutMsg(const Message &msg) {
579 return instance()->mainUi->layoutMsg(msg);
582 void Client::checkForHighlight(Message &msg) {
583 NotificationSettings notificationSettings;
584 const Network *net = network(msg.bufferInfo().networkId());
585 if(net && !net->myNick().isEmpty()) {
586 QStringList nickList;
587 if(notificationSettings.highlightNick() == NotificationSettings::CurrentNick) {
588 nickList << net->myNick();
589 } else if(notificationSettings.highlightNick() == NotificationSettings::AllNicks) {
590 nickList = identity(net->identity())->nicks();
592 foreach(QString nickname, nickList) {
593 QRegExp nickRegExp("^(.*\\W)?" + QRegExp::escape(nickname) + "(\\W.*)?$");
594 if((msg.type() & (Message::Plain | Message::Notice | Message::Action))
595 && !(msg.flags() & Message::Self)
596 && nickRegExp.exactMatch(msg.contents())) {
597 msg.setFlags(msg.flags() | Message::Highlight);
602 foreach(QVariant highlight, notificationSettings.highlightList()) {
603 QVariantMap highlightRule = highlight.toMap();
604 if(!highlightRule["enable"].toBool())
606 Qt::CaseSensitivity caseSensitivity = highlightRule["cs"].toBool() ? Qt::CaseSensitive : Qt::CaseInsensitive;
607 QString name = highlightRule["name"].toString();
609 if(highlightRule["regex"].toBool()) {
610 userRegExp = QRegExp(name, caseSensitivity);
612 userRegExp = QRegExp("^(.*\\W)?" + QRegExp::escape(name) + "(\\W.*)?$", caseSensitivity);
614 if((msg.type() & (Message::Plain | Message::Notice | Message::Action))
615 && !(msg.flags() & Message::Self)
616 && userRegExp.exactMatch(msg.contents())) {
617 msg.setFlags(msg.flags() | Message::Highlight);
624 void Client::updateLastSeenMsg(BufferId id, const MsgId &msgId) {
625 Buffer *b = buffer(id);
627 qWarning() << "Client::updateLastSeen(): Unknown buffer" << id;
630 b->setLastSeenMsg(msgId);
633 void Client::setBufferLastSeenMsg(BufferId id, const MsgId &msgId) {
636 bufferSyncer()->requestSetLastSeenMsg(id, msgId);
639 void Client::removeBuffer(BufferId id) {
640 if(!bufferSyncer()) return;
641 bufferSyncer()->requestRemoveBuffer(id);
644 void Client::bufferRemoved(BufferId bufferId) {
645 // first remove the buffer from hash. this prohibits further lastSeenUpdates
647 if(_buffers.contains(bufferId)) {
648 buff = _buffers.take(bufferId);
649 disconnect(buff, 0, this, 0);
652 // then we select a sane buffer (status buffer)
653 /* we have to manually select a buffer because otherwise inconsitent changes
654 * to the model might occur:
655 * the result of a buffer removal triggers a change in the selection model.
656 * the newly selected buffer might be a channel that hasn't been selected yet
657 * and a new nickview would be created (which never heard of the "rowsAboutToBeRemoved").
658 * this new view (and/or) its sort filter will then only receive a "rowsRemoved" signal.
660 QModelIndex current = bufferModel()->currentIndex();
661 if(current.data(NetworkModel::BufferIdRole).value<BufferId>() == bufferId) {
662 bufferModel()->setCurrentIndex(current.sibling(0,0));
665 // and remove it from the model
666 networkModel()->removeBuffer(bufferId);
672 void Client::bufferRenamed(BufferId bufferId, const QString &newName) {
673 QModelIndex bufferIndex = networkModel()->bufferIndex(bufferId);
674 if(bufferIndex.isValid()) {
675 networkModel()->setData(bufferIndex, newName, Qt::DisplayRole);