fixing auto expand issues with new networks
[quassel.git] / src / client / backlogrequester.cpp
1 /***************************************************************************
2  *   Copyright (C) 2005-09 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 #include "backlogrequester.h"
22
23 #include <QObject>
24
25 #include "backlogsettings.h"
26 #include "clientbacklogmanager.h"
27
28 BacklogRequester::BacklogRequester(bool buffering, ClientBacklogManager *backlogManager)
29   : backlogManager(backlogManager),
30     _isBuffering(buffering),
31     _totalBuffers(0)
32 {
33   Q_ASSERT(backlogManager);
34 }
35
36 void BacklogRequester::setWaitingBuffers(const QSet<BufferId> &buffers) {
37   _buffersWaiting = buffers;
38   _totalBuffers = _buffersWaiting.count();
39 }
40
41 void BacklogRequester::addWaitingBuffer(BufferId buffer) {
42   _buffersWaiting << buffer;
43   _totalBuffers++;
44 }
45
46 bool BacklogRequester::buffer(BufferId bufferId, const MessageList &messages) {
47   _bufferedMessages << messages;
48   _buffersWaiting.remove(bufferId);
49   return !_buffersWaiting.isEmpty();
50 }
51
52 // ========================================
53 //  FIXED BACKLOG REQUESTER
54 // ========================================
55 FixedBacklogRequester::FixedBacklogRequester(ClientBacklogManager *backlogManager)
56   : BacklogRequester(true, backlogManager)
57 {
58   BacklogSettings backlogSettings;
59   _backlogCount = backlogSettings.fixedBacklogAmount();
60 }
61
62 void FixedBacklogRequester::requestBacklog() {
63   QList<BufferId> allBuffers = allBufferIds();
64   setWaitingBuffers(allBuffers);
65   backlogManager->emitMessagesRequested(QObject::tr("Requesting a total of up to %1 backlog messages for %2 buffers").arg(_backlogCount * allBuffers.count()).arg(allBuffers.count()));
66   foreach(BufferId bufferId, allBuffers) {
67     backlogManager->requestBacklog(bufferId, -1, -1, _backlogCount);
68   }
69 }
70
71 // ========================================
72 //  GLOBAL UNREAD BACKLOG REQUESTER
73 // ========================================
74 GlobalUnreadBacklogRequester::GlobalUnreadBacklogRequester(ClientBacklogManager *backlogManager)
75   : BacklogRequester(false, backlogManager)
76 {
77   BacklogSettings backlogSettings;
78   _limit = backlogSettings.globalUnreadBacklogLimit();
79   _additional = backlogSettings.globalUnreadBacklogAdditional();
80 }
81
82 void GlobalUnreadBacklogRequester::requestBacklog() {
83   MsgId oldestUnreadMessage;
84   foreach(BufferId bufferId, allBufferIds()) {
85     MsgId msgId = Client::networkModel()->lastSeenMsgId(bufferId);
86     if(!oldestUnreadMessage.isValid() || oldestUnreadMessage > msgId)
87       oldestUnreadMessage = msgId;
88   }
89   backlogManager->emitMessagesRequested(QObject::tr("Requesting up to %1 of all unread backlog messages (plus additional %2)").arg(_limit).arg(_additional));
90   backlogManager->requestBacklogAll(oldestUnreadMessage, -1, _limit, _additional);
91 }
92
93 // ========================================
94 //  PER BUFFER UNREAD BACKLOG REQUESTER
95 // ========================================
96 PerBufferUnreadBacklogRequester::PerBufferUnreadBacklogRequester(ClientBacklogManager *backlogManager)
97   : BacklogRequester(true, backlogManager)
98 {
99   BacklogSettings backlogSettings;
100   _limit = backlogSettings.perBufferUnreadBacklogLimit();
101   _additional = backlogSettings.perBufferUnreadBacklogAdditional();
102 }
103
104 void PerBufferUnreadBacklogRequester::requestBacklog() {
105   QList<BufferId> allBuffers = allBufferIds();
106   setWaitingBuffers(allBuffers);
107   backlogManager->emitMessagesRequested(QObject::tr("Requesting a total of up to %1 unread backlog messages for %2 buffers").arg((_limit + _additional) * allBuffers.count()).arg(allBuffers.count()));
108   foreach(BufferId bufferId, allBuffers) {
109     backlogManager->requestBacklog(bufferId, Client::networkModel()->lastSeenMsgId(bufferId), -1, _limit, _additional);
110   }
111 }