1118 lines
28 KiB
C++
1118 lines
28 KiB
C++
/*
|
|
* This file Copyright (C) 2009-2016 Mnemosyne LLC
|
|
*
|
|
* It may be used under the GNU GPL versions 2 or 3
|
|
* or any future license endorsed by Mnemosyne LLC.
|
|
*
|
|
*/
|
|
|
|
#include <cassert>
|
|
#include <iostream>
|
|
|
|
#include <QApplication>
|
|
#include <QByteArray>
|
|
#include <QClipboard>
|
|
#include <QCoreApplication>
|
|
#include <QDebug>
|
|
#include <QDesktopServices>
|
|
#include <QFile>
|
|
#include <QFileInfo>
|
|
#include <QMessageBox>
|
|
#include <QStyle>
|
|
#include <QTextStream>
|
|
|
|
#include <libtransmission/transmission.h>
|
|
#include <libtransmission/session-id.h>
|
|
#include <libtransmission/utils.h> // tr_free
|
|
#include <libtransmission/variant.h>
|
|
|
|
#include "AddData.h"
|
|
#include "Prefs.h"
|
|
#include "RpcQueue.h"
|
|
#include "Session.h"
|
|
#include "SessionDialog.h"
|
|
#include "Torrent.h"
|
|
#include "Utils.h"
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
namespace
|
|
{
|
|
|
|
typedef Torrent::KeyList KeyList;
|
|
|
|
KeyList const& getInfoKeys()
|
|
{
|
|
return Torrent::getInfoKeys();
|
|
}
|
|
|
|
KeyList const& getStatKeys()
|
|
{
|
|
return Torrent::getStatKeys();
|
|
}
|
|
|
|
KeyList const& getExtraStatKeys()
|
|
{
|
|
return Torrent::getExtraStatKeys();
|
|
}
|
|
|
|
void addList(tr_variant* list, KeyList const& keys)
|
|
{
|
|
tr_variantListReserve(list, keys.size());
|
|
|
|
for (tr_quark const key : keys)
|
|
{
|
|
tr_variantListAddQuark(list, key);
|
|
}
|
|
}
|
|
|
|
// If this object is passed as "ids" (compared by address), then recently active torrents are queried.
|
|
QSet<int> const recentlyActiveIds = QSet<int>() << -1;
|
|
|
|
// If this object is passed as "ids" (compared by being empty), then all torrents are queried.
|
|
QSet<int> const allIds;
|
|
|
|
} // namespace
|
|
|
|
void Session::sessionSet(tr_quark const key, QVariant const& value)
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 1);
|
|
|
|
switch (value.type())
|
|
{
|
|
case QVariant::Bool:
|
|
tr_variantDictAddBool(&args, key, value.toBool());
|
|
break;
|
|
|
|
case QVariant::Int:
|
|
tr_variantDictAddInt(&args, key, value.toInt());
|
|
break;
|
|
|
|
case QVariant::Double:
|
|
tr_variantDictAddReal(&args, key, value.toDouble());
|
|
break;
|
|
|
|
case QVariant::String:
|
|
tr_variantDictAddStr(&args, key, value.toString().toUtf8().constData());
|
|
break;
|
|
|
|
default:
|
|
assert(false);
|
|
}
|
|
|
|
exec("session-set", &args);
|
|
}
|
|
|
|
void Session::portTest()
|
|
{
|
|
RpcQueue* q = new RpcQueue();
|
|
|
|
q->add([this]()
|
|
{
|
|
return exec("port-test", nullptr);
|
|
});
|
|
|
|
q->add([this](RpcResponse const& r)
|
|
{
|
|
bool isOpen = false;
|
|
|
|
if (r.success)
|
|
{
|
|
(void)tr_variantDictFindBool(r.args.get(), TR_KEY_port_is_open, &isOpen);
|
|
}
|
|
|
|
emit portTested(isOpen);
|
|
});
|
|
|
|
q->run();
|
|
}
|
|
|
|
void Session::copyMagnetLinkToClipboard(int torrentId)
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 2);
|
|
tr_variantListAddInt(tr_variantDictAddList(&args, TR_KEY_ids, 1), torrentId);
|
|
tr_variantListAddStr(tr_variantDictAddList(&args, TR_KEY_fields, 1), "magnetLink");
|
|
|
|
RpcQueue* q = new RpcQueue();
|
|
|
|
q->add([this, &args]()
|
|
{
|
|
return exec(TR_KEY_torrent_get, &args);
|
|
});
|
|
|
|
q->add([this](RpcResponse const& r)
|
|
{
|
|
tr_variant* torrents;
|
|
|
|
if (!tr_variantDictFindList(r.args.get(), TR_KEY_torrents, &torrents))
|
|
{
|
|
return;
|
|
}
|
|
|
|
tr_variant* const child = tr_variantListChild(torrents, 0);
|
|
char const* str;
|
|
|
|
if (child != nullptr && tr_variantDictFindStr(child, TR_KEY_magnetLink, &str, nullptr))
|
|
{
|
|
qApp->clipboard()->setText(QString::fromUtf8(str));
|
|
}
|
|
});
|
|
|
|
q->run();
|
|
}
|
|
|
|
void Session::updatePref(int key)
|
|
{
|
|
if (myPrefs.isCore(key))
|
|
{
|
|
switch (key)
|
|
{
|
|
case Prefs::ALT_SPEED_LIMIT_DOWN:
|
|
case Prefs::ALT_SPEED_LIMIT_ENABLED:
|
|
case Prefs::ALT_SPEED_LIMIT_TIME_BEGIN:
|
|
case Prefs::ALT_SPEED_LIMIT_TIME_DAY:
|
|
case Prefs::ALT_SPEED_LIMIT_TIME_ENABLED:
|
|
case Prefs::ALT_SPEED_LIMIT_TIME_END:
|
|
case Prefs::ALT_SPEED_LIMIT_UP:
|
|
case Prefs::BLOCKLIST_DATE:
|
|
case Prefs::BLOCKLIST_ENABLED:
|
|
case Prefs::BLOCKLIST_URL:
|
|
case Prefs::DHT_ENABLED:
|
|
case Prefs::DOWNLOAD_QUEUE_ENABLED:
|
|
case Prefs::DOWNLOAD_QUEUE_SIZE:
|
|
case Prefs::DSPEED:
|
|
case Prefs::DSPEED_ENABLED:
|
|
case Prefs::IDLE_LIMIT:
|
|
case Prefs::IDLE_LIMIT_ENABLED:
|
|
case Prefs::INCOMPLETE_DIR:
|
|
case Prefs::INCOMPLETE_DIR_ENABLED:
|
|
case Prefs::LPD_ENABLED:
|
|
case Prefs::PEER_LIMIT_GLOBAL:
|
|
case Prefs::PEER_LIMIT_TORRENT:
|
|
case Prefs::PEER_PORT:
|
|
case Prefs::PEER_PORT_RANDOM_ON_START:
|
|
case Prefs::QUEUE_STALLED_MINUTES:
|
|
case Prefs::PEX_ENABLED:
|
|
case Prefs::PORT_FORWARDING:
|
|
case Prefs::RENAME_PARTIAL_FILES:
|
|
case Prefs::SCRIPT_TORRENT_DONE_ENABLED:
|
|
case Prefs::SCRIPT_TORRENT_DONE_FILENAME:
|
|
case Prefs::START:
|
|
case Prefs::TRASH_ORIGINAL:
|
|
case Prefs::USPEED:
|
|
case Prefs::USPEED_ENABLED:
|
|
case Prefs::UTP_ENABLED:
|
|
sessionSet(myPrefs.getKey(key), myPrefs.variant(key));
|
|
break;
|
|
|
|
case Prefs::DOWNLOAD_DIR:
|
|
sessionSet(myPrefs.getKey(key), myPrefs.variant(key));
|
|
/* this will change the 'freespace' argument, so refresh */
|
|
refreshSessionInfo();
|
|
break;
|
|
|
|
case Prefs::RATIO:
|
|
sessionSet(TR_KEY_seedRatioLimit, myPrefs.variant(key));
|
|
break;
|
|
|
|
case Prefs::RATIO_ENABLED:
|
|
sessionSet(TR_KEY_seedRatioLimited, myPrefs.variant(key));
|
|
break;
|
|
|
|
case Prefs::ENCRYPTION:
|
|
{
|
|
int const i = myPrefs.variant(key).toInt();
|
|
|
|
switch (i)
|
|
{
|
|
case 0:
|
|
sessionSet(myPrefs.getKey(key), QLatin1String("tolerated"));
|
|
break;
|
|
|
|
case 1:
|
|
sessionSet(myPrefs.getKey(key), QLatin1String("preferred"));
|
|
break;
|
|
|
|
case 2:
|
|
sessionSet(myPrefs.getKey(key), QLatin1String("required"));
|
|
break;
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
case Prefs::RPC_AUTH_REQUIRED:
|
|
if (mySession != nullptr)
|
|
{
|
|
tr_sessionSetRPCPasswordEnabled(mySession, myPrefs.getBool(key));
|
|
}
|
|
|
|
break;
|
|
|
|
case Prefs::RPC_ENABLED:
|
|
if (mySession != nullptr)
|
|
{
|
|
tr_sessionSetRPCEnabled(mySession, myPrefs.getBool(key));
|
|
}
|
|
|
|
break;
|
|
|
|
case Prefs::RPC_PASSWORD:
|
|
if (mySession != nullptr)
|
|
{
|
|
tr_sessionSetRPCPassword(mySession, myPrefs.getString(key).toUtf8().constData());
|
|
}
|
|
|
|
break;
|
|
|
|
case Prefs::RPC_PORT:
|
|
if (mySession != nullptr)
|
|
{
|
|
tr_sessionSetRPCPort(mySession, myPrefs.getInt(key));
|
|
}
|
|
|
|
break;
|
|
|
|
case Prefs::RPC_USERNAME:
|
|
if (mySession != nullptr)
|
|
{
|
|
tr_sessionSetRPCUsername(mySession, myPrefs.getString(key).toUtf8().constData());
|
|
}
|
|
|
|
break;
|
|
|
|
case Prefs::RPC_WHITELIST_ENABLED:
|
|
if (mySession != nullptr)
|
|
{
|
|
tr_sessionSetRPCWhitelistEnabled(mySession, myPrefs.getBool(key));
|
|
}
|
|
|
|
break;
|
|
|
|
case Prefs::RPC_WHITELIST:
|
|
if (mySession != nullptr)
|
|
{
|
|
tr_sessionSetRPCWhitelist(mySession, myPrefs.getString(key).toUtf8().constData());
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
|
std::cerr << "unhandled pref: " << key << std::endl;
|
|
}
|
|
}
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
Session::Session(QString const& configDir, Prefs& prefs) :
|
|
myConfigDir(configDir),
|
|
myPrefs(prefs),
|
|
myBlocklistSize(-1),
|
|
mySession(nullptr),
|
|
myIsDefinitelyLocalSession(true)
|
|
{
|
|
myStats.ratio = TR_RATIO_NA;
|
|
myStats.uploadedBytes = 0;
|
|
myStats.downloadedBytes = 0;
|
|
myStats.filesAdded = 0;
|
|
myStats.sessionCount = 0;
|
|
myStats.secondsActive = 0;
|
|
myCumulativeStats = myStats;
|
|
|
|
connect(&myPrefs, SIGNAL(changed(int)), this, SLOT(updatePref(int)));
|
|
connect(&myRpc, SIGNAL(httpAuthenticationRequired()), this, SIGNAL(httpAuthenticationRequired()));
|
|
connect(&myRpc, SIGNAL(dataReadProgress()), this, SIGNAL(dataReadProgress()));
|
|
connect(&myRpc, SIGNAL(dataSendProgress()), this, SIGNAL(dataSendProgress()));
|
|
connect(&myRpc, SIGNAL(networkResponse(QNetworkReply::NetworkError, QString)), this,
|
|
SIGNAL(networkResponse(QNetworkReply::NetworkError, QString)));
|
|
}
|
|
|
|
Session::~Session()
|
|
{
|
|
stop();
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
void Session::stop()
|
|
{
|
|
myRpc.stop();
|
|
|
|
if (mySession != nullptr)
|
|
{
|
|
tr_sessionClose(mySession);
|
|
mySession = nullptr;
|
|
}
|
|
}
|
|
|
|
void Session::restart()
|
|
{
|
|
stop();
|
|
start();
|
|
}
|
|
|
|
void Session::start()
|
|
{
|
|
if (myPrefs.get<bool>(Prefs::SESSION_IS_REMOTE))
|
|
{
|
|
QUrl url;
|
|
url.setScheme(QLatin1String("http"));
|
|
url.setHost(myPrefs.get<QString>(Prefs::SESSION_REMOTE_HOST));
|
|
url.setPort(myPrefs.get<int>(Prefs::SESSION_REMOTE_PORT));
|
|
url.setPath(QLatin1String("/transmission/rpc"));
|
|
|
|
if (myPrefs.get<bool>(Prefs::SESSION_REMOTE_AUTH))
|
|
{
|
|
url.setUserName(myPrefs.get<QString>(Prefs::SESSION_REMOTE_USERNAME));
|
|
url.setPassword(myPrefs.get<QString>(Prefs::SESSION_REMOTE_PASSWORD));
|
|
}
|
|
|
|
myRpc.start(url);
|
|
}
|
|
else
|
|
{
|
|
tr_variant settings;
|
|
tr_variantInitDict(&settings, 0);
|
|
tr_sessionLoadSettings(&settings, myConfigDir.toUtf8().constData(), "qt");
|
|
mySession = tr_sessionInit(myConfigDir.toUtf8().constData(), true, &settings);
|
|
tr_variantFree(&settings);
|
|
|
|
myRpc.start(mySession);
|
|
|
|
tr_ctor* ctor = tr_ctorNew(mySession);
|
|
int torrentCount;
|
|
tr_torrent** torrents = tr_sessionLoadTorrents(mySession, ctor, &torrentCount);
|
|
tr_free(torrents);
|
|
tr_ctorFree(ctor);
|
|
}
|
|
|
|
emit sourceChanged();
|
|
}
|
|
|
|
bool Session::isServer() const
|
|
{
|
|
return mySession != nullptr;
|
|
}
|
|
|
|
bool Session::isLocal() const
|
|
{
|
|
if (!mySessionId.isEmpty())
|
|
{
|
|
return myIsDefinitelyLocalSession;
|
|
}
|
|
|
|
return myRpc.isLocal();
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
namespace
|
|
{
|
|
|
|
void addOptionalIds(tr_variant* args, QSet<int> const& ids)
|
|
{
|
|
if (&ids == &recentlyActiveIds)
|
|
{
|
|
tr_variantDictAddStr(args, TR_KEY_ids, "recently-active");
|
|
}
|
|
else if (!ids.isEmpty())
|
|
{
|
|
tr_variant* idList(tr_variantDictAddList(args, TR_KEY_ids, ids.size()));
|
|
|
|
for (int const i : ids)
|
|
{
|
|
tr_variantListAddInt(idList, i);
|
|
}
|
|
}
|
|
}
|
|
|
|
} // namespace
|
|
|
|
void Session::torrentSet(QSet<int> const& ids, tr_quark const key, double value)
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 2);
|
|
tr_variantDictAddReal(&args, key, value);
|
|
addOptionalIds(&args, ids);
|
|
|
|
exec(TR_KEY_torrent_set, &args);
|
|
}
|
|
|
|
void Session::torrentSet(QSet<int> const& ids, tr_quark const key, int value)
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 2);
|
|
tr_variantDictAddInt(&args, key, value);
|
|
addOptionalIds(&args, ids);
|
|
|
|
exec(TR_KEY_torrent_set, &args);
|
|
}
|
|
|
|
void Session::torrentSet(QSet<int> const& ids, tr_quark const key, bool value)
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 2);
|
|
tr_variantDictAddBool(&args, key, value);
|
|
addOptionalIds(&args, ids);
|
|
|
|
exec(TR_KEY_torrent_set, &args);
|
|
}
|
|
|
|
void Session::torrentSet(QSet<int> const& ids, tr_quark const key, QStringList const& value)
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 2);
|
|
addOptionalIds(&args, ids);
|
|
tr_variant* list(tr_variantDictAddList(&args, key, value.size()));
|
|
|
|
for (QString const& str : value)
|
|
{
|
|
tr_variantListAddStr(list, str.toUtf8().constData());
|
|
}
|
|
|
|
exec(TR_KEY_torrent_set, &args);
|
|
}
|
|
|
|
void Session::torrentSet(QSet<int> const& ids, tr_quark const key, QList<int> const& value)
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 2);
|
|
addOptionalIds(&args, ids);
|
|
tr_variant* list(tr_variantDictAddList(&args, key, value.size()));
|
|
|
|
for (int const i : value)
|
|
{
|
|
tr_variantListAddInt(list, i);
|
|
}
|
|
|
|
exec(TR_KEY_torrent_set, &args);
|
|
}
|
|
|
|
void Session::torrentSet(QSet<int> const& ids, tr_quark const key, QPair<int, QString> const& value)
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 2);
|
|
addOptionalIds(&args, ids);
|
|
tr_variant* list(tr_variantDictAddList(&args, key, 2));
|
|
tr_variantListAddInt(list, value.first);
|
|
tr_variantListAddStr(list, value.second.toUtf8().constData());
|
|
|
|
exec(TR_KEY_torrent_set, &args);
|
|
}
|
|
|
|
void Session::torrentSetLocation(QSet<int> const& ids, QString const& location, bool doMove)
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 3);
|
|
addOptionalIds(&args, ids);
|
|
tr_variantDictAddStr(&args, TR_KEY_location, location.toUtf8().constData());
|
|
tr_variantDictAddBool(&args, TR_KEY_move, doMove);
|
|
|
|
exec(TR_KEY_torrent_set_location, &args);
|
|
}
|
|
|
|
void Session::torrentRenamePath(QSet<int> const& ids, QString const& oldpath, QString const& newname)
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 2);
|
|
addOptionalIds(&args, ids);
|
|
tr_variantDictAddStr(&args, TR_KEY_path, oldpath.toUtf8().constData());
|
|
tr_variantDictAddStr(&args, TR_KEY_name, newname.toUtf8().constData());
|
|
|
|
RpcQueue* q = new RpcQueue();
|
|
|
|
q->add([this, &args]()
|
|
{
|
|
return exec("torrent-rename-path", &args);
|
|
},
|
|
[this](RpcResponse const& r)
|
|
{
|
|
char const* path = "(unknown)";
|
|
char const* name = "(unknown)";
|
|
tr_variantDictFindStr(r.args.get(), TR_KEY_path, &path, nullptr);
|
|
tr_variantDictFindStr(r.args.get(), TR_KEY_name, &name, nullptr);
|
|
|
|
QMessageBox* d = new QMessageBox(QMessageBox::Information, tr("Error Renaming Path"),
|
|
tr("<p><b>Unable to rename \"%1\" as \"%2\": %3.</b></p><p>Please correct the errors and try again.</p>").
|
|
arg(QString::fromUtf8(path)).arg(QString::fromUtf8(name)).arg(r.result), QMessageBox::Close,
|
|
qApp->activeWindow());
|
|
connect(d, SIGNAL(rejected()), d, SLOT(deleteLater()));
|
|
d->show();
|
|
});
|
|
|
|
q->add([this](RpcResponse const& r)
|
|
{
|
|
int64_t id = 0;
|
|
|
|
if (tr_variantDictFindInt(r.args.get(), TR_KEY_id, &id) && id != 0)
|
|
{
|
|
refreshTorrents(QSet<int>() << id, KeyList() << TR_KEY_fileStats << TR_KEY_files << TR_KEY_id << TR_KEY_name);
|
|
}
|
|
});
|
|
|
|
q->run();
|
|
}
|
|
|
|
void Session::refreshTorrents(QSet<int> const& ids, KeyList const& keys)
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 2);
|
|
addList(tr_variantDictAddList(&args, TR_KEY_fields, 0), keys);
|
|
addOptionalIds(&args, ids);
|
|
|
|
RpcQueue* q = new RpcQueue();
|
|
|
|
q->add([this, &args]()
|
|
{
|
|
return exec(TR_KEY_torrent_get, &args);
|
|
});
|
|
|
|
bool const allTorrents = ids.empty();
|
|
|
|
q->add([this, allTorrents](RpcResponse const& r)
|
|
{
|
|
tr_variant* torrents;
|
|
|
|
if (tr_variantDictFindList(r.args.get(), TR_KEY_torrents, &torrents))
|
|
{
|
|
emit torrentsUpdated(torrents, allTorrents);
|
|
}
|
|
|
|
if (tr_variantDictFindList(r.args.get(), TR_KEY_removed, &torrents))
|
|
{
|
|
emit torrentsRemoved(torrents);
|
|
}
|
|
});
|
|
|
|
q->run();
|
|
}
|
|
|
|
void Session::refreshExtraStats(QSet<int> const& ids)
|
|
{
|
|
refreshTorrents(ids, getStatKeys() + getExtraStatKeys());
|
|
}
|
|
|
|
void Session::sendTorrentRequest(char const* request, QSet<int> const& ids)
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 1);
|
|
addOptionalIds(&args, ids);
|
|
|
|
RpcQueue* q = new RpcQueue();
|
|
|
|
q->add([this, request, &args]()
|
|
{
|
|
return exec(request, &args);
|
|
});
|
|
|
|
q->add([this, ids]()
|
|
{
|
|
refreshTorrents(ids, getStatKeys());
|
|
});
|
|
|
|
q->run();
|
|
}
|
|
|
|
void Session::pauseTorrents(QSet<int> const& ids)
|
|
{
|
|
sendTorrentRequest("torrent-stop", ids);
|
|
}
|
|
|
|
void Session::startTorrents(QSet<int> const& ids)
|
|
{
|
|
sendTorrentRequest("torrent-start", ids);
|
|
}
|
|
|
|
void Session::startTorrentsNow(QSet<int> const& ids)
|
|
{
|
|
sendTorrentRequest("torrent-start-now", ids);
|
|
}
|
|
|
|
void Session::queueMoveTop(QSet<int> const& ids)
|
|
{
|
|
sendTorrentRequest("queue-move-top", ids);
|
|
}
|
|
|
|
void Session::queueMoveUp(QSet<int> const& ids)
|
|
{
|
|
sendTorrentRequest("queue-move-up", ids);
|
|
}
|
|
|
|
void Session::queueMoveDown(QSet<int> const& ids)
|
|
{
|
|
sendTorrentRequest("queue-move-down", ids);
|
|
}
|
|
|
|
void Session::queueMoveBottom(QSet<int> const& ids)
|
|
{
|
|
sendTorrentRequest("queue-move-bottom", ids);
|
|
}
|
|
|
|
void Session::refreshActiveTorrents()
|
|
{
|
|
refreshTorrents(recentlyActiveIds, getStatKeys());
|
|
}
|
|
|
|
void Session::refreshAllTorrents()
|
|
{
|
|
refreshTorrents(allIds, getStatKeys());
|
|
}
|
|
|
|
void Session::initTorrents(QSet<int> const& ids)
|
|
{
|
|
refreshTorrents(ids, getStatKeys() + getInfoKeys());
|
|
}
|
|
|
|
void Session::refreshSessionStats()
|
|
{
|
|
RpcQueue* q = new RpcQueue();
|
|
|
|
q->add([this]()
|
|
{
|
|
return exec("session-stats", nullptr);
|
|
});
|
|
|
|
q->add([this](RpcResponse const& r)
|
|
{
|
|
updateStats(r.args.get());
|
|
});
|
|
|
|
q->run();
|
|
}
|
|
|
|
void Session::refreshSessionInfo()
|
|
{
|
|
RpcQueue* q = new RpcQueue();
|
|
|
|
q->add([this]()
|
|
{
|
|
return exec("session-get", nullptr);
|
|
});
|
|
|
|
q->add([this](RpcResponse const& r)
|
|
{
|
|
updateInfo(r.args.get());
|
|
});
|
|
|
|
q->run();
|
|
}
|
|
|
|
void Session::updateBlocklist()
|
|
{
|
|
RpcQueue* q = new RpcQueue();
|
|
|
|
q->add([this]()
|
|
{
|
|
return exec("blocklist-update", nullptr);
|
|
});
|
|
|
|
q->add([this](RpcResponse const& r)
|
|
{
|
|
int64_t blocklistSize;
|
|
|
|
if (tr_variantDictFindInt(r.args.get(), TR_KEY_blocklist_size, &blocklistSize))
|
|
{
|
|
setBlocklistSize(blocklistSize);
|
|
}
|
|
});
|
|
|
|
q->run();
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
RpcResponseFuture Session::exec(tr_quark method, tr_variant* args)
|
|
{
|
|
return myRpc.exec(method, args);
|
|
}
|
|
|
|
RpcResponseFuture Session::exec(char const* method, tr_variant* args)
|
|
{
|
|
return myRpc.exec(method, args);
|
|
}
|
|
|
|
void Session::updateStats(tr_variant* d, tr_session_stats* stats)
|
|
{
|
|
int64_t i;
|
|
|
|
if (tr_variantDictFindInt(d, TR_KEY_uploadedBytes, &i))
|
|
{
|
|
stats->uploadedBytes = i;
|
|
}
|
|
|
|
if (tr_variantDictFindInt(d, TR_KEY_downloadedBytes, &i))
|
|
{
|
|
stats->downloadedBytes = i;
|
|
}
|
|
|
|
if (tr_variantDictFindInt(d, TR_KEY_filesAdded, &i))
|
|
{
|
|
stats->filesAdded = i;
|
|
}
|
|
|
|
if (tr_variantDictFindInt(d, TR_KEY_sessionCount, &i))
|
|
{
|
|
stats->sessionCount = i;
|
|
}
|
|
|
|
if (tr_variantDictFindInt(d, TR_KEY_secondsActive, &i))
|
|
{
|
|
stats->secondsActive = i;
|
|
}
|
|
|
|
stats->ratio = tr_getRatio(stats->uploadedBytes, stats->downloadedBytes);
|
|
}
|
|
|
|
void Session::updateStats(tr_variant* d)
|
|
{
|
|
tr_variant* c;
|
|
|
|
if (tr_variantDictFindDict(d, TR_KEY_current_stats, &c))
|
|
{
|
|
updateStats(c, &myStats);
|
|
}
|
|
|
|
if (tr_variantDictFindDict(d, TR_KEY_cumulative_stats, &c))
|
|
{
|
|
updateStats(c, &myCumulativeStats);
|
|
}
|
|
|
|
emit statsUpdated();
|
|
}
|
|
|
|
void Session::updateInfo(tr_variant* d)
|
|
{
|
|
int64_t i;
|
|
char const* str;
|
|
|
|
disconnect(&myPrefs, SIGNAL(changed(int)), this, SLOT(updatePref(int)));
|
|
|
|
for (int i = Prefs::FIRST_CORE_PREF; i <= Prefs::LAST_CORE_PREF; ++i)
|
|
{
|
|
tr_variant const* b(tr_variantDictFind(d, myPrefs.getKey(i)));
|
|
|
|
if (b == nullptr)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
if (i == Prefs::ENCRYPTION)
|
|
{
|
|
char const* val;
|
|
|
|
if (tr_variantGetStr(b, &val, nullptr))
|
|
{
|
|
if (qstrcmp(val, "required") == 0)
|
|
{
|
|
myPrefs.set(i, 2);
|
|
}
|
|
else if (qstrcmp(val, "preferred") == 0)
|
|
{
|
|
myPrefs.set(i, 1);
|
|
}
|
|
else if (qstrcmp(val, "tolerated") == 0)
|
|
{
|
|
myPrefs.set(i, 0);
|
|
}
|
|
}
|
|
|
|
continue;
|
|
}
|
|
|
|
switch (myPrefs.type(i))
|
|
{
|
|
case QVariant::Int:
|
|
{
|
|
int64_t val;
|
|
|
|
if (tr_variantGetInt(b, &val))
|
|
{
|
|
myPrefs.set(i, static_cast<int>(val));
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
case QVariant::Double:
|
|
{
|
|
double val;
|
|
|
|
if (tr_variantGetReal(b, &val))
|
|
{
|
|
myPrefs.set(i, val);
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
case QVariant::Bool:
|
|
{
|
|
bool val;
|
|
|
|
if (tr_variantGetBool(b, &val))
|
|
{
|
|
myPrefs.set(i, val);
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
case CustomVariantType::FilterModeType:
|
|
case CustomVariantType::SortModeType:
|
|
case QVariant::String:
|
|
{
|
|
char const* val;
|
|
|
|
if (tr_variantGetStr(b, &val, nullptr))
|
|
{
|
|
myPrefs.set(i, QString::fromUtf8(val));
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
bool b;
|
|
double x;
|
|
|
|
if (tr_variantDictFindBool(d, TR_KEY_seedRatioLimited, &b))
|
|
{
|
|
myPrefs.set(Prefs::RATIO_ENABLED, b);
|
|
}
|
|
|
|
if (tr_variantDictFindReal(d, TR_KEY_seedRatioLimit, &x))
|
|
{
|
|
myPrefs.set(Prefs::RATIO, x);
|
|
}
|
|
|
|
/* Use the C API to get settings that, for security reasons, aren't supported by RPC */
|
|
if (mySession != nullptr)
|
|
{
|
|
myPrefs.set(Prefs::RPC_ENABLED, tr_sessionIsRPCEnabled(mySession));
|
|
myPrefs.set(Prefs::RPC_AUTH_REQUIRED, tr_sessionIsRPCPasswordEnabled(mySession));
|
|
myPrefs.set(Prefs::RPC_PASSWORD, QString::fromUtf8(tr_sessionGetRPCPassword(mySession)));
|
|
myPrefs.set(Prefs::RPC_PORT, tr_sessionGetRPCPort(mySession));
|
|
myPrefs.set(Prefs::RPC_USERNAME, QString::fromUtf8(tr_sessionGetRPCUsername(mySession)));
|
|
myPrefs.set(Prefs::RPC_WHITELIST_ENABLED, tr_sessionGetRPCWhitelistEnabled(mySession));
|
|
myPrefs.set(Prefs::RPC_WHITELIST, QString::fromUtf8(tr_sessionGetRPCWhitelist(mySession)));
|
|
}
|
|
|
|
if (tr_variantDictFindInt(d, TR_KEY_blocklist_size, &i) && i != blocklistSize())
|
|
{
|
|
setBlocklistSize(i);
|
|
}
|
|
|
|
if (tr_variantDictFindStr(d, TR_KEY_version, &str, nullptr) && mySessionVersion != QString::fromUtf8(str))
|
|
{
|
|
mySessionVersion = QString::fromUtf8(str);
|
|
}
|
|
|
|
if (tr_variantDictFindStr(d, TR_KEY_session_id, &str, nullptr))
|
|
{
|
|
QString const sessionId = QString::fromUtf8(str);
|
|
|
|
if (mySessionId != sessionId)
|
|
{
|
|
mySessionId = sessionId;
|
|
myIsDefinitelyLocalSession = tr_session_id_is_local(str);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
mySessionId.clear();
|
|
}
|
|
|
|
// std::cerr << "Session::updateInfo end" << std::endl;
|
|
connect(&myPrefs, SIGNAL(changed(int)), this, SLOT(updatePref(int)));
|
|
|
|
emit sessionUpdated();
|
|
}
|
|
|
|
void Session::setBlocklistSize(int64_t i)
|
|
{
|
|
myBlocklistSize = i;
|
|
|
|
emit blocklistUpdated(i);
|
|
}
|
|
|
|
void Session::addTorrent(AddData const& addMe, tr_variant* args, bool trashOriginal)
|
|
{
|
|
assert(tr_variantDictFind(args, TR_KEY_filename) == nullptr);
|
|
assert(tr_variantDictFind(args, TR_KEY_metainfo) == nullptr);
|
|
|
|
if (tr_variantDictFind(args, TR_KEY_paused) == nullptr)
|
|
{
|
|
tr_variantDictAddBool(args, TR_KEY_paused, !myPrefs.getBool(Prefs::START));
|
|
}
|
|
|
|
switch (addMe.type)
|
|
{
|
|
case AddData::MAGNET:
|
|
tr_variantDictAddStr(args, TR_KEY_filename, addMe.magnet.toUtf8().constData());
|
|
break;
|
|
|
|
case AddData::URL:
|
|
tr_variantDictAddStr(args, TR_KEY_filename, addMe.url.toString().toUtf8().constData());
|
|
break;
|
|
|
|
case AddData::FILENAME: /* fall-through */
|
|
case AddData::METAINFO:
|
|
{
|
|
QByteArray const b64 = addMe.toBase64();
|
|
tr_variantDictAddRaw(args, TR_KEY_metainfo, b64.constData(), b64.size());
|
|
break;
|
|
}
|
|
|
|
default:
|
|
qWarning() << "Unhandled AddData type: " << addMe.type;
|
|
break;
|
|
}
|
|
|
|
RpcQueue* q = new RpcQueue();
|
|
|
|
q->add([this, args]()
|
|
{
|
|
return exec("torrent-add", args);
|
|
},
|
|
[this, addMe](RpcResponse const& r)
|
|
{
|
|
QMessageBox* d = new QMessageBox(QMessageBox::Warning, tr("Error Adding Torrent"),
|
|
QString::fromLatin1("<p><b>%1</b></p><p>%2</p>").arg(r.result).arg(addMe.readableName()), QMessageBox::Close,
|
|
qApp->activeWindow());
|
|
connect(d, SIGNAL(rejected()), d, SLOT(deleteLater()));
|
|
d->show();
|
|
});
|
|
|
|
q->add([this, addMe](RpcResponse const& r)
|
|
{
|
|
tr_variant* dup;
|
|
|
|
if (!tr_variantDictFindDict(r.args.get(), TR_KEY_torrent_duplicate, &dup))
|
|
{
|
|
return;
|
|
}
|
|
|
|
char const* str;
|
|
|
|
if (tr_variantDictFindStr(dup, TR_KEY_name, &str, nullptr))
|
|
{
|
|
QString const name = QString::fromUtf8(str);
|
|
QMessageBox* d = new QMessageBox(QMessageBox::Warning, tr("Add Torrent"),
|
|
tr("<p><b>Unable to add \"%1\".</b></p><p>It is a duplicate of \"%2\" which is already added.</p>").
|
|
arg(addMe.readableShortName()).arg(name), QMessageBox::Close, qApp->activeWindow());
|
|
connect(d, SIGNAL(rejected()), d, SLOT(deleteLater()));
|
|
d->show();
|
|
}
|
|
});
|
|
|
|
if (trashOriginal && addMe.type == AddData::FILENAME)
|
|
{
|
|
q->add([this, addMe]()
|
|
{
|
|
QFile original(addMe.filename);
|
|
original.setPermissions(QFile::ReadOwner | QFile::WriteOwner);
|
|
original.remove();
|
|
});
|
|
}
|
|
|
|
q->run();
|
|
}
|
|
|
|
void Session::addTorrent(AddData const& addMe)
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 3);
|
|
|
|
addTorrent(addMe, &args, myPrefs.getBool(Prefs::TRASH_ORIGINAL));
|
|
}
|
|
|
|
void Session::addNewlyCreatedTorrent(QString const& filename, QString const& localPath)
|
|
{
|
|
QByteArray const b64 = AddData(filename).toBase64();
|
|
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 3);
|
|
tr_variantDictAddStr(&args, TR_KEY_download_dir, localPath.toUtf8().constData());
|
|
tr_variantDictAddBool(&args, TR_KEY_paused, !myPrefs.getBool(Prefs::START));
|
|
tr_variantDictAddRaw(&args, TR_KEY_metainfo, b64.constData(), b64.size());
|
|
|
|
exec("torrent-add", &args);
|
|
}
|
|
|
|
void Session::removeTorrents(QSet<int> const& ids, bool deleteFiles)
|
|
{
|
|
if (!ids.isEmpty())
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 2);
|
|
addOptionalIds(&args, ids);
|
|
tr_variantDictAddInt(&args, TR_KEY_delete_local_data, deleteFiles);
|
|
|
|
exec("torrent-remove", &args);
|
|
}
|
|
}
|
|
|
|
void Session::verifyTorrents(QSet<int> const& ids)
|
|
{
|
|
if (!ids.isEmpty())
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 1);
|
|
addOptionalIds(&args, ids);
|
|
|
|
exec("torrent-verify", &args);
|
|
}
|
|
}
|
|
|
|
void Session::reannounceTorrents(QSet<int> const& ids)
|
|
{
|
|
if (!ids.isEmpty())
|
|
{
|
|
tr_variant args;
|
|
tr_variantInitDict(&args, 1);
|
|
addOptionalIds(&args, ids);
|
|
|
|
exec("torrent-reannounce", &args);
|
|
}
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
void Session::launchWebInterface()
|
|
{
|
|
QUrl url;
|
|
|
|
if (mySession == nullptr) // remote session
|
|
{
|
|
url = myRpc.url();
|
|
url.setPath(QLatin1String("/transmission/web/"));
|
|
}
|
|
else // local session
|
|
{
|
|
url.setScheme(QLatin1String("http"));
|
|
url.setHost(QLatin1String("localhost"));
|
|
url.setPort(myPrefs.getInt(Prefs::RPC_PORT));
|
|
}
|
|
|
|
QDesktopServices::openUrl(url);
|
|
}
|