1 /***************************************************************************
2 * Copyright (C) 2005-08 by the Quassel Project *
3 * devel@quassel-irc.org *
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. *
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. *
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 ***************************************************************************/
21 #ifndef NETWORKMODEL_H
22 #define NETWORKMODEL_H
26 #include "treemodel.h"
27 #include "bufferinfo.h"
33 #include "selectionmodelsynchronizer.h"
34 #include "modelpropertymapper.h"
35 #include "clientsettings.h"
36 #include "ircchannel.h"
41 class MappedSelectionModel;
42 class QAbstractItemView;
45 /*****************************************
47 *****************************************/
48 class NetworkItem : public PropertyMapItem {
50 Q_PROPERTY(QString networkName READ networkName)
51 Q_PROPERTY(QString currentServer READ currentServer)
52 Q_PROPERTY(int nickCount READ nickCount)
55 NetworkItem(const NetworkId &netid, AbstractTreeItem *parent = 0);
57 virtual QVariant data(int column, int row) const;
59 inline bool isActive() const { return (bool)_network ? _network->isConnected() : false; }
61 inline const NetworkId &networkId() const { return _networkId; }
62 inline QString networkName() const { return (bool)_network ? _network->networkName() : QString(); }
63 inline QString currentServer() const { return (bool)_network ? _network->currentServer() : QString(); }
64 inline int nickCount() const { return (bool)_network ? _network->ircUsers().count() : 0; }
66 virtual QString toolTip(int column) const;
68 BufferItem *findBufferItem(BufferId bufferId);
69 inline BufferItem *findBufferItem(const BufferInfo &bufferInfo) { return findBufferItem(bufferInfo.bufferId()); }
70 BufferItem *bufferItem(const BufferInfo &bufferInfo);
73 void setNetworkName(const QString &networkName);
74 void setCurrentServer(const QString &serverName);
76 void attachNetwork(Network *network);
77 void attachIrcChannel(IrcChannel *channel);
78 void attachIrcUser(IrcUser *ircUser);
83 QPointer<Network> _network;
86 /*****************************************
88 *****************************************/
89 class BufferItem : public PropertyMapItem {
91 Q_PROPERTY(QString bufferName READ bufferName WRITE setBufferName)
92 Q_PROPERTY(QString topic READ topic)
93 Q_PROPERTY(int nickCount READ nickCount)
96 BufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent = 0);
98 inline const BufferInfo &bufferInfo() const { return _bufferInfo; }
99 virtual QVariant data(int column, int role) const;
100 virtual bool setData(int column, const QVariant &value, int role);
102 inline BufferId bufferId() const { return _bufferInfo.bufferId(); }
103 inline BufferInfo::Type bufferType() const { return _bufferInfo.type(); }
105 void setBufferName(const QString &name);
106 virtual inline QString bufferName() const { return _bufferInfo.bufferName(); }
107 virtual inline QString topic() const { return QString(); }
108 virtual inline int nickCount() const { return 0; }
110 virtual inline bool isActive() const { return qobject_cast<NetworkItem *>(parent())->isActive(); }
112 inline const MsgId &lastSeenMsgId() const { return _lastSeenMsgId; }
113 inline const MsgId &lastSeenMarkerMsgId() const { return _lastSeenMarkerMsgId; }
114 void setLastSeenMsgId(const MsgId &msgId);
115 inline BufferInfo::ActivityLevel activityLevel() const { return _activity; }
116 void setActivityLevel(BufferInfo::ActivityLevel level);
117 void clearActivityLevel();
118 void updateActivityLevel(const Message &msg);
120 bool isCurrentBuffer() const;
121 virtual QString toolTip(int column) const;
124 virtual inline void setTopic(const QString &) { emit dataChanged(1); }
127 BufferInfo _bufferInfo;
128 BufferInfo::ActivityLevel _activity;
129 MsgId _lastSeenMsgId;
130 MsgId _lastSeenMarkerMsgId;
133 /*****************************************
135 *****************************************/
136 class StatusBufferItem : public BufferItem {
140 StatusBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent);
142 virtual QString toolTip(int column) const;
143 virtual inline QString bufferName() const { return tr("Status Buffer"); }
146 /*****************************************
148 *****************************************/
149 class QueryBufferItem : public BufferItem {
153 QueryBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent);
155 virtual QVariant data(int column, int role) const;
156 virtual inline bool isActive() const { return (bool)_ircUser; }
157 virtual QString toolTip(int column) const;
160 void attachIrcUser(IrcUser *ircUser);
161 void ircUserDestroyed();
167 /*****************************************
169 *****************************************/
170 class UserCategoryItem;
172 class ChannelBufferItem : public BufferItem {
176 ChannelBufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent);
178 virtual inline bool isActive() const { return (bool)_ircChannel; }
179 virtual QString toolTip(int column) const;
181 virtual inline QString topic() const { return (bool)_ircChannel ? _ircChannel->topic() : QString(); }
182 virtual inline int nickCount() const { return (bool)_ircChannel ? _ircChannel->ircUsers().count() : 0; }
184 void attachIrcChannel(IrcChannel *ircChannel);
187 void join(const QList<IrcUser *> &ircUsers);
188 void part(IrcUser *ircUser);
190 UserCategoryItem *findCategoryItem(int categoryId);
191 void addUserToCategory(IrcUser *ircUser);
192 void addUsersToCategory(const QList<IrcUser *> &ircUser);
193 void removeUserFromCategory(IrcUser *ircUser);
194 void userModeChanged(IrcUser *ircUser);
197 void ircChannelDestroyed();
200 IrcChannel *_ircChannel;
203 /*****************************************
204 * User Category Items (like @vh etc.)
205 *****************************************/
207 class UserCategoryItem : public PropertyMapItem {
209 Q_PROPERTY(QString categoryName READ categoryName)
212 UserCategoryItem(int category, AbstractTreeItem *parent);
214 QString categoryName() const;
215 inline int categoryId() const { return _category; }
216 virtual QVariant data(int column, int role) const;
218 IrcUserItem *findIrcUser(IrcUser *ircUser);
219 void addUsers(const QList<IrcUser *> &ircUser);
220 bool removeUser(IrcUser *ircUser);
222 static int categoryFromModes(const QString &modes);
227 static const QList<QChar> categories;
230 /*****************************************
232 *****************************************/
233 class IrcUserItem : public PropertyMapItem {
235 Q_PROPERTY(QString nickName READ nickName)
238 IrcUserItem(IrcUser *ircUser, AbstractTreeItem *parent);
240 inline QString nickName() const { return _ircUser ? _ircUser->nick() : QString(); }
241 inline bool isActive() const { return _ircUser ? !_ircUser->isAway() : false; }
243 inline IrcUser *ircUser() { return _ircUser; }
244 virtual QVariant data(int column, int role) const;
245 virtual QString toolTip(int column) const;
248 inline void ircUserDestroyed() { parent()->removeChild(this); }
251 QPointer<IrcUser> _ircUser;
255 /*****************************************
257 *****************************************/
258 class NetworkModel : public TreeModel {
263 BufferTypeRole = TreeModel::UserRole,
274 NetworkItemType = 0x01,
275 BufferItemType = 0x02,
276 UserCategoryItemType = 0x04,
277 IrcUserItemType = 0x08
279 Q_DECLARE_FLAGS(itemTypes, itemType)
281 NetworkModel(QObject *parent = 0);
282 static QList<QVariant> defaultHeader();
284 static bool mimeContainsBufferList(const QMimeData *mimeData);
285 static QList< QPair<NetworkId, BufferId> > mimeDataToBufferList(const QMimeData *mimeData);
287 virtual QStringList mimeTypes() const;
288 virtual QMimeData *mimeData(const QModelIndexList &) const;
289 virtual bool dropMimeData(const QMimeData *, Qt::DropAction, int, int, const QModelIndex &);
291 void attachNetwork(Network *network);
293 bool isBufferIndex(const QModelIndex &) const;
294 //Buffer *getBufferByIndex(const QModelIndex &) const;
295 QModelIndex networkIndex(NetworkId networkId);
296 QModelIndex bufferIndex(BufferId bufferId);
298 const Network *networkByIndex(const QModelIndex &index) const;
300 BufferInfo::ActivityLevel bufferActivity(const BufferInfo &buffer) const;
302 //! Finds a buffer with a given name in a given network
303 /** This performs a linear search through all BufferItems, hence it is expensive.
304 * @param networkId The network which we search in
305 * @param bufferName The bufferName we look for
306 * @return The id of the buffer if found, an invalid one else
308 BufferId bufferId(NetworkId networkId, const QString &bufferName) const;
310 QString bufferName(BufferId bufferId) const;
311 BufferInfo::Type bufferType(BufferId bufferId) const;
312 BufferInfo bufferInfo(BufferId bufferId) const;
313 MsgId lastSeenMsgId(BufferId bufferId) const;
314 MsgId lastSeenMarkerMsgId(BufferId bufferId) const;
315 NetworkId networkId(BufferId bufferId) const;
316 QString networkName(BufferId bufferId) const;
318 inline QList<BufferId> allBufferIds() const { return _bufferItemCache.keys(); }
321 void bufferUpdated(BufferInfo bufferInfo);
322 void removeBuffer(BufferId bufferId);
323 void setLastSeenMsgId(const BufferId &bufferId, const MsgId &msgId);
324 void setBufferActivity(const BufferId &bufferId, BufferInfo::ActivityLevel activity);
325 void clearBufferActivity(const BufferId &bufferId);
326 void updateBufferActivity(const Message &msg);
327 void networkRemoved(const NetworkId &networkId);
330 void setLastSeenMsg(BufferId bufferId, MsgId msgId);
333 void checkForRemovedBuffers(const QModelIndex &parent, int start, int end);
334 void checkForNewBuffers(const QModelIndex &parent, int start, int end);
337 int networkRow(NetworkId networkId);
338 NetworkItem *findNetworkItem(NetworkId networkId);
339 NetworkItem *networkItem(NetworkId networkId);
340 inline BufferItem *findBufferItem(const BufferInfo &bufferInfo) { return findBufferItem(bufferInfo.bufferId()); }
341 BufferItem *findBufferItem(BufferId bufferId);
342 BufferItem *bufferItem(const BufferInfo &bufferInfo);
344 QHash<BufferId, BufferItem *> _bufferItemCache;
346 Q_DECLARE_OPERATORS_FOR_FLAGS(NetworkModel::itemTypes)
348 #endif // NETWORKMODEL_H