src/common/networkinfo.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-07 by the Quassel IRC 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) 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 #include "networkinfo.h"
  21
  22 #include "signalproxy.h"
  23 #include "ircuser.h"
  24 #include "ircchannel.h"
  25
  26 #include <QDebug>
  27
  28 #include "util.h"
  29
  30 // ====================
  31 //  Public:
  32 // ====================
  33 NetworkInfo::NetworkInfo(const uint &networkid, QObject *parent)
  34   : QObject(parent),
  35     _networkId(networkid),
  36     _initialized(false),
  37     _myNick(QString()),
  38     _networkName(QString()),
  39     _currentServer(QString()),
  40     _prefixes(QString()),
  41     _prefixModes(QString()),
  42     _proxy(NULL)
  43 {
  44   setObjectName(QString::number(networkid));
  45 }
  46
  47 // I think this is unnecessary since IrcUsers have us as their daddy :)
  48 //NetworkInfo::~NetworkInfo() {
  49 //   QHashIterator<QString, IrcUser *> ircuser(_ircUsers);
  50 //   while (ircuser.hasNext()) {
  51 //     ircuser.next();
  52 //     delete ircuser.value();
  53 //   }
  54 //}
  55
  56 uint NetworkInfo::networkId() const {
  57   return _networkId;
  58 }
  59
  60 bool NetworkInfo::initialized() const {
  61   return _initialized;
  62 }
  63
  64 SignalProxy *NetworkInfo::proxy() const {
  65   return _proxy;
  66 }
  67
  68 void NetworkInfo::setProxy(SignalProxy *proxy) {
  69   _proxy = proxy;
  70   proxy->synchronize(this);
  71 }
  72
  73 bool NetworkInfo::isMyNick(const QString &nick) const {
  74   return (myNick().toLower() == nick.toLower());
  75 }
  76
  77 bool NetworkInfo::isMyNick(IrcUser *ircuser) const {
  78   return (ircuser->nick().toLower() == myNick());
  79 }
  80
  81 bool NetworkInfo::isChannelName(const QString &channelname) const {
  82   if(channelname.isEmpty())
  83     return false;
  84
  85   if(supports("CHANTYPES"))
  86     return support("CHANTYPES").contains(channelname[0]);
  87   else
  88     return QString("#&!+").contains(channelname[0]);
  89 }
  90
  91 QString NetworkInfo::prefixToMode(const QString &prefix) {
  92   if(prefixes().contains(prefix))
  93     return QString(prefixModes()[prefixes().indexOf(prefix)]);
  94   else
  95     return QString();
  96 }
  97
  98 QString NetworkInfo::prefixToMode(const QCharRef &prefix) {
  99   return prefixToMode(QString(prefix));
 100 }
 101
 102 QString NetworkInfo::modeToPrefix(const QString &mode) {
 103   if(prefixModes().contains(mode))
 104     return QString(prefixes()[prefixModes().indexOf(mode)]);
 105   else
 106     return QString();
 107 }
 108
 109 QString NetworkInfo::modeToPrefix(const QCharRef &mode) {
 110   return modeToPrefix(QString(mode));
 111 }
 112
 113 QString NetworkInfo::networkName() const {
 114   return _networkName;
 115 }
 116
 117 QString NetworkInfo::currentServer() const {
 118   return _currentServer;
 119 }
 120
 121 QString NetworkInfo::myNick() const {
 122   return _myNick;
 123 }
 124
 125 QStringList NetworkInfo::nicks() const {
 126   // we don't use _ircUsers.keys() since the keys may be
 127   // not up to date after a nick change
 128   QStringList nicks;
 129   foreach(IrcUser *ircuser, _ircUsers.values()) {
 130     nicks << ircuser->nick();
 131   }
 132   return nicks;
 133 }
 134
 135 QStringList NetworkInfo::channels() const {
 136   return _ircChannels.keys();
 137 }
 138
 139 QString NetworkInfo::prefixes() {
 140   if(_prefixes.isNull())
 141     determinePrefixes();
 142
 143   return _prefixes;
 144 }
 145
 146 QString NetworkInfo::prefixModes() {
 147   if(_prefixModes.isNull())
 148     determinePrefixes();
 149
 150   return _prefixModes;
 151 }
 152
 153 bool NetworkInfo::supports(const QString &param) const {
 154   return _supports.contains(param);
 155 }
 156
 157 QString NetworkInfo::support(const QString &param) const {
 158   QString support_ = param.toUpper();
 159   if(_supports.contains(support_))
 160     return _supports[support_];
 161   else
 162     return QString();
 163 }
 164
 165 IrcUser *NetworkInfo::newIrcUser(const QString &hostmask) {
 166   QString nick(nickFromMask(hostmask).toLower());
 167   if(!_ircUsers.contains(nick)) {
 168     IrcUser *ircuser = new IrcUser(hostmask, this);
 169     // mark IrcUser as already initialized to keep the SignalProxy from requesting initData
 170     if(initialized())
 171       ircuser->setInitialized();
 172     if(proxy())
 173       proxy()->synchronize(ircuser);
 174     else
 175       qWarning() << "unable to synchronize new IrcUser" << hostmask << "forgot to call NetworkInfo::setProxy(SignalProxy *)?";
 176
 177     connect(ircuser, SIGNAL(nickSet(QString)), this, SLOT(ircUserNickChanged(QString)));
 178     connect(ircuser, SIGNAL(initDone()), this, SIGNAL(ircUserInitDone()));
 179     connect(ircuser, SIGNAL(destroyed()), this, SLOT(ircUserDestroyed()));
 180     _ircUsers[nick] = ircuser;
 181     emit ircUserAdded(hostmask);
 182   }
 183   return _ircUsers[nick];
 184 }
 185
 186 void NetworkInfo::removeIrcUser(IrcUser *ircuser) {
 187   QString nick = _ircUsers.key(ircuser);
 188   if(nick.isNull())
 189     return;
 190
 191   _ircUsers.remove(nick);
 192   ircuser->deleteLater();
 193   emit ircUserRemoved(nick);
 194 }
 195
 196 void NetworkInfo::removeIrcUser(QString nick) {
 197   IrcUser *ircuser;
 198   if((ircuser = ircUser(nick)) != 0)
 199     removeIrcUser(ircuser);
 200 }
 201
 202 IrcUser *NetworkInfo::ircUser(QString nickname) const {
 203   nickname = nickname.toLower();
 204   if(_ircUsers.contains(nickname))
 205     return _ircUsers[nickname];
 206   else
 207     return 0;
 208 }
 209
 210 QList<IrcUser *> NetworkInfo::ircUsers() const {
 211   return _ircUsers.values();
 212 }
 213
 214 IrcChannel *NetworkInfo::newIrcChannel(const QString &channelname) {
 215   if(!_ircChannels.contains(channelname.toLower())) {
 216     IrcChannel *channel = new IrcChannel(channelname, this);
 217     // mark IrcUser as already initialized to keep the SignalProxy from requesting initData
 218     if(initialized())
 219       channel->setInitialized();
 220
 221     if(proxy())
 222       proxy()->synchronize(channel);
 223     else
 224       qWarning() << "unable to synchronize new IrcChannel" << channelname << "forgot to call NetworkInfo::setProxy(SignalProxy *)?";
 225
 226     connect(channel, SIGNAL(initDone()), this, SIGNAL(ircChannelInitDone()));
 227     connect(channel, SIGNAL(destroyed()), this, SLOT(channelDestroyed()));
 228     _ircChannels[channelname.toLower()] = channel;
 229     emit ircChannelAdded(channelname);
 230   }
 231   return _ircChannels[channelname.toLower()];
 232 }
 233
 234
 235 IrcChannel *NetworkInfo::ircChannel(QString channelname) {
 236   channelname = channelname.toLower();
 237   if(_ircChannels.contains(channelname))
 238     return _ircChannels[channelname];
 239   else
 240     return 0;
 241 }
 242
 243 QList<IrcChannel *> NetworkInfo::ircChannels() const {
 244   return _ircChannels.values();
 245 }
 246
 247 // ====================
 248 //  Public Slots:
 249 // ====================
 250 void NetworkInfo::setNetworkName(const QString &networkName) {
 251   _networkName = networkName;
 252   emit networkNameSet(networkName);
 253 }
 254
 255 void NetworkInfo::setCurrentServer(const QString &currentServer) {
 256   _currentServer = currentServer;
 257   emit currentServerSet(currentServer);
 258 }
 259
 260 void NetworkInfo::setMyNick(const QString &nickname) {
 261   _myNick = nickname;
 262   emit myNickSet(nickname);
 263 }
 264
 265 void NetworkInfo::addSupport(const QString &param, const QString &value) {
 266   if(!_supports.contains(param)) {
 267     _supports[param] = value;
 268     emit supportAdded(param, value);
 269   }
 270 }
 271
 272 void NetworkInfo::removeSupport(const QString &param) {
 273   if(_supports.contains(param)) {
 274     _supports.remove(param);
 275     emit supportRemoved(param);
 276   }
 277 }
 278
 279 QVariantMap NetworkInfo::initSupports() const {
 280   QVariantMap supports;
 281   QHashIterator<QString, QString> iter(_supports);
 282   while(iter.hasNext()) {
 283     iter.next();
 284     supports[iter.key()] = iter.value();
 285   }
 286   return supports;
 287 }
 288
 289 QStringList NetworkInfo::initIrcUsers() const {
 290   QStringList hostmasks;
 291   foreach(IrcUser *ircuser, ircUsers()) {
 292     hostmasks << ircuser->hostmask();
 293   }
 294   return hostmasks;
 295 }
 296
 297 QStringList NetworkInfo::initIrcChannels() const {
 298   return _ircChannels.keys();
 299 }
 300
 301 void NetworkInfo::initSetSupports(const QVariantMap &supports) {
 302   QMapIterator<QString, QVariant> iter(supports);
 303   while(iter.hasNext()) {
 304     iter.next();
 305     addSupport(iter.key(), iter.value().toString());
 306   }
 307 }
 308
 309 void NetworkInfo::initSetIrcUsers(const QStringList &hostmasks) {
 310   if(!_ircUsers.empty())
 311     return;
 312   foreach(QString hostmask, hostmasks) {
 313     newIrcUser(hostmask);
 314   }
 315 }
 316
 317 void NetworkInfo::initSetChannels(const QStringList &channels) {
 318   if(!_ircChannels.empty())
 319     return;
 320   foreach(QString channel, channels)
 321     newIrcChannel(channel);
 322 }
 323
 324 IrcUser *NetworkInfo::updateNickFromMask(const QString &mask) {
 325   QString nick(nickFromMask(mask));
 326   IrcUser *ircuser;
 327
 328   if(_ircUsers.contains(nick)) {
 329     ircuser = _ircUsers[nick];
 330     ircuser->updateHostmask(mask);
 331   } else {
 332     ircuser = newIrcUser(mask);
 333   }
 334   return ircuser;
 335 }
 336
 337 void NetworkInfo::ircUserNickChanged(QString newnick) {
 338   QString oldnick = _ircUsers.key(qobject_cast<IrcUser*>(sender()));
 339
 340   if(oldnick.isNull())
 341     return;
 342
 343   _ircUsers[newnick] = _ircUsers.take(oldnick);
 344
 345   if(myNick() == oldnick)
 346     setMyNick(newnick);
 347 }
 348
 349 void NetworkInfo::ircUserDestroyed() {
 350   IrcUser *ircuser = static_cast<IrcUser *>(sender());
 351   Q_ASSERT(ircuser);
 352   removeIrcUser(ircuser);
 353 }
 354
 355 void NetworkInfo::channelDestroyed() {
 356   IrcChannel *channel = static_cast<IrcChannel *>(sender());
 357   Q_ASSERT(channel);
 358   _ircChannels.remove(_ircChannels.key(channel));
 359 }
 360
 361 void NetworkInfo::setInitialized() {
 362   _initialized = true;
 363   emit initDone();
 364 }
 365
 366 // ====================
 367 //  Private:
 368 // ====================
 369 void NetworkInfo::determinePrefixes() {
 370   // seems like we have to construct them first
 371   QString PREFIX = support("PREFIX");
 372
 373   if(PREFIX.startsWith("(") && PREFIX.contains(")")) {
 374     _prefixes = PREFIX.section(")", 1);
 375     _prefixModes = PREFIX.mid(1).section(")", 0, 0);
 376   } else {
 377     QString defaultPrefixes("~&@%+");
 378     QString defaultPrefixModes("qaohv");
 379
 380     // we just assume that in PREFIX are only prefix chars stored
 381     for(int i = 0; i < defaultPrefixes.size(); i++) {
 382       if(PREFIX.contains(defaultPrefixes[i])) {
 383         _prefixes += defaultPrefixes[i];
 384         _prefixModes += defaultPrefixModes[i];
 385       }
 386     }
 387     // check for success
 388     if(!_prefixes.isNull())
 389       return;
 390
 391     // well... our assumption was obviously wrong...
 392     // check if it's only prefix modes
 393     for(int i = 0; i < defaultPrefixes.size(); i++) {
 394       if(PREFIX.contains(defaultPrefixModes[i])) {
 395         _prefixes += defaultPrefixes[i];
 396         _prefixModes += defaultPrefixModes[i];
 397       }
 398     }
 399     // now we've done all we've could...
 400   }
 401 }
 402