File indexing completed on 2024-05-05 04:59:14

 /***************************************************************************
  *   Copyright (C) 2008 by Joris Guisson and Ivan Vasic                    *
  *   joris.guisson@gmail.com                                               *
  *   ivasic@gmail.com                                                      *
  *                                                                         *
  *   This program is free software; you can redistribute it and/or modify  *
  *   it under the terms of the GNU General Public License as published by  *
  *   the Free Software Foundation; either version 2 of the License, or     *
  *   (at your option) any later version.                                   *
  *                                                                         *
  *   This program is distributed in the hope that it will be useful,       *
  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  *   GNU General Public License for more details.                          *
  *                                                                         *
  *   You should have received a copy of the GNU General Public License     *
  *   along with this program; if not, write to the                         *
  *   Free Software Foundation, Inc.,                                       *
  *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.          *
  ***************************************************************************/
 #include "webseedstab.h"
 
 #include <QHeaderView>
 
 #include <KConfigGroup>
 #include <KMessageBox>
 
 #include "webseedsmodel.h"
 #include <interfaces/torrentinterface.h>
 #include <interfaces/webseedinterface.h>
 
 using namespace bt;
 
 namespace kt
 {
 
 WebSeedsTab::WebSeedsTab(QWidget *parent)
     : QWidget(parent)
     , curr_tc(nullptr)
 {
     setupUi(this);
     connect(m_add, &QAbstractButton::clicked, this, &WebSeedsTab::addWebSeed);
     connect(m_remove, &QAbstractButton::clicked, this, &WebSeedsTab::removeWebSeed);
     m_add->setIcon(QIcon::fromTheme("list-add"));
     m_remove->setIcon(QIcon::fromTheme("list-remove"));
     m_add->setEnabled(false);
     m_remove->setEnabled(false);
     m_webseed_list->setEnabled(false);
     model = new WebSeedsModel(this);
     proxy_model = new QSortFilterProxyModel(this);
     proxy_model->setSourceModel(model);
     proxy_model->setSortRole(Qt::UserRole);
     m_webseed_list->setModel(proxy_model);
     m_webseed_list->setSortingEnabled(true);
 
     connect(m_webseed_list->selectionModel(),
             SIGNAL(selectionChanged(QItemSelection, QItemSelection)),
             this,
             SLOT(selectionChanged(QItemSelection, QItemSelection)));
 
     connect(m_webseed, &QLineEdit::textChanged, this, &WebSeedsTab::onWebSeedTextChanged);
 }
 
 WebSeedsTab::~WebSeedsTab()
 {
 }
 
 void WebSeedsTab::changeTC(bt::TorrentInterface *tc)
 {
     curr_tc = tc;
     model->changeTC(tc);
     m_add->setEnabled(curr_tc != nullptr);
     m_remove->setEnabled(curr_tc != nullptr);
     m_webseed_list->setEnabled(curr_tc != nullptr);
     m_webseed->setEnabled(curr_tc != nullptr);
     onWebSeedTextChanged(m_webseed->text());
 
     // see if we need to enable or disable the remove button
     if (curr_tc)
         selectionChanged(m_webseed_list->selectionModel()->selectedRows());
 }
 
 void WebSeedsTab::addWebSeed()
 {
     if (!curr_tc)
         return;
 
     QUrl url(m_webseed->text());
     if (curr_tc != nullptr && url.isValid() && url.scheme() == "http") {
         if (curr_tc->addWebSeed(url)) {
             model->changeTC(curr_tc);
             m_webseed->clear();
         } else {
             KMessageBox::error(this, i18n("Cannot add the webseed %1, it is already part of the list of webseeds.", url.toDisplayString()));
         }
     }
 }
 
 void WebSeedsTab::removeWebSeed()
 {
     if (!curr_tc)
         return;
 
     QModelIndexList idx_list = m_webseed_list->selectionModel()->selectedRows();
     foreach (const QModelIndex &idx, idx_list) {
         const WebSeedInterface *ws = curr_tc->getWebSeed(proxy_model->mapToSource(idx).row());
         if (ws && ws->isUserCreated()) {
             if (!curr_tc->removeWebSeed(ws->getUrl()))
                 KMessageBox::error(this, i18n("Cannot remove webseed %1, it is part of the torrent.", ws->getUrl().toDisplayString()));
         }
     }
 
     model->changeTC(curr_tc);
 }
 
 void WebSeedsTab::selectionChanged(const QModelIndexList &indexes)
 {
     foreach (const QModelIndex &idx, indexes) {
         const WebSeedInterface *ws = curr_tc->getWebSeed(proxy_model->mapToSource(idx).row());
         if (ws && ws->isUserCreated()) {
             m_remove->setEnabled(true);
             return;
         }
     }
 
     m_remove->setEnabled(false);
 }
 
 void WebSeedsTab::selectionChanged(const QItemSelection &selected, const QItemSelection &deselected)
 {
     Q_UNUSED(deselected)
     if (!curr_tc)
         return;
 
     selectionChanged(selected.indexes());
 }
 
 void WebSeedsTab::onWebSeedTextChanged(const QString &ws)
 {
     QUrl url(ws);
     m_add->setEnabled(curr_tc != nullptr && url.isValid() && url.scheme() == "http");
 }
 
 void WebSeedsTab::update()
 {
     if (model->update())
         proxy_model->invalidate();
 }
 
 void WebSeedsTab::saveState(KSharedConfigPtr cfg)
 {
     KConfigGroup g = cfg->group("WebSeedsTab");
     QByteArray s = m_webseed_list->header()->saveState();
     g.writeEntry("state", s.toBase64());
 }
 
 void WebSeedsTab::loadState(KSharedConfigPtr cfg)
 {
     KConfigGroup g = cfg->group("WebSeedsTab");
     QByteArray s = QByteArray::fromBase64(g.readEntry("state", QByteArray()));
     if (!s.isNull())
         m_webseed_list->header()->restoreState(s);
 }
 }
 
 #include "moc_webseedstab.cpp"