|
|
|
|
@ -1,391 +1,391 @@
@@ -1,391 +1,391 @@
|
|
|
|
|
/*
|
|
|
|
|
* Bittorrent Client using Qt4 and libtorrent. |
|
|
|
|
* Copyright (C) 2006 Christophe Dumez |
|
|
|
|
* |
|
|
|
|
* 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. |
|
|
|
|
* |
|
|
|
|
* In addition, as a special exception, the copyright holders give permission to |
|
|
|
|
* link this program with the OpenSSL project's "OpenSSL" library (or with |
|
|
|
|
* modified versions of it that use the same license as the "OpenSSL" library), |
|
|
|
|
* and distribute the linked executables. You must obey the GNU General Public |
|
|
|
|
* License in all respects for all of the code used other than "OpenSSL". If you |
|
|
|
|
* modify file(s), you may extend this exception to your version of the file(s), |
|
|
|
|
* but you are not obligated to do so. If you do not wish to do so, delete this |
|
|
|
|
* exception statement from your version. |
|
|
|
|
* |
|
|
|
|
* Contact : chris@qbittorrent.org |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
#include <QTreeWidgetItem> |
|
|
|
|
#include <QStringList> |
|
|
|
|
#include <QMenu> |
|
|
|
|
#include <QHash> |
|
|
|
|
#include <QAction> |
|
|
|
|
#include <QColor> |
|
|
|
|
#include <libtorrent/version.hpp> |
|
|
|
|
|
|
|
|
|
#include "trackerlist.h" |
|
|
|
|
#include "propertieswidget.h" |
|
|
|
|
#include "trackersadditiondlg.h" |
|
|
|
|
#include "misc.h" |
|
|
|
|
#include "bittorrent.h" |
|
|
|
|
#include "qinisettings.h" |
|
|
|
|
|
|
|
|
|
TrackerList::TrackerList(PropertiesWidget *properties): QTreeWidget(), properties(properties) { |
|
|
|
|
// Graphical settings
|
|
|
|
|
setRootIsDecorated(false); |
|
|
|
|
setAllColumnsShowFocus(true); |
|
|
|
|
setItemsExpandable(false); |
|
|
|
|
setSelectionMode(QAbstractItemView::ExtendedSelection); |
|
|
|
|
// Context menu
|
|
|
|
|
setContextMenuPolicy(Qt::CustomContextMenu); |
|
|
|
|
connect(this, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(showTrackerListMenu(QPoint))); |
|
|
|
|
// Set header
|
|
|
|
|
QStringList header; |
|
|
|
|
header << tr("URL"); |
|
|
|
|
header << tr("Status"); |
|
|
|
|
header << tr("Peers"); |
|
|
|
|
header << tr("Message"); |
|
|
|
|
setHeaderItem(new QTreeWidgetItem(header)); |
|
|
|
|
dht_item = new QTreeWidgetItem(QStringList("** "+tr("[DHT]")+" **")); |
|
|
|
|
insertTopLevelItem(0, dht_item); |
|
|
|
|
setRowColor(0, QColor("grey")); |
|
|
|
|
pex_item = new QTreeWidgetItem(QStringList("** "+tr("[PeX]")+" **")); |
|
|
|
|
insertTopLevelItem(1, pex_item); |
|
|
|
|
setRowColor(1, QColor("grey")); |
|
|
|
|
lsd_item = new QTreeWidgetItem(QStringList("** "+tr("[LSD]")+" **")); |
|
|
|
|
insertTopLevelItem(2, lsd_item); |
|
|
|
|
setRowColor(2, QColor("grey")); |
|
|
|
|
loadSettings(); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
TrackerList::~TrackerList() { |
|
|
|
|
saveSettings(); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
QList<QTreeWidgetItem*> TrackerList::getSelectedTrackerItems() const { |
|
|
|
|
QList<QTreeWidgetItem*> selected_items = selectedItems(); |
|
|
|
|
QList<QTreeWidgetItem*> selected_trackers; |
|
|
|
|
foreach(QTreeWidgetItem *item, selectedItems()) { |
|
|
|
|
if(indexOfTopLevelItem(item) >= NB_STICKY_ITEM) { // Ignore STICKY ITEMS
|
|
|
|
|
selected_trackers << item; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
return selected_trackers; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::setRowColor(int row, QColor color) { |
|
|
|
|
unsigned int nbColumns = columnCount(); |
|
|
|
|
QTreeWidgetItem *item = topLevelItem(row); |
|
|
|
|
for(unsigned int i=0; i<nbColumns; ++i) { |
|
|
|
|
item->setData(i, Qt::ForegroundRole, color); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::moveSelectionUp() { |
|
|
|
|
#if LIBTORRENT_VERSION_MINOR < 15 |
|
|
|
|
QTorrentHandle h = properties->getCurrentTorrent(); |
|
|
|
|
if(!h.is_valid()) { |
|
|
|
|
clear(); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
QList<QTreeWidgetItem *> selected_items = getSelectedTrackerItems(); |
|
|
|
|
if(selected_items.isEmpty()) return; |
|
|
|
|
bool change = false; |
|
|
|
|
foreach(QTreeWidgetItem *item, selected_items){ |
|
|
|
|
int index = indexOfTopLevelItem(item); |
|
|
|
|
if(index > NB_STICKY_ITEM) { |
|
|
|
|
insertTopLevelItem(index-1, takeTopLevelItem(index)); |
|
|
|
|
change = true; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
if(!change) return; |
|
|
|
|
// Restore selection
|
|
|
|
|
QItemSelectionModel *selection = selectionModel(); |
|
|
|
|
foreach(QTreeWidgetItem *item, selected_items) { |
|
|
|
|
selection->select(indexFromItem(item), QItemSelectionModel::Rows|QItemSelectionModel::Select); |
|
|
|
|
} |
|
|
|
|
setSelectionModel(selection); |
|
|
|
|
// Update torrent trackers
|
|
|
|
|
std::vector<announce_entry> trackers; |
|
|
|
|
for(int i=NB_STICKY_ITEM; i<topLevelItemCount(); ++i) { |
|
|
|
|
QString tracker_url = topLevelItem(i)->data(COL_URL, Qt::DisplayRole).toString(); |
|
|
|
|
announce_entry e(tracker_url.toStdString()); |
|
|
|
|
e.tier = i-NB_STICKY_ITEM; |
|
|
|
|
trackers.push_back(e); |
|
|
|
|
} |
|
|
|
|
h.replace_trackers(trackers); |
|
|
|
|
// Reannounce
|
|
|
|
|
h.force_reannounce(); |
|
|
|
|
#endif |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::moveSelectionDown() { |
|
|
|
|
#if LIBTORRENT_VERSION_MINOR < 15 |
|
|
|
|
QTorrentHandle h = properties->getCurrentTorrent(); |
|
|
|
|
if(!h.is_valid()) { |
|
|
|
|
clear(); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
QList<QTreeWidgetItem *> selected_items = getSelectedTrackerItems(); |
|
|
|
|
if(selected_items.isEmpty()) return; |
|
|
|
|
bool change = false; |
|
|
|
|
for(int i=selectedItems().size()-1; i>= 0; --i) { |
|
|
|
|
int index = indexOfTopLevelItem(selected_items.at(i)); |
|
|
|
|
if(index < topLevelItemCount()-1) { |
|
|
|
|
insertTopLevelItem(index+1, takeTopLevelItem(index)); |
|
|
|
|
change = true; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
if(!change) return; |
|
|
|
|
// Restore selection
|
|
|
|
|
QItemSelectionModel *selection = selectionModel(); |
|
|
|
|
foreach(QTreeWidgetItem *item, selected_items) { |
|
|
|
|
selection->select(indexFromItem(item), QItemSelectionModel::Rows|QItemSelectionModel::Select); |
|
|
|
|
} |
|
|
|
|
setSelectionModel(selection); |
|
|
|
|
// Update torrent trackers
|
|
|
|
|
std::vector<announce_entry> trackers; |
|
|
|
|
for(int i=NB_STICKY_ITEM; i<topLevelItemCount(); ++i) { |
|
|
|
|
QString tracker_url = topLevelItem(i)->data(COL_URL, Qt::DisplayRole).toString(); |
|
|
|
|
announce_entry e(tracker_url.toStdString()); |
|
|
|
|
e.tier = i-NB_STICKY_ITEM; |
|
|
|
|
trackers.push_back(e); |
|
|
|
|
} |
|
|
|
|
h.replace_trackers(trackers); |
|
|
|
|
// Reannounce
|
|
|
|
|
h.force_reannounce(); |
|
|
|
|
#endif |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::clear() { |
|
|
|
|
qDeleteAll(tracker_items.values()); |
|
|
|
|
tracker_items.clear(); |
|
|
|
|
dht_item->setText(COL_PEERS, ""); |
|
|
|
|
dht_item->setText(COL_STATUS, ""); |
|
|
|
|
dht_item->setText(COL_MSG, ""); |
|
|
|
|
pex_item->setText(COL_PEERS, ""); |
|
|
|
|
pex_item->setText(COL_STATUS, ""); |
|
|
|
|
pex_item->setText(COL_MSG, ""); |
|
|
|
|
lsd_item->setText(COL_PEERS, ""); |
|
|
|
|
lsd_item->setText(COL_STATUS, ""); |
|
|
|
|
lsd_item->setText(COL_MSG, ""); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::loadStickyItems(const QTorrentHandle &h) { |
|
|
|
|
// XXX: libtorrent should provide this info...
|
|
|
|
|
// Count peers from DHT, LSD, PeX
|
|
|
|
|
uint nb_dht=0, nb_lsd=0, nb_pex=0; |
|
|
|
|
std::vector<peer_info> peers; |
|
|
|
|
h.get_peer_info(peers); |
|
|
|
|
std::vector<peer_info>::iterator it; |
|
|
|
|
for(it=peers.begin(); it!=peers.end(); it++) { |
|
|
|
|
if(it->source & peer_info::dht) |
|
|
|
|
++nb_dht; |
|
|
|
|
if(it->source & peer_info::lsd) |
|
|
|
|
++nb_lsd; |
|
|
|
|
if(it->source & peer_info::pex) |
|
|
|
|
++nb_pex; |
|
|
|
|
} |
|
|
|
|
// load DHT information
|
|
|
|
|
if(properties->getBTSession()->isDHTEnabled() && h.has_metadata() && !h.priv()) { |
|
|
|
|
dht_item->setText(COL_STATUS, tr("Working")); |
|
|
|
|
} else { |
|
|
|
|
dht_item->setText(COL_STATUS, tr("Disabled")); |
|
|
|
|
} |
|
|
|
|
dht_item->setText(COL_PEERS, QString::number(nb_dht)); |
|
|
|
|
if(h.has_metadata() && h.priv()) { |
|
|
|
|
dht_item->setText(COL_MSG, tr("This torrent is private")); |
|
|
|
|
} |
|
|
|
|
// Load PeX Information
|
|
|
|
|
if(properties->getBTSession()->isPexEnabled()) |
|
|
|
|
pex_item->setText(COL_STATUS, tr("Working")); |
|
|
|
|
else |
|
|
|
|
pex_item->setText(COL_STATUS, tr("Disabled")); |
|
|
|
|
pex_item->setText(COL_PEERS, QString::number(nb_pex)); |
|
|
|
|
// Load LSD Information
|
|
|
|
|
if(properties->getBTSession()->isLSDEnabled()) |
|
|
|
|
lsd_item->setText(COL_STATUS, tr("Working")); |
|
|
|
|
else |
|
|
|
|
lsd_item->setText(COL_STATUS, tr("Disabled")); |
|
|
|
|
lsd_item->setText(COL_PEERS, QString::number(nb_lsd)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::loadTrackers() { |
|
|
|
|
// Load trackers from torrent handle
|
|
|
|
|
QTorrentHandle h = properties->getCurrentTorrent(); |
|
|
|
|
if(!h.is_valid()) return; |
|
|
|
|
loadStickyItems(h); |
|
|
|
|
// Load actual trackers information
|
|
|
|
|
QHash<QString, TrackerInfos> trackers_data = properties->getBTSession()->getTrackersInfo(h.hash()); |
|
|
|
|
QStringList old_trackers_urls = tracker_items.keys(); |
|
|
|
|
const std::vector<announce_entry> trackers = h.trackers(); |
|
|
|
|
for(std::vector<announce_entry>::const_iterator it = trackers.begin(); it != trackers.end(); it++) { |
|
|
|
|
QString tracker_url = misc::toQString(it->url); |
|
|
|
|
QTreeWidgetItem *item = tracker_items.value(tracker_url, 0); |
|
|
|
|
if(!item) { |
|
|
|
|
item = new QTreeWidgetItem(); |
|
|
|
|
item->setText(COL_URL, tracker_url); |
|
|
|
|
addTopLevelItem(item); |
|
|
|
|
tracker_items[tracker_url] = item; |
|
|
|
|
} else { |
|
|
|
|
old_trackers_urls.removeOne(tracker_url); |
|
|
|
|
} |
|
|
|
|
TrackerInfos data = trackers_data.value(tracker_url, TrackerInfos(tracker_url)); |
|
|
|
|
QString error_message = data.last_message.trimmed(); |
|
|
|
|
#if LIBTORRENT_VERSION_MINOR > 14 |
|
|
|
|
if(it->verified) { |
|
|
|
|
item->setText(COL_STATUS, tr("Working")); |
|
|
|
|
item->setText(COL_MSG, ""); |
|
|
|
|
} else { |
|
|
|
|
if(it->updating && it->fails == 0) { |
|
|
|
|
item->setText(COL_STATUS, tr("Updating...")); |
|
|
|
|
item->setText(COL_MSG, ""); |
|
|
|
|
} else { |
|
|
|
|
if(it->fails > 0) { |
|
|
|
|
item->setText(COL_STATUS, tr("Not working")); |
|
|
|
|
item->setText(COL_MSG, error_message); |
|
|
|
|
} else { |
|
|
|
|
item->setText(COL_STATUS, tr("Not contacted yet")); |
|
|
|
|
item->setText(COL_MSG, ""); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
#else |
|
|
|
|
if(data.verified) { |
|
|
|
|
item->setText(COL_STATUS, tr("Working")); |
|
|
|
|
item->setText(COL_MSG, ""); |
|
|
|
|
} else { |
|
|
|
|
if(data.fail_count > 0) { |
|
|
|
|
item->setText(COL_STATUS, tr("Not working")); |
|
|
|
|
item->setText(COL_MSG, error_message); |
|
|
|
|
} else { |
|
|
|
|
item->setText(COL_STATUS, tr("Not contacted yet")); |
|
|
|
|
item->setText(COL_MSG, ""); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
#endif |
|
|
|
|
item->setText(COL_PEERS, QString::number(trackers_data.value(tracker_url, TrackerInfos(tracker_url)).num_peers)); |
|
|
|
|
} |
|
|
|
|
// Remove old trackers
|
|
|
|
|
foreach(const QString &tracker, old_trackers_urls) { |
|
|
|
|
delete tracker_items.take(tracker); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Ask the user for new trackers and add them to the torrent
|
|
|
|
|
void TrackerList::askForTrackers(){ |
|
|
|
|
QTorrentHandle h = properties->getCurrentTorrent(); |
|
|
|
|
if(!h.is_valid()) return; |
|
|
|
|
QStringList trackers = TrackersAdditionDlg::askForTrackers(h); |
|
|
|
|
if(!trackers.empty()) { |
|
|
|
|
foreach(const QString& tracker, trackers) { |
|
|
|
|
announce_entry url(tracker.toStdString()); |
|
|
|
|
url.tier = 0; |
|
|
|
|
h.add_tracker(url); |
|
|
|
|
} |
|
|
|
|
// Reannounce to new trackers
|
|
|
|
|
h.force_reannounce(); |
|
|
|
|
// Reload tracker list
|
|
|
|
|
loadTrackers(); |
|
|
|
|
// XXX: I don't think this is necessary now
|
|
|
|
|
//BTSession->saveTrackerFile(h.hash());
|
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::deleteSelectedTrackers(){ |
|
|
|
|
QTorrentHandle h = properties->getCurrentTorrent(); |
|
|
|
|
if(!h.is_valid()) { |
|
|
|
|
clear(); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
QList<QTreeWidgetItem *> selected_items = getSelectedTrackerItems(); |
|
|
|
|
if(selected_items.isEmpty()) return; |
|
|
|
|
QStringList urls_to_remove; |
|
|
|
|
foreach(QTreeWidgetItem *item, selected_items){ |
|
|
|
|
QString tracker_url = item->data(COL_URL, Qt::DisplayRole).toString(); |
|
|
|
|
urls_to_remove << tracker_url; |
|
|
|
|
tracker_items.remove(tracker_url); |
|
|
|
|
delete item; |
|
|
|
|
} |
|
|
|
|
// Iterate of trackers and remove selected ones
|
|
|
|
|
std::vector<announce_entry> trackers = h.trackers(); |
|
|
|
|
std::vector<announce_entry>::iterator it = trackers.begin(); |
|
|
|
|
while(it != trackers.end()) { |
|
|
|
|
int index = urls_to_remove.indexOf(misc::toQString((*it).url)); |
|
|
|
|
if(index >= 0) { |
|
|
|
|
trackers.erase(it); |
|
|
|
|
urls_to_remove.removeAt(index); |
|
|
|
|
} else { |
|
|
|
|
it++; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
h.replace_trackers(trackers); |
|
|
|
|
h.force_reannounce(); |
|
|
|
|
// Reload Trackers
|
|
|
|
|
loadTrackers(); |
|
|
|
|
//XXX: I don't think this is necessary
|
|
|
|
|
//BTSession->saveTrackerFile(h.hash());
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::showTrackerListMenu(QPoint) { |
|
|
|
|
QTorrentHandle h = properties->getCurrentTorrent(); |
|
|
|
|
if(!h.is_valid() || !h.has_metadata()) return; |
|
|
|
|
//QList<QTreeWidgetItem*> selected_items = getSelectedTrackerItems();
|
|
|
|
|
QMenu menu; |
|
|
|
|
// Add actions
|
|
|
|
|
QAction *addAct = menu.addAction(QIcon(":/Icons/oxygen/list-add.png"), tr("Add a new tracker...")); |
|
|
|
|
QAction *delAct = 0; |
|
|
|
|
if(!getSelectedTrackerItems().isEmpty()) { |
|
|
|
|
delAct = menu.addAction(QIcon(":/Icons/oxygen/list-remove.png"), tr("Remove tracker")); |
|
|
|
|
} |
|
|
|
|
menu.addSeparator(); |
|
|
|
|
QAction *reannounceAct = menu.addAction(QIcon(":/Icons/oxygen/run-build.png"), tr("Force reannounce")); |
|
|
|
|
QAction *act = menu.exec(QCursor::pos()); |
|
|
|
|
if(act == 0) return; |
|
|
|
|
if(act == addAct) { |
|
|
|
|
askForTrackers(); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
if(act == delAct) { |
|
|
|
|
deleteSelectedTrackers(); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
if(act == reannounceAct) { |
|
|
|
|
properties->getCurrentTorrent().force_reannounce(); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::loadSettings() { |
|
|
|
|
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent")); |
|
|
|
|
QList<int> contentColsWidths = misc::intListfromStringList(settings.value(QString::fromUtf8("TorrentProperties/Trackers/trackersColsWidth")).toStringList()); |
|
|
|
|
if(!contentColsWidths.empty()) { |
|
|
|
|
for(int i=0; i<contentColsWidths.size(); ++i) { |
|
|
|
|
setColumnWidth(i, contentColsWidths.at(i)); |
|
|
|
|
} |
|
|
|
|
} else { |
|
|
|
|
setColumnWidth(0, 300); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::saveSettings() const { |
|
|
|
|
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent")); |
|
|
|
|
QStringList contentColsWidths; |
|
|
|
|
for(int i=0; i<columnCount(); ++i) { |
|
|
|
|
contentColsWidths << QString::number(columnWidth(i)); |
|
|
|
|
} |
|
|
|
|
settings.setValue(QString::fromUtf8("TorrentProperties/Trackers/trackersColsWidth"), contentColsWidths); |
|
|
|
|
} |
|
|
|
|
/*
|
|
|
|
|
* Bittorrent Client using Qt4 and libtorrent. |
|
|
|
|
* Copyright (C) 2006 Christophe Dumez |
|
|
|
|
* |
|
|
|
|
* 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. |
|
|
|
|
* |
|
|
|
|
* In addition, as a special exception, the copyright holders give permission to |
|
|
|
|
* link this program with the OpenSSL project's "OpenSSL" library (or with |
|
|
|
|
* modified versions of it that use the same license as the "OpenSSL" library), |
|
|
|
|
* and distribute the linked executables. You must obey the GNU General Public |
|
|
|
|
* License in all respects for all of the code used other than "OpenSSL". If you |
|
|
|
|
* modify file(s), you may extend this exception to your version of the file(s), |
|
|
|
|
* but you are not obligated to do so. If you do not wish to do so, delete this |
|
|
|
|
* exception statement from your version. |
|
|
|
|
* |
|
|
|
|
* Contact : chris@qbittorrent.org |
|
|
|
|
*/ |
|
|
|
|
|
|
|
|
|
#include <QTreeWidgetItem> |
|
|
|
|
#include <QStringList> |
|
|
|
|
#include <QMenu> |
|
|
|
|
#include <QHash> |
|
|
|
|
#include <QAction> |
|
|
|
|
#include <QColor> |
|
|
|
|
#include <libtorrent/version.hpp> |
|
|
|
|
|
|
|
|
|
#include "trackerlist.h" |
|
|
|
|
#include "propertieswidget.h" |
|
|
|
|
#include "trackersadditiondlg.h" |
|
|
|
|
#include "misc.h" |
|
|
|
|
#include "qbtsession.h" |
|
|
|
|
#include "qinisettings.h" |
|
|
|
|
|
|
|
|
|
TrackerList::TrackerList(PropertiesWidget *properties): QTreeWidget(), properties(properties) { |
|
|
|
|
// Graphical settings
|
|
|
|
|
setRootIsDecorated(false); |
|
|
|
|
setAllColumnsShowFocus(true); |
|
|
|
|
setItemsExpandable(false); |
|
|
|
|
setSelectionMode(QAbstractItemView::ExtendedSelection); |
|
|
|
|
// Context menu
|
|
|
|
|
setContextMenuPolicy(Qt::CustomContextMenu); |
|
|
|
|
connect(this, SIGNAL(customContextMenuRequested(QPoint)), this, SLOT(showTrackerListMenu(QPoint))); |
|
|
|
|
// Set header
|
|
|
|
|
QStringList header; |
|
|
|
|
header << tr("URL"); |
|
|
|
|
header << tr("Status"); |
|
|
|
|
header << tr("Peers"); |
|
|
|
|
header << tr("Message"); |
|
|
|
|
setHeaderItem(new QTreeWidgetItem(header)); |
|
|
|
|
dht_item = new QTreeWidgetItem(QStringList("** "+tr("[DHT]")+" **")); |
|
|
|
|
insertTopLevelItem(0, dht_item); |
|
|
|
|
setRowColor(0, QColor("grey")); |
|
|
|
|
pex_item = new QTreeWidgetItem(QStringList("** "+tr("[PeX]")+" **")); |
|
|
|
|
insertTopLevelItem(1, pex_item); |
|
|
|
|
setRowColor(1, QColor("grey")); |
|
|
|
|
lsd_item = new QTreeWidgetItem(QStringList("** "+tr("[LSD]")+" **")); |
|
|
|
|
insertTopLevelItem(2, lsd_item); |
|
|
|
|
setRowColor(2, QColor("grey")); |
|
|
|
|
loadSettings(); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
TrackerList::~TrackerList() { |
|
|
|
|
saveSettings(); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
QList<QTreeWidgetItem*> TrackerList::getSelectedTrackerItems() const { |
|
|
|
|
QList<QTreeWidgetItem*> selected_items = selectedItems(); |
|
|
|
|
QList<QTreeWidgetItem*> selected_trackers; |
|
|
|
|
foreach(QTreeWidgetItem *item, selectedItems()) { |
|
|
|
|
if(indexOfTopLevelItem(item) >= NB_STICKY_ITEM) { // Ignore STICKY ITEMS
|
|
|
|
|
selected_trackers << item; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
return selected_trackers; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::setRowColor(int row, QColor color) { |
|
|
|
|
unsigned int nbColumns = columnCount(); |
|
|
|
|
QTreeWidgetItem *item = topLevelItem(row); |
|
|
|
|
for(unsigned int i=0; i<nbColumns; ++i) { |
|
|
|
|
item->setData(i, Qt::ForegroundRole, color); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::moveSelectionUp() { |
|
|
|
|
#if LIBTORRENT_VERSION_MINOR < 15 |
|
|
|
|
QTorrentHandle h = properties->getCurrentTorrent(); |
|
|
|
|
if(!h.is_valid()) { |
|
|
|
|
clear(); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
QList<QTreeWidgetItem *> selected_items = getSelectedTrackerItems(); |
|
|
|
|
if(selected_items.isEmpty()) return; |
|
|
|
|
bool change = false; |
|
|
|
|
foreach(QTreeWidgetItem *item, selected_items){ |
|
|
|
|
int index = indexOfTopLevelItem(item); |
|
|
|
|
if(index > NB_STICKY_ITEM) { |
|
|
|
|
insertTopLevelItem(index-1, takeTopLevelItem(index)); |
|
|
|
|
change = true; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
if(!change) return; |
|
|
|
|
// Restore selection
|
|
|
|
|
QItemSelectionModel *selection = selectionModel(); |
|
|
|
|
foreach(QTreeWidgetItem *item, selected_items) { |
|
|
|
|
selection->select(indexFromItem(item), QItemSelectionModel::Rows|QItemSelectionModel::Select); |
|
|
|
|
} |
|
|
|
|
setSelectionModel(selection); |
|
|
|
|
// Update torrent trackers
|
|
|
|
|
std::vector<announce_entry> trackers; |
|
|
|
|
for(int i=NB_STICKY_ITEM; i<topLevelItemCount(); ++i) { |
|
|
|
|
QString tracker_url = topLevelItem(i)->data(COL_URL, Qt::DisplayRole).toString(); |
|
|
|
|
announce_entry e(tracker_url.toStdString()); |
|
|
|
|
e.tier = i-NB_STICKY_ITEM; |
|
|
|
|
trackers.push_back(e); |
|
|
|
|
} |
|
|
|
|
h.replace_trackers(trackers); |
|
|
|
|
// Reannounce
|
|
|
|
|
h.force_reannounce(); |
|
|
|
|
#endif |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::moveSelectionDown() { |
|
|
|
|
#if LIBTORRENT_VERSION_MINOR < 15 |
|
|
|
|
QTorrentHandle h = properties->getCurrentTorrent(); |
|
|
|
|
if(!h.is_valid()) { |
|
|
|
|
clear(); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
QList<QTreeWidgetItem *> selected_items = getSelectedTrackerItems(); |
|
|
|
|
if(selected_items.isEmpty()) return; |
|
|
|
|
bool change = false; |
|
|
|
|
for(int i=selectedItems().size()-1; i>= 0; --i) { |
|
|
|
|
int index = indexOfTopLevelItem(selected_items.at(i)); |
|
|
|
|
if(index < topLevelItemCount()-1) { |
|
|
|
|
insertTopLevelItem(index+1, takeTopLevelItem(index)); |
|
|
|
|
change = true; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
if(!change) return; |
|
|
|
|
// Restore selection
|
|
|
|
|
QItemSelectionModel *selection = selectionModel(); |
|
|
|
|
foreach(QTreeWidgetItem *item, selected_items) { |
|
|
|
|
selection->select(indexFromItem(item), QItemSelectionModel::Rows|QItemSelectionModel::Select); |
|
|
|
|
} |
|
|
|
|
setSelectionModel(selection); |
|
|
|
|
// Update torrent trackers
|
|
|
|
|
std::vector<announce_entry> trackers; |
|
|
|
|
for(int i=NB_STICKY_ITEM; i<topLevelItemCount(); ++i) { |
|
|
|
|
QString tracker_url = topLevelItem(i)->data(COL_URL, Qt::DisplayRole).toString(); |
|
|
|
|
announce_entry e(tracker_url.toStdString()); |
|
|
|
|
e.tier = i-NB_STICKY_ITEM; |
|
|
|
|
trackers.push_back(e); |
|
|
|
|
} |
|
|
|
|
h.replace_trackers(trackers); |
|
|
|
|
// Reannounce
|
|
|
|
|
h.force_reannounce(); |
|
|
|
|
#endif |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::clear() { |
|
|
|
|
qDeleteAll(tracker_items.values()); |
|
|
|
|
tracker_items.clear(); |
|
|
|
|
dht_item->setText(COL_PEERS, ""); |
|
|
|
|
dht_item->setText(COL_STATUS, ""); |
|
|
|
|
dht_item->setText(COL_MSG, ""); |
|
|
|
|
pex_item->setText(COL_PEERS, ""); |
|
|
|
|
pex_item->setText(COL_STATUS, ""); |
|
|
|
|
pex_item->setText(COL_MSG, ""); |
|
|
|
|
lsd_item->setText(COL_PEERS, ""); |
|
|
|
|
lsd_item->setText(COL_STATUS, ""); |
|
|
|
|
lsd_item->setText(COL_MSG, ""); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::loadStickyItems(const QTorrentHandle &h) { |
|
|
|
|
// XXX: libtorrent should provide this info...
|
|
|
|
|
// Count peers from DHT, LSD, PeX
|
|
|
|
|
uint nb_dht=0, nb_lsd=0, nb_pex=0; |
|
|
|
|
std::vector<peer_info> peers; |
|
|
|
|
h.get_peer_info(peers); |
|
|
|
|
std::vector<peer_info>::iterator it; |
|
|
|
|
for(it=peers.begin(); it!=peers.end(); it++) { |
|
|
|
|
if(it->source & peer_info::dht) |
|
|
|
|
++nb_dht; |
|
|
|
|
if(it->source & peer_info::lsd) |
|
|
|
|
++nb_lsd; |
|
|
|
|
if(it->source & peer_info::pex) |
|
|
|
|
++nb_pex; |
|
|
|
|
} |
|
|
|
|
// load DHT information
|
|
|
|
|
if(properties->getBTSession()->isDHTEnabled() && h.has_metadata() && !h.priv()) { |
|
|
|
|
dht_item->setText(COL_STATUS, tr("Working")); |
|
|
|
|
} else { |
|
|
|
|
dht_item->setText(COL_STATUS, tr("Disabled")); |
|
|
|
|
} |
|
|
|
|
dht_item->setText(COL_PEERS, QString::number(nb_dht)); |
|
|
|
|
if(h.has_metadata() && h.priv()) { |
|
|
|
|
dht_item->setText(COL_MSG, tr("This torrent is private")); |
|
|
|
|
} |
|
|
|
|
// Load PeX Information
|
|
|
|
|
if(properties->getBTSession()->isPexEnabled()) |
|
|
|
|
pex_item->setText(COL_STATUS, tr("Working")); |
|
|
|
|
else |
|
|
|
|
pex_item->setText(COL_STATUS, tr("Disabled")); |
|
|
|
|
pex_item->setText(COL_PEERS, QString::number(nb_pex)); |
|
|
|
|
// Load LSD Information
|
|
|
|
|
if(properties->getBTSession()->isLSDEnabled()) |
|
|
|
|
lsd_item->setText(COL_STATUS, tr("Working")); |
|
|
|
|
else |
|
|
|
|
lsd_item->setText(COL_STATUS, tr("Disabled")); |
|
|
|
|
lsd_item->setText(COL_PEERS, QString::number(nb_lsd)); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::loadTrackers() { |
|
|
|
|
// Load trackers from torrent handle
|
|
|
|
|
QTorrentHandle h = properties->getCurrentTorrent(); |
|
|
|
|
if(!h.is_valid()) return; |
|
|
|
|
loadStickyItems(h); |
|
|
|
|
// Load actual trackers information
|
|
|
|
|
QHash<QString, TrackerInfos> trackers_data = properties->getBTSession()->getTrackersInfo(h.hash()); |
|
|
|
|
QStringList old_trackers_urls = tracker_items.keys(); |
|
|
|
|
const std::vector<announce_entry> trackers = h.trackers(); |
|
|
|
|
for(std::vector<announce_entry>::const_iterator it = trackers.begin(); it != trackers.end(); it++) { |
|
|
|
|
QString tracker_url = misc::toQString(it->url); |
|
|
|
|
QTreeWidgetItem *item = tracker_items.value(tracker_url, 0); |
|
|
|
|
if(!item) { |
|
|
|
|
item = new QTreeWidgetItem(); |
|
|
|
|
item->setText(COL_URL, tracker_url); |
|
|
|
|
addTopLevelItem(item); |
|
|
|
|
tracker_items[tracker_url] = item; |
|
|
|
|
} else { |
|
|
|
|
old_trackers_urls.removeOne(tracker_url); |
|
|
|
|
} |
|
|
|
|
TrackerInfos data = trackers_data.value(tracker_url, TrackerInfos(tracker_url)); |
|
|
|
|
QString error_message = data.last_message.trimmed(); |
|
|
|
|
#if LIBTORRENT_VERSION_MINOR > 14 |
|
|
|
|
if(it->verified) { |
|
|
|
|
item->setText(COL_STATUS, tr("Working")); |
|
|
|
|
item->setText(COL_MSG, ""); |
|
|
|
|
} else { |
|
|
|
|
if(it->updating && it->fails == 0) { |
|
|
|
|
item->setText(COL_STATUS, tr("Updating...")); |
|
|
|
|
item->setText(COL_MSG, ""); |
|
|
|
|
} else { |
|
|
|
|
if(it->fails > 0) { |
|
|
|
|
item->setText(COL_STATUS, tr("Not working")); |
|
|
|
|
item->setText(COL_MSG, error_message); |
|
|
|
|
} else { |
|
|
|
|
item->setText(COL_STATUS, tr("Not contacted yet")); |
|
|
|
|
item->setText(COL_MSG, ""); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
#else |
|
|
|
|
if(data.verified) { |
|
|
|
|
item->setText(COL_STATUS, tr("Working")); |
|
|
|
|
item->setText(COL_MSG, ""); |
|
|
|
|
} else { |
|
|
|
|
if(data.fail_count > 0) { |
|
|
|
|
item->setText(COL_STATUS, tr("Not working")); |
|
|
|
|
item->setText(COL_MSG, error_message); |
|
|
|
|
} else { |
|
|
|
|
item->setText(COL_STATUS, tr("Not contacted yet")); |
|
|
|
|
item->setText(COL_MSG, ""); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
#endif |
|
|
|
|
item->setText(COL_PEERS, QString::number(trackers_data.value(tracker_url, TrackerInfos(tracker_url)).num_peers)); |
|
|
|
|
} |
|
|
|
|
// Remove old trackers
|
|
|
|
|
foreach(const QString &tracker, old_trackers_urls) { |
|
|
|
|
delete tracker_items.take(tracker); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Ask the user for new trackers and add them to the torrent
|
|
|
|
|
void TrackerList::askForTrackers(){ |
|
|
|
|
QTorrentHandle h = properties->getCurrentTorrent(); |
|
|
|
|
if(!h.is_valid()) return; |
|
|
|
|
QStringList trackers = TrackersAdditionDlg::askForTrackers(h); |
|
|
|
|
if(!trackers.empty()) { |
|
|
|
|
foreach(const QString& tracker, trackers) { |
|
|
|
|
announce_entry url(tracker.toStdString()); |
|
|
|
|
url.tier = 0; |
|
|
|
|
h.add_tracker(url); |
|
|
|
|
} |
|
|
|
|
// Reannounce to new trackers
|
|
|
|
|
h.force_reannounce(); |
|
|
|
|
// Reload tracker list
|
|
|
|
|
loadTrackers(); |
|
|
|
|
// XXX: I don't think this is necessary now
|
|
|
|
|
//BTSession->saveTrackerFile(h.hash());
|
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::deleteSelectedTrackers(){ |
|
|
|
|
QTorrentHandle h = properties->getCurrentTorrent(); |
|
|
|
|
if(!h.is_valid()) { |
|
|
|
|
clear(); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
QList<QTreeWidgetItem *> selected_items = getSelectedTrackerItems(); |
|
|
|
|
if(selected_items.isEmpty()) return; |
|
|
|
|
QStringList urls_to_remove; |
|
|
|
|
foreach(QTreeWidgetItem *item, selected_items){ |
|
|
|
|
QString tracker_url = item->data(COL_URL, Qt::DisplayRole).toString(); |
|
|
|
|
urls_to_remove << tracker_url; |
|
|
|
|
tracker_items.remove(tracker_url); |
|
|
|
|
delete item; |
|
|
|
|
} |
|
|
|
|
// Iterate of trackers and remove selected ones
|
|
|
|
|
std::vector<announce_entry> trackers = h.trackers(); |
|
|
|
|
std::vector<announce_entry>::iterator it = trackers.begin(); |
|
|
|
|
while(it != trackers.end()) { |
|
|
|
|
int index = urls_to_remove.indexOf(misc::toQString((*it).url)); |
|
|
|
|
if(index >= 0) { |
|
|
|
|
trackers.erase(it); |
|
|
|
|
urls_to_remove.removeAt(index); |
|
|
|
|
} else { |
|
|
|
|
it++; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
h.replace_trackers(trackers); |
|
|
|
|
h.force_reannounce(); |
|
|
|
|
// Reload Trackers
|
|
|
|
|
loadTrackers(); |
|
|
|
|
//XXX: I don't think this is necessary
|
|
|
|
|
//BTSession->saveTrackerFile(h.hash());
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::showTrackerListMenu(QPoint) { |
|
|
|
|
QTorrentHandle h = properties->getCurrentTorrent(); |
|
|
|
|
if(!h.is_valid() || !h.has_metadata()) return; |
|
|
|
|
//QList<QTreeWidgetItem*> selected_items = getSelectedTrackerItems();
|
|
|
|
|
QMenu menu; |
|
|
|
|
// Add actions
|
|
|
|
|
QAction *addAct = menu.addAction(QIcon(":/Icons/oxygen/list-add.png"), tr("Add a new tracker...")); |
|
|
|
|
QAction *delAct = 0; |
|
|
|
|
if(!getSelectedTrackerItems().isEmpty()) { |
|
|
|
|
delAct = menu.addAction(QIcon(":/Icons/oxygen/list-remove.png"), tr("Remove tracker")); |
|
|
|
|
} |
|
|
|
|
menu.addSeparator(); |
|
|
|
|
QAction *reannounceAct = menu.addAction(QIcon(":/Icons/oxygen/run-build.png"), tr("Force reannounce")); |
|
|
|
|
QAction *act = menu.exec(QCursor::pos()); |
|
|
|
|
if(act == 0) return; |
|
|
|
|
if(act == addAct) { |
|
|
|
|
askForTrackers(); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
if(act == delAct) { |
|
|
|
|
deleteSelectedTrackers(); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
if(act == reannounceAct) { |
|
|
|
|
properties->getCurrentTorrent().force_reannounce(); |
|
|
|
|
return; |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::loadSettings() { |
|
|
|
|
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent")); |
|
|
|
|
QList<int> contentColsWidths = misc::intListfromStringList(settings.value(QString::fromUtf8("TorrentProperties/Trackers/trackersColsWidth")).toStringList()); |
|
|
|
|
if(!contentColsWidths.empty()) { |
|
|
|
|
for(int i=0; i<contentColsWidths.size(); ++i) { |
|
|
|
|
setColumnWidth(i, contentColsWidths.at(i)); |
|
|
|
|
} |
|
|
|
|
} else { |
|
|
|
|
setColumnWidth(0, 300); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
void TrackerList::saveSettings() const { |
|
|
|
|
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent")); |
|
|
|
|
QStringList contentColsWidths; |
|
|
|
|
for(int i=0; i<columnCount(); ++i) { |
|
|
|
|
contentColsWidths << QString::number(columnWidth(i)); |
|
|
|
|
} |
|
|
|
|
settings.setValue(QString::fromUtf8("TorrentProperties/Trackers/trackersColsWidth"), contentColsWidths); |
|
|
|
|
} |
|
|
|
|
|
Christophe Dumez
14 years ago