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