src/client/client.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-07 by The Quassel Team                             *
   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) any later version.                                   *
   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 "networkinfo.h"
  24 #include "ircuser.h"
  25 #include "ircchannel.h"
  26
  27 #include "message.h"
  28
  29 #include "bufferinfo.h"
  30 #include "buffertreemodel.h"
  31 #include "quasselui.h"
  32 #include "signalproxy.h"
  33 #include "synchronizer.h"
  34 #include "util.h"
  35
  36 QPointer<Client> Client::instanceptr = 0;
  37
  38 // ==============================
  39 //  public Static Methods
  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 }
  50
  51 void Client::init(AbstractUi *ui) {
  52   instance()->mainUi = ui;
  53   instance()->init();
  54 }
  55
  56 QList<NetworkInfo *> Client::networkInfos() {
  57   return instance()->_networkInfo.values();
  58 }
  59
  60 NetworkInfo *Client::networkInfo(uint networkid) {
  61   if(instance()->_networkInfo.contains(networkid))
  62     return instance()->_networkInfo[networkid];
  63   else
  64     return 0;
  65 }
  66
  67 QList<BufferInfo> Client::allBufferInfos() {
  68   QList<BufferInfo> bufferids;
  69   foreach(Buffer *buffer, buffers()) {
  70     bufferids << buffer->bufferInfo();
  71   }
  72   return bufferids;
  73 }
  74
  75 QList<Buffer *> Client::buffers() {
  76   return instance()->_buffers.values();
  77 }
  78
  79 Buffer *Client::buffer(uint bufferUid) {
  80   if(instance()->_buffers.contains(bufferUid))
  81     return instance()->_buffers[bufferUid];
  82   else
  83     return 0;
  84 }
  85
  86 Buffer *Client::buffer(BufferInfo id) {
  87   Buffer *buff = buffer(id.uid());
  88
  89   if(!buff) {
  90     Client *client = Client::instance();
  91     buff = new Buffer(id, client);
  92
  93     connect(buff, SIGNAL(userInput(BufferInfo, QString)),
  94             client, SLOT(userInput(BufferInfo, QString)));
  95     connect(buff, SIGNAL(bufferUpdated(Buffer *)),
  96             client, SIGNAL(bufferUpdated(Buffer *)));
  97     connect(buff, SIGNAL(destroyed()),
  98             client, SLOT(bufferDestroyed()));
  99     client->_buffers[id.uid()] = buff;
 100     emit client->bufferUpdated(buff);
 101   }
 102   Q_ASSERT(buff);
 103   return buff;
 104 }
 105
 106 // FIXME switch to netids!
 107 // WHEN IS THIS NEEDED ANYHOW!?
 108 BufferInfo Client::bufferInfo(QString net, QString buf) {
 109   foreach(Buffer *buffer_, buffers()) {
 110     BufferInfo bufferInfo = buffer_->bufferInfo();
 111     if(bufferInfo.network() == net && bufferInfo.buffer() == buf)
 112       return bufferInfo;
 113   }
 114   Q_ASSERT(false);  // should never happen!
 115   return BufferInfo();
 116 }
 117
 118 BufferInfo Client::statusBufferInfo(QString net) {
 119   return bufferInfo(net, "");
 120 }
 121
 122 BufferTreeModel *Client::bufferModel() {
 123   return instance()->_bufferModel;
 124 }
 125
 126 SignalProxy *Client::signalProxy() {
 127   return instance()->_signalProxy;
 128 }
 129
 130 // ==============================
 131 //  Constructor / Decon
 132 // ==============================
 133 Client::Client(QObject *parent)
 134   : QObject(parent),
 135     socket(0),
 136     _signalProxy(new SignalProxy(SignalProxy::Client, this)),
 137     mainUi(0),
 138     _bufferModel(0),
 139     connectedToCore(false)
 140 {
 141 }
 142
 143 Client::~Client() {
 144 }
 145
 146 void Client::init() {
 147   blockSize = 0;
 148
 149   _bufferModel = new BufferTreeModel(this);
 150
 151   connect(this, SIGNAL(bufferSelected(Buffer *)),
 152           _bufferModel, SLOT(selectBuffer(Buffer *)));
 153   connect(this, SIGNAL(bufferUpdated(Buffer *)),
 154           _bufferModel, SLOT(bufferUpdated(Buffer *)));
 155   connect(this, SIGNAL(bufferActivity(Buffer::ActivityLevel, Buffer *)),
 156           _bufferModel, SLOT(bufferActivity(Buffer::ActivityLevel, Buffer *)));
 157
 158   SignalProxy *p = signalProxy();
 159   p->attachSignal(this, SIGNAL(sendSessionData(const QString &, const QVariant &)),
 160                   SIGNAL(clientSessionDataChanged(const QString &, const QVariant &)));
 161   p->attachSlot(SIGNAL(coreSessionDataChanged(const QString &, const QVariant &)),
 162                 this, SLOT(recvSessionData(const QString &, const QVariant &)));
 163   p->attachSlot(SIGNAL(coreState(const QVariant &)),
 164                 this, SLOT(recvCoreState(const QVariant &)));
 165   p->attachSlot(SIGNAL(networkConnected(uint)),
 166                 this, SLOT(networkConnected(uint)));
 167   p->attachSlot(SIGNAL(networkDisconnected(uint)),
 168                 this, SLOT(networkDisconnected(uint)));
 169   p->attachSlot(SIGNAL(displayMsg(const Message &)),
 170                 this, SLOT(recvMessage(const Message &)));
 171   p->attachSlot(SIGNAL(displayStatusMsg(QString, QString)),
 172                 this, SLOT(recvStatusMsg(QString, QString)));
 173
 174
 175   p->attachSlot(SIGNAL(backlogData(BufferInfo, const QVariantList &, bool)), this, SLOT(recvBacklogData(BufferInfo, const QVariantList &, bool)));
 176   p->attachSlot(SIGNAL(bufferInfoUpdated(BufferInfo)), this, SLOT(updateBufferInfo(BufferInfo)));
 177   p->attachSignal(this, SIGNAL(sendInput(BufferInfo, QString)));
 178   p->attachSignal(this, SIGNAL(requestNetworkStates()));
 179
 180   connect(mainUi, SIGNAL(connectToCore(const QVariantMap &)), this, SLOT(connectToCore(const QVariantMap &)));
 181   connect(mainUi, SIGNAL(disconnectFromCore()), this, SLOT(disconnectFromCore()));
 182   connect(this, SIGNAL(connected()), mainUi, SLOT(connectedToCore()));
 183   connect(this, SIGNAL(disconnected()), mainUi, SLOT(disconnectedFromCore()));
 184
 185   layoutTimer = new QTimer(this);
 186   layoutTimer->setInterval(0);
 187   layoutTimer->setSingleShot(false);
 188   connect(layoutTimer, SIGNAL(timeout()), this, SLOT(layoutMsg()));
 189
 190 }
 191
 192 bool Client::isConnected() {
 193   return instance()->connectedToCore;
 194 }
 195
 196 void Client::fakeInput(uint bufferUid, QString message) {
 197   Buffer *buff = buffer(bufferUid);
 198   if(!buff)
 199     qWarning() << "No Buffer with uid" << bufferUid << "can't send Input" << message;
 200   else
 201     emit instance()->sendInput(buff->bufferInfo(), message);
 202 }
 203
 204 void Client::fakeInput(BufferInfo bufferInfo, QString message) {
 205   fakeInput(bufferInfo, message);
 206 }
 207
 208 void Client::connectToCore(const QVariantMap &conn) {
 209   // TODO implement SSL
 210   coreConnectionInfo = conn;
 211   if(isConnected()) {
 212     emit coreConnectionError(tr("Already connected to Core!"));
 213     return;
 214   }
 215
 216   if(socket != 0)
 217     socket->deleteLater();
 218
 219   if(conn["Host"].toString().isEmpty()) {
 220     clientMode = LocalCore;
 221     socket = new QBuffer(this);
 222     connect(socket, SIGNAL(readyRead()), this, SLOT(coreHasData()));
 223     socket->open(QIODevice::ReadWrite);
 224     //QVariant state = connectToLocalCore(coreConnectionInfo["User"].toString(), coreConnectionInfo["Password"].toString());
 225     //syncToCore(state);
 226     coreSocketConnected();
 227   } else {
 228     clientMode = RemoteCore;
 229     emit coreConnectionMsg(tr("Connecting..."));
 230     Q_ASSERT(!socket);
 231     QTcpSocket *sock = new QTcpSocket(this);
 232     socket = sock;
 233     connect(sock, SIGNAL(readyRead()), this, SLOT(coreHasData()));
 234     connect(sock, SIGNAL(connected()), this, SLOT(coreSocketConnected()));
 235     connect(sock, SIGNAL(disconnected()), this, SLOT(coreSocketDisconnected()));
 236     connect(signalProxy(), SIGNAL(disconnected()), this, SLOT(coreSocketDisconnected()));
 237     //connect(sock, SIGNAL(stateChanged(QAbstractSocket::SocketState)), this, SLOT(coreSocketStateChanged(QAbstractSocket::SocketState)));
 238     connect(sock, SIGNAL(error(QAbstractSocket::SocketError)), this, SLOT(coreSocketError(QAbstractSocket::SocketError)));
 239     sock->connectToHost(conn["Host"].toString(), conn["Port"].toUInt());
 240   }
 241 }
 242
 243 void Client::disconnectFromCore() {
 244   socket->close();
 245   if(clientMode == LocalCore) {
 246     coreSocketDisconnected();
 247   }
 248 }
 249
 250 void Client::coreSocketConnected() {
 251   connect(this, SIGNAL(recvPartialItem(uint, uint)), this, SIGNAL(coreConnectionProgress(uint, uint)));
 252   emit coreConnectionMsg(tr("Synchronizing to core..."));
 253   QVariantMap clientInit;
 254   clientInit["GuiProtocol"] = GUI_PROTOCOL;
 255   clientInit["User"] = coreConnectionInfo["User"].toString();
 256   clientInit["Password"] = coreConnectionInfo["Password"].toString();
 257   writeDataToDevice(socket, clientInit);
 258 }
 259
 260 void Client::coreSocketDisconnected() {
 261   instance()->connectedToCore = false;
 262   emit disconnected();
 263   socket->deleteLater();
 264   blockSize = 0;
 265
 266   /* Clear internal data. Hopefully nothing relies on it at this point. */
 267   _bufferModel->clear();
 268   foreach(Buffer *buffer, _buffers.values()) {
 269     buffer->deleteLater();
 270   }
 271   _buffers.clear();
 272
 273   foreach(NetworkInfo *networkinfo, _networkInfo.values()) {
 274     networkinfo->deleteLater();
 275   }
 276   _networkInfo.clear();
 277
 278   coreConnectionInfo.clear();
 279   sessionData.clear();
 280   layoutQueue.clear();
 281   layoutTimer->stop();
 282 }
 283
 284 void Client::coreSocketStateChanged(QAbstractSocket::SocketState state) {
 285   if(state == QAbstractSocket::UnconnectedState) coreSocketDisconnected();
 286 }
 287
 288 void Client::recvCoreState(const QVariant &state) {
 289   disconnect(this, SIGNAL(recvPartialItem(uint, uint)), this, SIGNAL(coreConnectionProgress(uint, uint)));
 290   disconnect(socket, 0, this, 0);  // rest of communication happens through SignalProxy
 291   signalProxy()->addPeer(socket);
 292   syncToCore(state);
 293 }
 294
 295 // TODO: auth errors
 296 void Client::syncToCore(const QVariant &coreState) {
 297   if(!coreState.toMap().contains("SessionState")) {
 298     emit coreConnectionError(tr("Invalid data received from core!"));
 299     disconnectFromCore();
 300     return;
 301   }
 302
 303   QVariantMap sessionState = coreState.toMap()["SessionState"].toMap();
 304
 305   // store sessionData
 306   QVariantMap sessData = sessionState["SessionData"].toMap();
 307   foreach(QString key, sessData.keys())
 308     recvSessionData(key, sessData[key]);
 309
 310   // store Buffer details
 311   QVariantList coreBuffers = sessionState["Buffers"].toList();
 312   /* make lookups by id faster */
 313   foreach(QVariant vid, coreBuffers) {
 314     buffer(vid.value<BufferInfo>()); // create all buffers, so we see them in the network views
 315   }
 316
 317   // create networkInfo objects
 318   QVariantList networkids = sessionState["Networks"].toList();
 319   foreach(QVariant networkid, networkids) {
 320     networkConnected(networkid.toUInt());
 321   }
 322
 323   instance()->connectedToCore = true;
 324   updateCoreConnectionProgress();
 325 }
 326
 327 void Client::updateCoreConnectionProgress() {
 328   // we'll do this in three steps:
 329   // 1.) networks
 330   // 2.) channels
 331   // 3.) ircusers
 332
 333   int numNets = networkInfos().count();
 334   int numNetsWaiting = 0;
 335
 336   int numIrcUsers = 0;
 337   int numIrcUsersWaiting = 0;
 338
 339   int numChannels = 0;
 340   int numChannelsWaiting = 0;
 341
 342   foreach(NetworkInfo *net, networkInfos()) {
 343     if(! net->initialized())
 344       numNetsWaiting++;
 345
 346     numIrcUsers += net->ircUsers().count();
 347     foreach(IrcUser *user, net->ircUsers()) {
 348       if(! user->initialized())
 349         numIrcUsersWaiting++;
 350     }
 351
 352     numChannels += net->ircChannels().count();
 353     foreach(IrcChannel *channel, net->ircChannels()) {
 354       if(! channel->initialized())
 355         numChannelsWaiting++;
 356     }
 357   }
 358
 359   if(numNetsWaiting > 0) {
 360     emit coreConnectionMsg(tr("Requesting network states..."));
 361     emit coreConnectionProgress(numNets - numNetsWaiting, numNets);
 362     return;
 363   }
 364
 365   if(numIrcUsersWaiting > 0) {
 366     emit coreConnectionMsg(tr("Requesting User states..."));
 367     emit coreConnectionProgress(numIrcUsers - numIrcUsersWaiting, numIrcUsers);
 368     return;
 369   }
 370
 371   if(numChannelsWaiting > 0) {
 372     emit coreConnectionMsg(tr("Requesting Channel states..."));
 373     emit coreConnectionProgress(numChannels - numChannelsWaiting, numChannels);
 374     return;
 375   }
 376
 377   emit coreConnectionProgress(1,1);
 378   emit connected();
 379
 380   foreach(NetworkInfo *net, networkInfos()) {
 381     disconnect(net, 0, this, SLOT(updateCoreConnectionProgress()));
 382   }
 383
 384   // signalProxy()->dumpProxyStats();
 385 }
 386
 387 void Client::recvSessionData(const QString &key, const QVariant &data) {
 388   sessionData[key] = data;
 389   emit sessionDataChanged(key, data);
 390   emit sessionDataChanged(key);
 391 }
 392
 393 void Client::storeSessionData(const QString &key, const QVariant &data) {
 394   // Not sure if this is a good idea, but we'll try it anyway:
 395   // Calling this function only sends a signal to core. Data is stored upon reception of the update signal,
 396   // rather than immediately.
 397   emit instance()->sendSessionData(key, data);
 398 }
 399
 400 QVariant Client::retrieveSessionData(const QString &key, const QVariant &def) {
 401   if(instance()->sessionData.contains(key)) return instance()->sessionData[key];
 402   else return def;
 403 }
 404
 405 QStringList Client::sessionDataKeys() {
 406   return instance()->sessionData.keys();
 407 }
 408
 409 void Client::coreSocketError(QAbstractSocket::SocketError) {
 410   emit coreConnectionError(socket->errorString());
 411   socket->deleteLater();
 412 }
 413
 414 void Client::coreHasData() {
 415   QVariant item;
 416   if(readDataFromDevice(socket, blockSize, item)) {
 417     emit recvPartialItem(1,1);
 418     recvCoreState(item);
 419     blockSize = 0;
 420     return;
 421   }
 422   if(blockSize > 0) {
 423     emit recvPartialItem(socket->bytesAvailable(), blockSize);
 424   }
 425 }
 426
 427 void Client::networkConnected(uint netid) {
 428   // TODO: create statusBuffer / switch to networkids
 429   //BufferInfo id = statusBufferInfo(net);
 430   //Buffer *b = buffer(id);
 431   //b->setActive(true);
 432
 433   NetworkInfo *netinfo = new NetworkInfo(netid, signalProxy(), this);
 434   connect(netinfo, SIGNAL(initDone()), this, SLOT(updateCoreConnectionProgress()));
 435   connect(netinfo, SIGNAL(ircUserInitDone()), this, SLOT(updateCoreConnectionProgress()));
 436   connect(netinfo, SIGNAL(ircChannelInitDone()), this, SLOT(updateCoreConnectionProgress()));
 437   connect(netinfo, SIGNAL(destroyed()), this, SLOT(networkInfoDestroyed()));
 438   _networkInfo[netid] = netinfo;
 439 }
 440
 441 void Client::networkDisconnected(uint networkid) {
 442   foreach(Buffer *buffer, buffers()) {
 443     if(buffer->bufferInfo().networkId() != networkid)
 444       continue;
 445
 446     //buffer->displayMsg(Message(bufferid, Message::Server, tr("Server disconnected."))); FIXME
 447     buffer->setActive(false);
 448   }
 449
 450   Q_ASSERT(networkInfo(networkid));
 451   if(!networkInfo(networkid)->initialized()) {
 452     qDebug() << "Network" << networkid << "disconnected while not yet initialized!";
 453     updateCoreConnectionProgress();
 454   }
 455 }
 456
 457 void Client::updateBufferInfo(BufferInfo id) {
 458   buffer(id)->updateBufferInfo(id);
 459 }
 460
 461 void Client::bufferDestroyed() {
 462   Buffer *buffer = static_cast<Buffer *>(sender());
 463   uint bufferUid = buffer->uid();
 464   if(_buffers.contains(bufferUid))
 465     _buffers.remove(bufferUid);
 466 }
 467
 468 void Client::networkInfoDestroyed() {
 469   NetworkInfo *netinfo = static_cast<NetworkInfo *>(sender());
 470   uint networkId = netinfo->networkId();
 471   if(_networkInfo.contains(networkId))
 472     _networkInfo.remove(networkId);
 473 }
 474
 475 void Client::recvMessage(const Message &msg) {
 476   Buffer *b = buffer(msg.buffer());
 477
 478   Buffer::ActivityLevel level = Buffer::OtherActivity;
 479   if(msg.type() == Message::Plain || msg.type() == Message::Notice){
 480     level |= Buffer::NewMessage;
 481   }
 482   if(msg.flags() & Message::Highlight){
 483     level |= Buffer::Highlight;
 484   }
 485   emit bufferActivity(level, b);
 486
 487   b->appendMsg(msg);
 488 }
 489
 490 void Client::recvStatusMsg(QString /*net*/, QString /*msg*/) {
 491   //recvMessage(net, Message::server("", QString("[STATUS] %1").arg(msg)));
 492 }
 493
 494 void Client::recvBacklogData(BufferInfo id, QVariantList msgs, bool /*done*/) {
 495   Buffer *b = buffer(id);
 496   foreach(QVariant v, msgs) {
 497     Message msg = v.value<Message>();
 498     b->prependMsg(msg);
 499     if(!layoutQueue.contains(b)) layoutQueue.append(b);
 500   }
 501   if(layoutQueue.count() && !layoutTimer->isActive()) layoutTimer->start();
 502 }
 503
 504 void Client::layoutMsg() {
 505   if(layoutQueue.count()) {
 506     Buffer *b = layoutQueue.takeFirst();  // TODO make this the current buffer
 507     if(b->layoutMsg())
 508       layoutQueue.append(b);  // Buffer has more messages in its queue --> Round Robin
 509   }
 510
 511   if(!layoutQueue.count())
 512     layoutTimer->stop();
 513 }
 514
 515 AbstractUiMsg *Client::layoutMsg(const Message &msg) {
 516   return instance()->mainUi->layoutMsg(msg);
 517 }
 518
 519 void Client::userInput(BufferInfo id, QString msg) {
 520   emit sendInput(id, msg);
 521 }
 522