c9a1594fafb020c2171585cd269fdba6ea3f74d9
[quassel.git] / src / uisupport / networkmodelcontroller.cpp
1 /***************************************************************************
2  *   Copyright (C) 2005-2015 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 <QComboBox>
22 #include <QDialogButtonBox>
23 #include <QGridLayout>
24 #include <QIcon>
25 #include <QLabel>
26 #include <QLineEdit>
27 #include <QInputDialog>
28 #include <QMessageBox>
29 #include <QPushButton>
30
31 #include "networkmodelcontroller.h"
32
33 #include "buffermodel.h"
34 #include "buffersettings.h"
35 #include "clientidentity.h"
36 #include "network.h"
37 #include "util.h"
38 #include "clientignorelistmanager.h"
39 #include "client.h"
40
41 NetworkModelController::NetworkModelController(QObject *parent)
42     : QObject(parent),
43     _actionCollection(new ActionCollection(this)),
44     _messageFilter(0),
45     _receiver(0)
46 {
47     connect(_actionCollection, SIGNAL(actionTriggered(QAction *)), SLOT(actionTriggered(QAction *)));
48 }
49
50
51 NetworkModelController::~NetworkModelController()
52 {
53 }
54
55
56 Action *NetworkModelController::registerAction(ActionType type, const QString &text, bool checkable)
57 {
58     return registerAction(type, QPixmap(), text, checkable);
59 }
60
61
62 Action *NetworkModelController::registerAction(ActionType type, const QIcon &icon, const QString &text, bool checkable)
63 {
64     Action *act;
65     if (icon.isNull())
66         act = new Action(text, this);
67     else
68         act = new Action(icon, text, this);
69
70     act->setCheckable(checkable);
71     act->setData(type);
72
73     _actionCollection->addAction(QString::number(type, 16), act);
74     _actionByType[type] = act;
75     return act;
76 }
77
78
79 /******** Helper Functions ***********************************************************************/
80
81 void NetworkModelController::setIndexList(const QModelIndex &index)
82 {
83     _indexList = QList<QModelIndex>() << index;
84 }
85
86
87 void NetworkModelController::setIndexList(const QList<QModelIndex> &list)
88 {
89     _indexList = list;
90 }
91
92
93 void NetworkModelController::setMessageFilter(MessageFilter *filter)
94 {
95     _messageFilter = filter;
96 }
97
98
99 void NetworkModelController::setContextItem(const QString &contextItem)
100 {
101     _contextItem = contextItem;
102 }
103
104
105 void NetworkModelController::setSlot(QObject *receiver, const char *method)
106 {
107     _receiver = receiver;
108     _method = method;
109 }
110
111
112 bool NetworkModelController::checkRequirements(const QModelIndex &index, ItemActiveStates requiredActiveState)
113 {
114     if (!index.isValid())
115         return false;
116
117     ItemActiveStates isActive = index.data(NetworkModel::ItemActiveRole).toBool()
118                                 ? ActiveState
119                                 : InactiveState;
120
121     if (!(isActive & requiredActiveState))
122         return false;
123
124     return true;
125 }
126
127
128 QString NetworkModelController::nickName(const QModelIndex &index) const
129 {
130     IrcUser *ircUser = qobject_cast<IrcUser *>(index.data(NetworkModel::IrcUserRole).value<QObject *>());
131     if (ircUser)
132         return ircUser->nick();
133
134     BufferInfo bufferInfo = index.data(NetworkModel::BufferInfoRole).value<BufferInfo>();
135     if (!bufferInfo.isValid())
136         return QString();
137     if (bufferInfo.type() != BufferInfo::QueryBuffer)
138         return QString();
139
140     return bufferInfo.bufferName(); // FIXME this might break with merged queries maybe
141 }
142
143
144 BufferId NetworkModelController::findQueryBuffer(const QModelIndex &index, const QString &predefinedNick) const
145 {
146     NetworkId networkId = index.data(NetworkModel::NetworkIdRole).value<NetworkId>();
147     if (!networkId.isValid())
148         return BufferId();
149
150     QString nick = predefinedNick.isEmpty() ? nickName(index) : predefinedNick;
151     if (nick.isEmpty())
152         return BufferId();
153
154     return findQueryBuffer(networkId, nick);
155 }
156
157
158 BufferId NetworkModelController::findQueryBuffer(NetworkId networkId, const QString &nick) const
159 {
160     return Client::networkModel()->bufferId(networkId, nick);
161 }
162
163
164 void NetworkModelController::removeBuffers(const QModelIndexList &indexList)
165 {
166     QList<BufferInfo> inactive;
167     foreach(QModelIndex index, indexList) {
168         BufferInfo info = index.data(NetworkModel::BufferInfoRole).value<BufferInfo>();
169         if (info.isValid()) {
170             if (info.type() == BufferInfo::QueryBuffer
171                 || (info.type() == BufferInfo::ChannelBuffer && !index.data(NetworkModel::ItemActiveRole).toBool()))
172                 inactive << info;
173         }
174     }
175     QString msg;
176     if (inactive.count()) {
177         msg = tr("Do you want to delete the following buffer(s) permanently?", 0, inactive.count());
178         msg += "<ul>";
179         int count = 0;
180         foreach(BufferInfo info, inactive) {
181             if (count < 10) {
182                 msg += QString("<li>%1</li>").arg(info.bufferName());
183                 count++;
184             }
185             else
186                 break;
187         }
188         msg += "</ul>";
189         if (count > 9 && inactive.size() - count != 0)
190             msg += tr("...and <b>%1</b> more<br><br>").arg(inactive.size() - count);
191         msg += tr("<b>Note:</b> This will delete all related data, including all backlog data, from the core's database and cannot be undone.");
192         if (inactive.count() != indexList.count())
193             msg += tr("<br>Active channel buffers cannot be deleted, please part the channel first.");
194
195         if (QMessageBox::question(0, tr("Remove buffers permanently?"), msg, QMessageBox::Yes|QMessageBox::No, QMessageBox::No) == QMessageBox::Yes) {
196             foreach(BufferInfo info, inactive)
197             Client::removeBuffer(info.bufferId());
198         }
199     }
200 }
201
202
203 void NetworkModelController::handleExternalAction(ActionType type, QAction *action)
204 {
205     Q_UNUSED(type);
206     if (receiver() && method()) {
207         if (!QMetaObject::invokeMethod(receiver(), method(), Q_ARG(QAction *, action)))
208             qWarning() << "NetworkModelActionController::handleExternalAction(): Could not invoke slot" << receiver() << method();
209     }
210 }
211
212
213 /******** Handle Actions *************************************************************************/
214
215 void NetworkModelController::actionTriggered(QAction *action)
216 {
217     ActionType type = (ActionType)action->data().toInt();
218     if (type > 0) {
219         if (type & NetworkMask)
220             handleNetworkAction(type, action);
221         else if (type & BufferMask)
222             handleBufferAction(type, action);
223         else if (type & HideMask)
224             handleHideAction(type, action);
225         else if (type & GeneralMask)
226             handleGeneralAction(type, action);
227         else if (type & NickMask)
228             handleNickAction(type, action);
229         else if (type & ExternalMask)
230             handleExternalAction(type, action);
231         else
232             qWarning() << "NetworkModelController::actionTriggered(): Unhandled action!";
233     }
234 }
235
236
237 void NetworkModelController::handleNetworkAction(ActionType type, QAction *)
238 {
239     if (type == NetworkConnectAll || type == NetworkDisconnectAll) {
240         foreach(NetworkId id, Client::networkIds()) {
241             const Network *net = Client::network(id);
242             if (type == NetworkConnectAll && net->connectionState() == Network::Disconnected)
243                 net->requestConnect();
244             if (type == NetworkDisconnectAll && net->connectionState() != Network::Disconnected)
245                 net->requestDisconnect();
246         }
247         return;
248     }
249
250     if (!indexList().count())
251         return;
252
253     const Network *network = Client::network(indexList().at(0).data(NetworkModel::NetworkIdRole).value<NetworkId>());
254     Q_CHECK_PTR(network);
255     if (!network)
256         return;
257
258     switch (type) {
259     case NetworkConnect:
260         network->requestConnect();
261         break;
262     case NetworkDisconnect:
263         network->requestDisconnect();
264         break;
265     default:
266         break;
267     }
268 }
269
270
271 void NetworkModelController::handleBufferAction(ActionType type, QAction *)
272 {
273     if (type == BufferRemove) {
274         removeBuffers(indexList());
275     }
276     else {
277         QList<BufferInfo> bufferList; // create temp list because model indexes might change
278         foreach(QModelIndex index, indexList()) {
279             BufferInfo bufferInfo = index.data(NetworkModel::BufferInfoRole).value<BufferInfo>();
280             if (bufferInfo.isValid())
281                 bufferList << bufferInfo;
282         }
283
284         foreach(BufferInfo bufferInfo, bufferList) {
285             switch (type) {
286             case BufferJoin:
287                 Client::userInput(bufferInfo, QString("/JOIN %1").arg(bufferInfo.bufferName()));
288                 break;
289             case BufferPart:
290             {
291                 QString reason = Client::identity(Client::network(bufferInfo.networkId())->identity())->partReason();
292                 Client::userInput(bufferInfo, QString("/PART %1").arg(reason));
293                 break;
294             }
295             case BufferSwitchTo:
296                 Client::bufferModel()->switchToBuffer(bufferInfo.bufferId());
297                 break;
298             default:
299                 break;
300             }
301         }
302     }
303 }
304
305
306 void NetworkModelController::handleHideAction(ActionType type, QAction *action)
307 {
308     Q_UNUSED(action)
309
310     if (type == HideJoinPartQuit) {
311         bool anyChecked = NetworkModelController::action(HideJoin)->isChecked();
312         anyChecked |= NetworkModelController::action(HidePart)->isChecked();
313         anyChecked |= NetworkModelController::action(HideQuit)->isChecked();
314
315         // If any are checked, uncheck them all.
316         // If none are checked, check them all.
317         bool newCheckedState = !anyChecked;
318         NetworkModelController::action(HideJoin)->setChecked(newCheckedState);
319         NetworkModelController::action(HidePart)->setChecked(newCheckedState);
320         NetworkModelController::action(HideQuit)->setChecked(newCheckedState);
321     }
322
323     int filter = 0;
324     if (NetworkModelController::action(HideJoin)->isChecked())
325         filter |= Message::Join | Message::NetsplitJoin;
326     if (NetworkModelController::action(HidePart)->isChecked())
327         filter |= Message::Part;
328     if (NetworkModelController::action(HideQuit)->isChecked())
329         filter |= Message::Quit | Message::NetsplitQuit;
330     if (NetworkModelController::action(HideNick)->isChecked())
331         filter |= Message::Nick;
332     if (NetworkModelController::action(HideMode)->isChecked())
333         filter |= Message::Mode;
334     if (NetworkModelController::action(HideDayChange)->isChecked())
335         filter |= Message::DayChange;
336     if (NetworkModelController::action(HideTopic)->isChecked())
337         filter |= Message::Topic;
338
339     switch (type) {
340     case HideJoinPartQuit:
341     case HideJoin:
342     case HidePart:
343     case HideQuit:
344     case HideNick:
345     case HideMode:
346     case HideDayChange:
347     case HideTopic:
348         if (_messageFilter)
349             BufferSettings(_messageFilter->idString()).setMessageFilter(filter);
350         else {
351             foreach(QModelIndex index, _indexList) {
352                 BufferId bufferId = index.data(NetworkModel::BufferIdRole).value<BufferId>();
353                 if (!bufferId.isValid())
354                     continue;
355                 BufferSettings(bufferId).setMessageFilter(filter);
356             }
357         }
358         return;
359     case HideApplyToAll:
360         BufferSettings().setMessageFilter(filter);
361         [[fallthrough]];
362     case HideUseDefaults:
363         if (_messageFilter)
364             BufferSettings(_messageFilter->idString()).removeFilter();
365         else {
366             foreach(QModelIndex index, _indexList) {
367                 BufferId bufferId = index.data(NetworkModel::BufferIdRole).value<BufferId>();
368                 if (!bufferId.isValid())
369                     continue;
370                 BufferSettings(bufferId).removeFilter();
371             }
372         }
373         return;
374     default:
375         return;
376     };
377 }
378
379
380 void NetworkModelController::handleGeneralAction(ActionType type, QAction *action)
381 {
382     Q_UNUSED(action)
383
384     if (!indexList().count())
385         return;
386     NetworkId networkId = indexList().at(0).data(NetworkModel::NetworkIdRole).value<NetworkId>();
387
388     switch (type) {
389     case JoinChannel:
390     {
391         QString channelName = contextItem();
392         QString channelPassword;
393         if (channelName.isEmpty()) {
394             JoinDlg dlg(indexList().first());
395             if (dlg.exec() == QDialog::Accepted) {
396                 channelName = dlg.channelName();
397                 networkId = dlg.networkId();
398                 channelPassword = dlg.channelPassword();
399             }
400         }
401         if (!channelName.isEmpty()) {
402             if (!channelPassword.isEmpty())
403                 Client::instance()->userInput(BufferInfo::fakeStatusBuffer(networkId), QString("/JOIN %1 %2").arg(channelName).arg(channelPassword));
404             else
405                 Client::instance()->userInput(BufferInfo::fakeStatusBuffer(networkId), QString("/JOIN %1").arg(channelName));
406         }
407         break;
408     }
409     case ShowChannelList:
410         if (networkId.isValid())
411             emit showChannelList(networkId);
412         break;
413     case ShowIgnoreList:
414         if (networkId.isValid())
415             emit showIgnoreList(QString());
416         break;
417     default:
418         break;
419     }
420 }
421
422
423 void NetworkModelController::handleNickAction(ActionType type, QAction *action)
424 {
425     foreach(QModelIndex index, indexList()) {
426         NetworkId networkId = index.data(NetworkModel::NetworkIdRole).value<NetworkId>();
427         if (!networkId.isValid())
428             continue;
429         QString nick = nickName(index);
430         if (nick.isEmpty())
431             continue;
432         BufferInfo bufferInfo = index.data(NetworkModel::BufferInfoRole).value<BufferInfo>();
433         if (!bufferInfo.isValid())
434             continue;
435
436         switch (type) {
437         case NickWhois:
438             Client::userInput(bufferInfo, QString("/WHOIS %1 %1").arg(nick));
439             break;
440         case NickCtcpVersion:
441             Client::userInput(bufferInfo, QString("/CTCP %1 VERSION").arg(nick));
442             break;
443         case NickCtcpPing:
444             Client::userInput(bufferInfo, QString("/CTCP %1 PING").arg(nick));
445             break;
446         case NickCtcpTime:
447             Client::userInput(bufferInfo, QString("/CTCP %1 TIME").arg(nick));
448             break;
449         case NickCtcpClientinfo:
450             Client::userInput(bufferInfo, QString("/CTCP %1 CLIENTINFO").arg(nick));
451             break;
452         case NickOp:
453             Client::userInput(bufferInfo, QString("/OP %1").arg(nick));
454             break;
455         case NickDeop:
456             Client::userInput(bufferInfo, QString("/DEOP %1").arg(nick));
457             break;
458         case NickHalfop:
459             Client::userInput(bufferInfo, QString("/HALFOP %1").arg(nick));
460             break;
461         case NickDehalfop:
462             Client::userInput(bufferInfo, QString("/DEHALFOP %1").arg(nick));
463             break;
464         case NickVoice:
465             Client::userInput(bufferInfo, QString("/VOICE %1").arg(nick));
466             break;
467         case NickDevoice:
468             Client::userInput(bufferInfo, QString("/DEVOICE %1").arg(nick));
469             break;
470         case NickKick:
471             Client::userInput(bufferInfo, QString("/KICK %1").arg(nick));
472             break;
473         case NickBan:
474             Client::userInput(bufferInfo, QString("/BAN %1").arg(nick));
475             break;
476         case NickKickBan:
477             Client::userInput(bufferInfo, QString("/BAN %1").arg(nick));
478             Client::userInput(bufferInfo, QString("/KICK %1").arg(nick));
479             break;
480         case NickSwitchTo:
481         case NickQuery:
482             Client::bufferModel()->switchToOrStartQuery(networkId, nick);
483             break;
484         case NickIgnoreUser:
485         {
486             IrcUser *ircUser = qobject_cast<IrcUser *>(index.data(NetworkModel::IrcUserRole).value<QObject *>());
487             if (!ircUser)
488                 break;
489             Client::ignoreListManager()->requestAddIgnoreListItem(IgnoreListManager::SenderIgnore,
490                 action->property("ignoreRule").toString(),
491                 false, IgnoreListManager::SoftStrictness,
492                 IgnoreListManager::NetworkScope,
493                 ircUser->network()->networkName(), true);
494             break;
495         }
496         case NickIgnoreHost:
497         {
498             IrcUser *ircUser = qobject_cast<IrcUser *>(index.data(NetworkModel::IrcUserRole).value<QObject *>());
499             if (!ircUser)
500                 break;
501             Client::ignoreListManager()->requestAddIgnoreListItem(IgnoreListManager::SenderIgnore,
502                 action->property("ignoreRule").toString(),
503                 false, IgnoreListManager::SoftStrictness,
504                 IgnoreListManager::NetworkScope,
505                 ircUser->network()->networkName(), true);
506             break;
507         }
508         case NickIgnoreDomain:
509         {
510             IrcUser *ircUser = qobject_cast<IrcUser *>(index.data(NetworkModel::IrcUserRole).value<QObject *>());
511             if (!ircUser)
512                 break;
513             Client::ignoreListManager()->requestAddIgnoreListItem(IgnoreListManager::SenderIgnore,
514                 action->property("ignoreRule").toString(),
515                 false, IgnoreListManager::SoftStrictness,
516                 IgnoreListManager::NetworkScope,
517                 ircUser->network()->networkName(), true);
518             break;
519         }
520         case NickIgnoreCustom:
521             // forward that to mainwin since we can access the settingspage only from there
522             emit showIgnoreList(action->property("ignoreRule").toString());
523             break;
524         case NickIgnoreToggleEnabled0:
525         case NickIgnoreToggleEnabled1:
526         case NickIgnoreToggleEnabled2:
527         case NickIgnoreToggleEnabled3:
528         case NickIgnoreToggleEnabled4:
529             Client::ignoreListManager()->requestToggleIgnoreRule(action->property("ignoreRule").toString());
530             break;
531         default:
532             qWarning() << "Unhandled nick action";
533         }
534     }
535 }
536
537
538 /***************************************************************************************************************
539  * JoinDlg
540  ***************************************************************************************************************/
541
542 NetworkModelController::JoinDlg::JoinDlg(const QModelIndex &index, QWidget *parent) : QDialog(parent)
543 {
544     setWindowIcon(QIcon::fromTheme("irc-join-channel"));
545     setWindowTitle(tr("Join Channel"));
546
547     QGridLayout *layout = new QGridLayout(this);
548     layout->addWidget(new QLabel(tr("Network:")), 0, 0);
549     layout->addWidget(networks = new QComboBox, 0, 1);
550     layout->addWidget(new QLabel(tr("Channel:")), 1, 0);
551     layout->addWidget(channel = new QLineEdit, 1, 1);
552     layout->addWidget(new QLabel(tr("Password:")), 2, 0);
553     layout->addWidget(password = new QLineEdit, 2, 1);
554     layout->addWidget(buttonBox = new QDialogButtonBox(QDialogButtonBox::Ok|QDialogButtonBox::Cancel), 3, 0, 1, 2);
555     setLayout(layout);
556
557     channel->setFocus();
558     buttonBox->button(QDialogButtonBox::Ok)->setEnabled(false);
559     networks->setInsertPolicy(QComboBox::InsertAlphabetically);
560     password->setEchoMode(QLineEdit::Password);
561
562     connect(buttonBox, SIGNAL(accepted()), SLOT(accept()));
563     connect(buttonBox, SIGNAL(rejected()), SLOT(reject()));
564     connect(channel, SIGNAL(textChanged(QString)), SLOT(on_channel_textChanged(QString)));
565
566     foreach(NetworkId id, Client::networkIds()) {
567         const Network *net = Client::network(id);
568         if (net->isConnected()) {
569             networks->addItem(net->networkName(), QVariant::fromValue<NetworkId>(id));
570         }
571     }
572
573     if (index.isValid()) {
574         NetworkId networkId = index.data(NetworkModel::NetworkIdRole).value<NetworkId>();
575         if (networkId.isValid()) {
576             networks->setCurrentIndex(networks->findText(Client::network(networkId)->networkName()));
577             if (index.data(NetworkModel::BufferTypeRole) == BufferInfo::ChannelBuffer
578                 && !index.data(NetworkModel::ItemActiveRole).toBool())
579                 channel->setText(index.data(Qt::DisplayRole).toString());
580         }
581     }
582 }
583
584
585 NetworkId NetworkModelController::JoinDlg::networkId() const
586 {
587     return networks->itemData(networks->currentIndex()).value<NetworkId>();
588 }
589
590
591 QString NetworkModelController::JoinDlg::channelName() const
592 {
593     return channel->text();
594 }
595
596
597 QString NetworkModelController::JoinDlg::channelPassword() const
598 {
599     return password->text();
600 }
601
602
603 void NetworkModelController::JoinDlg::on_channel_textChanged(const QString &text)
604 {
605     buttonBox->button(QDialogButtonBox::Ok)->setEnabled(!text.isEmpty());
606 }