making the requester type configurable
[quassel.git] / src / client / clientbacklogmanager.cpp
1 /***************************************************************************
2  *   Copyright (C) 2005-08 by the Quassel IRC Team                         *
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 "clientbacklogmanager.h"
22
23 #include "abstractmessageprocessor.h"
24 #include "backlogsettings.h"
25 #include "backlogrequester.h"
26 #include "client.h"
27
28 #include <ctime>
29
30 #include <QDebug>
31
32 ClientBacklogManager::ClientBacklogManager(QObject *parent)
33   : BacklogManager(parent),
34     _requester(0)
35 {
36 }
37
38 void ClientBacklogManager::receiveBacklog(BufferId bufferId, MsgId first, MsgId last, int limit, int additional, QVariantList msgs) {
39   Q_UNUSED(first) Q_UNUSED(last) Q_UNUSED(limit) Q_UNUSED(additional)
40
41   if(msgs.isEmpty())
42     return;
43
44   emit messagesReceived(bufferId, msgs.count());
45
46   MessageList msglist;
47   foreach(QVariant v, msgs) {
48     Message msg = v.value<Message>();
49     msg.setFlags(msg.flags() | Message::Backlog);
50     msglist << msg;
51   }
52
53   if(isBuffering()) {
54     if(!_requester->buffer(bufferId, msglist)) {
55       // this was the last part to buffer
56       stopBuffering();
57       reset();
58     }
59   } else {
60     dispatchMessages(msglist);
61   }
62 }
63
64 void ClientBacklogManager::receiveBacklogAll(MsgId first, MsgId last, int limit, int additional, QVariantList msgs) {
65   Q_UNUSED(first) Q_UNUSED(last) Q_UNUSED(limit) Q_UNUSED(additional)
66
67   MessageList msglist;
68   foreach(QVariant v, msgs) {
69     Message msg = v.value<Message>();
70     msg.setFlags(msg.flags() | Message::Backlog);
71     msglist << msg;
72   }
73
74   dispatchMessages(msglist);
75   reset();
76 }
77
78 void ClientBacklogManager::requestInitialBacklog() {
79   if(_requester) {
80     qWarning() << "ClientBacklogManager::requestInitialBacklog() called twice in the same session! (Backlog has already been requested)";
81     return;
82   }
83
84   BacklogSettings settings;
85   switch(settings.requesterType()) {
86   case BacklogRequester::GlobalUnread:
87     _requester = new GlobalUnreadBacklogRequester(this);
88     break;
89   case BacklogRequester::PerBufferUnread:
90     _requester = new PerBufferUnreadBacklogRequester(this);
91     break;
92   case BacklogRequester::PerBufferFixed:
93   default:
94     _requester = new FixedBacklogRequester(this);
95   };
96
97   _requester->requestBacklog();
98 }
99
100 void ClientBacklogManager::stopBuffering() {
101   Q_ASSERT(_requester);
102
103   dispatchMessages(_requester->bufferedMessages(), true);
104 }
105
106 bool ClientBacklogManager::isBuffering() {
107   return _requester && _requester->isBuffering();
108 }
109
110 void ClientBacklogManager::dispatchMessages(const MessageList &messages, bool sort) {
111   MessageList msgs = messages;
112
113   clock_t start_t = clock();
114   if(sort)
115     qSort(msgs);
116   Client::messageProcessor()->process(msgs);
117   clock_t end_t = clock();
118
119   emit messagesProcessed(tr("Processed %1 messages in %2 seconds.").arg(msgs.count()).arg((float)(end_t - start_t) / CLOCKS_PER_SEC));
120 }
121
122 void ClientBacklogManager::reset() {
123   delete _requester;
124   _requester = 0;
125 }