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