fixing crash on disconnect

[quassel.git] / src / client / backlogrequester.h

diff --git a/src/client/backlogrequester.h b/src/client/backlogrequester.h
index 5ef71b2..c3c6832 100644 (file)
--- a/src/client/backlogrequester.h
+++ b/src/client/backlogrequester.h
@@ -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  *
@@ -24,33 +24,85 @@
 #include <QList>
 
 #include "client.h"
+#include "message.h"
 #include "networkmodel.h"
 #include "types.h"
 
-class BacklogManager;
+class ClientBacklogManager;
 
 class BacklogRequester {
 public:
-  BacklogRequester(BacklogManager *backlogManger);
+  enum RequesterTypes {
+    InvalidRequester = 0,
+    PerBufferFixed,
+    PerBufferUnread,
+    GlobalUnread
+  };
+
+  BacklogRequester(bool buffering, ClientBacklogManager *backlogManger);
   virtual inline ~BacklogRequester() {}
 
+  inline bool isBuffering() { return _isBuffering; }
+  inline const QList<Message> &bufferedMessages() { return _bufferedMessages; }
+
+  inline int buffersWaiting() const { return _buffersWaiting.count(); }
+  inline int totalBuffers() const { return _totalBuffers; }
+  //! returns false if it was the last missing backlogpart
+  bool buffer(BufferId bufferId, const MessageList &messages);
+  
   virtual void requestBacklog() = 0;
 
 protected:
   inline QList<BufferId> allBufferIds() const { return Client::networkModel()->allBufferIds(); }
-  BacklogManager *backlogManager;
-};
+  inline void setWaitingBuffers(const QList<BufferId> &buffers) { setWaitingBuffers(buffers.toSet()); }
+  void setWaitingBuffers(const QSet<BufferId> &buffers);
+  void addWaitingBuffer(BufferId buffer);
+
+  ClientBacklogManager *backlogManager;
 
+private:
+  bool _isBuffering;
+  MessageList _bufferedMessages;
+  int _totalBuffers;
+  QSet<BufferId> _buffersWaiting;
+};
 
+// ========================================
+//  FIXED BACKLOG REQUESTER
+// ========================================
 class FixedBacklogRequester : public BacklogRequester {
 public:
-  FixedBacklogRequester(BacklogManager *backlogManager);
+  FixedBacklogRequester(ClientBacklogManager *backlogManager);
+  virtual void requestBacklog();
+
+private:
+  int _backlogCount;
+};
 
+// ========================================
+//  GLOBAL UNREAD BACKLOG REQUESTER
+// ========================================
+class GlobalUnreadBacklogRequester : public BacklogRequester {
+public:
+  GlobalUnreadBacklogRequester(ClientBacklogManager *backlogManager);
   virtual void requestBacklog();
 
 private:
-  static const int backlogCount;
+  int _limit;
+  int _additional;
 };
 
+// ========================================
+//  PER BUFFER UNREAD BACKLOG REQUESTER
+// ========================================
+class PerBufferUnreadBacklogRequester : public BacklogRequester {
+public:
+  PerBufferUnreadBacklogRequester(ClientBacklogManager *backlogManager);
+  virtual void requestBacklog();
+
+private:
+  int _limit;
+  int _additional;
+};
 
 #endif //BACKLOGREQUESTER_H