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 private:
 137   NetworkId _networkId;
 138
 139   QPointer<Network> _network;
 140 };
 141
 142 /*****************************************
 143 *  User Category Items (like @vh etc.)
 144 *****************************************/
 145 class UserCategoryItem : public PropertyMapItem {
 146   Q_OBJECT
 147   Q_PROPERTY(QString categoryId READ categoryId)
 148
 149 public:
 150   UserCategoryItem(int category, AbstractTreeItem *parent);
 151
 152   QString categoryId();
 153   virtual quint64 id() const;
 154   virtual QVariant data(int column, int role) const;
 155
 156   void addUser(IrcUser *ircUser);
 157
 158   static int categoryFromModes(const QString &modes);
 159
 160 private:
 161   int _category;
 162
 163   struct Category {
 164     QChar mode;
 165     QString displayString;
 166     inline Category(QChar mode_, QString displayString_) : mode(mode_), displayString(displayString_) {};
 167   };
 168
 169   static const QList<Category> categories;
 170 };
 171
 172 /*****************************************
 173 *  Irc User Items
 174 *****************************************/
 175 class IrcUserItem : public PropertyMapItem {
 176   Q_OBJECT
 177   Q_PROPERTY(QString nickName READ nickName)
 178
 179 public:
 180   IrcUserItem(IrcUser *ircUser, AbstractTreeItem *parent);
 181
 182   QString nickName() const;
 183   IrcUser *ircUser();
 184   virtual quint64 id() const;
 185   virtual QVariant data(int column, int role) const;
 186   virtual QString toolTip(int column) const;
 187
 188 private slots:
 189   void setNick(QString newNick);
 190
 191 private:
 192   IrcUser *_ircUser;
 193 };
 194
 195
 196 /*****************************************
 197  * NetworkModel
 198  *****************************************/
 199 class NetworkModel : public TreeModel {
 200   Q_OBJECT
 201
 202 public:
 203   enum myRoles {
 204     BufferTypeRole = Qt::UserRole,
 205     ItemActiveRole,
 206     BufferActivityRole,
 207     BufferIdRole,
 208     NetworkIdRole,
 209     BufferInfoRole,
 210     ItemTypeRole
 211   };
 212
 213   enum itemTypes {
 214     NetworkItemType,
 215     BufferItemType,
 216     UserCategoryItemType,
 217     IrcUserItemType
 218   };
 219
 220   NetworkModel(QObject *parent = 0);
 221   static QList<QVariant> defaultHeader();
 222
 223   static bool mimeContainsBufferList(const QMimeData *mimeData);
 224   static QList< QPair<NetworkId, BufferId> > mimeDataToBufferList(const QMimeData *mimeData);
 225
 226   virtual QStringList mimeTypes() const;
 227   virtual QMimeData *mimeData(const QModelIndexList &) const;
 228   virtual bool dropMimeData(const QMimeData *, Qt::DropAction, int, int, const QModelIndex &);
 229
 230   void attachNetwork(Network *network);
 231
 232   bool isBufferIndex(const QModelIndex &) const;
 233   //Buffer *getBufferByIndex(const QModelIndex &) const;
 234   QModelIndex bufferIndex(BufferId bufferId);
 235
 236 public slots:
 237   void bufferUpdated(BufferInfo bufferInfo);
 238   void updateBufferActivity(const Message &msg);
 239
 240 private:
 241   QModelIndex networkIndex(NetworkId networkId);
 242   NetworkItem *networkItem(NetworkId networkId);
 243   NetworkItem *existsNetworkItem(NetworkId networkId);
 244   BufferItem *bufferItem(const BufferInfo &bufferInfo);
 245   BufferItem *existsBufferItem(const BufferInfo &bufferInfo);
 246
 247 };
 248
 249 #endif // NETWORKMODEL_H