kget/transfer-plugins/bittorrent/btadvanceddetailswidget.cpp

   1 /* This file is part of the KDE project
   2
   3    Copyright (C) 2007 Lukas Appelhans <l.appelhans@gmx.de>
   4    Copyright (C) 2007 Joris Guisson   <joris.guisson@gmail.com>
   5
   6    This program is free software; you can redistribute it and/or
   7    modify it under the terms of the GNU General Public
   8    License as published by the Free Software Foundation; either
   9    version 2 of the License, or (at your option) any later version.
  10 */
  11
  12 #include "btadvanceddetailswidget.h"
  13
  14 #include <interfaces/torrentinterface.h>
  15 #include <interfaces/trackerslist.h>
  16
  17 #include "bttransferhandler.h"
  18 #include "btfiletreeview.h"
  19 #include "bittorrentsettings.h"
  20
  21 #include <kdebug.h>
  22 #include <kmessagebox.h>
  23 #include <kurl.h>
  24 #include <kglobal.h>
  25 #include <kdialog.h>
  26 #include <QTimer>
  27
  28 BTAdvancedDetailsWidget::BTAdvancedDetailsWidget(BTTransferHandler * transfer)
  29     : m_transfer(transfer)
  30 {
  31     /**A lot of code is from KTorrent's infowidget by Joris Guisson, thx for the nice work**/
  32     setupUi(this);
  33
  34     tc = m_transfer->torrentControl();
  35
  36     tc->setMonitor(this);
  37
  38     init();
  39
  40     transfer->addObserver(this);
  41     //This updates the widget with the right values
  42     transferChangedEvent(transfer);
  43 }
  44
  45 BTAdvancedDetailsWidget::~BTAdvancedDetailsWidget()
  46 {
  47     m_transfer->delObserver(this);
  48 }
  49
  50 void BTAdvancedDetailsWidget::init()
  51 {
  52     setWindowTitle(i18n("Advanced-Details for %1", m_transfer->source().fileName()));
  53
  54     fileTreeView = new BTFileTreeView(tc, tabWidget->widget(0));
  55     tabWidget->widget(0)->layout()->addWidget(fileTreeView);
  56
  57     updateTrackerList();
  58
  59     const bt::TorrentStats & s = tc->getStats();
  60     totalChunksLabel->setText(QString::number(s.total_chunks));
  61     sizeChunkLabel->setText(KGlobal::locale()->formatByteSize(s.chunk_size));
  62
  63     /**Set Column-widths**/
  64
  65     QList<int> fileColumnWidths = BittorrentSettings::fileColumnWidths();
  66     if (!fileColumnWidths.isEmpty())
  67     {
  68         int j = 0;
  69         foreach(int i, fileColumnWidths)
  70         {
  71             fileTreeView->setColumnWidth(j, i);
  72             j++;
  73         }
  74     }
  75     else
  76     {
  77         fileTreeView->setColumnWidth(0 , 250);
  78     }
  79
  80     QList<int> peersColumnWidths = BittorrentSettings::peersColumnWidths();
  81     if (!peersColumnWidths.isEmpty())
  82     {
  83         int j = 0;
  84         foreach(int i, peersColumnWidths)
  85         {
  86             peersTreeWidget->setColumnWidth(j, i);
  87             j++;
  88         }
  89     }
  90     else
  91     {
  92         peersTreeWidget->setColumnWidth(0 , 250);
  93     }
  94
  95     QList<int> chunksColumnWidths = BittorrentSettings::chunksColumnWidths();
  96     if (!chunksColumnWidths.isEmpty())
  97     {
  98         int j = 0;
  99         foreach(int i, chunksColumnWidths)
 100         {
 101             chunkTreeWidget->setColumnWidth(j, i);
 102             j++;
 103         }
 104     }
 105     else
 106     {
 107         chunkTreeWidget->setColumnWidth(0 , 250);
 108     }
 109
 110     connect(deleteTrackerButton, SIGNAL(clicked()), SLOT(deleteTracker()));
 111     connect(updateTrackerButton, SIGNAL(clicked()), SLOT(updateTracker()));
 112     connect(addTrackerButton, SIGNAL(clicked()), SLOT(addTracker()));
 113     connect(changeTrackerButton, SIGNAL(clicked()), SLOT(changeTracker()));
 114     connect(defaultTrackerButton, SIGNAL(clicked()), SLOT(setDefaultTracker()));
 115 }
 116
 117 void BTAdvancedDetailsWidget::transferChangedEvent(TransferHandler * transfer)
 118 {
 119     kDebug(5001);
 120     TransferHandler::ChangesFlags transferFlags = m_transfer->changesFlags(this);
 121
 122     if (m_transfer->status() == Job::Running)
 123     {
 124         updateChunkView();
 125         peersTreeWidget->update();
 126         updateTrackerGUI();
 127     }
 128
 129     m_transfer->resetChangesFlags(this);
 130 }
 131
 132 void BTAdvancedDetailsWidget::updateTracker()
 133 {
 134     kDebug(5001);
 135     tc->updateTracker();
 136     if (tc->getStats().trackerstatus != "Ok")
 137         QTimer::singleShot(1000, this, SLOT(updateTracker()));
 138
 139     QTimer::singleShot(1000, this, SLOT(updateTrackerGUI()));
 140 }
 141
 142 void BTAdvancedDetailsWidget::updateTrackerGUI()
 143 {
 144     kDebug(5001);
 145     const bt::TorrentStats & s = tc->getStats();
 146
 147     if (s.running)
 148     {
 149         QTime t;
 150         t = t.addSecs(tc->getTimeToNextTrackerUpdate());
 151         trackerUpdateTime->setText(t.toString("mm:ss"));
 152     }
 153
 154     //Update manual annunce button
 155     updateTrackerButton->setEnabled(s.running); // && tc->announceAllowed()
 156     // only enable change when we can actually change and the torrent is running
 157     changeTrackerButton->setEnabled(s.running && tc->getTrackersList()->getTrackerURLs().size() > 1);
 158
 159     trackerStatus->setText("<b>" + s.trackerstatus + "</b>");
 160
 161     if (tc->getTrackersList())
 162         trackerUrl->setText("<b>" + tc->getTrackersList()->getTrackerURL().prettyUrl() + "</b>");
 163     else
 164         trackerUrl->clear();
 165 }
 166
 167 void BTAdvancedDetailsWidget::addTracker()
 168 {
 169     kDebug(5001);
 170     addTrackerDialog = new KDialog(this);
 171     QWidget * widget = new QWidget(this);
 172     addTrackerWidget.setupUi(widget);
 173     addTrackerDialog->setMainWidget(widget);
 174     addTrackerDialog->setButtons(KDialog::Ok | KDialog::Cancel);
 175     addTrackerDialog->exec();
 176
 177     if (addTrackerDialog->result() == 1)//Dialog accepted
 178     {
 179         m_transfer->addTracker(addTrackerWidget.lineEdit->text());
 180     }
 181
 182     QTimer::singleShot(1000, this, SLOT(updateTrackerList()));
 183 }
 184
 185 void BTAdvancedDetailsWidget::updateTrackerList()
 186 {
 187     kDebug(5001);
 188     trackerList->clear();
 189     const KUrl::List trackers = tc->getTrackersList()->getTrackerURLs();
 190
 191     if (trackers.empty())
 192     {
 193         trackerList->addItem(tc->getTrackersList()->getTrackerURL().prettyUrl());
 194     }
 195     else
 196     {
 197         foreach (KUrl u,trackers)
 198             trackerList->addItem(u.prettyUrl());
 199     }
 200     updateTracker();
 201 }
 202
 203 void BTAdvancedDetailsWidget::deleteTracker()
 204 {
 205     kDebug(5001);
 206     QListWidgetItem* current = trackerList->currentItem();
 207     if(!current)
 208         return;
 209
 210     KUrl url(current->text());
 211     if(tc->getTrackersList()->removeTracker(url))
 212         delete current;
 213     else
 214         KMessageBox::sorry(0, i18n("Cannot remove torrent default tracker."));
 215 }
 216
 217 void BTAdvancedDetailsWidget::setDefaultTracker()
 218 {
 219     kDebug(5001);
 220     tc->getTrackersList()->restoreDefault();
 221     tc->updateTracker();
 222
 223     // update the list of trackers
 224     trackerList->clear();
 225
 226     const KUrl::List trackers = tc->getTrackersList()->getTrackerURLs();
 227     if(trackers.empty())
 228         return;
 229
 230     foreach (KUrl u,trackers)
 231         trackerList->addItem(u.prettyUrl());
 232 }
 233
 234 void BTAdvancedDetailsWidget::changeTracker()
 235 {
 236     kDebug(5001);
 237     QListWidgetItem* current = trackerList->currentItem();
 238     if(!current)
 239         return;
 240
 241     KUrl url(current->text());
 242     tc->getTrackersList()->setTracker(url);
 243     tc->updateTracker();
 244 }
 245
 246 void BTAdvancedDetailsWidget::peerAdded(bt::PeerInterface* peer)
 247 {
 248     kDebug(5001);
 249     peersTreeWidget->peerAdded(peer);
 250 }
 251
 252 void BTAdvancedDetailsWidget::peerRemoved(bt::PeerInterface* peer)
 253 {
 254     kDebug(5001);
 255     peersTreeWidget->peerRemoved(peer);
 256 }
 257
 258 void BTAdvancedDetailsWidget::downloadStarted(bt::ChunkDownloadInterface* chunk)
 259 {
 260     kDebug(5001);
 261     items.insert(chunk, new ChunkDownloadViewItem(chunkTreeWidget, chunk, tc));
 262 }
 263
 264 void BTAdvancedDetailsWidget::downloadRemoved(bt::ChunkDownloadInterface* chunk)
 265 {
 266     kDebug(5001);
 267     ChunkDownloadViewItem* v = items.find(chunk);
 268     if (v)
 269     {
 270         items.erase(chunk);
 271         delete v;
 272     }
 273 }
 274
 275 void BTAdvancedDetailsWidget::stopped()
 276 {
 277     kDebug(5001);
 278     if (m_transfer->status() != Job::Running)
 279     { //Cleanup Chunk- and Peers-View when a transfer stops
 280         peersTreeWidget->removeAll();
 281         chunkTreeWidget->clear();
 282         items.clear();
 283     }
 284     tc->setMonitor(0);
 285     deleteLater();
 286 }
 287
 288 void BTAdvancedDetailsWidget::destroyed()
 289 {
 290     kDebug(5001);
 291     peersTreeWidget->removeAll();
 292     chunkTreeWidget->clear();
 293     items.clear();
 294     tc->setMonitor(0);
 295     deleteLater();
 296 }
 297
 298 void BTAdvancedDetailsWidget::updateChunkView()
 299 {
 300     kDebug(5001);
 301
 302     if (items.count() == 0 || !tc)
 303         return;
 304
 305     bt::PtrMap<bt::ChunkDownloadInterface*,ChunkDownloadViewItem>::iterator i;
 306
 307     for (i = items.begin();  i != items.end(); i++)
 308     {
 309         if (i->second)
 310             i->second->update(false);
 311     }
 312
 313     const bt::TorrentStats & s = tc->getStats();
 314     downloadingChunksLabel->setText(QString::number(s.num_chunks_downloading));
 315     downloadedChunksLabel->setText(QString::number(s.num_chunks_downloaded));
 316     excludedChunksLabel->setText(QString::number(s.num_chunks_excluded));
 317     leftChunksLabel->setText(QString::number(s.num_chunks_left));
 318 }
 319
 320 void BTAdvancedDetailsWidget::hideEvent(QHideEvent * event)
 321 {
 322     kDebug(5001) << "Save config";
 323     QList<int>  fileColumnWidths;
 324     for (int i = 0; i<1; i++)
 325     {
 326         fileColumnWidths.append(fileTreeView->columnWidth(i));
 327     }
 328     BittorrentSettings::setFileColumnWidths(fileColumnWidths);
 329
 330     QList<int>  peersColumnWidths;
 331     for (int i = 0; i<13; i++)
 332     {
 333         peersColumnWidths.append(peersTreeWidget->columnWidth(i));
 334     }
 335     BittorrentSettings::setPeersColumnWidths(peersColumnWidths);
 336
 337     QList<int>  chunksColumnWidths;
 338     for (int i = 0; i<3; i++)
 339     {
 340         chunksColumnWidths.append(chunkTreeWidget->columnWidth(i));
 341     }
 342     BittorrentSettings::setChunksColumnWidths(chunksColumnWidths);
 343     BittorrentSettings::self()->writeConfig();
 344 }
 345
 346 #include "btadvanceddetailswidget.moc"
 347