transmission/qt/options.cc

577 lines
17 KiB
C++

/*
* This file Copyright (C) Mnemosyne LLC
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 2
* as published by the Free Software Foundation.
*
* http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
*
* $Id$
*/
#include <cstdio>
#include <iostream>
#include <QApplication>
#include <QCheckBox>
#include <QComboBox>
#include <QDialogButtonBox>
#include <QEvent>
#include <QFileDialog>
#include <QFileIconProvider>
#include <QFileInfo>
#include <QGridLayout>
#include <QLabel>
#include <QMessageBox>
#include <QPushButton>
#include <QResizeEvent>
#include <QSet>
#include <QVBoxLayout>
#include <QWidget>
#include <libtransmission/transmission.h>
#include <libtransmission/bencode.h>
#include <libtransmission/utils.h> /* mime64 */
#include "add-data.h"
#include "file-tree.h"
#include "hig.h"
#include "options.h"
#include "prefs.h"
#include "session.h"
#include "torrent.h"
#include "utils.h"
/***
****
***/
void
FileAdded :: executed( int64_t tag, const QString& result, struct tr_benc * arguments )
{
Q_UNUSED( arguments );
if( tag != myTag )
return;
if( result == "success" )
if( !myDelFile.isEmpty( ) )
QFile( myDelFile ).remove( );
if( result != "success" ) {
QString text = result;
for( int i=0, n=text.size(); i<n; ++i )
if( !i || text[i-1].isSpace() )
text[i] = text[i].toUpper();
QMessageBox::warning( QApplication::activeWindow(),
tr( "Error Adding Torrent" ),
QString("<p><b>%1</b></p><p>%2</p>").arg(text).arg(myName) );
}
deleteLater();
}
/***
****
***/
Options :: Options( Session& session, const Prefs& prefs, const AddData& addme, QWidget * parent ):
QDialog( parent, Qt::Dialog ),
mySession( session ),
myAdd( addme ),
myHaveInfo( false ),
myDestinationButton( 0 ),
myVerifyButton( 0 ),
myVerifyFile( 0 ),
myVerifyHash( QCryptographicHash::Sha1 )
{
setWindowTitle( tr( "Add Torrent" ) );
QFontMetrics fontMetrics( font( ) );
QGridLayout * layout = new QGridLayout( this );
int row = 0;
const int iconSize( style( )->pixelMetric( QStyle :: PM_SmallIconSize ) );
QIcon fileIcon = style( )->standardIcon( QStyle::SP_FileIcon );
const QPixmap filePixmap = fileIcon.pixmap( iconSize );
QPushButton * p;
int width = fontMetrics.size( 0, "This is a pretty long torrent filename indeed.torrent" ).width( );
QLabel * l = new QLabel( tr( "&Torrent file:" ) );
layout->addWidget( l, row, 0, Qt::AlignLeft );
p = myFileButton = new QPushButton;
p->setIcon( filePixmap );
p->setMinimumWidth( width );
p->setStyleSheet( "text-align: left; padding-left: 5; padding-right: 5" );
p->installEventFilter( this );
layout->addWidget( p, row, 1 );
l->setBuddy( p );
connect( p, SIGNAL(clicked(bool)), this, SLOT(onFilenameClicked()));
if( session.isLocal( ) )
{
const QFileIconProvider iconProvider;
const QIcon folderIcon = iconProvider.icon( QFileIconProvider::Folder );
const QPixmap folderPixmap = folderIcon.pixmap( iconSize );
l = new QLabel( tr( "&Destination folder:" ) );
layout->addWidget( l, ++row, 0, Qt::AlignLeft );
myDestination.setPath( prefs.getString( Prefs :: DOWNLOAD_DIR ) );
p = myDestinationButton = new QPushButton;
p->setIcon( folderPixmap );
p->setStyleSheet( "text-align: left; padding-left: 5; padding-right: 5" );
p->installEventFilter( this );
layout->addWidget( p, row, 1 );
l->setBuddy( p );
connect( p, SIGNAL(clicked(bool)), this, SLOT(onDestinationClicked()));
}
myTree = new FileTreeView;
layout->addWidget( myTree, ++row, 0, 1, 2 );
if( !session.isLocal( ) )
myTree->hideColumn( 1 ); // hide the % done, since we've no way of knowing
QComboBox * m = new QComboBox;
m->addItem( tr( "High" ), TR_PRI_HIGH );
m->addItem( tr( "Normal" ), TR_PRI_NORMAL );
m->addItem( tr( "Low" ), TR_PRI_LOW );
m->setCurrentIndex( 1 ); // Normal
myPriorityCombo = m;
l = new QLabel( tr( "Torrent &priority:" ) );
l->setBuddy( m );
layout->addWidget( l, ++row, 0, Qt::AlignLeft );
layout->addWidget( m, row, 1 );
if( session.isLocal( ) )
{
p = myVerifyButton = new QPushButton( tr( "&Verify Local Data" ) );
layout->addWidget( p, ++row, 0, Qt::AlignLeft );
}
QCheckBox * c;
c = myStartCheck = new QCheckBox( tr( "&Start when added" ) );
c->setChecked( prefs.getBool( Prefs :: START ) );
layout->addWidget( c, ++row, 0, 1, 2, Qt::AlignLeft );
c = myTrashCheck = new QCheckBox( tr( "Mo&ve .torrent file to the trash" ) );
c->setChecked( prefs.getBool( Prefs :: TRASH_ORIGINAL ) );
layout->addWidget( c, ++row, 0, 1, 2, Qt::AlignLeft );
QDialogButtonBox * b = new QDialogButtonBox( QDialogButtonBox::Ok|QDialogButtonBox::Cancel, Qt::Horizontal, this );
connect( b, SIGNAL(rejected()), this, SLOT(deleteLater()) );
connect( b, SIGNAL(accepted()), this, SLOT(onAccepted()) );
layout->addWidget( b, ++row, 0, 1, 2 );
layout->setRowStretch( 2, 2 );
layout->setColumnStretch( 1, 2 );
layout->setSpacing( HIG :: PAD );
connect( myTree, SIGNAL(priorityChanged(const QSet<int>&,int)), this, SLOT(onPriorityChanged(const QSet<int>&,int)));
connect( myTree, SIGNAL(wantedChanged(const QSet<int>&,bool)), this, SLOT(onWantedChanged(const QSet<int>&,bool)));
if( session.isLocal( ) )
connect( myVerifyButton, SIGNAL(clicked(bool)), this, SLOT(onVerify()));
connect( &myVerifyTimer, SIGNAL(timeout()), this, SLOT(onTimeout()));
reload( );
}
Options :: ~Options( )
{
clearInfo( );
}
/***
****
***/
void
Options :: refreshButton( QPushButton * p, const QString& text, int width )
{
if( width <= 0 ) width = p->width( );
width -= 15;
QFontMetrics fontMetrics( font( ) );
QString str = fontMetrics.elidedText( text, Qt::ElideRight, width );
p->setText( str );
}
void
Options :: refreshFileButton( int width )
{
QString text;
switch( myAdd.type )
{
case AddData::FILENAME: text = QFileInfo(myAdd.filename).baseName(); break;
case AddData::URL: text = myAdd.url.toString(); break;
case AddData::MAGNET: text = myAdd.magnet; break;
default: break;
}
refreshButton( myFileButton, text, width );
}
void
Options :: refreshDestinationButton( int width )
{
if( myDestinationButton != 0 )
refreshButton( myDestinationButton, myDestination.absolutePath(), width );
}
bool
Options :: eventFilter( QObject * o, QEvent * event )
{
if( o==myFileButton && event->type() == QEvent::Resize )
{
refreshFileButton( dynamic_cast<QResizeEvent*>(event)->size().width() );
}
if( o==myDestinationButton && event->type() == QEvent::Resize )
{
refreshDestinationButton( dynamic_cast<QResizeEvent*>(event)->size().width() );
}
return false;
}
/***
****
***/
void
Options :: clearInfo( )
{
if( myHaveInfo )
tr_metainfoFree( &myInfo );
myHaveInfo = false;
myFiles.clear( );
}
void
Options :: reload( )
{
clearInfo( );
clearVerify( );
tr_ctor * ctor = tr_ctorNew( 0 );
switch( myAdd.type ) {
case AddData::MAGNET: tr_ctorSetMetainfoFromMagnetLink( ctor, myAdd.magnet.toUtf8().constData() ); break;
case AddData::FILENAME: tr_ctorSetMetainfoFromFile( ctor, myAdd.filename.toUtf8().constData() ); break;
case AddData::METAINFO: tr_ctorSetMetainfo( ctor, (const uint8_t*)myAdd.metainfo.constData(), myAdd.metainfo.size() ); break;
default: break;
}
const int err = tr_torrentParse( ctor, &myInfo );
myHaveInfo = !err;
tr_ctorFree( ctor );
myTree->clear( );
myFiles.clear( );
myPriorities.clear( );
myWanted.clear( );
if( myHaveInfo )
{
myPriorities.insert( 0, myInfo.fileCount, TR_PRI_NORMAL );
myWanted.insert( 0, myInfo.fileCount, true );
for( tr_file_index_t i=0; i<myInfo.fileCount; ++i ) {
TrFile file;
file.index = i;
file.priority = myPriorities[i];
file.wanted = myWanted[i];
file.size = myInfo.files[i].length;
file.have = 0;
file.filename = QString::fromUtf8( myInfo.files[i].name );
myFiles.append( file );
}
}
myTree->update( myFiles );
}
void
Options :: onPriorityChanged( const QSet<int>& fileIndices, int priority )
{
foreach( int i, fileIndices )
myPriorities[i] = priority;
}
void
Options :: onWantedChanged( const QSet<int>& fileIndices, bool isWanted )
{
foreach( int i, fileIndices )
myWanted[i] = isWanted;
}
void
Options :: onAccepted( )
{
// rpc spec section 3.4 "adding a torrent"
const int64_t tag = mySession.getUniqueTag( );
tr_benc top;
tr_bencInitDict( &top, 3 );
tr_bencDictAddStr( &top, "method", "torrent-add" );
tr_bencDictAddInt( &top, "tag", tag );
tr_benc * args( tr_bencDictAddDict( &top, "arguments", 10 ) );
// "download-dir"
if( myDestinationButton )
tr_bencDictAddStr( args, "download-dir", myDestination.absolutePath().toUtf8().constData() );
// "metainfo"
switch( myAdd.type )
{
case AddData::MAGNET:
tr_bencDictAddStr( args, "filename", myAdd.magnet.toUtf8().constData() );
break;
case AddData::URL:
tr_bencDictAddStr( args, "filename", myAdd.url.toString().toUtf8().constData() );
break;
case AddData::FILENAME:
case AddData::METAINFO: {
const QByteArray b64 = myAdd.toBase64( );
tr_bencDictAddRaw( args, "metainfo", b64.constData(), b64.size() );
break;
}
default:
std::cerr << "unhandled AddData.type: " << myAdd.type << std::endl;
}
// paused
tr_bencDictAddBool( args, "paused", !myStartCheck->isChecked( ) );
// priority
const int index = myPriorityCombo->currentIndex( );
const int priority = myPriorityCombo->itemData(index).toInt( );
tr_bencDictAddInt( args, "bandwidthPriority", priority );
// files-unwanted
int count = myWanted.count( false );
if( count > 0 ) {
tr_benc * l = tr_bencDictAddList( args, "files-unwanted", count );
for( int i=0, n=myWanted.size(); i<n; ++i )
if( myWanted.at(i) == false )
tr_bencListAddInt( l, i );
}
// priority-low
count = myPriorities.count( TR_PRI_LOW );
if( count > 0 ) {
tr_benc * l = tr_bencDictAddList( args, "priority-low", count );
for( int i=0, n=myPriorities.size(); i<n; ++i )
if( myPriorities.at(i) == TR_PRI_LOW )
tr_bencListAddInt( l, i );
}
// priority-high
count = myPriorities.count( TR_PRI_HIGH );
if( count > 0 ) {
tr_benc * l = tr_bencDictAddList( args, "priority-high", count );
for( int i=0, n=myPriorities.size(); i<n; ++i )
if( myPriorities.at(i) == TR_PRI_HIGH )
tr_bencListAddInt( l, i );
}
// maybe delete the source .torrent
FileAdded * fileAdded = new FileAdded( tag, myAdd.readableName() );
if( myTrashCheck->isChecked( ) && ( myAdd.type==AddData::FILENAME ) )
fileAdded->setFileToDelete( myAdd.filename );
connect( &mySession, SIGNAL(executed(int64_t,const QString&, struct tr_benc*)),
fileAdded, SLOT(executed(int64_t,const QString&, struct tr_benc*)));
//std::cerr << tr_bencToStr(&top,TR_FMT_JSON,NULL) << std::endl;
mySession.exec( &top );
tr_bencFree( &top );
deleteLater( );
}
void
Options :: onFilenameClicked( )
{
if( myAdd.type == AddData::FILENAME )
{
QFileDialog * d = new QFileDialog( this,
tr( "Add Torrent" ),
QFileInfo(myAdd.filename).absolutePath(),
tr( "Torrent Files (*.torrent);;All Files (*.*)" ) );
d->setFileMode( QFileDialog::ExistingFile );
connect( d, SIGNAL(filesSelected(const QStringList&)), this, SLOT(onFilesSelected(const QStringList&)) );
d->show( );
}
}
void
Options :: onFilesSelected( const QStringList& files )
{
if( files.size() == 1 )
{
myAdd.set( files.at(0) );
refreshFileButton( );
reload( );
}
}
void
Options :: onDestinationClicked( )
{
QFileDialog * d = new QFileDialog( this,
tr( "Select Destination" ),
myDestination.absolutePath( ) );
d->setFileMode( QFileDialog::Directory );
connect( d, SIGNAL(filesSelected(const QStringList&)), this, SLOT(onDestinationsSelected(const QStringList&)) );
d->show( );
}
void
Options :: onDestinationsSelected( const QStringList& destinations )
{
if( destinations.size() == 1 )
{
const QString& destination( destinations.first( ) );
myDestination.setPath( destination );
refreshDestinationButton( );
}
}
/***
****
**** VERIFY
****
***/
void
Options :: clearVerify( )
{
myVerifyHash.reset( );
myVerifyFile.close( );
myVerifyFilePos = 0;
myVerifyFlags.clear( );
myVerifyFileIndex = 0;
myVerifyPieceIndex = 0;
myVerifyPiecePos = 0;
myVerifyTimer.stop( );
for( int i=0, n=myFiles.size(); i<n; ++i )
myFiles[i].have = 0;
myTree->update( myFiles );
}
void
Options :: onVerify( )
{
//std::cerr << "starting to verify..." << std::endl;
clearVerify( );
myVerifyFlags.insert( 0, myInfo.pieceCount, false );
myVerifyTimer.setSingleShot( false );
myVerifyTimer.start( 0 );
}
namespace
{
uint64_t getPieceSize( const tr_info * info, tr_piece_index_t pieceIndex )
{
if( pieceIndex != info->pieceCount - 1 )
return info->pieceSize;
return info->totalSize % info->pieceSize;
}
}
void
Options :: onTimeout( )
{
const tr_file * file = &myInfo.files[myVerifyFileIndex];
if( !myVerifyFilePos && !myVerifyFile.isOpen( ) )
{
const QFileInfo fileInfo( myDestination, QString::fromUtf8( file->name ) );
myVerifyFile.setFileName( fileInfo.absoluteFilePath( ) );
//std::cerr << "opening file" << qPrintable(fileInfo.absoluteFilePath()) << std::endl;
myVerifyFile.open( QIODevice::ReadOnly );
}
int64_t leftInPiece = getPieceSize( &myInfo, myVerifyPieceIndex ) - myVerifyPiecePos;
int64_t leftInFile = file->length - myVerifyFilePos;
int64_t bytesThisPass = std::min( leftInFile, leftInPiece );
bytesThisPass = std::min( bytesThisPass, (int64_t)sizeof( myVerifyBuf ) );
if( myVerifyFile.isOpen() && myVerifyFile.seek( myVerifyFilePos ) ) {
int64_t numRead = myVerifyFile.read( myVerifyBuf, bytesThisPass );
if( numRead == bytesThisPass )
myVerifyHash.addData( myVerifyBuf, numRead );
}
leftInPiece -= bytesThisPass;
leftInFile -= bytesThisPass;
myVerifyPiecePos += bytesThisPass;
myVerifyFilePos += bytesThisPass;
myVerifyBins[myVerifyFileIndex] += bytesThisPass;
if( leftInPiece == 0 )
{
const QByteArray result( myVerifyHash.result( ) );
const bool matches = !memcmp( result.constData(),
myInfo.pieces[myVerifyPieceIndex].hash,
SHA_DIGEST_LENGTH );
myVerifyFlags[myVerifyPieceIndex] = matches;
myVerifyPiecePos = 0;
++myVerifyPieceIndex;
myVerifyHash.reset( );
FileList changedFiles;
if( matches ) {
mybins_t::const_iterator i;
for( i=myVerifyBins.begin(); i!=myVerifyBins.end(); ++i ) {
TrFile& f( myFiles[i.key( )] );
f.have += i.value( );
changedFiles.append( f );
}
}
myTree->update( changedFiles );
myVerifyBins.clear( );
}
if( leftInFile == 0 )
{
//std::cerr << "closing file" << std::endl;
myVerifyFile.close( );
++myVerifyFileIndex;
myVerifyFilePos = 0;
}
bool done = myVerifyPieceIndex >= myInfo.pieceCount;
if( done )
{
uint64_t have = 0;
foreach( const TrFile& f, myFiles )
have += f.have;
if( !have ) // everything failed
{
// did the user accidentally specify the child directory instead of the parent?
const QStringList tokens = QString(file->name).split('/');
if( !tokens.empty() && myDestination.dirName()==tokens.at(0) )
{
// move up one directory and try again
myDestination.cdUp( );
refreshDestinationButton( -1 );
onVerify( );
done = false;
}
}
}
if( done )
myVerifyTimer.stop( );
}