qBittorrent/src/FinishedTorrents.cpp

/*
 * 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.
 *
 * Contact : chris@qbittorrent.org
 */
#include "FinishedTorrents.h"
#include "misc.h"
#include "GUI.h"
#include <QFile>

FinishedTorrents::FinishedTorrents(QObject *parent, bittorrent *BTSession){
  setupUi(this);
  nbFinished = 0;
  ((GUI*)parent)->setTabText(1, tr("Finished") +" (0)");
  this->parent = parent;
  this->BTSession = BTSession;
  finishedListModel = new QStandardItemModel(0,9);
  finishedListModel->setHeaderData(NAME, Qt::Horizontal, tr("Name", "i.e: file name"));
  finishedListModel->setHeaderData(SIZE, Qt::Horizontal, tr("Size", "i.e: file size"));
  finishedListModel->setHeaderData(PROGRESS, Qt::Horizontal, tr("Progress", "i.e: % downloaded"));
  finishedListModel->setHeaderData(DLSPEED, Qt::Horizontal, tr("DL Speed", "i.e: Download speed"));
  finishedListModel->setHeaderData(UPSPEED, Qt::Horizontal, tr("UP Speed", "i.e: Upload speed"));
  finishedListModel->setHeaderData(SEEDSLEECH, Qt::Horizontal, tr("Seeds/Leechs", "i.e: full/partial sources"));
  finishedListModel->setHeaderData(STATUS, Qt::Horizontal, tr("Status"));
  finishedListModel->setHeaderData(ETA, Qt::Horizontal, tr("ETA", "i.e: Estimated Time of Arrival / Time left"));
  finishedList->setModel(finishedListModel);
  // Hide hash column
  finishedList->hideColumn(HASH);
  finishedListDelegate = new DLListDelegate();
  finishedList->setItemDelegate(finishedListDelegate);
}

FinishedTorrents::~FinishedTorrents(){
  delete finishedListDelegate;
  delete finishedListModel;
}

void FinishedTorrents::addFinishedSHA(QString hash){
  if(finishedSHAs.indexOf(hash) == -1) {
    finishedSHAs << hash;
    int row = finishedListModel->rowCount();
    torrent_handle h = BTSession->getTorrentHandle(hash);
    // Adding torrent to download list
    finishedListModel->insertRow(row);
    finishedListModel->setData(finishedListModel->index(row, NAME), QVariant(h.name().c_str()));
    finishedListModel->setData(finishedListModel->index(row, SIZE), QVariant((qlonglong)h.get_torrent_info().total_size()));
    finishedListModel->setData(finishedListModel->index(row, DLSPEED), QVariant((double)0.));
    finishedListModel->setData(finishedListModel->index(row, UPSPEED), QVariant((double)0.));
    finishedListModel->setData(finishedListModel->index(row, SEEDSLEECH), QVariant("0/0"));
    finishedListModel->setData(finishedListModel->index(row, ETA), QVariant((qlonglong)-1));
    finishedListModel->setData(finishedListModel->index(row, HASH), QVariant(hash));
    finishedListModel->setData(finishedListModel->index(row, PROGRESS), QVariant((double)1.));
    // Start the torrent if it was paused
    if(h.is_paused()) {
      h.resume();
      if(QFile::exists(misc::qBittorrentPath()+"BT_backup"+QDir::separator()+hash+".paused")) {
        QFile::remove(misc::qBittorrentPath()+"BT_backup"+QDir::separator()+hash+".paused");
      }
    }
    finishedListModel->setData(finishedListModel->index(row, STATUS), QVariant(tr("Finished", "i.e: Torrent has finished downloading")));
    finishedListModel->setData(finishedListModel->index(row, NAME), QVariant(QIcon(":/Icons/skin/seeding.png")), Qt::DecorationRole);
    setRowColor(row, "orange");
    // Create .finished file
    QFile finished_file(misc::qBittorrentPath()+"BT_backup"+QDir::separator()+hash+".finished");
    finished_file.open(QIODevice::WriteOnly | QIODevice::Text);
    finished_file.close();
    // Update the number of finished torrents
    ++nbFinished;
    ((GUI*)parent)->setTabText(1, tr("Finished") +" ("+QString(misc::toString(nbFinished).c_str())+")");
  } else {
    qDebug("Problem: this torrent (%s) has finished twice...", hash.toStdString().c_str());
  }
}

// Set the color of a row in data model
void FinishedTorrents::setRowColor(int row, const QString& color){
  for(int i=0; i<finishedListModel->columnCount(); ++i){
    finishedListModel->setData(finishedListModel->index(row, i), QVariant(QColor(color)), Qt::TextColorRole);
  }
}

void FinishedTorrents::updateFinishedList(){
//   qDebug("Updating finished list");
  QString hash;
  foreach(hash, finishedSHAs){
    torrent_handle h = BTSession->getTorrentHandle(hash);
    if(!h.is_valid()){
      qDebug("Problem: This torrent is not valid in finished list");
      continue;
    }
    if(h.is_paused()) continue;
    torrent_status torrentStatus = h.status();
    if(torrentStatus.state == torrent_status::downloading) {
      // What are you doing here, go back to download tab!
      qDebug("Info: a torrent was moved from finished to download tab");
      deleteFromFinishedList(hash);
      continue;
    }
    QList<QStandardItem *> items = finishedListModel->findItems(hash, Qt::MatchExactly, HASH );
    if(items.size() != 1){
      qDebug("Problem: Can't find torrent in finished list");
      continue;
    }
    int row = items.at(0)->row();
    finishedListModel->setData(finishedListModel->index(row, UPSPEED), QVariant((double)torrentStatus.upload_payload_rate));
    finishedListModel->setData(finishedListModel->index(row, SEEDSLEECH), QVariant(QString(misc::toString(torrentStatus.num_seeds, true).c_str())+"/"+QString(misc::toString(torrentStatus.num_peers - torrentStatus.num_seeds, true).c_str())));
  }
}

QStringList FinishedTorrents::getFinishedSHAs(){
  return finishedSHAs;
}

// Will move it to download tab
void FinishedTorrents::deleteFromFinishedList(QString hash){
  finishedSHAs.removeAll(hash);
  QList<QStandardItem *> items = finishedListModel->findItems(hash, Qt::MatchExactly, HASH );
  if(items.size() != 1){
    qDebug("Problem: Can't delete torrent from finished list");
    return;
  }
  finishedListModel->removeRow(finishedListModel->indexFromItem(items.at(0)).row());
  QFile::remove(misc::qBittorrentPath()+"BT_backup"+QDir::separator()+hash+".finished");
  --nbFinished;
  ((GUI*)parent)->setTabText(1, tr("Finished") +" ("+QString(misc::toString(nbFinished).c_str())+")");
}