4cfd5b68ec91af21ecb58a97ab0ce4e87df0d7fc
[quassel.git] / src / qtui / nicklistwidget.cpp
1 /***************************************************************************
2  *   Copyright (C) 2005-08 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 "nicklistwidget.h"
22
23 #include "buffer.h"
24 #include "nickview.h"
25 #include "client.h"
26 #include "networkmodel.h"
27 #include "buffermodel.h"
28 #include "nickviewfilter.h"
29
30 NickListWidget::NickListWidget(QWidget *parent)
31   : QWidget(parent),
32     _bufferModel(0),
33     _selectionModel(0)
34 {
35   ui.setupUi(this);
36 }
37
38 void NickListWidget::setModel(BufferModel *bufferModel) {
39   if(_bufferModel) {
40     disconnect(_bufferModel, 0, this, 0);
41   }
42   
43   _bufferModel = bufferModel;
44
45   if(bufferModel) {
46     connect(bufferModel, SIGNAL(rowsAboutToBeRemoved(QModelIndex, int, int)),
47             this, SLOT(rowsAboutToBeRemoved(QModelIndex, int, int)));
48   }
49 }
50
51 void NickListWidget::setSelectionModel(QItemSelectionModel *selectionModel) {
52   if(_selectionModel) {
53     disconnect(_selectionModel, 0, this, 0);
54   }
55
56   _selectionModel = selectionModel;
57
58   if(selectionModel) {
59     connect(selectionModel, SIGNAL(currentChanged(QModelIndex, QModelIndex)),
60             this, SLOT(currentChanged(QModelIndex, QModelIndex)));
61   }
62 }
63
64 void NickListWidget::currentChanged(const QModelIndex &current, const QModelIndex &previous) {
65   BufferInfo::Type bufferType = (BufferInfo::Type)current.data(NetworkModel::BufferTypeRole).toInt();
66   BufferId newBufferId = current.data(NetworkModel::BufferIdRole).value<BufferId>();
67   BufferId oldBufferId = previous.data(NetworkModel::BufferIdRole).value<BufferId>();
68
69   if(bufferType != BufferInfo::ChannelBuffer) {
70     ui.stackedWidget->setCurrentWidget(ui.emptyPage);
71     return;
72   }
73
74   if(newBufferId == oldBufferId)
75     return;
76
77   if(nickViews.contains(newBufferId)) {
78     ui.stackedWidget->setCurrentWidget(nickViews.value(newBufferId));
79   } else {
80     NickView *view = new NickView(this);
81     NickViewFilter *filter = new NickViewFilter(newBufferId, Client::networkModel());
82     view->setModel(filter);
83     QModelIndex source_current = Client::bufferModel()->mapToSource(current);
84     view->setRootIndex(filter->mapFromSource(source_current));
85     view->expandAll();
86     nickViews[newBufferId] = view;
87     ui.stackedWidget->addWidget(view);
88     ui.stackedWidget->setCurrentWidget(view);
89   }
90 }
91
92 void NickListWidget::rowsAboutToBeRemoved(const QModelIndex &parent, int start, int end) {
93   Q_ASSERT(model());
94   if(!parent.isValid()) {
95     // ok this means that whole networks are about to be removed
96     // we can't determine which buffers are affect, so we hope that all nets are removed
97     // this is the most common case (for example disconnecting from the core or terminating the clint)
98     NickView *nickView;
99     QHash<BufferId, NickView *>::iterator iter = nickViews.begin();
100     while(iter != nickViews.end()) {
101       nickView = *iter;
102       iter = nickViews.erase(iter);
103       ui.stackedWidget->removeWidget(nickView);
104       QAbstractItemModel *model = nickView->model();
105       nickView->setModel(0);
106       if(QSortFilterProxyModel *filter = qobject_cast<QSortFilterProxyModel *>(model))
107         filter->setSourceModel(0);
108       model->deleteLater();
109       nickView->deleteLater();
110     }
111   } else {
112     // check if there are explicitly buffers removed
113     for(int i = start; i <= end; i++) {
114       QVariant variant = parent.child(i,0).data(NetworkModel::BufferIdRole);
115       if(!variant.isValid())
116         continue;
117
118       BufferId bufferId = qVariantValue<BufferId>(variant);
119       removeBuffer(bufferId);
120     }
121   }
122 }
123
124 void NickListWidget::removeBuffer(BufferId bufferId) {
125   if(!nickViews.contains(bufferId))
126     return;
127
128   NickView *view = nickViews.take(bufferId);
129   ui.stackedWidget->removeWidget(view);
130   QAbstractItemModel *model = view->model();
131   view->setModel(0);
132   if(QSortFilterProxyModel *filter = qobject_cast<QSortFilterProxyModel *>(model))
133     filter->setSourceModel(0);
134   model->deleteLater();
135   view->deleteLater();
136 }
137
138 QSize NickListWidget::sizeHint() const {
139   QWidget *currentWidget = ui.stackedWidget->currentWidget();
140   if(!currentWidget || currentWidget == ui.emptyPage)
141     return QSize(100, height());
142   else
143     return currentWidget->sizeHint();
144 }