/* * 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 #include #include #include #include #include #include #include #include #include #include #include #include "propertieswidget.h" #include "TransferListWidget.h" #include "torrentPersistentData.h" #include "realprogressbar.h" #include "realprogressbarthread.h" #include "arborescence.h" #include "bittorrent.h" #include "PropListDelegate.h" #include "TrackersAdditionDlg.h" #define DEFAULT_BUTTON_CSS "QPushButton {border: 1px solid rgb(85, 81, 91);border-radius: 3px;padding: 2px;}" #define SELECTED_BUTTON_CSS "QPushButton {border: 1px solid rgb(85, 81, 91);border-radius: 3px;padding: 2px;background-color: rgb(255, 208, 105);}" PropertiesWidget::PropertiesWidget(QWidget *parent, TransferListWidget *transferList, bittorrent* BTSession): QWidget(parent), transferList(transferList), BTSession(BTSession) { setupUi(this); state = VISIBLE; QSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent")); if(!settings.value("TorrentProperties/Visible", false).toBool()) { reduce(); } else { main_infos_button->setStyleSheet(SELECTED_BUTTON_CSS); setEnabled(false); } // Set Properties list model PropListModel = new QStandardItemModel(0,5); PropListModel->setHeaderData(NAME, Qt::Horizontal, tr("File name")); PropListModel->setHeaderData(SIZE, Qt::Horizontal, tr("Size")); PropListModel->setHeaderData(PROGRESS, Qt::Horizontal, tr("Progress")); PropListModel->setHeaderData(PRIORITY, Qt::Horizontal, tr("Priority")); filesList->setModel(PropListModel); filesList->hideColumn(INDEX); PropDelegate = new PropListDelegate(0); filesList->setItemDelegate(PropDelegate); // QActions actionIgnored = new QAction(this); actionNormal = new QAction(this); actionMaximum = new QAction(this); actionHigh = new QAction(this); // SIGNAL/SLOTS connect(filesList, SIGNAL(clicked(const QModelIndex&)), filesList, SLOT(edit(const QModelIndex&))); connect(collapseAllButton, SIGNAL(clicked()), filesList, SLOT(collapseAll())); connect(expandAllButton, SIGNAL(clicked()), filesList, SLOT(expandAll())); connect(filesList, SIGNAL(customContextMenuRequested(const QPoint&)), this, SLOT(displayFilesListMenu(const QPoint&))); connect(addTracker_button, SIGNAL(clicked()), this, SLOT(askForTracker())); connect(removeTracker_button, SIGNAL(clicked()), this, SLOT(deleteSelectedTrackers())); connect(riseTracker_button, SIGNAL(clicked()), this, SLOT(riseSelectedTracker())); connect(lowerTracker_button, SIGNAL(clicked()), this, SLOT(lowerSelectedTracker())); connect(actionIgnored, SIGNAL(triggered()), this, SLOT(ignoreSelection())); connect(actionNormal, SIGNAL(triggered()), this, SLOT(normalSelection())); connect(actionHigh, SIGNAL(triggered()), this, SLOT(highSelection())); connect(actionMaximum, SIGNAL(triggered()), this, SLOT(maximumSelection())); connect(addWS_button, SIGNAL(clicked()), this, SLOT(askWebSeed())); connect(deleteWS_button, SIGNAL(clicked()), this, SLOT(deleteSelectedUrlSeeds())); connect(transferList, SIGNAL(currentTorrentChanged(QTorrentHandle&)), this, SLOT(loadTorrentInfos(QTorrentHandle &))); connect(incrementalDownload, SIGNAL(stateChanged(int)), this, SLOT(setIncrementalDownload(int))); connect(PropListModel, SIGNAL(itemChanged(QStandardItem*)), this, SLOT(updatePriorities(QStandardItem*))); connect(PropDelegate, SIGNAL(filteredFilesChanged()), this, SLOT(filteredFilesChanged())); // Downloaded pieces progress bar progressBar = new RealProgressBar(this); progressBar->setForegroundColor(Qt::blue); progressBarVbox = new QVBoxLayout(RealProgressBox); progressBarVbox->addWidget(progressBar); // Pointers init progressBarUpdater = 0; arb = 0; // Dynamic data refresher refreshTimer = new QTimer(this); connect(refreshTimer, SIGNAL(timeout()), this, SLOT(loadDynamicData())); refreshTimer->start(10000); // 10sec } PropertiesWidget::~PropertiesWidget() { QSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent")); settings.setValue("TorrentProperties/Visible", state==VISIBLE); delete refreshTimer; if(progressBarUpdater) delete progressBarUpdater; delete progressBar; delete progressBarVbox; delete PropListModel; if(arb) delete arb; // Delete QActions delete actionIgnored; delete actionNormal; delete actionMaximum; delete actionHigh; } void PropertiesWidget::reduce() { if(state == VISIBLE) { stackedProperties->setFixedHeight(0); state = REDUCED; } } void PropertiesWidget::slide() { if(state == REDUCED) { stackedProperties->setFixedHeight(232); state = VISIBLE; } } void PropertiesWidget::clear() { save_path->clear(); creator->clear(); hash_lbl->clear(); comment_lbl->clear(); incrementalDownload->setChecked(false); trackersURLS->clear(); trackerURL->clear(); progressBar->setProgress(QRealArray()); failed->clear(); upTotal->clear(); dlTotal->clear(); shareRatio->clear(); setEnabled(false); } void PropertiesWidget::loadTorrentInfos(QTorrentHandle &_h) { h = _h; if(!h.is_valid()) { clear(); return; } setEnabled(true); if(progressBarUpdater) { delete progressBarUpdater; progressBarUpdater = 0; } if(arb != 0) { delete arb; arb = 0; } try { // Save path save_path->setText(TorrentPersistentData::getSavePath(h.hash())); // Author QString author = h.creator().trimmed(); if(author.isEmpty()) author = tr("Unknown"); creator->setText(author); // Hash hash_lbl->setText(h.hash()); // Comment comment_lbl->setText(h.comment()); // Sequential download incrementalDownload->setChecked(TorrentPersistentData::isSequentialDownload(h.hash())); // Trackers loadTrackers(); // URL seeds loadUrlSeeds(); // downloaded pieces updater progressBarUpdater = new RealProgressBarThread(progressBar, h); progressBarUpdater->start(); // Create arborescence (Tree representation of files in the torrent) std::vector fp; h.file_progress(fp); std::vector files_priority = loadFilesPriorities(); // List files in torrent arborescence *arb = new arborescence(h.get_torrent_info(), fp, files_priority); addFilesToTree(arb->getRoot(), PropListModel->invisibleRootItem()); } catch(invalid_handle e) { } // Load dynamic data loadDynamicData(); } void PropertiesWidget::loadDynamicData() { if(!h.is_valid()) return; try { // Session infos failed->setText(misc::friendlyUnit(h.total_failed_bytes())); upTotal->setText(misc::friendlyUnit(h.total_payload_upload())); dlTotal->setText(misc::friendlyUnit(h.total_payload_download())); // Update ratio info float ratio; if(h.total_payload_download() == 0){ if(h.total_payload_upload() == 0) ratio = 1.; else ratio = 10.; // Max ratio }else{ ratio = (double)h.total_payload_upload()/(double)h.total_payload_download(); if(ratio > 10.){ ratio = 10.; } } shareRatio->setText(QString(QByteArray::number(ratio, 'f', 1))); // Downloaded pieces if(progressBarUpdater) progressBarUpdater->refresh(); } catch(invalid_handle e) {} } void PropertiesWidget::setIncrementalDownload(int checkboxState) { if(!h.is_valid()) return; h.set_sequential_download(checkboxState == Qt::Checked); TorrentPersistentData::saveSequentialStatus(h); } void PropertiesWidget::loadTrackers() { if(!h.is_valid()) return; //Trackers std::vector trackers = h.trackers(); trackersURLS->clear(); QHash errors = BTSession->getTrackersErrors(h.hash()); unsigned int nbTrackers = trackers.size(); for(unsigned int i=0; isetForeground(QBrush(QColor("red"))); // Set tooltip QString msg=""; unsigned int i=0; foreach(QString word, errors[current_tracker].split(" ")) { if(i > 0 && i%5!=1) msg += " "; msg += word; if(i> 0 && i%5==0) msg += "\n"; ++i; } item->setToolTip(msg); } else { item->setForeground(QBrush(QColor("green"))); } } QString tracker = h.current_tracker().trimmed(); if(!tracker.isEmpty()){ trackerURL->setText(tracker); }else{ trackerURL->setText(tr("None - Unreachable?")); } } void PropertiesWidget::loadUrlSeeds(){ QStringList already_added; listWebSeeds->clear(); QVariantList url_seeds = TorrentPersistentData::getUrlSeeds(h.hash()); foreach(const QVariant &var_url_seed, url_seeds){ QString url_seed = var_url_seed.toString(); if(!url_seed.isEmpty()) { new QListWidgetItem(url_seed, listWebSeeds); already_added << url_seed; } } // Load the hard-coded url seeds QStringList hc_seeds = h.url_seeds(); // Add hard coded url seeds foreach(const QString &hc_seed, hc_seeds){ if(!already_added.contains(hc_seed)){ new QListWidgetItem(hc_seed, listWebSeeds); } } } /* Tab buttons */ QPushButton* PropertiesWidget::getButtonFromIndex(int index) { switch(index) { case TRACKERS_TAB: return trackers_button; case URLSEEDS_TAB: return url_seeds_button; case FILES_TAB: return files_button; default: return main_infos_button; } } void PropertiesWidget::on_main_infos_button_clicked() { if(state == VISIBLE && stackedProperties->currentIndex() == MAIN_TAB) { reduce(); } else { slide(); getButtonFromIndex(stackedProperties->currentIndex())->setStyleSheet(DEFAULT_BUTTON_CSS); stackedProperties->setCurrentIndex(MAIN_TAB); main_infos_button->setStyleSheet(SELECTED_BUTTON_CSS); } } void PropertiesWidget::on_trackers_button_clicked() { if(state == VISIBLE && stackedProperties->currentIndex() == TRACKERS_TAB) { reduce(); } else { slide(); getButtonFromIndex(stackedProperties->currentIndex())->setStyleSheet(DEFAULT_BUTTON_CSS); stackedProperties->setCurrentIndex(TRACKERS_TAB); trackers_button->setStyleSheet(SELECTED_BUTTON_CSS); } } void PropertiesWidget::on_url_seeds_button_clicked() { if(state == VISIBLE && stackedProperties->currentIndex() == URLSEEDS_TAB) { reduce(); } else { slide(); getButtonFromIndex(stackedProperties->currentIndex())->setStyleSheet(DEFAULT_BUTTON_CSS); stackedProperties->setCurrentIndex(URLSEEDS_TAB); url_seeds_button->setStyleSheet(SELECTED_BUTTON_CSS); } } void PropertiesWidget::on_files_button_clicked() { if(state == VISIBLE && stackedProperties->currentIndex() == FILES_TAB) { reduce(); } else { slide(); getButtonFromIndex(stackedProperties->currentIndex())->setStyleSheet(DEFAULT_BUTTON_CSS); stackedProperties->setCurrentIndex(FILES_TAB); files_button->setStyleSheet(SELECTED_BUTTON_CSS); } } // priority is the new priority of given item void PropertiesWidget::updateParentsPriority(QStandardItem *item, int priority) { QStandardItem *parent = item->parent(); if(!parent) return; // Check if children have different priorities // then folder must have NORMAL priority unsigned int rowCount = parent->rowCount(); for(unsigned int i=0; ichild(i, PRIORITY)->text().toInt() != priority) { QStandardItem *grandFather = parent->parent(); if(!grandFather) { grandFather = PropListModel->invisibleRootItem(); } QStandardItem *parentPrio = grandFather->child(parent->row(), PRIORITY); if(parentPrio->text().toInt() != NORMAL) { parentPrio->setText(misc::toQString(NORMAL)); setItemColor(parentPrio->index(), "green"); // Recursively update ancesters of this parent too updateParentsPriority(grandFather->child(parent->row()), priority); } return; } } // All the children have the same priority // Parent folder should have the same priority too QStandardItem *grandFather = parent->parent(); if(!grandFather) { grandFather = PropListModel->invisibleRootItem(); } QStandardItem *parentPrio = grandFather->child(parent->row(), PRIORITY); if(parentPrio->text().toInt() != priority) { parentPrio->setText(misc::toQString(priority)); if(priority == IGNORED) setItemColor(parentPrio->index(), "red"); else setItemColor(parentPrio->index(), "green"); // Recursively update ancesters of this parent too updateParentsPriority(grandFather->child(parent->row()), priority); } } void PropertiesWidget::updateChildrenPriority(QStandardItem *item, int priority) { QStandardItem *parent = item->parent(); if(!parent) { parent = PropListModel->invisibleRootItem(); } parent = parent->child(item->row()); unsigned int rowCount = parent->rowCount(); for(unsigned int i=0; ichild(i, PRIORITY); if(childPrio->text().toInt() != priority) { childPrio->setText(misc::toQString(priority)); if(priority == IGNORED) setItemColor(childPrio->index(), "red"); else setItemColor(childPrio->index(), "green"); // recursively update children of this child too updateChildrenPriority(parent->child(i), priority); } } } void PropertiesWidget::updatePriorities(QStandardItem *item) { qDebug("Priority changed"); // First we disable the signal/slot on item edition // temporarily so that it doesn't mess with our manual updates disconnect(PropListModel, SIGNAL(itemChanged(QStandardItem*)), this, SLOT(updatePriorities(QStandardItem*))); QStandardItem *parent = item->parent(); if(!parent) { parent = PropListModel->invisibleRootItem(); } int priority = parent->child(item->row(), PRIORITY)->text().toInt(); if(priority == IGNORED) setItemColor(item->index(), "red"); else setItemColor(item->index(), "green"); // Update parents priorities updateParentsPriority(item, priority); // If this is not a directory, then there are // no children to update if(parent->child(item->row(), INDEX)->text().toInt() == -1) { // Updating children qDebug("Priority changed for a folder to %d", priority); updateChildrenPriority(item, priority); } // Reconnect the signal/slot on item edition so that we // get future updates connect(PropListModel, SIGNAL(itemChanged(QStandardItem*)), this, SLOT(updatePriorities(QStandardItem*))); } std::vector PropertiesWidget::loadFilesPriorities(){ std::vector fp; QVariantList files_priority = TorrentPersistentData::getFilesPriority(h.hash()); if(files_priority.empty()) { for(int i=0; i 7){ // Normal priority as default priority = 1; } fp.push_back(priority); } } return fp; } bool PropertiesWidget::allFiltered() const { unsigned int nbRows = PropListModel->rowCount(); for(unsigned int i=0; idata(PropListModel->index(i, PRIORITY)).toInt() != IGNORED) return false; } return true; } void PropertiesWidget::getPriorities(QStandardItem *parent, int *priorities) { qDebug("In getPriorities"); unsigned int nbRows = parent->rowCount(); for(unsigned int i=0; ichild(i, INDEX); int index = item->text().toInt(); if(index < 0) { getPriorities(parent->child(i, NAME), priorities); } else { item = parent->child(i, PRIORITY); priorities[index] = item->text().toInt(); qDebug("File at index %d has priority %d", index, priorities[index]); } } } void PropertiesWidget::displayFilesListMenu(const QPoint&){ if(h.get_torrent_info().num_files() == 1) return; QMenu myFilesLlistMenu(this); QModelIndex index; // Enable/disable pause/start action given the DL state QModelIndexList selectedIndexes = filesList->selectionModel()->selectedIndexes(); myFilesLlistMenu.setTitle(tr("Priority")); myFilesLlistMenu.addAction(actionIgnored); myFilesLlistMenu.addAction(actionNormal); myFilesLlistMenu.addAction(actionHigh); myFilesLlistMenu.addAction(actionMaximum); // Call menu myFilesLlistMenu.exec(QCursor::pos()); } void PropertiesWidget::ignoreSelection(){ QModelIndexList selectedIndexes = filesList->selectionModel()->selectedIndexes(); foreach(const QModelIndex &index, selectedIndexes){ if(index.column() == PRIORITY){ if(PropListModel->data(index) != QVariant(IGNORED)){ PropListModel->setData(index, QVariant(IGNORED)); transferList->updateTorrentSizeAndProgress(h.hash()); setItemColor(index, "red"); } } } } void PropertiesWidget::normalSelection(){ QModelIndexList selectedIndexes = filesList->selectionModel()->selectedIndexes(); foreach(const QModelIndex &index, selectedIndexes){ if(index.column() == PRIORITY){ if(PropListModel->data(index) != QVariant(NORMAL)){ PropListModel->setData(index, QVariant(NORMAL)); transferList->updateTorrentSizeAndProgress(h.hash()); setItemColor(index, "green"); } } } } void PropertiesWidget::highSelection(){ QModelIndexList selectedIndexes = filesList->selectionModel()->selectedIndexes(); foreach(const QModelIndex &index, selectedIndexes){ if(index.column() == PRIORITY){ if(PropListModel->data(index) != QVariant(HIGH)){ PropListModel->setData(index, QVariant(HIGH)); transferList->updateTorrentSizeAndProgress(h.hash()); setItemColor(index, "green"); } } } } void PropertiesWidget::maximumSelection(){ QModelIndexList selectedIndexes = filesList->selectionModel()->selectedIndexes(); foreach(const QModelIndex &index, selectedIndexes){ if(index.column() == PRIORITY){ if(PropListModel->data(index) != QVariant(MAXIMUM)){ PropListModel->setData(index, QVariant(MAXIMUM)); transferList->updateTorrentSizeAndProgress(h.hash()); setItemColor(index, "green"); } } } } void PropertiesWidget::askWebSeed(){ bool ok; // Ask user for a new url seed QString url_seed = QInputDialog::getText(this, tr("New url seed", "New HTTP source"), tr("New url seed:"), QLineEdit::Normal, QString::fromUtf8("http://www."), &ok); if(!ok) return; qDebug("Adding %s web seed", url_seed.toLocal8Bit().data()); if(!listWebSeeds->findItems(url_seed, Qt::MatchFixedString).empty()) { QMessageBox::warning(this, tr("qBittorrent"), tr("This url seed is already in the list."), QMessageBox::Ok); return; } h.add_url_seed(url_seed); TorrentPersistentData::saveUrlSeeds(h); // Refresh the seeds list loadUrlSeeds(); } // Ask the user for a new tracker // and add it to the download list // if it is not already in it void PropertiesWidget::askForTracker(){ TrackersAddDlg *dlg = new TrackersAddDlg(this); connect(dlg, SIGNAL(TrackersToAdd(QStringList)), this, SLOT(addTrackerList(QStringList))); } void PropertiesWidget::addTrackerList(QStringList myTrackers) { // Add the trackers to the list std::vector trackers = h.trackers(); foreach(const QString& tracker, myTrackers) { announce_entry new_tracker(misc::toString(tracker.trimmed().toLocal8Bit().data())); new_tracker.tier = 0; // Will be fixed a bit later trackers.push_back(new_tracker); misc::fixTrackersTiers(trackers); } h.replace_trackers(trackers); h.force_reannounce(); // Reload Trackers loadTrackers(); BTSession->saveTrackerFile(h.hash()); } void PropertiesWidget::deleteSelectedUrlSeeds(){ QList selectedItems = listWebSeeds->selectedItems(); bool change = false; foreach(QListWidgetItem *item, selectedItems){ QString url_seed = item->text(); h.remove_url_seed(url_seed); change = true; } if(change){ // Save them to disk TorrentPersistentData::saveUrlSeeds(h); // Refresh list loadUrlSeeds(); } } void PropertiesWidget::deleteSelectedTrackers(){ QList selectedItems = trackersURLS->selectedItems(); if(!selectedItems.size()) return; std::vector trackers = h.trackers(); unsigned int nbTrackers = trackers.size(); if(nbTrackers == (unsigned int) selectedItems.size()){ QMessageBox::warning(this, tr("qBittorrent"), tr("Trackers list can't be empty."), QMessageBox::Ok); return; } foreach(QListWidgetItem *item, selectedItems){ QString url = item->text(); for(unsigned int i=0; isaveTrackerFile(h.hash()); } void PropertiesWidget::riseSelectedTracker(){ unsigned int i = 0; std::vector trackers = h.trackers(); QList selectedItems = trackersURLS->selectedItems(); bool change = false; unsigned int nbTrackers = trackers.size(); foreach(QListWidgetItem *item, selectedItems){ QString url = item->text(); for(i=0; i 0){ announce_entry tmp = trackers[i]; trackers[i] = trackers[i-1]; trackers[i-1] = tmp; change = true; } break; } } } if(change){ misc::fixTrackersTiers(trackers); h.replace_trackers(trackers); h.force_reannounce(); // Reload Trackers loadTrackers(); trackersURLS->item(i-1)->setSelected(true); BTSession->saveTrackerFile(h.hash()); } } void PropertiesWidget::lowerSelectedTracker(){ unsigned int i = 0; std::vector trackers = h.trackers(); QList selectedItems = trackersURLS->selectedItems(); bool change = false; unsigned int nbTrackers = trackers.size(); foreach(QListWidgetItem *item, selectedItems){ QString url = item->text(); for(i=0; iitem(i+1)->setSelected(true); BTSession->saveTrackerFile(h.hash()); } } void PropertiesWidget::setItemColor(QModelIndex index, QString color){ for(int i=0; icolumnCount(); ++i){ PropListModel->setData(index.sibling(index.row(), i), QVariant(QColor(color)), Qt::ForegroundRole); } } void PropertiesWidget::on_changeSavePathButton_clicked() { QString dir; QDir saveDir(h.save_path()); if(saveDir.exists()){ dir = QFileDialog::getExistingDirectory(this, tr("Choose save path"), h.save_path()); }else{ dir = QFileDialog::getExistingDirectory(this, tr("Choose save path"), QDir::homePath()); } if(!dir.isNull()){ // Check if savePath exists QDir savePath(dir); if(!savePath.exists()){ if(!savePath.mkpath(savePath.path())){ QMessageBox::critical(0, tr("Save path creation error"), tr("Could not create the save path")); return; } } // Save savepath TorrentPersistentData::saveSavePath(h.hash(), savePath.path()); // Actually move storage if(!BTSession->useTemporaryFolder() || h.is_seed()) h.move_storage(savePath.path()); // Update save_path in dialog save_path->setText(savePath.path()); } } void PropertiesWidget::filteredFilesChanged() { if(h.is_valid()) transferList->updateTorrentSizeAndProgress(h.hash()); } void PropertiesWidget::addFilesToTree(torrent_file *root, QStandardItem *parent) { QList child; // Name QStandardItem *first; if(root->isDir()) { first = new QStandardItem(QIcon(":/Icons/oxygen/folder.png"), root->name()); } else { first = new QStandardItem(QIcon(":/Icons/oxygen/file.png"), root->name()); } child << first; // Size child << new QStandardItem(misc::toQString(root->getSize())); // Progress child << new QStandardItem(misc::toQString(root->getProgress())); // Prio child << new QStandardItem(misc::toQString(root->getPriority())); // INDEX child << new QStandardItem(misc::toQString(root->getIndex())); // Add the child to the tree parent->appendRow(child); // Set row color if(root->getPriority() == IGNORED) setItemColor(first->index(), "red"); else setItemColor(first->index(), "green"); // Add childs foreach(torrent_file *childFile, root->getChildren()) { addFilesToTree(childFile, first); } }