- New tracker list (displays tracker status and error/warning messages)

src/trackerlist.h

@@ -0,0 +1,196 @@
+/*
+ * 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
+ */
+
+#ifndef TRACKERLIST_H
+#define TRACKERLIST_H
+
+#include <QTreeWidget>
+#include <QTreeWidgetItem>
+#include <QStringList>
+#include <QMenu>
+#include <QHash>
+#include <QAction>
+#include "propertieswidget.h"
+#include "trackersadditiondlg.h"
+#include "misc.h"
+#include "bittorrent.h"
+
+enum TrackerListColumn {COL_URL, COL_STATUS, COL_MSG};
+
+class TrackerList: public QTreeWidget {
+  Q_OBJECT
+
+private:
+  PropertiesWidget *properties;
+  QHash<QString, QTreeWidgetItem*> tracker_items;
+
+public:
+  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("Message");
+    setHeaderItem(new QTreeWidgetItem(header));
+  }
+
+  ~TrackerList() {
+
+  }
+
+public slots:
+
+  void loadTrackers() {
+    QStringList old_trackers_urls = tracker_items.keys();
+    // Load trackers from torrent handle
+    QTorrentHandle h = properties->getCurrentTorrent();
+    if(!h.is_valid()) return;
+    QHash<QString, QString> errors = properties->getBTSession()->getTrackersErrors(h.hash());
+    std::vector<announce_entry> trackers = h.trackers();
+    std::vector<announce_entry>::iterator it;
+    for(it = trackers.begin(); it != trackers.end(); it++) {
+      QStringList item_list;
+      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);
+      }
+      if((*it).verified) {
+        item->setText(COL_STATUS, tr("Working"));
+      } else {
+        if((*it).updating) {
+          item->setText(COL_STATUS, tr("Updating..."));
+        } else {
+          if((*it).fails > 0) {
+            item->setText(COL_STATUS, tr("Not working"));
+          } else {
+            item->setText(COL_STATUS, tr("Not contacted yet"));
+          }
+        }
+      }
+      item->setText(COL_MSG, errors.value(tracker_url, ""));
+    }
+    // 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 askForTrackers(){
+    QStringList trackers = TrackersAdditionDlg::asForTrackers();
+    if(!trackers.empty()) {
+      QTorrentHandle h = properties->getCurrentTorrent();
+      if(!h.is_valid()) return;
+      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 deleteSelectedTrackers(){
+    QTorrentHandle h = properties->getCurrentTorrent();
+    if(!h.is_valid()) {
+      clear();
+      return;
+    }
+    QList<QTreeWidgetItem *> selected_items = selectedItems();
+    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 showTrackerListMenu(QPoint) {
+    QList<QTreeWidgetItem*> selected_items = selectedItems();
+    QMenu menu;
+    // Add actions
+    QAction *addAct = menu.addAction(QIcon(":/Icons/oxygen/list-add.png"), tr("Add a new tracker"));
+    QAction *delAct = 0;
+    if(!selectedItems().isEmpty()) {
+      delAct = menu.addAction(QIcon(":/Icons/oxygen/list-remove.png"), "Remove tracker");
+    }
+    QAction *act = menu.exec(QCursor::pos());
+    if(act == addAct) {
+      askForTrackers();
+      return;
+    }
+    if(act == delAct) {
+      deleteSelectedTrackers();
+      return;
+    }
+  }
+
+};
+
+#endif // TRACKERLIST_H