src/client/networkmodel.h

   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 #ifndef NETWORKMODEL_H
  22 #define NETWORKMODEL_H
  23
  24 #include <QtCore>
  25
  26 #include "treemodel.h"
  27 #include "buffer.h"
  28
  29 #include <QPointer>
  30
  31 class BufferInfo;
  32
  33 #include "selectionmodelsynchronizer.h"
  34 #include "modelpropertymapper.h"
  35 class MappedSelectionModel;
  36 class QAbstractItemView;
  37 class Network;
  38 class IrcChannel;
  39 class IrcUser;
  40
  41 /*****************************************
  42  *  Fancy Buffer Items
  43  *****************************************/
  44 class BufferItem : public PropertyMapItem {
  45   Q_OBJECT
  46   Q_PROPERTY(QString bufferName READ bufferName)
  47   Q_PROPERTY(QString topic READ topic)
  48   Q_PROPERTY(int nickCount READ nickCount)
  49
  50 public:
  51   BufferItem(BufferInfo bufferInfo, AbstractTreeItem *parent = 0);
  52
  53   const BufferInfo &bufferInfo() const;
  54   virtual quint64 id() const;
  55   virtual QVariant data(int column, int role) const;
  56   virtual bool setData(int column, const QVariant &value, int role);
  57
  58   void attachIrcChannel(IrcChannel *ircChannel);
  59
  60   QString bufferName() const;
  61   QString topic() const;
  62   int nickCount() const;
  63
  64   enum Type {
  65     StatusType,
  66     ChannelType,
  67     QueryType
  68   };
  69
  70   bool isStatusBuffer() const;
  71   Type bufferType() const;
  72
  73   bool isActive() const;
  74
  75   enum Activity {
  76     NoActivity = 0x00,
  77     OtherActivity = 0x01,
  78     NewMessage = 0x02,
  79     Highlight = 0x40
  80   };
  81   Q_DECLARE_FLAGS(ActivityLevel, Activity)
  82
  83   ActivityLevel activity() const;
  84   void setActivity(const ActivityLevel &level);
  85   void updateActivity(const ActivityLevel &level);
  86
  87 public slots:
  88   void setTopic(const QString &topic);
  89   void join(IrcUser *ircUser);
  90   void part(IrcUser *ircUser);
  91
  92   void addUserToCategory(IrcUser *ircUser);
  93   void removeUserFromCategory(IrcUser *ircUser);
  94   void userModeChanged(IrcUser *ircUser);
  95
  96 private slots:
  97   void ircChannelDestroyed();
  98
  99 private:
 100   BufferInfo _bufferInfo;
 101   ActivityLevel _activity;
 102   Type _type;
 103
 104   QPointer<IrcChannel> _ircChannel;
 105 };
 106 Q_DECLARE_OPERATORS_FOR_FLAGS(BufferItem::ActivityLevel)
 107
 108 /*****************************************
 109  *  Network Items
 110  *****************************************/
 111 class NetworkItem : public PropertyMapItem {
 112   Q_OBJECT
 113   Q_PROPERTY(QString networkName READ networkName)
 114   Q_PROPERTY(QString currentServer READ currentServer)
 115   Q_PROPERTY(int nickCount READ nickCount)
 116
 117 public:
 118   NetworkItem(const NetworkId &netid, AbstractTreeItem *parent = 0);
 119
 120   virtual quint64 id() const;
 121   virtual QVariant data(int column, int row) const;
 122
 123   bool isActive() const;
 124
 125   QString networkName() const;
 126   QString currentServer() const;
 127   int nickCount() const;
 128
 129 public slots:
 130   void setNetworkName(const QString &networkName);
 131   void setCurrentServer(const QString &serverName);
 132
 133   void attachNetwork(Network *network);
 134   void attachIrcChannel(const QString &channelName);
 135
 136   void setActive(bool connected);
 137
 138 private:
 139   NetworkId _networkId;
 140
 141   QPointer<Network> _network;
 142 };
 143
 144 /*****************************************
 145 *  User Category Items (like @vh etc.)
 146 *****************************************/
 147 class UserCategoryItem : public PropertyMapItem {
 148   Q_OBJECT
 149   Q_PROPERTY(QString categoryId READ categoryId)
 150
 151 public:
 152   UserCategoryItem(int category, AbstractTreeItem *parent);
 153
 154   QString categoryId();
 155   virtual quint64 id() const;
 156   virtual QVariant data(int column, int role) const;
 157
 158   void addUser(IrcUser *ircUser);
 159
 160   static int categoryFromModes(const QString &modes);
 161
 162 private:
 163   int _category;
 164
 165   struct Category {
 166     QChar mode;
 167     QString displayString;
 168     inline Category(QChar mode_, QString displayString_) : mode(mode_), displayString(displayString_) {};
 169   };
 170
 171   static const QList<Category> categories;
 172 };
 173
 174 /*****************************************
 175 *  Irc User Items
 176 *****************************************/
 177 class IrcUserItem : public PropertyMapItem {
 178   Q_OBJECT
 179   Q_PROPERTY(QString nickName READ nickName)
 180
 181 public:
 182   IrcUserItem(IrcUser *ircUser, AbstractTreeItem *parent);
 183
 184   QString nickName() const;
 185   IrcUser *ircUser();
 186   virtual quint64 id() const;
 187   virtual QVariant data(int column, int role) const;
 188   virtual QString toolTip(int column) const;
 189
 190 private slots:
 191   void setNick(QString newNick);
 192
 193 private:
 194   IrcUser *_ircUser;
 195 };
 196
 197
 198 /*****************************************
 199  * NetworkModel
 200  *****************************************/
 201 class NetworkModel : public TreeModel {
 202   Q_OBJECT
 203
 204 public:
 205   enum myRoles {
 206     BufferTypeRole = Qt::UserRole,
 207     ItemActiveRole,
 208     BufferActivityRole,
 209     BufferIdRole,
 210     NetworkIdRole,
 211     BufferInfoRole,
 212     ItemTypeRole
 213   };
 214
 215   enum itemTypes {
 216     NetworkItemType,
 217     BufferItemType,
 218     UserCategoryItemType,
 219     IrcUserItemType
 220   };
 221
 222   NetworkModel(QObject *parent = 0);
 223   static QList<QVariant> defaultHeader();
 224
 225   static bool mimeContainsBufferList(const QMimeData *mimeData);
 226   static QList< QPair<NetworkId, BufferId> > mimeDataToBufferList(const QMimeData *mimeData);
 227
 228   virtual QStringList mimeTypes() const;
 229   virtual QMimeData *mimeData(const QModelIndexList &) const;
 230   virtual bool dropMimeData(const QMimeData *, Qt::DropAction, int, int, const QModelIndex &);
 231
 232   void attachNetwork(Network *network);
 233
 234   bool isBufferIndex(const QModelIndex &) const;
 235   //Buffer *getBufferByIndex(const QModelIndex &) const;
 236   QModelIndex bufferIndex(BufferId bufferId);
 237
 238 public slots:
 239   void bufferUpdated(BufferInfo bufferInfo);
 240   void updateBufferActivity(const Message &msg);
 241
 242 private:
 243   QModelIndex networkIndex(NetworkId networkId);
 244   NetworkItem *networkItem(NetworkId networkId);
 245   NetworkItem *existsNetworkItem(NetworkId networkId);
 246   BufferItem *bufferItem(const BufferInfo &bufferInfo);
 247   BufferItem *existsBufferItem(const BufferInfo &bufferInfo);
 248
 249 };
 250
 251 #endif // NETWORKMODEL_H