src/client/client.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 "client.h"
  22
  23 #include "bufferinfo.h"
  24 #include "buffersyncer.h"
  25 #include "global.h"
  26 #include "identity.h"
  27 #include "ircchannel.h"
  28 #include "ircuser.h"
  29 #include "message.h"
  30 #include "network.h"
  31 #include "networkmodel.h"
  32 #include "buffermodel.h"
  33 #include "quasselui.h"
  34 #include "signalproxy.h"
  35 #include "util.h"
  36
  37 QPointer<Client> Client::instanceptr = 0;
  38 AccountId Client::_currentCoreAccount = 0;
  39
  40 /*** Initialization/destruction ***/
  41
  42 Client *Client::instance() {
  43   if(!instanceptr)
  44     instanceptr = new Client();
  45   return instanceptr;
  46 }
  47
  48 void Client::destroy() {
  49   //delete instanceptr;
  50   instanceptr->deleteLater();
  51 }
  52
  53 void Client::init(AbstractUi *ui) {
  54   instance()->mainUi = ui;
  55   instance()->init();
  56 }
  57
  58 Client::Client(QObject *parent)
  59   : QObject(parent),
  60     socket(0),
  61     _signalProxy(new SignalProxy(SignalProxy::Client, this)),
  62     mainUi(0),
  63     _networkModel(0),
  64     _bufferModel(0),
  65     _bufferSyncer(0),
  66     _connectedToCore(false),
  67     _syncedToCore(false)
  68 {
  69   _monitorBuffer = new Buffer(BufferInfo(), this);
  70 }
  71
  72 Client::~Client() {
  73   disconnectFromCore();
  74 }
  75
  76 void Client::init() {
  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)));
  83
  84   _bufferModel = new BufferModel(_networkModel);
  85
  86   SignalProxy *p = signalProxy();
  87
  88   p->attachSlot(SIGNAL(displayMsg(const Message &)), this, SLOT(recvMessage(const Message &)));
  89   p->attachSlot(SIGNAL(displayStatusMsg(QString, QString)), this, SLOT(recvStatusMsg(QString, QString)));
  90
  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()));
  95
  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)));
 101
 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)));
 107
 108   connect(p, SIGNAL(disconnected()), this, SLOT(disconnectFromCore()));
 109
 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()));
 114
 115   layoutTimer = new QTimer(this);
 116   layoutTimer->setInterval(0);
 117   layoutTimer->setSingleShot(false);
 118   connect(layoutTimer, SIGNAL(timeout()), this, SLOT(layoutMsg()));
 119
 120 }
 121
 122 /*** public static methods ***/
 123
 124 AccountId Client::currentCoreAccount() {
 125   return _currentCoreAccount;
 126 }
 127
 128 void Client::setCurrentCoreAccount(AccountId id) {
 129   _currentCoreAccount = id;
 130 }
 131
 132 QList<BufferInfo> Client::allBufferInfos() {
 133   QList<BufferInfo> bufferids;
 134   foreach(Buffer *buffer, buffers()) {
 135     bufferids << buffer->bufferInfo();
 136   }
 137   return bufferids;
 138 }
 139
 140 QList<Buffer *> Client::buffers() {
 141   return instance()->_buffers.values();
 142 }
 143
 144
 145 // FIXME remove
 146 Buffer *Client::buffer(BufferId bufferId) {
 147   if(instance()->_buffers.contains(bufferId))
 148     return instance()->_buffers[bufferId];
 149   else
 150     return 0;
 151 }
 152
 153 // FIXME remove
 154 Buffer *Client::buffer(BufferInfo bufferInfo) {
 155   Buffer *buff = buffer(bufferInfo.bufferId());
 156
 157   if(!buff) {
 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);
 164   }
 165   Q_ASSERT(buff);
 166   return buff;
 167 }
 168
 169 bool Client::isConnected() {
 170   return instance()->_connectedToCore;
 171 }
 172
 173 bool Client::isSynced() {
 174   return instance()->_syncedToCore;
 175 }
 176
 177 /*** Network handling ***/
 178
 179 QList<NetworkId> Client::networkIds() {
 180   return instance()->_networks.keys();
 181 }
 182
 183 const Network * Client::network(NetworkId networkid) {
 184   if(instance()->_networks.contains(networkid)) return instance()->_networks[networkid];
 185   else return 0;
 186 }
 187
 188 void Client::createNetwork(const NetworkInfo &info) {
 189   emit instance()->requestCreateNetwork(info);
 190 }
 191
 192 void Client::updateNetwork(const NetworkInfo &info) {
 193   emit instance()->requestUpdateNetwork(info);
 194 }
 195
 196 void Client::removeNetwork(NetworkId id) {
 197   emit instance()->requestRemoveNetwork(id);
 198 }
 199
 200 void Client::addNetwork(Network *net) {
 201   net->setProxy(signalProxy());
 202   signalProxy()->synchronize(net);
 203   networkModel()->attachNetwork(net);
 204   connect(net, SIGNAL(destroyed()), instance(), SLOT(networkDestroyed()));
 205   instance()->_networks[net->networkId()] = net;
 206   emit instance()->networkCreated(net->networkId());
 207 }
 208
 209 void Client::coreNetworkCreated(NetworkId id) {
 210   if(_networks.contains(id)) {
 211     qWarning() << "Creation of already existing network requested!";
 212     return;
 213   }
 214   Network *net = new Network(id, this);
 215   addNetwork(net);
 216 }
 217
 218 void Client::coreNetworkRemoved(NetworkId id) {
 219   if(!_networks.contains(id)) return;
 220   Network *net = _networks.take(id);
 221   emit networkRemoved(net->networkId());
 222   net->deleteLater();
 223 }
 224
 225 /*** Identity handling ***/
 226
 227 QList<IdentityId> Client::identityIds() {
 228   return instance()->_identities.keys();
 229 }
 230
 231 const Identity * Client::identity(IdentityId id) {
 232   if(instance()->_identities.contains(id)) return instance()->_identities[id];
 233   else return 0;
 234 }
 235
 236 void Client::createIdentity(const Identity &id) {
 237   emit instance()->requestCreateIdentity(id);
 238 }
 239
 240 void Client::updateIdentity(const Identity &id) {
 241   emit instance()->requestUpdateIdentity(id);
 242 }
 243
 244 void Client::removeIdentity(IdentityId id) {
 245   emit instance()->requestRemoveIdentity(id);
 246 }
 247
 248 void Client::coreIdentityCreated(const Identity &other) {
 249   if(!_identities.contains(other.id())) {
 250     Identity *identity = new Identity(other, this);
 251     _identities[other.id()] = identity;
 252     identity->setInitialized();
 253     signalProxy()->synchronize(identity);
 254     emit identityCreated(other.id());
 255   } else {
 256     qWarning() << tr("Identity already exists in client!");
 257   }
 258 }
 259
 260 void Client::coreIdentityRemoved(IdentityId id) {
 261   if(_identities.contains(id)) {
 262     emit identityRemoved(id);
 263     Identity *i = _identities.take(id);
 264     i->deleteLater();
 265   }
 266 }
 267
 268 /***  ***/
 269 void Client::userInput(BufferInfo bufferInfo, QString message) {
 270   emit instance()->sendInput(bufferInfo, message);
 271 }
 272
 273 /*** core connection stuff ***/
 274
 275 void Client::setConnectedToCore(QIODevice *sock, AccountId id) {
 276   socket = sock;
 277   signalProxy()->addPeer(socket);
 278   _connectedToCore = true;
 279   setCurrentCoreAccount(id);
 280 }
 281
 282 void Client::setSyncedToCore() {
 283   // create buffersyncer
 284   Q_ASSERT(!_bufferSyncer);
 285   _bufferSyncer = new BufferSyncer(this);
 286   connect(bufferSyncer(), SIGNAL(lastSeenSet(BufferId, const QDateTime &)), this, SLOT(updateLastSeen(BufferId, const QDateTime &)));
 287   connect(bufferSyncer(), SIGNAL(bufferRemoved(BufferId)), this, SLOT(bufferRemoved(BufferId)));
 288   connect(bufferSyncer(), SIGNAL(bufferRenamed(BufferId, QString)), this, SLOT(bufferRenamed(BufferId, QString)));
 289   signalProxy()->synchronize(bufferSyncer());
 290
 291   _syncedToCore = true;
 292   emit connected();
 293   emit coreConnectionStateChanged(true);
 294 }
 295
 296 void Client::disconnectFromCore() {
 297   if(!isConnected())
 298     return;
 299
 300   if(socket) {
 301     socket->close();
 302     socket->deleteLater();
 303   }
 304   _connectedToCore = false;
 305   _syncedToCore = false;
 306   setCurrentCoreAccount(0);
 307   emit disconnected();
 308   emit coreConnectionStateChanged(false);
 309
 310   // Clear internal data. Hopefully nothing relies on it at this point.
 311   if(_bufferSyncer) {
 312     _bufferSyncer->deleteLater();
 313     _bufferSyncer = 0;
 314   }
 315   _networkModel->clear();
 316
 317   QHash<BufferId, Buffer *>::iterator bufferIter =  _buffers.begin();
 318   while(bufferIter != _buffers.end()) {
 319     Buffer *buffer = bufferIter.value();
 320     disconnect(buffer, SIGNAL(destroyed()), this, 0);
 321     bufferIter = _buffers.erase(bufferIter);
 322     buffer->deleteLater();
 323   }
 324   Q_ASSERT(_buffers.isEmpty());
 325
 326   QHash<NetworkId, Network*>::iterator netIter = _networks.begin();
 327   while(netIter != _networks.end()) {
 328     Network *net = netIter.value();
 329     emit networkRemoved(net->networkId());
 330     disconnect(net, SIGNAL(destroyed()), this, 0);
 331     netIter = _networks.erase(netIter);
 332     net->deleteLater();
 333   }
 334   Q_ASSERT(_networks.isEmpty());
 335
 336   QHash<IdentityId, Identity*>::iterator idIter = _identities.begin();
 337   while(idIter != _identities.end()) {
 338     Identity *id = idIter.value();
 339     emit identityRemoved(id->id());
 340     idIter = _identities.erase(idIter);
 341     id->deleteLater();
 342   }
 343   Q_ASSERT(_identities.isEmpty());
 344
 345   layoutQueue.clear();
 346   layoutTimer->stop();
 347 }
 348
 349 void Client::setCoreConfiguration(const QVariantMap &settings) {
 350   SignalProxy::writeDataToDevice(socket, settings);
 351 }
 352
 353 /*** ***/
 354
 355 void Client::updateBufferInfo(BufferInfo id) {
 356   emit bufferUpdated(id);
 357 }
 358
 359 void Client::bufferDestroyed() {
 360   Buffer *buffer = static_cast<Buffer *>(sender());
 361   QHash<BufferId, Buffer *>::iterator iter = _buffers.begin();
 362   while(iter != _buffers.end()) {
 363     if(iter.value() == buffer) {
 364       iter = _buffers.erase(iter);
 365       break;
 366     }
 367     iter++;
 368   }
 369 }
 370
 371 void Client::networkDestroyed() {
 372   Network *net = static_cast<Network *>(sender());
 373   QHash<NetworkId, Network *>::iterator netIter = _networks.begin();
 374   while(netIter != _networks.end()) {
 375     if(*netIter == net) {
 376       netIter = _networks.erase(netIter);
 377       break;
 378     } else {
 379       netIter++;
 380     }
 381   }
 382 }
 383
 384 void Client::recvMessage(const Message &message) {
 385   Message msg = message;
 386   Buffer *b;
 387
 388   if(msg.type() == Message::Error) {
 389     b = buffer(msg.bufferInfo().bufferId());
 390     if(!b) {
 391       // FIXME: if buffer doesn't exist, forward the message to the status or current buffer
 392       b = buffer(msg.bufferInfo());
 393     }
 394   } else {
 395     b = buffer(msg.bufferInfo());
 396   }
 397
 398   checkForHighlight(msg);
 399   b->appendMsg(msg);
 400   //bufferModel()->updateBufferActivity(msg);
 401
 402   if(msg.type() == Message::Plain || msg.type() == Message::Notice || msg.type() == Message::Action) {
 403     const Network *net = network(msg.bufferInfo().networkId());
 404     QString networkName = net != 0
 405       ? net->networkName() + ":"
 406       : QString();
 407     QString sender = networkName + msg.bufferInfo().bufferName() + ":" + msg.sender();
 408     Message mmsg = Message(msg.timestamp(), msg.bufferInfo(), msg.type(), msg.text(), sender, msg.flags());
 409     monitorBuffer()->appendMsg(mmsg);
 410   }
 411 }
 412
 413 void Client::recvStatusMsg(QString /*net*/, QString /*msg*/) {
 414   //recvMessage(net, Message::server("", QString("[STATUS] %1").arg(msg)));
 415 }
 416
 417 void Client::recvBacklogData(BufferInfo id, QVariantList msgs, bool /*done*/) {
 418   Buffer *b = buffer(id);
 419   foreach(QVariant v, msgs) {
 420     Message msg = v.value<Message>();
 421     checkForHighlight(msg);
 422     b->prependMsg(msg);
 423     //networkModel()->updateBufferActivity(msg);
 424     if(!layoutQueue.contains(b)) layoutQueue.append(b);
 425   }
 426   if(layoutQueue.count() && !layoutTimer->isActive()) layoutTimer->start();
 427 }
 428
 429 void Client::layoutMsg() {
 430   if(layoutQueue.count()) {
 431     Buffer *b = layoutQueue.takeFirst();  // TODO make this the current buffer
 432     if(b->layoutMsg())
 433       layoutQueue.append(b);  // Buffer has more messages in its queue --> Round Robin
 434   }
 435
 436   if(!layoutQueue.count())
 437     layoutTimer->stop();
 438 }
 439
 440 AbstractUiMsg *Client::layoutMsg(const Message &msg) {
 441   return instance()->mainUi->layoutMsg(msg);
 442 }
 443
 444 void Client::checkForHighlight(Message &msg) {
 445   const Network *net = network(msg.bufferInfo().networkId());
 446   if(net && !net->myNick().isEmpty()) {
 447     QRegExp nickRegExp("^(.*\\W)?" + QRegExp::escape(net->myNick()) + "(\\W.*)?$");
 448     if((msg.type() & (Message::Plain | Message::Notice | Message::Action)) && nickRegExp.exactMatch(msg.text()))
 449       msg.setFlags(msg.flags() | Message::Highlight);
 450   }
 451 }
 452
 453 void Client::updateLastSeen(BufferId id, const QDateTime &lastSeen) {
 454   Buffer *b = buffer(id);
 455   if(!b) {
 456     qWarning() << "Client::updateLastSeen(): Unknown buffer" << id;
 457     return;
 458   }
 459   b->setLastSeen(lastSeen);
 460 }
 461
 462 void Client::setBufferLastSeen(BufferId id, const QDateTime &lastSeen) {
 463   if(!bufferSyncer()) return;
 464   bufferSyncer()->requestSetLastSeen(id, lastSeen);
 465 }
 466
 467 void Client::removeBuffer(BufferId id) {
 468   if(!bufferSyncer()) return;
 469   bufferSyncer()->requestRemoveBuffer(id);
 470 }
 471
 472 void Client::bufferRemoved(BufferId bufferId) {
 473   QModelIndex current = bufferModel()->currentIndex();
 474   if(current.data(NetworkModel::BufferIdRole).value<BufferId>() == bufferId) {
 475     // select the status buffer if the currently displayed buffer is about to be removed
 476     bufferModel()->setCurrentIndex(current.sibling(0,0));
 477   }
 478
 479   networkModel()->removeBuffer(bufferId);
 480   if(_buffers.contains(bufferId)) {
 481     Buffer *buff = _buffers.take(bufferId);
 482     disconnect(buff, 0, this, 0);
 483     buff->deleteLater();
 484   }
 485 }
 486
 487 void Client::bufferRenamed(BufferId bufferId, const QString &newName) {
 488   QModelIndex bufferIndex = networkModel()->bufferIndex(bufferId);
 489   if(bufferIndex.isValid()) {
 490     networkModel()->setData(bufferIndex, newName, Qt::DisplayRole);
 491   }
 492 }