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 "buffersyncer.h"
27 #include "ircchannel.h"
31 #include "networkmodel.h"
32 #include "buffermodel.h"
33 #include "quasselui.h"
34 #include "signalproxy.h"
37 QPointer<Client> Client::instanceptr = 0;
38 AccountId Client::_currentCoreAccount = 0;
40 /*** Initialization/destruction ***/
42 Client *Client::instance() {
44 instanceptr = new Client();
48 void Client::destroy() {
50 instanceptr->deleteLater();
53 void Client::init(AbstractUi *ui) {
54 instance()->mainUi = ui;
58 Client::Client(QObject *parent)
61 _signalProxy(new SignalProxy(SignalProxy::Client, this)),
66 _connectedToCore(false),
69 _monitorBuffer = new Buffer(BufferInfo(), this);
77 _currentCoreAccount = 0;
78 _networkModel = new NetworkModel(this);
79 connect(this, SIGNAL(bufferUpdated(BufferInfo)),
80 _networkModel, SLOT(bufferUpdated(BufferInfo)));
81 connect(this, SIGNAL(networkRemoved(NetworkId)),
82 _networkModel, SLOT(networkRemoved(NetworkId)));
84 _bufferModel = new BufferModel(_networkModel);
86 SignalProxy *p = signalProxy();
88 p->attachSlot(SIGNAL(displayMsg(const Message &)), this, SLOT(recvMessage(const Message &)));
89 p->attachSlot(SIGNAL(displayStatusMsg(QString, QString)), this, SLOT(recvStatusMsg(QString, QString)));
91 p->attachSlot(SIGNAL(backlogData(BufferInfo, const QVariantList &, bool)), this, SLOT(recvBacklogData(BufferInfo, const QVariantList &, bool)));
92 p->attachSlot(SIGNAL(bufferInfoUpdated(BufferInfo)), this, SLOT(updateBufferInfo(BufferInfo)));
93 p->attachSignal(this, SIGNAL(sendInput(BufferInfo, QString)));
94 p->attachSignal(this, SIGNAL(requestNetworkStates()));
96 p->attachSignal(this, SIGNAL(requestCreateIdentity(const Identity &)), SIGNAL(createIdentity(const Identity &)));
97 p->attachSignal(this, SIGNAL(requestUpdateIdentity(const Identity &)), SIGNAL(updateIdentity(const Identity &)));
98 p->attachSignal(this, SIGNAL(requestRemoveIdentity(IdentityId)), SIGNAL(removeIdentity(IdentityId)));
99 p->attachSlot(SIGNAL(identityCreated(const Identity &)), this, SLOT(coreIdentityCreated(const Identity &)));
100 p->attachSlot(SIGNAL(identityRemoved(IdentityId)), this, SLOT(coreIdentityRemoved(IdentityId)));
102 p->attachSignal(this, SIGNAL(requestCreateNetwork(const NetworkInfo &)), SIGNAL(createNetwork(const NetworkInfo &)));
103 p->attachSignal(this, SIGNAL(requestUpdateNetwork(const NetworkInfo &)), SIGNAL(updateNetwork(const NetworkInfo &)));
104 p->attachSignal(this, SIGNAL(requestRemoveNetwork(NetworkId)), SIGNAL(removeNetwork(NetworkId)));
105 p->attachSlot(SIGNAL(networkCreated(NetworkId)), this, SLOT(coreNetworkCreated(NetworkId)));
106 p->attachSlot(SIGNAL(networkRemoved(NetworkId)), this, SLOT(coreNetworkRemoved(NetworkId)));
108 connect(p, SIGNAL(disconnected()), this, SLOT(disconnectFromCore()));
110 //connect(mainUi, SIGNAL(connectToCore(const QVariantMap &)), this, SLOT(connectToCore(const QVariantMap &)));
111 connect(mainUi, SIGNAL(disconnectFromCore()), this, SLOT(disconnectFromCore()));
112 connect(this, SIGNAL(connected()), mainUi, SLOT(connectedToCore()));
113 connect(this, SIGNAL(disconnected()), mainUi, SLOT(disconnectedFromCore()));
115 layoutTimer = new QTimer(this);
116 layoutTimer->setInterval(0);
117 layoutTimer->setSingleShot(false);
118 connect(layoutTimer, SIGNAL(timeout()), this, SLOT(layoutMsg()));
122 /*** public static methods ***/
124 AccountId Client::currentCoreAccount() {
125 return _currentCoreAccount;
128 void Client::setCurrentCoreAccount(AccountId id) {
129 _currentCoreAccount = id;
132 QList<BufferInfo> Client::allBufferInfos() {
133 QList<BufferInfo> bufferids;
134 foreach(Buffer *buffer, buffers()) {
135 bufferids << buffer->bufferInfo();
140 QList<Buffer *> Client::buffers() {
141 return instance()->_buffers.values();
146 Buffer *Client::buffer(BufferId bufferId) {
147 if(instance()->_buffers.contains(bufferId))
148 return instance()->_buffers[bufferId];
154 Buffer *Client::buffer(BufferInfo bufferInfo) {
155 Buffer *buff = buffer(bufferInfo.bufferId());
158 Client *client = Client::instance();
159 buff = new Buffer(bufferInfo, client);
160 connect(buff, SIGNAL(destroyed()),
161 client, SLOT(bufferDestroyed()));
162 client->_buffers[bufferInfo.bufferId()] = buff;
163 emit client->bufferUpdated(bufferInfo);
165 // I don't like this: but currently there isn't really a prettier way:
166 QModelIndex bufferIdx = networkModel()->bufferIndex(bufferInfo.bufferId());
167 bufferModel()->setCurrentIndex(bufferModel()->mapFromSource(bufferIdx));
173 bool Client::isConnected() {
174 return instance()->_connectedToCore;
177 bool Client::isSynced() {
178 return instance()->_syncedToCore;
181 /*** Network handling ***/
183 QList<NetworkId> Client::networkIds() {
184 return instance()->_networks.keys();
187 const Network * Client::network(NetworkId networkid) {
188 if(instance()->_networks.contains(networkid)) return instance()->_networks[networkid];
192 void Client::createNetwork(const NetworkInfo &info) {
193 emit instance()->requestCreateNetwork(info);
196 void Client::updateNetwork(const NetworkInfo &info) {
197 emit instance()->requestUpdateNetwork(info);
200 void Client::removeNetwork(NetworkId id) {
201 emit instance()->requestRemoveNetwork(id);
204 void Client::addNetwork(Network *net) {
205 net->setProxy(signalProxy());
206 signalProxy()->synchronize(net);
207 networkModel()->attachNetwork(net);
208 connect(net, SIGNAL(destroyed()), instance(), SLOT(networkDestroyed()));
209 instance()->_networks[net->networkId()] = net;
210 emit instance()->networkCreated(net->networkId());
213 void Client::coreNetworkCreated(NetworkId id) {
214 if(_networks.contains(id)) {
215 qWarning() << "Creation of already existing network requested!";
218 Network *net = new Network(id, this);
222 void Client::coreNetworkRemoved(NetworkId id) {
223 if(!_networks.contains(id)) return;
224 Network *net = _networks.take(id);
225 emit networkRemoved(net->networkId());
229 /*** Identity handling ***/
231 QList<IdentityId> Client::identityIds() {
232 return instance()->_identities.keys();
235 const Identity * Client::identity(IdentityId id) {
236 if(instance()->_identities.contains(id)) return instance()->_identities[id];
240 void Client::createIdentity(const Identity &id) {
241 emit instance()->requestCreateIdentity(id);
244 void Client::updateIdentity(const Identity &id) {
245 emit instance()->requestUpdateIdentity(id);
248 void Client::removeIdentity(IdentityId id) {
249 emit instance()->requestRemoveIdentity(id);
252 void Client::coreIdentityCreated(const Identity &other) {
253 if(!_identities.contains(other.id())) {
254 Identity *identity = new Identity(other, this);
255 _identities[other.id()] = identity;
256 identity->setInitialized();
257 signalProxy()->synchronize(identity);
258 emit identityCreated(other.id());
260 qWarning() << tr("Identity already exists in client!");
264 void Client::coreIdentityRemoved(IdentityId id) {
265 if(_identities.contains(id)) {
266 emit identityRemoved(id);
267 Identity *i = _identities.take(id);
273 void Client::userInput(BufferInfo bufferInfo, QString message) {
274 emit instance()->sendInput(bufferInfo, message);
277 /*** core connection stuff ***/
279 void Client::setConnectedToCore(QIODevice *sock, AccountId id) {
281 signalProxy()->addPeer(socket);
282 _connectedToCore = true;
283 setCurrentCoreAccount(id);
286 void Client::setSyncedToCore() {
287 // create buffersyncer
288 Q_ASSERT(!_bufferSyncer);
289 _bufferSyncer = new BufferSyncer(this);
290 connect(bufferSyncer(), SIGNAL(lastSeenSet(BufferId, const QDateTime &)), this, SLOT(updateLastSeen(BufferId, const QDateTime &)));
291 connect(bufferSyncer(), SIGNAL(bufferRemoved(BufferId)), this, SLOT(bufferRemoved(BufferId)));
292 connect(bufferSyncer(), SIGNAL(bufferRenamed(BufferId, QString)), this, SLOT(bufferRenamed(BufferId, QString)));
293 signalProxy()->synchronize(bufferSyncer());
295 _syncedToCore = true;
297 emit coreConnectionStateChanged(true);
300 void Client::disconnectFromCore() {
306 socket->deleteLater();
308 _connectedToCore = false;
309 _syncedToCore = false;
310 setCurrentCoreAccount(0);
312 emit coreConnectionStateChanged(false);
314 // Clear internal data. Hopefully nothing relies on it at this point.
316 _bufferSyncer->deleteLater();
319 _networkModel->clear();
321 QHash<BufferId, Buffer *>::iterator bufferIter = _buffers.begin();
322 while(bufferIter != _buffers.end()) {
323 Buffer *buffer = bufferIter.value();
324 disconnect(buffer, SIGNAL(destroyed()), this, 0);
325 bufferIter = _buffers.erase(bufferIter);
326 buffer->deleteLater();
328 Q_ASSERT(_buffers.isEmpty());
330 QHash<NetworkId, Network*>::iterator netIter = _networks.begin();
331 while(netIter != _networks.end()) {
332 Network *net = netIter.value();
333 emit networkRemoved(net->networkId());
334 disconnect(net, SIGNAL(destroyed()), this, 0);
335 netIter = _networks.erase(netIter);
338 Q_ASSERT(_networks.isEmpty());
340 QHash<IdentityId, Identity*>::iterator idIter = _identities.begin();
341 while(idIter != _identities.end()) {
342 Identity *id = idIter.value();
343 emit identityRemoved(id->id());
344 idIter = _identities.erase(idIter);
347 Q_ASSERT(_identities.isEmpty());
353 void Client::setCoreConfiguration(const QVariantMap &settings) {
354 SignalProxy::writeDataToDevice(socket, settings);
359 void Client::updateBufferInfo(BufferInfo id) {
360 emit bufferUpdated(id);
363 void Client::bufferDestroyed() {
364 Buffer *buffer = static_cast<Buffer *>(sender());
365 QHash<BufferId, Buffer *>::iterator iter = _buffers.begin();
366 while(iter != _buffers.end()) {
367 if(iter.value() == buffer) {
368 iter = _buffers.erase(iter);
375 void Client::networkDestroyed() {
376 Network *net = static_cast<Network *>(sender());
377 QHash<NetworkId, Network *>::iterator netIter = _networks.begin();
378 while(netIter != _networks.end()) {
379 if(*netIter == net) {
380 netIter = _networks.erase(netIter);
388 void Client::recvMessage(const Message &message) {
389 Message msg = message;
392 if(msg.type() == Message::Error) {
393 b = buffer(msg.bufferInfo().bufferId());
395 // FIXME: if buffer doesn't exist, forward the message to the status or current buffer
396 b = buffer(msg.bufferInfo());
399 b = buffer(msg.bufferInfo());
402 checkForHighlight(msg);
404 //bufferModel()->updateBufferActivity(msg);
406 if(msg.type() == Message::Plain || msg.type() == Message::Notice || msg.type() == Message::Action) {
407 const Network *net = network(msg.bufferInfo().networkId());
408 QString networkName = net != 0
409 ? net->networkName() + ":"
411 QString sender = networkName + msg.bufferInfo().bufferName() + ":" + msg.sender();
412 Message mmsg = Message(msg.timestamp(), msg.bufferInfo(), msg.type(), msg.text(), sender, msg.flags());
413 monitorBuffer()->appendMsg(mmsg);
417 void Client::recvStatusMsg(QString /*net*/, QString /*msg*/) {
418 //recvMessage(net, Message::server("", QString("[STATUS] %1").arg(msg)));
421 void Client::recvBacklogData(BufferInfo id, QVariantList msgs, bool /*done*/) {
422 Buffer *b = buffer(id);
423 foreach(QVariant v, msgs) {
424 Message msg = v.value<Message>();
425 checkForHighlight(msg);
427 //networkModel()->updateBufferActivity(msg);
428 if(!layoutQueue.contains(b)) layoutQueue.append(b);
430 if(layoutQueue.count() && !layoutTimer->isActive()) layoutTimer->start();
433 void Client::layoutMsg() {
434 if(layoutQueue.count()) {
435 Buffer *b = layoutQueue.takeFirst(); // TODO make this the current buffer
437 layoutQueue.append(b); // Buffer has more messages in its queue --> Round Robin
440 if(!layoutQueue.count())
444 AbstractUiMsg *Client::layoutMsg(const Message &msg) {
445 return instance()->mainUi->layoutMsg(msg);
448 void Client::checkForHighlight(Message &msg) {
449 const Network *net = network(msg.bufferInfo().networkId());
450 if(net && !net->myNick().isEmpty()) {
451 QRegExp nickRegExp("^(.*\\W)?" + QRegExp::escape(net->myNick()) + "(\\W.*)?$");
452 if((msg.type() & (Message::Plain | Message::Notice | Message::Action)) && nickRegExp.exactMatch(msg.text()))
453 msg.setFlags(msg.flags() | Message::Highlight);
457 void Client::updateLastSeen(BufferId id, const QDateTime &lastSeen) {
458 Buffer *b = buffer(id);
460 qWarning() << "Client::updateLastSeen(): Unknown buffer" << id;
463 b->setLastSeen(lastSeen);
466 void Client::setBufferLastSeen(BufferId id, const QDateTime &lastSeen) {
467 if(!bufferSyncer()) return;
468 bufferSyncer()->requestSetLastSeen(id, lastSeen);
471 void Client::removeBuffer(BufferId id) {
472 if(!bufferSyncer()) return;
473 bufferSyncer()->requestRemoveBuffer(id);
476 void Client::bufferRemoved(BufferId bufferId) {
477 QModelIndex current = bufferModel()->currentIndex();
478 if(current.data(NetworkModel::BufferIdRole).value<BufferId>() == bufferId) {
479 // select the status buffer if the currently displayed buffer is about to be removed
480 bufferModel()->setCurrentIndex(current.sibling(0,0));
483 networkModel()->removeBuffer(bufferId);
484 if(_buffers.contains(bufferId)) {
485 Buffer *buff = _buffers.take(bufferId);
486 disconnect(buff, 0, this, 0);
491 void Client::bufferRenamed(BufferId bufferId, const QString &newName) {
492 QModelIndex bufferIndex = networkModel()->bufferIndex(bufferId);
493 if(bufferIndex.isValid()) {
494 networkModel()->setData(bufferIndex, newName, Qt::DisplayRole);