Christophe Dumez
18 years ago
7 changed files with 588 additions and 167 deletions
@ -0,0 +1,163 @@ |
|||||||
|
<ui version="4.0" > |
||||||
|
<class>addTorrentDialog</class> |
||||||
|
<widget class="QDialog" name="addTorrentDialog" > |
||||||
|
<property name="geometry" > |
||||||
|
<rect> |
||||||
|
<x>0</x> |
||||||
|
<y>0</y> |
||||||
|
<width>440</width> |
||||||
|
<height>389</height> |
||||||
|
</rect> |
||||||
|
</property> |
||||||
|
<property name="windowTitle" > |
||||||
|
<string>Torrent addition dialog</string> |
||||||
|
</property> |
||||||
|
<layout class="QVBoxLayout" > |
||||||
|
<property name="margin" > |
||||||
|
<number>9</number> |
||||||
|
</property> |
||||||
|
<property name="spacing" > |
||||||
|
<number>6</number> |
||||||
|
</property> |
||||||
|
<item> |
||||||
|
<widget class="QLabel" name="fileNameLbl" > |
||||||
|
<property name="text" > |
||||||
|
<string/> |
||||||
|
</property> |
||||||
|
</widget> |
||||||
|
</item> |
||||||
|
<item> |
||||||
|
<widget class="Line" name="line" > |
||||||
|
<property name="orientation" > |
||||||
|
<enum>Qt::Horizontal</enum> |
||||||
|
</property> |
||||||
|
</widget> |
||||||
|
</item> |
||||||
|
<item> |
||||||
|
<widget class="QLabel" name="savePathLbl" > |
||||||
|
<property name="text" > |
||||||
|
<string>Save path:</string> |
||||||
|
</property> |
||||||
|
</widget> |
||||||
|
</item> |
||||||
|
<item> |
||||||
|
<layout class="QHBoxLayout" > |
||||||
|
<property name="margin" > |
||||||
|
<number>0</number> |
||||||
|
</property> |
||||||
|
<property name="spacing" > |
||||||
|
<number>6</number> |
||||||
|
</property> |
||||||
|
<item> |
||||||
|
<widget class="QLineEdit" name="savePathTxt" /> |
||||||
|
</item> |
||||||
|
<item> |
||||||
|
<widget class="QToolButton" name="browseButton" > |
||||||
|
<property name="text" > |
||||||
|
<string>...</string> |
||||||
|
</property> |
||||||
|
</widget> |
||||||
|
</item> |
||||||
|
</layout> |
||||||
|
</item> |
||||||
|
<item> |
||||||
|
<widget class="QLabel" name="torrentContentLbl" > |
||||||
|
<property name="text" > |
||||||
|
<string>Torrent content:</string> |
||||||
|
</property> |
||||||
|
</widget> |
||||||
|
</item> |
||||||
|
<item> |
||||||
|
<widget class="QTreeWidget" name="torrentContentList" > |
||||||
|
<property name="selectionMode" > |
||||||
|
<enum>QAbstractItemView::ExtendedSelection</enum> |
||||||
|
</property> |
||||||
|
<property name="indentation" > |
||||||
|
<number>2</number> |
||||||
|
</property> |
||||||
|
<column> |
||||||
|
<property name="text" > |
||||||
|
<string>File name</string> |
||||||
|
</property> |
||||||
|
</column> |
||||||
|
<column> |
||||||
|
<property name="text" > |
||||||
|
<string>File size</string> |
||||||
|
</property> |
||||||
|
</column> |
||||||
|
<column> |
||||||
|
<property name="text" > |
||||||
|
<string>Selected</string> |
||||||
|
</property> |
||||||
|
</column> |
||||||
|
</widget> |
||||||
|
</item> |
||||||
|
<item> |
||||||
|
<widget class="QCheckBox" name="checkIncrementalDL" > |
||||||
|
<property name="text" > |
||||||
|
<string>Download in correct order (slower but good for previewing)</string> |
||||||
|
</property> |
||||||
|
</widget> |
||||||
|
</item> |
||||||
|
<item> |
||||||
|
<widget class="QCheckBox" name="addInPause" > |
||||||
|
<property name="text" > |
||||||
|
<string>Add to download list in paused state</string> |
||||||
|
</property> |
||||||
|
</widget> |
||||||
|
</item> |
||||||
|
<item> |
||||||
|
<layout class="QHBoxLayout" > |
||||||
|
<property name="margin" > |
||||||
|
<number>0</number> |
||||||
|
</property> |
||||||
|
<property name="spacing" > |
||||||
|
<number>6</number> |
||||||
|
</property> |
||||||
|
<item> |
||||||
|
<spacer> |
||||||
|
<property name="orientation" > |
||||||
|
<enum>Qt::Horizontal</enum> |
||||||
|
</property> |
||||||
|
<property name="sizeHint" > |
||||||
|
<size> |
||||||
|
<width>40</width> |
||||||
|
<height>20</height> |
||||||
|
</size> |
||||||
|
</property> |
||||||
|
</spacer> |
||||||
|
</item> |
||||||
|
<item> |
||||||
|
<widget class="QPushButton" name="OkButton" > |
||||||
|
<property name="text" > |
||||||
|
<string>Add</string> |
||||||
|
</property> |
||||||
|
</widget> |
||||||
|
</item> |
||||||
|
<item> |
||||||
|
<widget class="QPushButton" name="CancelButton" > |
||||||
|
<property name="text" > |
||||||
|
<string>Cancel</string> |
||||||
|
</property> |
||||||
|
</widget> |
||||||
|
</item> |
||||||
|
<item> |
||||||
|
<spacer> |
||||||
|
<property name="orientation" > |
||||||
|
<enum>Qt::Horizontal</enum> |
||||||
|
</property> |
||||||
|
<property name="sizeHint" > |
||||||
|
<size> |
||||||
|
<width>40</width> |
||||||
|
<height>20</height> |
||||||
|
</size> |
||||||
|
</property> |
||||||
|
</spacer> |
||||||
|
</item> |
||||||
|
</layout> |
||||||
|
</item> |
||||||
|
</layout> |
||||||
|
</widget> |
||||||
|
<resources/> |
||||||
|
<connections/> |
||||||
|
</ui> |
||||||
@ -0,0 +1,214 @@ |
|||||||
|
/*
|
||||||
|
* 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 |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef TORRENTADDITION_H |
||||||
|
#define TORRENTADDITION_H |
||||||
|
|
||||||
|
#include <QDir> |
||||||
|
#include <QFileDialog> |
||||||
|
#include <QFile> |
||||||
|
#include <fstream> |
||||||
|
#include <QMessageBox> |
||||||
|
|
||||||
|
#include <libtorrent/session.hpp> |
||||||
|
#include <libtorrent/bencode.hpp> |
||||||
|
#include "misc.h" |
||||||
|
#include "ui_addTorrentDialog.h" |
||||||
|
|
||||||
|
#define NAME 0 |
||||||
|
#define SIZE 1 |
||||||
|
#define SELECTED 2 |
||||||
|
|
||||||
|
using namespace libtorrent; |
||||||
|
|
||||||
|
class torrentAdditionDialog : public QDialog, private Ui_addTorrentDialog{ |
||||||
|
Q_OBJECT |
||||||
|
|
||||||
|
private: |
||||||
|
QString fileName; |
||||||
|
QString filePath; |
||||||
|
QList<bool> selection; |
||||||
|
bool fromScanDir; |
||||||
|
QString from_url; |
||||||
|
|
||||||
|
public: |
||||||
|
// Check if the torrent in already in download list before calling this dialog
|
||||||
|
torrentAdditionDialog(QWidget *parent, QString filePath, bool fromScanDir=false, QString from_url=QString::null) : QDialog(parent), filePath(filePath), fromScanDir(fromScanDir), from_url(from_url){ |
||||||
|
setupUi(this); |
||||||
|
setAttribute(Qt::WA_DeleteOnClose); |
||||||
|
//TODO: Remember last entered path
|
||||||
|
QString home = QDir::homePath(); |
||||||
|
if(home[home.length()-1] != QDir::separator()){ |
||||||
|
home += QDir::separator(); |
||||||
|
} |
||||||
|
savePathTxt->setText(home+"qBT_dir"); |
||||||
|
std::ifstream in(filePath.toStdString().c_str(), std::ios_base::binary); |
||||||
|
in.unsetf(std::ios_base::skipws); |
||||||
|
try{ |
||||||
|
// Decode torrent file
|
||||||
|
entry e = bdecode(std::istream_iterator<char>(in), std::istream_iterator<char>()); |
||||||
|
// Getting torrent file informations
|
||||||
|
torrent_info t(e); |
||||||
|
// Setting file name
|
||||||
|
fileName = QString(t.name().c_str()); |
||||||
|
fileNameLbl->setText("<center><b>"+fileName+"</b></center>"); |
||||||
|
// List files in torrent
|
||||||
|
for(int i=0; i<t.num_files(); ++i){ |
||||||
|
QStringList line; |
||||||
|
line << QString(t.file_at(i).path.leaf().c_str()); |
||||||
|
line << misc::friendlyUnit((qlonglong)t.file_at(i).size); |
||||||
|
selection << true; |
||||||
|
line << tr("True"); |
||||||
|
torrentContentList->addTopLevelItem(new QTreeWidgetItem(line)); |
||||||
|
setLineColor(torrentContentList->topLevelItemCount()-1, "green"); |
||||||
|
} |
||||||
|
}catch (invalid_torrent_file&){ // Raised by torrent_info constructor
|
||||||
|
// Display warning to tell user we can't decode the torrent file
|
||||||
|
if(!from_url.isNull()){ |
||||||
|
emit setInfoBar(tr("Unable to decode torrent file:")+" '"+from_url+"'", "red"); |
||||||
|
}else{ |
||||||
|
emit setInfoBar(tr("Unable to decode torrent file:")+" '"+filePath+"'", "red"); |
||||||
|
} |
||||||
|
emit setInfoBar(tr("This file is either corrupted or this isn't a torrent."),"red"); |
||||||
|
if(fromScanDir){ |
||||||
|
//Rename file extension so that it won't display error message more than once
|
||||||
|
QFile::rename(filePath,filePath+".corrupt"); |
||||||
|
} |
||||||
|
close(); |
||||||
|
} |
||||||
|
//Connects
|
||||||
|
connect(torrentContentList, SIGNAL(itemDoubleClicked(QTreeWidgetItem*, int)), this, SLOT(toggleSelection(QTreeWidgetItem*, int))); |
||||||
|
show(); |
||||||
|
} |
||||||
|
|
||||||
|
public slots: |
||||||
|
void on_browseButton_clicked(){ |
||||||
|
QString dir = QFileDialog::getExistingDirectory(this, tr("Choose save path"), QDir::homePath()); |
||||||
|
if(!dir.isNull()){ |
||||||
|
savePathTxt->setText(dir); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void on_CancelButton_clicked(){ |
||||||
|
close(); |
||||||
|
} |
||||||
|
|
||||||
|
void toggleSelection(QTreeWidgetItem *item, int){ |
||||||
|
int row = torrentContentList->indexOfTopLevelItem(item); |
||||||
|
if(row == -1){ |
||||||
|
return; |
||||||
|
} |
||||||
|
if(selection.at(row)){ |
||||||
|
setLineColor(row, "red"); |
||||||
|
item->setText(SELECTED, tr("False")); |
||||||
|
selection.replace(row, false); |
||||||
|
}else{ |
||||||
|
setLineColor(row, "green"); |
||||||
|
item->setText(SELECTED, tr("True")); |
||||||
|
selection.replace(row, true); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void saveFilteredFiles(){ |
||||||
|
QFile pieces_file(misc::qBittorrentPath()+"BT_backup"+QDir::separator()+fileName+".pieces"); |
||||||
|
// First, remove old file
|
||||||
|
pieces_file.remove(); |
||||||
|
// Write new files
|
||||||
|
if(!pieces_file.open(QIODevice::WriteOnly | QIODevice::Text)){ |
||||||
|
std::cerr << "Error: Could not save filtered pieces\n"; |
||||||
|
return; |
||||||
|
} |
||||||
|
for(int i=0; i<torrentContentList->topLevelItemCount(); ++i){ |
||||||
|
if(selection.at(i)){ |
||||||
|
pieces_file.write(QByteArray("0\n")); |
||||||
|
}else{ |
||||||
|
pieces_file.write(QByteArray("1\n")); |
||||||
|
} |
||||||
|
} |
||||||
|
pieces_file.close(); |
||||||
|
} |
||||||
|
|
||||||
|
void setLineColor(int row, QString color){ |
||||||
|
QTreeWidgetItem *item = torrentContentList->topLevelItem(row); |
||||||
|
for(int i=0; i<item->columnCount(); ++i){ |
||||||
|
item->setData(i, Qt::ForegroundRole, QVariant(QColor(color))); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void on_OkButton_clicked(){ |
||||||
|
QDir savePath(savePathTxt->text()); |
||||||
|
if(savePathTxt->text().trimmed().isEmpty()){ |
||||||
|
QMessageBox::critical(0, tr("Empty save path"), tr("Please enter a save path")); |
||||||
|
return; |
||||||
|
} |
||||||
|
// Check if savePath exists
|
||||||
|
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
|
||||||
|
QFile savepath_file(misc::qBittorrentPath()+"BT_backup"+QDir::separator()+fileName+".savepath"); |
||||||
|
savepath_file.open(QIODevice::WriteOnly | QIODevice::Text); |
||||||
|
savepath_file.write(QByteArray(savePath.path().toStdString().c_str())); |
||||||
|
savepath_file.close(); |
||||||
|
// Create .incremental file if necessary
|
||||||
|
if(checkIncrementalDL->isChecked()){ |
||||||
|
QFile incremental_file(misc::qBittorrentPath()+"BT_backup"+QDir::separator()+fileName+".incremental"); |
||||||
|
incremental_file.open(QIODevice::WriteOnly | QIODevice::Text); |
||||||
|
incremental_file.close(); |
||||||
|
}else{ |
||||||
|
QFile::remove(misc::qBittorrentPath()+"BT_backup"+QDir::separator()+fileName+".incremental"); |
||||||
|
} |
||||||
|
// Create .paused file if necessary
|
||||||
|
if(addInPause->isChecked()){ |
||||||
|
QFile paused_file(misc::qBittorrentPath()+"BT_backup"+QDir::separator()+fileName+".paused"); |
||||||
|
paused_file.open(QIODevice::WriteOnly | QIODevice::Text); |
||||||
|
paused_file.close(); |
||||||
|
}else{ |
||||||
|
QFile::remove(misc::qBittorrentPath()+"BT_backup"+QDir::separator()+fileName+".paused"); |
||||||
|
} |
||||||
|
// Check if there is at least one selected file
|
||||||
|
bool selected_file = false; |
||||||
|
for(int i=0; i<torrentContentList->topLevelItemCount(); ++i){ |
||||||
|
if(selection.at(i)){ |
||||||
|
selected_file = true; |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
if(!selected_file){ |
||||||
|
QMessageBox::critical(0, tr("Invalid file selection"), tr("You must select at least one file in the torrent")); |
||||||
|
return; |
||||||
|
} |
||||||
|
// save filtered files
|
||||||
|
saveFilteredFiles(); |
||||||
|
// Add to download list
|
||||||
|
emit torrentAddition(filePath, fromScanDir, from_url); |
||||||
|
close(); |
||||||
|
} |
||||||
|
|
||||||
|
signals: |
||||||
|
void setInfoBar(QString message, QString color); |
||||||
|
void torrentAddition(const QString& filePath, bool fromScanDir, const QString& from_url); |
||||||
|
}; |
||||||
|
|
||||||
|
#endif |
||||||
Loading…
Reference in new issue