uisupport: Fix invalid model segfault from index
[quassel.git] / src / qtui / nicklistwidget.cpp
1 /***************************************************************************
2  *   Copyright (C) 2005-2020 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  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
19  ***************************************************************************/
20
21 #include "nicklistwidget.h"
22
23 #include <QAbstractButton>
24 #include <QAction>
25 #include <QDebug>
26 #include <QEvent>
27
28 #include "buffermodel.h"
29 #include "client.h"
30 #include "networkmodel.h"
31 #include "nickview.h"
32 #include "nickviewfilter.h"
33 #include "qtuisettings.h"
34
35 NickListWidget::NickListWidget(QWidget* parent)
36     : AbstractItemView(parent)
37 {
38     ui.setupUi(this);
39 }
40
41 QDockWidget* NickListWidget::dock() const
42 {
43     auto* dock = qobject_cast<QDockWidget*>(parent());
44     if (dock)
45         return dock;
46     else
47         return nullptr;
48 }
49
50 void NickListWidget::hideEvent(QHideEvent* event)
51 {
52     emit nickSelectionChanged(QModelIndexList());
53     AbstractItemView::hideEvent(event);
54 }
55
56 void NickListWidget::showEvent(QShowEvent* event)
57 {
58     auto* view = qobject_cast<NickView*>(ui.stackedWidget->currentWidget());
59     if (view)
60         emit nickSelectionChanged(view->selectedIndexes());
61
62     AbstractItemView::showEvent(event);
63 }
64
65 void NickListWidget::showWidget(bool visible)
66 {
67     if (!selectionModel())
68         return;
69
70     QModelIndex currentIndex = selectionModel()->currentIndex();
71     if (currentIndex.data(NetworkModel::BufferTypeRole) == BufferInfo::ChannelBuffer) {
72         QDockWidget* dock_ = dock();
73         if (!dock_)
74             return;
75
76         if (visible)
77             dock_->show();
78         else
79             dock_->close();
80     }
81 }
82
83 void NickListWidget::setVisible(bool visible)
84 {
85     QWidget::setVisible(visible);
86     QDockWidget* dock_ = dock();
87     if (!dock_)
88         return;
89
90     if (visible)
91         dock_->show();
92     else
93         dock_->close();
94 }
95
96 void NickListWidget::currentChanged(const QModelIndex& current, const QModelIndex& previous)
97 {
98     BufferInfo::Type bufferType = (BufferInfo::Type)current.data(NetworkModel::BufferTypeRole).toInt();
99     BufferId newBufferId = current.data(NetworkModel::BufferIdRole).value<BufferId>();
100     BufferId oldBufferId = previous.data(NetworkModel::BufferIdRole).value<BufferId>();
101
102     if (bufferType != BufferInfo::ChannelBuffer) {
103         ui.stackedWidget->setCurrentWidget(ui.emptyPage);
104         emit nickSelectionChanged(QModelIndexList());
105         return;
106     }
107
108     // See NickListDock::NickListDock() below
109     //   if(bufferType != BufferInfo::ChannelBuffer) {
110     //     ui.stackedWidget->setCurrentWidget(ui.emptyPage);
111     //     QDockWidget *dock_ = dock();
112     //     if(dock_) {
113     //       dock_->close();
114     //     }
115     //     return;
116     //   } else {
117     //     QDockWidget *dock_ = dock();
118     //     if(dock_ && dock_->toggleViewAction()->isChecked()) {
119     //       dock_->show();
120     //     }
121     //   }
122
123     if (newBufferId == oldBufferId)
124         return;
125
126     NickView* view;
127     if (nickViews.contains(newBufferId)) {
128         view = nickViews.value(newBufferId);
129         ui.stackedWidget->setCurrentWidget(view);
130     }
131     else {
132         view = new NickView(this);
133         auto* filter = new NickViewFilter(newBufferId, Client::networkModel());
134         view->setModel(filter);
135         QModelIndex source_current = Client::bufferModel()->mapToSource(current);
136         view->setRootIndex(filter->mapFromSource(source_current));
137         nickViews[newBufferId] = view;
138         ui.stackedWidget->addWidget(view);
139         ui.stackedWidget->setCurrentWidget(view);
140         connect(view, &NickView::selectionUpdated, this, &NickListWidget::onNickSelectionChanged);
141     }
142     emit nickSelectionChanged(view->selectedIndexes());
143 }
144
145 void NickListWidget::onNickSelectionChanged()
146 {
147     auto* view = qobject_cast<NickView*>(sender());
148     Q_ASSERT(view);
149     if (view != ui.stackedWidget->currentWidget()) {
150         qDebug() << "Nick selection of hidden view changed!";
151         return;
152     }
153     emit nickSelectionChanged(view->selectedIndexes());
154 }
155
156 void NickListWidget::rowsAboutToBeRemoved(const QModelIndex& parent, int start, int end)
157 {
158     Q_ASSERT(model());
159     if (!parent.isValid()) {
160         // ok this means that whole networks are about to be removed
161         // we can't determine which buffers are affect, so we hope that all nets are removed
162         // this is the most common case (for example disconnecting from the core or terminating the clint)
163         NickView* nickView;
164         QHash<BufferId, NickView*>::iterator iter = nickViews.begin();
165         while (iter != nickViews.end()) {
166             nickView = *iter;
167             iter = nickViews.erase(iter);
168             ui.stackedWidget->removeWidget(nickView);
169             QAbstractItemModel* model = nickView->model();
170             nickView->setModel(nullptr);
171             if (auto* filter = qobject_cast<QSortFilterProxyModel*>(model))
172                 filter->setSourceModel(nullptr);
173             model->deleteLater();
174             nickView->deleteLater();
175         }
176     }
177     else {
178         // check if there are explicitly buffers removed
179         // Make sure model is valid first
180         if (!parent.model()) {
181             return;
182         }
183         for (int i = start; i <= end; i++) {
184             QVariant variant = parent.model()->index(i, 0, parent).data(NetworkModel::BufferIdRole);
185             if (!variant.isValid())
186                 continue;
187
188             BufferId bufferId = variant.value<BufferId>();
189             removeBuffer(bufferId);
190         }
191     }
192 }
193
194 void NickListWidget::removeBuffer(BufferId bufferId)
195 {
196     if (!nickViews.contains(bufferId))
197         return;
198
199     NickView* view = nickViews.take(bufferId);
200     ui.stackedWidget->removeWidget(view);
201     QAbstractItemModel* model = view->model();
202     view->setModel(nullptr);
203     if (auto* filter = qobject_cast<QSortFilterProxyModel*>(model))
204         filter->setSourceModel(nullptr);
205     model->deleteLater();
206     view->deleteLater();
207 }
208
209 QSize NickListWidget::sizeHint() const
210 {
211     QWidget* currentWidget = ui.stackedWidget->currentWidget();
212     if (!currentWidget || currentWidget == ui.emptyPage)
213         return {100, height()};
214     else
215         return currentWidget->sizeHint();
216 }
217
218 // ==============================
219 //  NickList Dock
220 // ==============================
221 NickListDock::NickListDock(const QString& title, QWidget* parent)
222     : QDockWidget(title, parent)
223 {
224     // THIS STUFF IS NEEDED FOR NICKLIST AUTOHIDE...
225     // AS THIS BRINGS LOTS OF FUCKUPS WITH IT IT'S DEACTIVATED FOR NOW...
226
227     //   QAction *toggleView = toggleViewAction();
228     //   disconnect(toggleView, SIGNAL(triggered(bool)), this, 0);
229     //   toggleView->setChecked(QtUiSettings().value("ShowNickList", QVariant(true)).toBool());
230
231     //   // reconnecting the closebuttons clicked signal to the action
232     //   foreach(QAbstractButton *button, findChildren<QAbstractButton *>()) {
233     //     if(disconnect(button, SIGNAL(clicked()), this, SLOT(close())))
234     //       connect(button, SIGNAL(clicked()), toggleView, SLOT(trigger()));
235     //   }
236 }
237
238 void NickListDock::setLocked(bool locked)
239 {
240     if (locked) {
241         setFeatures(nullptr);
242     }
243     else {
244         setFeatures(QDockWidget::DockWidgetClosable | QDockWidget::DockWidgetMovable | QDockWidget::DockWidgetFloatable);
245     }
246 }
247
248 // NickListDock::~NickListDock() {
249 //   QtUiSettings().setValue("ShowNickList", toggleViewAction()->isChecked());
250 // }
251
252 // bool NickListDock::event(QEvent *event) {
253 //   switch (event->type()) {
254 //   case QEvent::Hide:
255 //   case QEvent::Show:
256 //     emit visibilityChanged(event->type() == QEvent::Show);
257 //     return QWidget::event(event);
258 //   default:
259 //     return QDockWidget::event(event);
260 //   }
261 // }