Fixes #410 - away log (you'll find it in the views menu)
[quassel.git] / src / client / networkmodel.h
index 6ac06f8..f763a21 100644 (file)
@@ -1,5 +1,5 @@
 /***************************************************************************
- *   Copyright (C) 2005-08 by the Quassel Project                          *
+ *   Copyright (C) 2005-09 by the Quassel Project                          *
  *   devel@quassel-irc.org                                                 *
  *                                                                         *
  *   This program is free software; you can redistribute it and/or modify  *
 #ifndef NETWORKMODEL_H
 #define NETWORKMODEL_H
 
-#include <QtCore>
-
+#include "bufferinfo.h"
+#include "clientsettings.h"
+#include "message.h"
+#include "network.h"
 #include "treemodel.h"
-#include "buffer.h"
 
-#include <QPointer>
+class BufferItem;
+class StatusBufferItem;
 
-class BufferInfo;
+/*****************************************
+ *  Network Items
+ *****************************************/
+class NetworkItem : public PropertyMapItem {
+  Q_OBJECT
+  Q_PROPERTY(QString networkName READ networkName)
+  Q_PROPERTY(QString currentServer READ currentServer)
+  Q_PROPERTY(int nickCount READ nickCount)
 
-#include "selectionmodelsynchronizer.h"
-#include "modelpropertymapper.h"
-#include "clientsettings.h"
+public:
+  NetworkItem(const NetworkId &netid, AbstractTreeItem *parent = 0);
+
+  virtual QVariant data(int column, int row) const;
+
+  inline bool isActive() const { return (bool)_network ? _network->isConnected() : false; }
+
+  inline const NetworkId &networkId() const { return _networkId; }
+  inline QString networkName() const { return (bool)_network ? _network->networkName() : QString(); }
+  inline QString currentServer() const { return (bool)_network ? _network->currentServer() : QString(); }
+  inline int nickCount() const { return (bool)_network ? _network->ircUsers().count() : 0; }
+
+  virtual QString toolTip(int column) const;
+
+  BufferItem *findBufferItem(BufferId bufferId);
+  inline BufferItem *findBufferItem(const BufferInfo &bufferInfo) { return findBufferItem(bufferInfo.bufferId()); }
+  BufferItem *bufferItem(const BufferInfo &bufferInfo);
+
+public slots:
+  void setNetworkName(const QString &networkName);
+  void setCurrentServer(const QString &serverName);
+
+  void attachNetwork(Network *network);
+  void attachIrcChannel(IrcChannel *channel);
+  void attachIrcUser(IrcUser *ircUser);
+
+signals:
+  void networkDataChanged(int column = -1);
+
+private slots:
+  void onBeginRemoveChilds(int start, int end);
+  
+private:
+  NetworkId _networkId;
+  StatusBufferItem *_statusBufferItem;
 
-class MappedSelectionModel;
-class QAbstractItemView;
-class Network;
-class IrcChannel;
-class IrcUser;
+  QPointer<Network> _network;
+};
 
 /*****************************************
  *  Fancy Buffer Items
@@ -50,105 +88,135 @@ class BufferItem : public PropertyMapItem {
   Q_PROPERTY(int nickCount READ nickCount)
 
 public:
-  BufferItem(BufferInfo bufferInfo, AbstractTreeItem *parent = 0);
+  BufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent = 0);
 
-  const BufferInfo &bufferInfo() const;
-  virtual quint64 id() const;
+  inline const BufferInfo &bufferInfo() const { return _bufferInfo; }
   virtual QVariant data(int column, int role) const;
   virtual bool setData(int column, const QVariant &value, int role);
 
-  void attachIrcChannel(IrcChannel *ircChannel);
-
-  QString bufferName() const;
-  void setBufferName(const QString &name);
-  QString topic() const;
-  int nickCount() const;
+  inline BufferId bufferId() const { return _bufferInfo.bufferId(); }
+  inline BufferInfo::Type bufferType() const { return _bufferInfo.type(); }
 
-  bool isStatusBuffer() const;
-  BufferInfo::Type bufferType() const;
+  virtual void setBufferName(const QString &name);
+  virtual inline QString bufferName() const { return _bufferInfo.bufferName(); }
+  virtual inline QString topic() const { return QString(); }
+  virtual inline int nickCount() const { return 0; }
 
-  bool isActive() const;
+  virtual inline bool isActive() const { return qobject_cast<NetworkItem *>(parent())->isActive(); }
 
-  inline Buffer::ActivityLevel activityLevel() const { return _activity; }
-  bool setActivityLevel(Buffer::ActivityLevel level);
-  void updateActivityLevel(Buffer::ActivityLevel level);
+  inline const MsgId &lastSeenMsgId() const { return _lastSeenMsgId; }
+  inline const MsgId &lastSeenMarkerMsgId() const { return _lastSeenMarkerMsgId; }
+  void setLastSeenMsgId(const MsgId &msgId);
+  inline BufferInfo::ActivityLevel activityLevel() const { return _activity; }
+  void setActivityLevel(BufferInfo::ActivityLevel level);
+  void clearActivityLevel();
+  void updateActivityLevel(const Message &msg);
 
-  void setLastMsgInsert(QDateTime msgDate);
-  bool setLastSeen();
-  QDateTime lastSeen();
+  bool isCurrentBuffer() const;
+  virtual QString toolTip(int column) const;
 
 public slots:
-  void setTopic(const QString &topic);
-  void join(IrcUser *ircUser);
-  void part(IrcUser *ircUser);
-
-  void addUserToCategory(IrcUser *ircUser);
-  void removeUserFromCategory(IrcUser *ircUser);
-  void userModeChanged(IrcUser *ircUser);
-
-private slots:
-  void ircChannelDestroyed();
-  void ircUserDestroyed();
+  virtual inline void setTopic(const QString &) { emit dataChanged(1); }
 
 private:
   BufferInfo _bufferInfo;
-  QString _bufferName;
-  Buffer::ActivityLevel _activity;
-
-  QPointer<IrcChannel> _ircChannel;
+  BufferInfo::ActivityLevel _activity;
+  MsgId _lastSeenMsgId;
+  MsgId _lastSeenMarkerMsgId;
 };
 
+/*****************************************
+*  StatusBufferItem
+*****************************************/
+class StatusBufferItem : public BufferItem {
+  Q_OBJECT
+
+public:
+  StatusBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent);
+
+  virtual QString toolTip(int column) const;
+  virtual inline QString bufferName() const { return tr("Status Buffer"); }
+};
 
 /*****************************************
- *  Network Items
- *****************************************/
-class NetworkItem : public PropertyMapItem {
+*  QueryBufferItem
+*****************************************/
+class QueryBufferItem : public BufferItem {
   Q_OBJECT
-  Q_PROPERTY(QString networkName READ networkName)
-  Q_PROPERTY(QString currentServer READ currentServer)
-  Q_PROPERTY(int nickCount READ nickCount)
-    
+
 public:
-  NetworkItem(const NetworkId &netid, AbstractTreeItem *parent = 0);
+  QueryBufferItem(const BufferInfo &bufferInfo, NetworkItem *parent);
 
-  virtual quint64 id() const;
-  inline const NetworkId &networkId() const { return _networkId; }
-  virtual QVariant data(int column, int row) const;
+  virtual QVariant data(int column, int role) const;
+  virtual bool setData(int column, const QVariant &value, int role);
+
+  virtual inline bool isActive() const { return (bool)_ircUser; }
+  virtual QString toolTip(int column) const;
+
+  virtual void setBufferName(const QString &name);
 
-  bool isActive() const;
-  
-  QString networkName() const;
-  QString currentServer() const;
-  int nickCount() const;
-  
 public slots:
-  void setNetworkName(const QString &networkName);
-  void setCurrentServer(const QString &serverName);
+  void setIrcUser(IrcUser *ircUser);
+  void removeIrcUser();
 
-  void attachNetwork(Network *network);
-  void attachIrcChannel(const QString &channelName);
-  
 private:
-  NetworkId _networkId;
+  IrcUser *_ircUser;
+};
 
-  QPointer<Network> _network;
+/*****************************************
+*  ChannelBufferItem
+*****************************************/
+class UserCategoryItem;
+
+class ChannelBufferItem : public BufferItem {
+  Q_OBJECT
+
+public:
+  ChannelBufferItem(const BufferInfo &bufferInfo, AbstractTreeItem *parent);
+
+  virtual QVariant data(int column, int role) const;
+  virtual inline bool isActive() const { return (bool)_ircChannel; }
+  virtual QString toolTip(int column) const;
+
+  virtual inline QString topic() const { return (bool)_ircChannel ? _ircChannel->topic() : QString(); }
+  virtual inline int nickCount() const { return (bool)_ircChannel ? _ircChannel->ircUsers().count() : 0; }
+
+  void attachIrcChannel(IrcChannel *ircChannel);
+
+public slots:
+  void join(const QList<IrcUser *> &ircUsers);
+  void part(IrcUser *ircUser);
+
+  UserCategoryItem *findCategoryItem(int categoryId);
+  void addUserToCategory(IrcUser *ircUser);
+  void addUsersToCategory(const QList<IrcUser *> &ircUser);
+  void removeUserFromCategory(IrcUser *ircUser);
+  void userModeChanged(IrcUser *ircUser);
+
+private slots:
+  void ircChannelParted();
+
+private:
+  IrcChannel *_ircChannel;
 };
 
 /*****************************************
 *  User Category Items (like @vh etc.)
 *****************************************/
+class IrcUserItem;
 class UserCategoryItem : public PropertyMapItem {
   Q_OBJECT
-  Q_PROPERTY(QString categoryId READ categoryId)
-    
+  Q_PROPERTY(QString categoryName READ categoryName)
+
 public:
   UserCategoryItem(int category, AbstractTreeItem *parent);
 
-  QString categoryId();
-  virtual quint64 id() const;
+  QString categoryName() const;
+  inline int categoryId() const { return _category; }
   virtual QVariant data(int column, int role) const;
-  
-  void addUser(IrcUser *ircUser);
+
+  IrcUserItem *findIrcUser(IrcUser *ircUser);
+  void addUsers(const QList<IrcUser *> &ircUser);
   bool removeUser(IrcUser *ircUser);
 
   static int categoryFromModes(const QString &modes);
@@ -156,13 +224,7 @@ public:
 private:
   int _category;
 
-  struct Category {
-    QChar mode;
-    QString displayString;
-    inline Category(QChar mode_, QString displayString_) : mode(mode_), displayString(displayString_) {};
-  };
-
-  static const QList<Category> categories;
+  static const QList<QChar> categories;
 };
 
 /*****************************************
@@ -171,25 +233,22 @@ private:
 class IrcUserItem : public PropertyMapItem {
   Q_OBJECT
   Q_PROPERTY(QString nickName READ nickName)
-    
+
 public:
   IrcUserItem(IrcUser *ircUser, AbstractTreeItem *parent);
 
-  QString nickName() const;
-  bool isActive() const;
+  inline QString nickName() const { return _ircUser ? _ircUser->nick() : QString(); }
+  inline bool isActive() const { return _ircUser ? !_ircUser->isAway() : false; }
 
-  IrcUser *ircUser();
-  virtual quint64 id() const;
+  inline IrcUser *ircUser() { return _ircUser; }
   virtual QVariant data(int column, int role) const;
   virtual QString toolTip(int column) const;
 
 private slots:
-  void setNick(QString newNick);
-  void setAway(bool);
+  inline void ircUserQuited() { parent()->removeChild(this); }
 
 private:
   QPointer<IrcUser> _ircUser;
-  quint64 _id;
 };
 
 
@@ -200,22 +259,26 @@ class NetworkModel : public TreeModel {
   Q_OBJECT
 
 public:
-  enum myRoles {
-    BufferTypeRole = Qt::UserRole,
+  enum Role {
+    BufferTypeRole = TreeModel::UserRole,
     ItemActiveRole,
     BufferActivityRole,
     BufferIdRole,
     NetworkIdRole,
     BufferInfoRole,
-    ItemTypeRole
+    ItemTypeRole,
+    UserAwayRole,
+    IrcUserRole,
+    IrcChannelRole
   };
 
-  enum itemTypes {
-    NetworkItemType,
-    BufferItemType,
-    UserCategoryItemType,
-    IrcUserItemType
+  enum ItemType {
+    NetworkItemType = 0x01,
+    BufferItemType = 0x02,
+    UserCategoryItemType = 0x04,
+    IrcUserItemType = 0x08
   };
+  Q_DECLARE_FLAGS(ItemTypes, ItemType)
 
   NetworkModel(QObject *parent = 0);
   static QList<QVariant> defaultHeader();
@@ -225,31 +288,67 @@ public:
 
   virtual QStringList mimeTypes() const;
   virtual QMimeData *mimeData(const QModelIndexList &) const;
-  virtual bool dropMimeData(const QMimeData *, Qt::DropAction, int, int, const QModelIndex &);
 
   void attachNetwork(Network *network);
 
   bool isBufferIndex(const QModelIndex &) const;
   //Buffer *getBufferByIndex(const QModelIndex &) const;
+  QModelIndex networkIndex(NetworkId networkId);
   QModelIndex bufferIndex(BufferId bufferId);
 
   const Network *networkByIndex(const QModelIndex &index) const;
 
-  Buffer::ActivityLevel bufferActivity(const BufferInfo &buffer) const;
+  BufferInfo::ActivityLevel bufferActivity(const BufferInfo &buffer) const;
+
+  //! Finds a buffer with a given name in a given network
+  /** This performs a linear search through all BufferItems, hence it is expensive.
+   *  @param networkId  The network which we search in
+   *  @param bufferName The bufferName we look for
+   *  @return The id of the buffer if found, an invalid one else
+   */
+  BufferId bufferId(NetworkId networkId, const QString &bufferName, Qt::CaseSensitivity cs = Qt::CaseInsensitive) const;
+
+  QString bufferName(BufferId bufferId) const;
+  BufferInfo::Type bufferType(BufferId bufferId) const;
+  BufferInfo bufferInfo(BufferId bufferId) const;
+  MsgId lastSeenMsgId(BufferId bufferId) const;
+  MsgId lastSeenMarkerMsgId(BufferId bufferId) const;
+  NetworkId networkId(BufferId bufferId) const;
+  QString networkName(BufferId bufferId) const;
+
+  inline QList<BufferId> allBufferIds() const { return _bufferItemCache.keys(); }
+  QList<BufferId> allBufferIdsSorted() const;
+  void sortBufferIds(QList<BufferId> &bufferIds) const;
 
 public slots:
   void bufferUpdated(BufferInfo bufferInfo);
   void removeBuffer(BufferId bufferId);
-  void setBufferActivity(const BufferInfo &buffer, Buffer::ActivityLevel activity);
+  MsgId lastSeenMsgId(const BufferId &bufferId);
+  void setLastSeenMsgId(const BufferId &bufferId, const MsgId &msgId);
+  void setBufferActivity(const BufferId &bufferId, BufferInfo::ActivityLevel activity);
+  void clearBufferActivity(const BufferId &bufferId);
+  void updateBufferActivity(const Message &msg);
   void networkRemoved(const NetworkId &networkId);
-  
+
+signals:
+  void setLastSeenMsg(BufferId bufferId, MsgId msgId);
+
+private slots:
+  void checkForRemovedBuffers(const QModelIndex &parent, int start, int end);
+  void checkForNewBuffers(const QModelIndex &parent, int start, int end);
+
 private:
-  QModelIndex networkIndex(NetworkId networkId);
+  int networkRow(NetworkId networkId) const;
+  NetworkItem *findNetworkItem(NetworkId networkId) const;
   NetworkItem *networkItem(NetworkId networkId);
-  NetworkItem *existsNetworkItem(NetworkId networkId);
+  inline BufferItem *findBufferItem(const BufferInfo &bufferInfo) const { return findBufferItem(bufferInfo.bufferId()); }
+  BufferItem *findBufferItem(BufferId bufferId) const;
   BufferItem *bufferItem(const BufferInfo &bufferInfo);
-  BufferItem *existsBufferItem(const BufferInfo &bufferInfo);
 
+  static bool bufferItemLessThan(const BufferItem *left, const BufferItem *right);
+
+  QHash<BufferId, BufferItem *> _bufferItemCache;
 };
+Q_DECLARE_OPERATORS_FOR_FLAGS(NetworkModel::ItemTypes)
 
 #endif // NETWORKMODEL_H