b1cc15f0511c1af29a4f5fe75b367629d8b27647
[quassel.git] / 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
57   void attachIrcChannel(IrcChannel *ircChannel);
58
59   QString bufferName() const;
60   QString topic() const;
61   int nickCount() const;
62
63   enum Type {
64     StatusType,
65     ChannelType,
66     QueryType
67   };
68   
69   bool isStatusBuffer() const;
70   Type bufferType() const;
71
72   bool isActive() const;
73   
74   enum Activity {
75     NoActivity = 0x00,
76     OtherActivity = 0x01,
77     NewMessage = 0x02,
78     Highlight = 0x40
79   };
80   Q_DECLARE_FLAGS(ActivityLevel, Activity)
81
82   ActivityLevel activity() const;
83   void setActivity(const ActivityLevel &level);
84   void addActivity(const ActivityLevel &level);
85
86 public slots:
87   void setTopic(const QString &topic);
88   void join(IrcUser *ircUser);
89   void part(IrcUser *ircUser);
90
91   void addUserToCategory(IrcUser *ircUser);
92   void removeUserFromCategory(IrcUser *ircUser);
93   void userModeChanged(IrcUser *ircUser);
94                                          
95 private slots:
96   void ircChannelDestroyed();
97   
98 private:
99   BufferInfo _bufferInfo;
100   ActivityLevel _activity;
101   Type _type;
102
103   QPointer<IrcChannel> _ircChannel;
104 };
105 Q_DECLARE_OPERATORS_FOR_FLAGS(BufferItem::ActivityLevel)
106
107 /*****************************************
108  *  Network Items
109  *****************************************/
110 class NetworkItem : public PropertyMapItem {
111   Q_OBJECT
112   Q_PROPERTY(QString networkName READ networkName)
113   Q_PROPERTY(QString currentServer READ currentServer)
114   Q_PROPERTY(int nickCount READ nickCount)
115     
116 public:
117   NetworkItem(const NetworkId &netid, AbstractTreeItem *parent = 0);
118
119   virtual QVariant data(int column, int row) const;
120   virtual quint64 id() const;
121
122   bool isActive() const;
123   
124   QString networkName() const;
125   QString currentServer() const;
126   int nickCount() const;
127   
128 public slots:
129   void setNetworkName(const QString &networkName);
130   void setCurrentServer(const QString &serverName);
131
132   void attachNetwork(Network *network);
133   void attachIrcChannel(const QString &channelName);
134   
135 private:
136   NetworkId _networkId;
137
138   QPointer<Network> _network;
139 };
140
141 /*****************************************
142 *  User Category Items (like @vh etc.)
143 *****************************************/
144 class UserCategoryItem : public PropertyMapItem {
145   Q_OBJECT
146   Q_PROPERTY(QString categoryId READ categoryId)
147     
148 public:
149   UserCategoryItem(int category, AbstractTreeItem *parent);
150
151   QString categoryId();
152   virtual quint64 id() const;
153
154   void addUser(IrcUser *ircUser);
155
156   static int categoryFromModes(const QString &modes);
157
158 private slots:
159   void checkNoChilds();
160
161 private:
162   int _category;
163
164   struct Category {
165     QChar mode;
166     QString displayString;
167     inline Category(QChar mode_, QString displayString_) : mode(mode_), displayString(displayString_) {};
168   };
169
170   static const QList<Category> categories;
171 };
172
173 /*****************************************
174 *  Irc User Items
175 *****************************************/
176 class IrcUserItem : public PropertyMapItem {
177   Q_OBJECT
178   Q_PROPERTY(QString nickName READ nickName)
179     
180 public:
181   IrcUserItem(IrcUser *ircUser, AbstractTreeItem *parent);
182
183   QString nickName();
184   IrcUser *ircUser();
185   virtual quint64 id() const;
186
187 private slots:
188   void setNick(QString newNick);
189   void ircUserDestroyed();
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     BufferIdRole,
207     NetworkIdRole,
208     ItemTypeRole
209   };
210
211   enum itemTypes {
212     AbstractItemType,
213     SimpleItemType,
214     NetworkItemType,
215     BufferItemType,
216     NickItemType
217   };
218     
219   NetworkModel(QObject *parent = 0);
220   static QList<QVariant> defaultHeader();
221
222   static bool mimeContainsBufferList(const QMimeData *mimeData);
223   static QList< QPair<NetworkId, BufferId> > mimeDataToBufferList(const QMimeData *mimeData);
224
225   virtual QStringList mimeTypes() const;
226   virtual QMimeData *mimeData(const QModelIndexList &) const;
227   virtual bool dropMimeData(const QMimeData *, Qt::DropAction, int, int, const QModelIndex &);
228
229   void attachNetwork(Network *network);
230
231   bool isBufferIndex(const QModelIndex &) const;
232   //Buffer *getBufferByIndex(const QModelIndex &) const;
233   QModelIndex bufferIndex(BufferId bufferId);
234
235 public slots:
236   void bufferUpdated(BufferInfo bufferInfo);
237   void bufferActivity(BufferItem::ActivityLevel, BufferInfo bufferInfo);
238
239 private:
240   QModelIndex networkIndex(NetworkId networkId);
241   NetworkItem *networkItem(NetworkId networkId);
242   NetworkItem *existsNetworkItem(NetworkId networkId);
243   BufferItem *bufferItem(const BufferInfo &bufferInfo);
244   BufferItem *existsBufferItem(const BufferInfo &bufferInfo);
245
246 };
247
248 #endif // NETWORKMODEL_H