transmission/qt/MakeDialog.cc

284 lines
7.6 KiB
C++

/*
* This file Copyright (C) 2009-2015 Mnemosyne LLC
*
* It may be used under the GNU GPL versions 2 or 3
* or any future license endorsed by Mnemosyne LLC.
*
*/
#include <QDir>
#include <QFileInfo>
#include <QMimeData>
#include <QPushButton>
#include <QTimer>
#include <libtransmission/transmission.h>
#include <libtransmission/makemeta.h>
#include <libtransmission/utils.h>
#include "ColumnResizer.h"
#include "Formatter.h"
#include "MakeDialog.h"
#include "Session.h"
#include "Utils.h"
#include "ui_MakeProgressDialog.h"
namespace
{
class MakeProgressDialog: public BaseDialog
{
Q_OBJECT
public:
MakeProgressDialog (Session& session, tr_metainfo_builder& builder, QWidget * parent = nullptr);
private slots:
void onButtonBoxClicked (QAbstractButton *);
void onProgress ();
private:
Session& mySession;
tr_metainfo_builder& myBuilder;
Ui::MakeProgressDialog ui;
QTimer myTimer;
};
}
MakeProgressDialog::MakeProgressDialog (Session& session, tr_metainfo_builder& builder, QWidget * parent):
BaseDialog (parent),
mySession (session),
myBuilder (builder)
{
ui.setupUi (this);
connect (ui.dialogButtons, SIGNAL (clicked (QAbstractButton *)),
this, SLOT (onButtonBoxClicked (QAbstractButton *)));
connect (&myTimer, SIGNAL (timeout ()), this, SLOT (onProgress ()));
myTimer.start (100);
onProgress ();
}
void
MakeProgressDialog::onButtonBoxClicked (QAbstractButton * button)
{
switch (ui.dialogButtons->standardButton (button))
{
case QDialogButtonBox::Open:
mySession.addNewlyCreatedTorrent (QString::fromUtf8 (myBuilder.outputFile),
QFileInfo (QString::fromUtf8 (myBuilder.top)).dir ().path ());
break;
case QDialogButtonBox::Abort:
myBuilder.abortFlag = true;
break;
default: // QDialogButtonBox::Ok:
break;
}
close ();
}
void
MakeProgressDialog::onProgress ()
{
// progress bar
const tr_metainfo_builder& b = myBuilder;
const double denom = b.pieceCount ? b.pieceCount : 1;
ui.progressBar->setValue (static_cast<int> ((100.0 * b.pieceIndex) / denom));
// progress label
const QString top = QString::fromUtf8 (b.top);
const QString base (QFileInfo (top).completeBaseName ());
QString str;
if (!b.isDone)
str = tr ("Creating \"%1\"").arg (base);
else if (b.result == TR_MAKEMETA_OK)
str = tr ("Created \"%1\"!").arg (base);
else if (b.result == TR_MAKEMETA_URL)
str = tr ("Error: invalid announce URL \"%1\"").arg (QString::fromUtf8 (b.errfile));
else if (b.result == TR_MAKEMETA_CANCELLED)
str = tr ("Cancelled");
else if (b.result == TR_MAKEMETA_IO_READ)
str = tr ("Error reading \"%1\": %2").arg (QString::fromUtf8 (b.errfile)).
arg (QString::fromLocal8Bit (tr_strerror (b.my_errno)));
else if (b.result == TR_MAKEMETA_IO_WRITE)
str = tr ("Error writing \"%1\": %2").arg (QString::fromUtf8 (b.errfile)).
arg (QString::fromLocal8Bit (tr_strerror (b.my_errno)));
ui.progressLabel->setText (str);
// buttons
ui.dialogButtons->button (QDialogButtonBox::Abort)->setEnabled (!b.isDone);
ui.dialogButtons->button (QDialogButtonBox::Ok)->setEnabled (b.isDone);
ui.dialogButtons->button (QDialogButtonBox::Open)->setEnabled (b.isDone && b.result == TR_MAKEMETA_OK);
}
#include "MakeDialog.moc"
/***
****
***/
void
MakeDialog::makeTorrent ()
{
if (myBuilder == nullptr)
return;
// get the tiers
int tier = 0;
QVector<tr_tracker_info> trackers;
for (const QString& line: ui.trackersEdit->toPlainText ().split (QLatin1Char ('\n')))
{
const QString announceUrl = line.trimmed ();
if (announceUrl.isEmpty ())
{
++tier;
}
else
{
tr_tracker_info tmp;
tmp.announce = tr_strdup (announceUrl.toUtf8 ().constData ());
tmp.tier = tier;
trackers.append (tmp);
}
}
// the file to create
const QString path = QString::fromUtf8 (myBuilder->top);
const QString torrentName = QFileInfo (path).completeBaseName () + QLatin1String (".torrent");
const QString target = QDir (ui.destinationButton->path ()).filePath (torrentName);
// comment
QString comment;
if (ui.commentCheck->isChecked ())
comment = ui.commentEdit->text ();
// start making the torrent
tr_makeMetaInfo (myBuilder.get (),
target.toUtf8 ().constData (),
trackers.isEmpty () ? NULL : trackers.data (),
trackers.size (),
comment.isEmpty () ? NULL : comment.toUtf8 ().constData (),
ui.privateCheck->isChecked ());
// pop up the dialog
MakeProgressDialog * dialog = new MakeProgressDialog (mySession, *myBuilder, this);
dialog->setAttribute (Qt::WA_DeleteOnClose);
dialog->open ();
}
/***
****
***/
QString
MakeDialog::getSource () const
{
return (ui.sourceFileRadio->isChecked () ? ui.sourceFileButton : ui.sourceFolderButton)->path ();
}
/***
****
***/
void
MakeDialog::onSourceChanged ()
{
myBuilder.reset ();
const QString filename = getSource ();
if (!filename.isEmpty ())
myBuilder.reset (tr_metaInfoBuilderCreate (filename.toUtf8 ().constData ()));
QString text;
if (myBuilder == nullptr)
{
text = tr ("<i>No source selected</i>");
}
else
{
QString files = tr ("%Ln File(s)", 0, myBuilder->fileCount);
QString pieces = tr ("%Ln Piece(s)", 0, myBuilder->pieceCount);
text = tr ("%1 in %2; %3 @ %4")
.arg (Formatter::sizeToString (myBuilder->totalSize))
.arg (files)
.arg (pieces)
.arg (Formatter::sizeToString (myBuilder->pieceSize));
}
ui.sourceSizeLabel->setText (text);
}
MakeDialog::MakeDialog (Session& session, QWidget * parent):
BaseDialog (parent),
mySession (session),
myBuilder (nullptr, &tr_metaInfoBuilderFree)
{
ui.setupUi (this);
ui.destinationButton->setMode (PathButton::DirectoryMode);
ui.destinationButton->setPath (QDir::homePath ());
ui.sourceFolderButton->setMode (PathButton::DirectoryMode);
ui.sourceFileButton->setMode (PathButton::FileMode);
ColumnResizer * cr (new ColumnResizer (this));
cr->addLayout (ui.filesSectionLayout);
cr->addLayout (ui.propertiesSectionLayout);
cr->update ();
resize (minimumSizeHint ());
connect (ui.sourceFolderRadio, SIGNAL (toggled (bool)), this, SLOT (onSourceChanged ()));
connect (ui.sourceFolderButton, SIGNAL (pathChanged (QString)), this, SLOT (onSourceChanged ()));
connect (ui.sourceFileRadio, SIGNAL (toggled (bool)), this, SLOT (onSourceChanged ()));
connect (ui.sourceFileButton, SIGNAL (pathChanged (QString)), this, SLOT (onSourceChanged ()));
connect (ui.dialogButtons, SIGNAL (accepted ()), this, SLOT (makeTorrent ()));
connect (ui.dialogButtons, SIGNAL (rejected ()), this, SLOT (close ()));
onSourceChanged ();
}
MakeDialog::~MakeDialog ()
{
}
/***
****
***/
void
MakeDialog::dragEnterEvent (QDragEnterEvent * event)
{
const QMimeData * mime = event->mimeData ();
if (!mime->urls ().isEmpty () && QFileInfo (mime->urls ().front ().path ()).exists ())
event->acceptProposedAction ();
}
void
MakeDialog::dropEvent (QDropEvent * event)
{
const QString filename = event->mimeData ()->urls ().front ().path ();
const QFileInfo fileInfo (filename);
if (fileInfo.exists ())
{
if (fileInfo.isDir ())
{
ui.sourceFolderRadio->setChecked (true);
ui.sourceFolderButton->setPath (filename);
}
else // it's a file
{
ui.sourceFileRadio->setChecked (true);
ui.sourceFileButton->setPath (filename);
}
}
}