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 "abstractmessageprocessor.h"
  24 #include "bufferinfo.h"
  25 #include "buffermodel.h"
  26 #include "buffersettings.h"
  27 #include "buffersyncer.h"
  28 #include "bufferviewmanager.h"
  29 #include "clientbacklogmanager.h"
  30 #include "clientirclisthelper.h"
  31 #include "global.h"
  32 #include "identity.h"
  33 #include "ircchannel.h"
  34 #include "ircuser.h"
  35 #include "message.h"
  36 #include "messagemodel.h"
  37 #include "network.h"
  38 #include "networkmodel.h"
  39 #include "quasselui.h"
  40 #include "signalproxy.h"
  41 #include "util.h"
  42
  43 QPointer<Client> Client::instanceptr = 0;
  44 AccountId Client::_currentCoreAccount = 0;
  45
  46 /*** Initialization/destruction ***/
  47
  48 Client *Client::instance() {
  49   if(!instanceptr)
  50     instanceptr = new Client();
  51   return instanceptr;
  52 }
  53
  54 void Client::destroy() {
  55   //delete instanceptr;
  56   instanceptr->deleteLater();
  57 }
  58
  59 void Client::init(AbstractUi *ui) {
  60   instance()->mainUi = ui;
  61   instance()->init();
  62 }
  63
  64 Client::Client(QObject *parent)
  65   : QObject(parent),
  66     socket(0),
  67     _signalProxy(new SignalProxy(SignalProxy::Client, this)),
  68     mainUi(0),
  69     _networkModel(0),
  70     _bufferModel(0),
  71     _bufferSyncer(0),
  72     _backlogManager(new ClientBacklogManager(this)),
  73     _bufferViewManager(0),
  74     _ircListHelper(new ClientIrcListHelper(this)),
  75     _messageModel(0),
  76     _messageProcessor(0),
  77     _connectedToCore(false),
  78     _syncedToCore(false)
  79 {
  80   _monitorBuffer = new Buffer(BufferInfo(), this);
  81   _signalProxy->synchronize(_ircListHelper);
  82
  83   connect(_backlogManager, SIGNAL(backlog(BufferId, const QVariantList &)),
  84           this, SLOT(receiveBacklog(BufferId, const QVariantList &)));
  85 }
  86
  87 Client::~Client() {
  88   disconnectFromCore();
  89 }
  90
  91 void Client::init() {
  92   _currentCoreAccount = 0;
  93   _networkModel = new NetworkModel(this);
  94
  95   connect(this, SIGNAL(bufferUpdated(BufferInfo)),
  96           _networkModel, SLOT(bufferUpdated(BufferInfo)));
  97   connect(this, SIGNAL(networkRemoved(NetworkId)),
  98           _networkModel, SLOT(networkRemoved(NetworkId)));
  99
 100   _bufferModel = new BufferModel(_networkModel);
 101   _messageModel = mainUi->createMessageModel(this);
 102   _messageProcessor = mainUi->createMessageProcessor(this);
 103
 104   SignalProxy *p = signalProxy();
 105
 106   p->attachSlot(SIGNAL(displayMsg(const Message &)), this, SLOT(recvMessage(const Message &)));
 107   p->attachSlot(SIGNAL(displayStatusMsg(QString, QString)), this, SLOT(recvStatusMsg(QString, QString)));
 108
 109   p->attachSlot(SIGNAL(bufferInfoUpdated(BufferInfo)), this, SLOT(updateBufferInfo(BufferInfo)));
 110   p->attachSignal(this, SIGNAL(sendInput(BufferInfo, QString)));
 111   p->attachSignal(this, SIGNAL(requestNetworkStates()));
 112
 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)));
 118
 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)));
 124
 125   connect(p, SIGNAL(disconnected()), this, SLOT(disconnectFromCore()));
 126
 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()));
 131
 132 }
 133
 134 /*** public static methods ***/
 135
 136 AccountId Client::currentCoreAccount() {
 137   return _currentCoreAccount;
 138 }
 139
 140 void Client::setCurrentCoreAccount(AccountId id) {
 141   _currentCoreAccount = id;
 142 }
 143
 144 QList<BufferInfo> Client::allBufferInfos() {
 145   QList<BufferInfo> bufferids;
 146   foreach(Buffer *buffer, buffers()) {
 147     bufferids << buffer->bufferInfo();
 148   }
 149   return bufferids;
 150 }
 151
 152 QList<Buffer *> Client::buffers() {
 153   return instance()->_buffers.values();
 154 }
 155
 156
 157 Buffer *Client::statusBuffer(const NetworkId &networkId) const {
 158   if(_statusBuffers.contains(networkId))
 159     return _statusBuffers[networkId];
 160   else
 161     return 0;
 162 }
 163
 164 Buffer *Client::buffer(BufferId bufferId) {
 165   if(instance()->_buffers.contains(bufferId))
 166     return instance()->_buffers[bufferId];
 167   else
 168     return 0;
 169 }
 170
 171 Buffer *Client::buffer(BufferInfo bufferInfo) {
 172   Buffer *buff = buffer(bufferInfo.bufferId());
 173
 174   if(!buff) {
 175     Client *client = Client::instance();
 176     buff = new Buffer(bufferInfo, client);
 177     connect(buff, SIGNAL(destroyed()), client, SLOT(bufferDestroyed()));
 178     client->_buffers[bufferInfo.bufferId()] = buff;
 179     if(bufferInfo.type() == BufferInfo::StatusBuffer)
 180       client->_statusBuffers[bufferInfo.networkId()] = buff;
 181
 182     emit client->bufferUpdated(bufferInfo);
 183
 184     // I don't like this: but currently there isn't really a prettier way:
 185     if(isSynced()) {  // this slows down syncing a lot, so disable it during sync
 186       QModelIndex bufferIdx = networkModel()->bufferIndex(bufferInfo.bufferId());
 187       bufferModel()->setCurrentIndex(bufferModel()->mapFromSource(bufferIdx));
 188     }
 189   }
 190   Q_ASSERT(buff);
 191   return buff;
 192 }
 193
 194 bool Client::isConnected() {
 195   return instance()->_connectedToCore;
 196 }
 197
 198 bool Client::isSynced() {
 199   return instance()->_syncedToCore;
 200 }
 201
 202 /*** Network handling ***/
 203
 204 QList<NetworkId> Client::networkIds() {
 205   return instance()->_networks.keys();
 206 }
 207
 208 const Network * Client::network(NetworkId networkid) {
 209   if(instance()->_networks.contains(networkid)) return instance()->_networks[networkid];
 210   else return 0;
 211 }
 212
 213 void Client::createNetwork(const NetworkInfo &info) {
 214   emit instance()->requestCreateNetwork(info);
 215 }
 216
 217 void Client::updateNetwork(const NetworkInfo &info) {
 218   emit instance()->requestUpdateNetwork(info);
 219 }
 220
 221 void Client::removeNetwork(NetworkId id) {
 222   emit instance()->requestRemoveNetwork(id);
 223 }
 224
 225 void Client::addNetwork(Network *net) {
 226   net->setProxy(signalProxy());
 227   signalProxy()->synchronize(net);
 228   networkModel()->attachNetwork(net);
 229   connect(net, SIGNAL(destroyed()), instance(), SLOT(networkDestroyed()));
 230   instance()->_networks[net->networkId()] = net;
 231   emit instance()->networkCreated(net->networkId());
 232 }
 233
 234 void Client::coreNetworkCreated(NetworkId id) {
 235   if(_networks.contains(id)) {
 236     qWarning() << "Creation of already existing network requested!";
 237     return;
 238   }
 239   Network *net = new Network(id, this);
 240   addNetwork(net);
 241 }
 242
 243 void Client::coreNetworkRemoved(NetworkId id) {
 244   if(!_networks.contains(id))
 245     return;
 246   Network *net = _networks.take(id);
 247   emit networkRemoved(net->networkId());
 248   net->deleteLater();
 249 }
 250
 251 /*** Identity handling ***/
 252
 253 QList<IdentityId> Client::identityIds() {
 254   return instance()->_identities.keys();
 255 }
 256
 257 const Identity * Client::identity(IdentityId id) {
 258   if(instance()->_identities.contains(id)) return instance()->_identities[id];
 259   else return 0;
 260 }
 261
 262 void Client::createIdentity(const Identity &id) {
 263   emit instance()->requestCreateIdentity(id);
 264 }
 265
 266 void Client::updateIdentity(const Identity &id) {
 267   emit instance()->requestUpdateIdentity(id);
 268 }
 269
 270 void Client::removeIdentity(IdentityId id) {
 271   emit instance()->requestRemoveIdentity(id);
 272 }
 273
 274 void Client::coreIdentityCreated(const Identity &other) {
 275   if(!_identities.contains(other.id())) {
 276     Identity *identity = new Identity(other, this);
 277     _identities[other.id()] = identity;
 278     identity->setInitialized();
 279     signalProxy()->synchronize(identity);
 280     emit identityCreated(other.id());
 281   } else {
 282     qWarning() << tr("Identity already exists in client!");
 283   }
 284 }
 285
 286 void Client::coreIdentityRemoved(IdentityId id) {
 287   if(_identities.contains(id)) {
 288     emit identityRemoved(id);
 289     Identity *i = _identities.take(id);
 290     i->deleteLater();
 291   }
 292 }
 293
 294 /***  ***/
 295 void Client::userInput(BufferInfo bufferInfo, QString message) {
 296   emit instance()->sendInput(bufferInfo, message);
 297 }
 298
 299 /*** core connection stuff ***/
 300
 301 void Client::setConnectedToCore(QIODevice *sock, AccountId id) {
 302   socket = sock;
 303   signalProxy()->addPeer(socket);
 304   _connectedToCore = true;
 305   setCurrentCoreAccount(id);
 306 }
 307
 308 void Client::setSyncedToCore() {
 309   // create buffersyncer
 310   Q_ASSERT(!_bufferSyncer);
 311   _bufferSyncer = new BufferSyncer(this);
 312   connect(bufferSyncer(), SIGNAL(lastSeenMsgSet(BufferId, MsgId)), this, SLOT(updateLastSeenMsg(BufferId, MsgId)));
 313   connect(bufferSyncer(), SIGNAL(bufferRemoved(BufferId)), this, SLOT(bufferRemoved(BufferId)));
 314   connect(bufferSyncer(), SIGNAL(bufferRenamed(BufferId, QString)), this, SLOT(bufferRenamed(BufferId, QString)));
 315   signalProxy()->synchronize(bufferSyncer());
 316
 317   // attach backlog manager
 318   signalProxy()->synchronize(backlogManager());
 319
 320   // create a new BufferViewManager
 321   _bufferViewManager = new BufferViewManager(signalProxy(), this);
 322
 323   _syncedToCore = true;
 324   emit connected();
 325   emit coreConnectionStateChanged(true);
 326 }
 327
 328 void Client::setSecuredConnection() {
 329   emit securedConnection();
 330 }
 331
 332 void Client::disconnectFromCore() {
 333   if(!isConnected())
 334     return;
 335   _connectedToCore = false;
 336
 337   if(socket) {
 338     socket->close();
 339     socket->deleteLater();
 340   }
 341   _syncedToCore = false;
 342   emit disconnected();
 343   emit coreConnectionStateChanged(false);
 344
 345   // Clear internal data. Hopefully nothing relies on it at this point.
 346   setCurrentCoreAccount(0);
 347
 348   if(_bufferSyncer) {
 349     _bufferSyncer->deleteLater();
 350     _bufferSyncer = 0;
 351   }
 352
 353   if(_bufferViewManager) {
 354     _bufferViewManager->deleteLater();
 355     _bufferViewManager = 0;
 356   }
 357
 358   _messageModel->clear();
 359   _networkModel->clear();
 360
 361   QHash<BufferId, Buffer *>::iterator bufferIter =  _buffers.begin();
 362   while(bufferIter != _buffers.end()) {
 363     Buffer *buffer = bufferIter.value();
 364     disconnect(buffer, SIGNAL(destroyed()), this, 0);
 365     bufferIter = _buffers.erase(bufferIter);
 366     buffer->deleteLater();
 367   }
 368   Q_ASSERT(_buffers.isEmpty());
 369
 370   _statusBuffers.clear();
 371
 372   QHash<NetworkId, Network*>::iterator netIter = _networks.begin();
 373   while(netIter != _networks.end()) {
 374     Network *net = netIter.value();
 375     emit networkRemoved(net->networkId());
 376     disconnect(net, SIGNAL(destroyed()), this, 0);
 377     netIter = _networks.erase(netIter);
 378     net->deleteLater();
 379   }
 380   Q_ASSERT(_networks.isEmpty());
 381
 382   QHash<IdentityId, Identity*>::iterator idIter = _identities.begin();
 383   while(idIter != _identities.end()) {
 384     Identity *id = idIter.value();
 385     emit identityRemoved(id->id());
 386     idIter = _identities.erase(idIter);
 387     id->deleteLater();
 388   }
 389   Q_ASSERT(_identities.isEmpty());
 390
 391 }
 392
 393 void Client::setCoreConfiguration(const QVariantMap &settings) {
 394   SignalProxy::writeDataToDevice(socket, settings);
 395 }
 396
 397 /*** ***/
 398
 399 void Client::updateBufferInfo(BufferInfo id) {
 400   emit bufferUpdated(id);
 401 }
 402
 403 void Client::bufferDestroyed() {
 404   Buffer *buffer = static_cast<Buffer *>(sender());
 405   QHash<BufferId, Buffer *>::iterator iter = _buffers.begin();
 406   while(iter != _buffers.end()) {
 407     if(iter.value() == buffer) {
 408       iter = _buffers.erase(iter);
 409       break;
 410     }
 411     iter++;
 412   }
 413
 414   QHash<NetworkId, Buffer *>::iterator statusIter = _statusBuffers.begin();
 415   while(statusIter != _statusBuffers.end()) {
 416     if(statusIter.value() == buffer) {
 417       statusIter = _statusBuffers.erase(statusIter);
 418       break;
 419     }
 420     statusIter++;
 421   }
 422 }
 423
 424 void Client::networkDestroyed() {
 425   Network *net = static_cast<Network *>(sender());
 426   QHash<NetworkId, Network *>::iterator netIter = _networks.begin();
 427   while(netIter != _networks.end()) {
 428     if(*netIter == net) {
 429       netIter = _networks.erase(netIter);
 430       break;
 431     } else {
 432       netIter++;
 433     }
 434   }
 435 }
 436
 437 // Hmm... we never used this...
 438 void Client::recvStatusMsg(QString /*net*/, QString /*msg*/) {
 439   //recvMessage(net, Message::server("", QString("[STATUS] %1").arg(msg)));
 440 }
 441
 442 void Client::recvMessage(const Message &msg_) {
 443   Message msg = msg_;
 444   messageProcessor()->process(msg);
 445 }
 446
 447 void Client::receiveBacklog(BufferId bufferId, const QVariantList &msgs) {
 448   //QTime start = QTime::currentTime();
 449   QList<Message> msglist;
 450   foreach(QVariant v, msgs) {
 451     msglist << v.value<Message>();
 452   }
 453   messageProcessor()->process(msglist);
 454   //qDebug() << "processed" << msgs.count() << "backlog lines in" << start.msecsTo(QTime::currentTime());
 455 }
 456
 457 void Client::updateLastSeenMsg(BufferId id, const MsgId &msgId) {
 458   Buffer *b = buffer(id);
 459   if(!b) {
 460     qWarning() << "Client::updateLastSeen(): Unknown buffer" << id;
 461     return;
 462   }
 463   b->setLastSeenMsg(msgId);
 464 }
 465
 466 void Client::setBufferLastSeenMsg(BufferId id, const MsgId &msgId) {
 467   if(!bufferSyncer())
 468     return;
 469   bufferSyncer()->requestSetLastSeenMsg(id, msgId);
 470 }
 471
 472 void Client::removeBuffer(BufferId id) {
 473   if(!bufferSyncer()) return;
 474   bufferSyncer()->requestRemoveBuffer(id);
 475 }
 476
 477 void Client::bufferRemoved(BufferId bufferId) {
 478   // first remove the buffer from hash. this prohibits further lastSeenUpdates
 479   Buffer *buff = 0;
 480   if(_buffers.contains(bufferId)) {
 481     buff = _buffers.take(bufferId);
 482     disconnect(buff, 0, this, 0);
 483   }
 484
 485   // then we select a sane buffer (status buffer)
 486   /* we have to manually select a buffer because otherwise inconsitent changes
 487    * to the model might occur:
 488    * the result of a buffer removal triggers a change in the selection model.
 489    * the newly selected buffer might be a channel that hasn't been selected yet
 490    * and a new nickview would be created (which never heard of the "rowsAboutToBeRemoved").
 491    * this new view (and/or) its sort filter will then only receive a "rowsRemoved" signal.
 492    */
 493   QModelIndex current = bufferModel()->currentIndex();
 494   if(current.data(NetworkModel::BufferIdRole).value<BufferId>() == bufferId) {
 495     bufferModel()->setCurrentIndex(current.sibling(0,0));
 496   }
 497
 498   // and remove it from the model
 499   networkModel()->removeBuffer(bufferId);
 500
 501   if(buff)
 502     buff->deleteLater();
 503 }
 504
 505 void Client::bufferRenamed(BufferId bufferId, const QString &newName) {
 506   QModelIndex bufferIndex = networkModel()->bufferIndex(bufferId);
 507   if(bufferIndex.isValid()) {
 508     networkModel()->setData(bufferIndex, newName, Qt::DisplayRole);
 509   }
 510 }