github/qBittorrent
1
0
Fork 0
mirror of https://github.com/qbittorrent/qBittorrent synced 2025-08-19 21:03:30 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Follow project coding style. Issue #2192.

Browse source
This commit is contained in:
sledgehammer999 2015-01-27 15:30:49 +02:00
commit 70985480d6
2 changed files with 447 additions and 375 deletions
Download patch file Download diff file Expand all files Collapse all files

612
src/torrentpersistentdata.cpp
View file

@ -46,392 +46,462 @@ QHash<QString, bool> HiddenData::data = QHash<QString, bool>();
unsigned int HiddenData::metadata_counter = 0; unsigned int HiddenData::metadata_counter = 0;
TorrentPersistentData* TorrentPersistentData::m_instance = 0; TorrentPersistentData* TorrentPersistentData::m_instance = 0;
bool TorrentTempData::hasTempData(const QString &hash) { TorrentTempData::TorrentData::TorrentData()
return data.contains(hash); : sequential(false)
, seed(false)
, add_paused(Preferences::instance()->addTorrentsInPause())
{
} }
void TorrentTempData::deleteTempData(const QString &hash) { TorrentTempData::TorrentMoveState::TorrentMoveState(QString oldPath, QString newPath)
data.remove(hash); : oldPath(oldPath)
, newPath(newPath)
{
} }
void TorrentTempData::setFilesPriority(const QString &hash, const std::vector<int> &pp) { bool TorrentTempData::hasTempData(const QString &hash)
data[hash].files_priority = pp; {
return data.contains(hash);
} }
void TorrentTempData::setFilesPath(const QString &hash, const QStringList &path_list) { void TorrentTempData::deleteTempData(const QString &hash)
data[hash].path_list = path_list; {
data.remove(hash);
} }
void TorrentTempData::setSavePath(const QString &hash, const QString &save_path) { void TorrentTempData::setFilesPriority(const QString &hash, const std::vector<int> &pp)
data[hash].save_path = save_path; {
data[hash].files_priority = pp;
} }
void TorrentTempData::setLabel(const QString &hash, const QString &label) { void TorrentTempData::setFilesPath(const QString &hash, const QStringList &path_list)
data[hash].label = label; {
data[hash].path_list = path_list;
} }
void TorrentTempData::setSequential(const QString &hash, const bool &sequential) { void TorrentTempData::setSavePath(const QString &hash, const QString &save_path)
data[hash].sequential = sequential; {
data[hash].save_path = save_path;
} }
bool TorrentTempData::isSequential(const QString &hash) { void TorrentTempData::setLabel(const QString &hash, const QString &label)
return data.value(hash).sequential; {
data[hash].label = label;
} }
void TorrentTempData::setSeedingMode(const QString &hash, const bool &seed) { void TorrentTempData::setSequential(const QString &hash, const bool &sequential)
data[hash].seed = seed; {
data[hash].sequential = sequential;
} }
bool TorrentTempData::isSeedingMode(const QString &hash) { bool TorrentTempData::isSequential(const QString &hash)
return data.value(hash).seed; {
return data.value(hash).sequential;
} }
QString TorrentTempData::getSavePath(const QString &hash) { void TorrentTempData::setSeedingMode(const QString &hash, const bool &seed)
return data.value(hash).save_path; {
data[hash].seed = seed;
} }
QStringList TorrentTempData::getFilesPath(const QString &hash) { bool TorrentTempData::isSeedingMode(const QString &hash)
return data.value(hash).path_list; {
return data.value(hash).seed;
} }
QString TorrentTempData::getLabel(const QString &hash) { QString TorrentTempData::getSavePath(const QString &hash)
return data.value(hash).label; {
return data.value(hash).save_path;
} }
void TorrentTempData::getFilesPriority(const QString &hash, std::vector<int> &fp) { QStringList TorrentTempData::getFilesPath(const QString &hash)
fp = data.value(hash).files_priority; {
return data.value(hash).path_list;
} }
bool TorrentTempData::isMoveInProgress(const QString &hash) { QString TorrentTempData::getLabel(const QString &hash)
return torrentMoveStates.find(hash) != torrentMoveStates.end(); {
return data.value(hash).label;
} }
void TorrentTempData::enqueueMove(const QString &hash, const QString &queuedPath) { void TorrentTempData::getFilesPriority(const QString &hash, std::vector<int> &fp)
QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash); {
if (i == torrentMoveStates.end()) { fp = data.value(hash).files_priority;
Q_ASSERT(false);
return;
}
i->queuedPath = queuedPath;
} }
void TorrentTempData::startMove(const QString &hash, const QString &oldPath, const QString& newPath) { bool TorrentTempData::isMoveInProgress(const QString &hash)
QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash); {
if (i != torrentMoveStates.end()) { return torrentMoveStates.find(hash) != torrentMoveStates.end();
Q_ASSERT(false);
return;
}
torrentMoveStates.insert(hash, TorrentMoveState(oldPath, newPath));
} }
void TorrentTempData::finishMove(const QString &hash) { void TorrentTempData::enqueueMove(const QString &hash, const QString &queuedPath)
QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash); {
if (i == torrentMoveStates.end()) { QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash);
Q_ASSERT(false); if (i == torrentMoveStates.end()) {
return; Q_ASSERT(false);
} return;
torrentMoveStates.erase(i); }
i->queuedPath = queuedPath;
} }
QString TorrentTempData::getOldPath(const QString &hash) { void TorrentTempData::startMove(const QString &hash, const QString &oldPath, const QString& newPath)
QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash); {
if (i == torrentMoveStates.end()) { QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash);
Q_ASSERT(false); if (i != torrentMoveStates.end()) {
return QString(); Q_ASSERT(false);
} return;
return i->oldPath; }
torrentMoveStates.insert(hash, TorrentMoveState(oldPath, newPath));
} }
QString TorrentTempData::getNewPath(const QString &hash) { void TorrentTempData::finishMove(const QString &hash)
QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash); {
if (i == torrentMoveStates.end()) { QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash);
Q_ASSERT(false); if (i == torrentMoveStates.end()) {
return QString(); Q_ASSERT(false);
} return;
return i->newPath; }
torrentMoveStates.erase(i);
} }
QString TorrentTempData::getQueuedPath(const QString &hash) { QString TorrentTempData::getOldPath(const QString &hash)
QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash); {
if (i == torrentMoveStates.end()) { QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash);
Q_ASSERT(false); if (i == torrentMoveStates.end()) {
return QString(); Q_ASSERT(false);
} return QString();
return i->queuedPath; }
return i->oldPath;
} }
void TorrentTempData::setAddPaused(const QString &hash, const bool &paused) { QString TorrentTempData::getNewPath(const QString &hash)
data[hash].add_paused = paused; {
QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash);
if (i == torrentMoveStates.end()) {
Q_ASSERT(false);
return QString();
}
return i->newPath;
} }
bool TorrentTempData::isAddPaused(const QString &hash) { QString TorrentTempData::getQueuedPath(const QString &hash)
return data.value(hash).add_paused; {
QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash);
if (i == torrentMoveStates.end()) {
Q_ASSERT(false);
return QString();
}
return i->queuedPath;
} }
void HiddenData::addData(const QString &hash) { void TorrentTempData::setAddPaused(const QString &hash, const bool &paused)
data[hash] = false; {
data[hash].add_paused = paused;
} }
bool HiddenData::hasData(const QString &hash) { bool TorrentTempData::isAddPaused(const QString &hash)
return data.contains(hash); {
return data.value(hash).add_paused;
} }
void HiddenData::deleteData(const QString &hash) { void HiddenData::addData(const QString &hash)
if (data.value(hash, false)) {
metadata_counter--; data[hash] = false;
data.remove(hash);
} }
int HiddenData::getSize() { bool HiddenData::hasData(const QString &hash)
return data.size(); {
return data.contains(hash);
} }
int HiddenData::getDownloadingSize() { void HiddenData::deleteData(const QString &hash)
return data.size() - metadata_counter; {
if (data.value(hash, false))
metadata_counter--;
data.remove(hash);
} }
void HiddenData::gotMetadata(const QString &hash) { int HiddenData::getSize()
if (!data.contains(hash)) {
return; return data.size();
data[hash] = true; }
metadata_counter++;
int HiddenData::getDownloadingSize()
{
return data.size() - metadata_counter;
}
void HiddenData::gotMetadata(const QString &hash)
{
if (!data.contains(hash))
return;
data[hash] = true;
metadata_counter++;
} }
TorrentPersistentData::TorrentPersistentData() TorrentPersistentData::TorrentPersistentData()
: all_data(QIniSettings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")).value("torrents").toHash()) : all_data(QIniSettings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")).value("torrents").toHash())
, dirty(false) , dirty(false)
{ {
connect(&timer, SIGNAL(timeout()), SLOT(saveImpl())); connect(&timer, SIGNAL(timeout()), SLOT(saveImpl()));
m_instance = this; m_instance = this;
} }
TorrentPersistentData::~TorrentPersistentData() { TorrentPersistentData::~TorrentPersistentData()
save(); {
m_instance = 0; save();
m_instance = 0;
} }
void TorrentPersistentData::save() { void TorrentPersistentData::save()
if (dirty) {
saveImpl(); if (dirty)
saveImpl();
} }
TorrentPersistentData& TorrentPersistentData::instance() TorrentPersistentData& TorrentPersistentData::instance()
{ {
Q_ASSERT(m_instance); Q_ASSERT(m_instance);
return *m_instance; return *m_instance;
} }
bool TorrentPersistentData::isKnownTorrent(QString hash) { bool TorrentPersistentData::isKnownTorrent(QString hash)
return all_data.contains(hash); {
return all_data.contains(hash);
} }
QStringList TorrentPersistentData::knownTorrents() { QStringList TorrentPersistentData::knownTorrents()
return all_data.keys(); {
return all_data.keys();
} }
void TorrentPersistentData::setRatioLimit(const QString &hash, const qreal &ratio) { void TorrentPersistentData::setRatioLimit(const QString &hash, const qreal &ratio)
QHash<QString, QVariant> data = all_data.value(hash).toHash(); {
data["max_ratio"] = ratio; QHash<QString, QVariant> data = all_data.value(hash).toHash();
all_data[hash] = data; data["max_ratio"] = ratio;
markDirty();
}
qreal TorrentPersistentData::getRatioLimit(const QString &hash) {
const QHash<QString, QVariant> data = all_data.value(hash).toHash();
return data.value("max_ratio", USE_GLOBAL_RATIO).toReal();
}
bool TorrentPersistentData::hasPerTorrentRatioLimit() {
QHash<QString, QVariant>::ConstIterator it = all_data.constBegin();
QHash<QString, QVariant>::ConstIterator itend = all_data.constEnd();
for ( ; it != itend; ++it) {
if (it.value().toHash().value("max_ratio", USE_GLOBAL_RATIO).toReal() >= 0) {
return true;
}
}
return false;
}
void TorrentPersistentData::setAddedDate(const QString &hash, const QDateTime &time) {
QHash<QString, QVariant> data = all_data.value(hash).toHash();
if (!data.contains("add_date")) {
data["add_date"] = time;
all_data[hash] = data; all_data[hash] = data;
markDirty(); markDirty();
}
} }
QDateTime TorrentPersistentData::getAddedDate(const QString &hash) { qreal TorrentPersistentData::getRatioLimit(const QString &hash)
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); {
QDateTime dt = data.value("add_date").toDateTime(); const QHash<QString, QVariant> data = all_data.value(hash).toHash();
if (!dt.isValid()) { return data.value("max_ratio", USE_GLOBAL_RATIO).toReal();
setAddedDate(hash, QDateTime::currentDateTime());
dt = QDateTime::currentDateTime();
}
return dt;
} }
void TorrentPersistentData::setErrorState(const QString &hash, const bool has_error) { bool TorrentPersistentData::hasPerTorrentRatioLimit()
QHash<QString, QVariant> data = all_data.value(hash).toHash(); {
data["has_error"] = has_error; QHash<QString, QVariant>::ConstIterator it = all_data.constBegin();
all_data[hash] = data; QHash<QString, QVariant>::ConstIterator itend = all_data.constEnd();
markDirty(); for (; it != itend; ++it)
if (it.value().toHash().value("max_ratio", USE_GLOBAL_RATIO).toReal() >= 0)
return true;
return false;
} }
bool TorrentPersistentData::hasError(const QString &hash) { void TorrentPersistentData::setAddedDate(const QString &hash, const QDateTime &time)
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); {
return data.value("has_error", false).toBool(); QHash<QString, QVariant> data = all_data.value(hash).toHash();
if (!data.contains("add_date")) {
data["add_date"] = time;
all_data[hash] = data;
markDirty();
}
} }
QDateTime TorrentPersistentData::getSeedDate(const QString &hash) { QDateTime TorrentPersistentData::getAddedDate(const QString &hash)
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); {
return data.value("seed_date").toDateTime(); const QHash<QString, QVariant> data = all_data.value(hash).toHash();
QDateTime dt = data.value("add_date").toDateTime();
if (!dt.isValid()) {
setAddedDate(hash, QDateTime::currentDateTime());
dt = QDateTime::currentDateTime();
}
return dt;
} }
void TorrentPersistentData::deletePersistentData(const QString &hash) { void TorrentPersistentData::setErrorState(const QString &hash, const bool has_error)
if (all_data.contains(hash)) { {
all_data.remove(hash); QHash<QString, QVariant> data = all_data.value(hash).toHash();
data["has_error"] = has_error;
all_data[hash] = data;
markDirty(); markDirty();
}
} }
void TorrentPersistentData::saveTorrentPersistentData(const QTorrentHandle &h, const QString &save_path, const bool is_magnet) { bool TorrentPersistentData::hasError(const QString &hash)
Q_ASSERT(h.is_valid()); {
qDebug("Saving persistent data for %s", qPrintable(h.hash())); const QHash<QString, QVariant> data = all_data.value(hash).toHash();
// Save persistent data return data.value("has_error", false).toBool();
QHash<QString, QVariant> data = all_data.value(h.hash()).toHash();
data["is_magnet"] = is_magnet;
if (is_magnet) {
data["magnet_uri"] = misc::toQString(make_magnet_uri(h));
}
data["seed"] = h.is_seed();
data["priority"] = h.queue_position();
if (save_path.isEmpty()) {
qDebug("TorrentPersistantData: save path is %s", qPrintable(h.save_path()));
data["save_path"] = h.save_path();
} else {
qDebug("TorrentPersistantData: overriding save path is %s", qPrintable(save_path));
data["save_path"] = save_path; // Override torrent save path (e.g. because it is a temp dir)
}
// Label
data["label"] = TorrentTempData::getLabel(h.hash());
// Save data
all_data[h.hash()] = data;
markDirty();
qDebug("TorrentPersistentData: Saving save_path %s, hash: %s", qPrintable(h.save_path()), qPrintable(h.hash()));
// Set Added date
setAddedDate(h.hash(), QDateTime::currentDateTime());
// Finally, remove temp data
TorrentTempData::deleteTempData(h.hash());
} }
void TorrentPersistentData::saveSavePath(const QString &hash, const QString &save_path) { QDateTime TorrentPersistentData::getSeedDate(const QString &hash)
Q_ASSERT(!hash.isEmpty()); {
qDebug("TorrentPersistentData::saveSavePath(%s)", qPrintable(save_path)); const QHash<QString, QVariant> data = all_data.value(hash).toHash();
QHash<QString, QVariant> data = all_data.value(hash).toHash(); return data.value("seed_date").toDateTime();
data["save_path"] = save_path;
all_data[hash] = data;
markDirty();
qDebug("TorrentPersistentData: Saving save_path: %s, hash: %s", qPrintable(save_path), qPrintable(hash));
} }
void TorrentPersistentData::saveLabel(const QString &hash, const QString &label) { void TorrentPersistentData::deletePersistentData(const QString &hash)
Q_ASSERT(!hash.isEmpty()); {
QHash<QString, QVariant> data = all_data.value(hash).toHash(); if (all_data.contains(hash)) {
data["label"] = label; all_data.remove(hash);
all_data[hash] = data; markDirty();
markDirty(); }
} }
void TorrentPersistentData::saveName(const QString &hash, const QString &name) { void TorrentPersistentData::saveTorrentPersistentData(const QTorrentHandle &h, const QString &save_path, const bool is_magnet)
Q_ASSERT(!hash.isEmpty()); {
QHash<QString, QVariant> data = all_data.value(hash).toHash(); Q_ASSERT(h.is_valid());
data["name"] = name; qDebug("Saving persistent data for %s", qPrintable(h.hash()));
all_data[hash] = data; // Save persistent data
markDirty(); QHash<QString, QVariant> data = all_data.value(h.hash()).toHash();
} data["is_magnet"] = is_magnet;
if (is_magnet)
void TorrentPersistentData::savePriority(const QTorrentHandle &h) { data["magnet_uri"] = misc::toQString(make_magnet_uri(h));
QHash<QString, QVariant> data = all_data[h.hash()].toHash(); data["seed"] = h.is_seed();
data["priority"] = h.queue_position(); data["priority"] = h.queue_position();
all_data[h.hash()] = data; if (save_path.isEmpty()) {
markDirty(); qDebug("TorrentPersistantData: save path is %s", qPrintable(h.save_path()));
} data["save_path"] = h.save_path();
}
void TorrentPersistentData::savePriority(const QString &hash, const int &queue_pos) { else {
QHash<QString, QVariant> data = all_data[hash].toHash(); qDebug("TorrentPersistantData: overriding save path is %s", qPrintable(save_path));
data["priority"] = queue_pos; data["save_path"] = save_path; // Override torrent save path (e.g. because it is a temp dir)
all_data[hash] = data; }
markDirty(); // Label
} data["label"] = TorrentTempData::getLabel(h.hash());
// Save data
void TorrentPersistentData::saveSeedStatus(const QTorrentHandle &h) {
QHash<QString, QVariant> data = all_data[h.hash()].toHash();
bool was_seed = data.value("seed", false).toBool();
if (was_seed != h.is_seed()) {
data["seed"] = !was_seed;
all_data[h.hash()] = data; all_data[h.hash()] = data;
markDirty(); markDirty();
} qDebug("TorrentPersistentData: Saving save_path %s, hash: %s", qPrintable(h.save_path()), qPrintable(h.hash()));
// Set Added date
setAddedDate(h.hash(), QDateTime::currentDateTime());
// Finally, remove temp data
TorrentTempData::deleteTempData(h.hash());
} }
void TorrentPersistentData::saveSeedStatus(const QString &hash, const bool seedStatus) { void TorrentPersistentData::saveSavePath(const QString &hash, const QString &save_path)
QHash<QString, QVariant> data = all_data[hash].toHash(); {
data["seed"] = seedStatus; Q_ASSERT(!hash.isEmpty());
all_data[hash] = data; qDebug("TorrentPersistentData::saveSavePath(%s)", qPrintable(save_path));
markDirty(); QHash<QString, QVariant> data = all_data.value(hash).toHash();
data["save_path"] = save_path;
all_data[hash] = data;
markDirty();
qDebug("TorrentPersistentData: Saving save_path: %s, hash: %s", qPrintable(save_path), qPrintable(hash));
} }
QString TorrentPersistentData::getSavePath(const QString &hash) { void TorrentPersistentData::saveLabel(const QString &hash, const QString &label)
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); {
//qDebug("TorrentPersistentData: getSavePath %s", data["save_path"].toString().toLocal8Bit().data()); Q_ASSERT(!hash.isEmpty());
return data.value("save_path").toString(); QHash<QString, QVariant> data = all_data.value(hash).toHash();
data["label"] = label;
all_data[hash] = data;
markDirty();
} }
QString TorrentPersistentData::getLabel(const QString &hash) { void TorrentPersistentData::saveName(const QString &hash, const QString &name)
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); {
return data.value("label", "").toString(); Q_ASSERT(!hash.isEmpty());
QHash<QString, QVariant> data = all_data.value(hash).toHash();
data["name"] = name;
all_data[hash] = data;
markDirty();
} }
QString TorrentPersistentData::getName(const QString &hash) { void TorrentPersistentData::savePriority(const QTorrentHandle &h)
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); {
return data.value("name", "").toString(); QHash<QString, QVariant> data = all_data[h.hash()].toHash();
data["priority"] = h.queue_position();
all_data[h.hash()] = data;
markDirty();
} }
int TorrentPersistentData::getPriority(const QString &hash) { void TorrentPersistentData::savePriority(const QString &hash, const int &queue_pos)
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); {
return data.value("priority", -1).toInt(); QHash<QString, QVariant> data = all_data[hash].toHash();
data["priority"] = queue_pos;
all_data[hash] = data;
markDirty();
} }
bool TorrentPersistentData::isSeed(const QString &hash) { void TorrentPersistentData::saveSeedStatus(const QTorrentHandle &h)
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); {
return data.value("seed", false).toBool(); QHash<QString, QVariant> data = all_data[h.hash()].toHash();
bool was_seed = data.value("seed", false).toBool();
if (was_seed != h.is_seed()) {
data["seed"] = !was_seed;
all_data[h.hash()] = data;
markDirty();
}
} }
bool TorrentPersistentData::isMagnet(const QString &hash) { void TorrentPersistentData::saveSeedStatus(const QString &hash, const bool seedStatus)
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); {
return data.value("is_magnet", false).toBool(); QHash<QString, QVariant> data = all_data[hash].toHash();
data["seed"] = seedStatus;
all_data[hash] = data;
markDirty();
} }
QString TorrentPersistentData::getMagnetUri(const QString &hash) { QString TorrentPersistentData::getSavePath(const QString &hash)
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); {
Q_ASSERT(data.value("is_magnet", false).toBool()); const QHash<QString, QVariant> data = all_data.value(hash).toHash();
return data.value("magnet_uri").toString(); //qDebug("TorrentPersistentData: getSavePath %s", data["save_path"].toString().toLocal8Bit().data());
return data.value("save_path").toString();
} }
void TorrentPersistentData::markDirty() { QString TorrentPersistentData::getLabel(const QString &hash)
if (!dirty) { {
dirty = true; const QHash<QString, QVariant> data = all_data.value(hash).toHash();
timer.start(1000); return data.value("label", "").toString();
}
} }
void TorrentPersistentData::saveImpl() { QString TorrentPersistentData::getName(const QString &hash)
Q_ASSERT(dirty); {
const QHash<QString, QVariant> data = all_data.value(hash).toHash();
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); return data.value("name", "").toString();
settings.setValue("torrents", all_data); }
dirty = false;
timer.stop(); int TorrentPersistentData::getPriority(const QString &hash)
{
const QHash<QString, QVariant> data = all_data.value(hash).toHash();
return data.value("priority", -1).toInt();
}
bool TorrentPersistentData::isSeed(const QString &hash)
{
const QHash<QString, QVariant> data = all_data.value(hash).toHash();
return data.value("seed", false).toBool();
}
bool TorrentPersistentData::isMagnet(const QString &hash)
{
const QHash<QString, QVariant> data = all_data.value(hash).toHash();
return data.value("is_magnet", false).toBool();
}
QString TorrentPersistentData::getMagnetUri(const QString &hash)
{
const QHash<QString, QVariant> data = all_data.value(hash).toHash();
Q_ASSERT(data.value("is_magnet", false).toBool());
return data.value("magnet_uri").toString();
}
void TorrentPersistentData::markDirty()
{
if (!dirty) {
dirty = true;
timer.start(1000);
}
}
void TorrentPersistentData::saveImpl()
{
Q_ASSERT(dirty);
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume"));
settings.setValue("torrents", all_data);
dirty = false;
timer.stop();
} }

210
src/torrentpersistentdata.h
View file

@ -44,136 +44,138 @@ QT_END_NAMESPACE
class QTorrentHandle; class QTorrentHandle;
class TorrentTempData { class TorrentTempData
// This class stores strings w/o modifying separators {
// This class stores strings w/o modifying separators
public: public:
static bool hasTempData(const QString &hash); static bool hasTempData(const QString &hash);
static void deleteTempData(const QString &hash); static void deleteTempData(const QString &hash);
static void setFilesPriority(const QString &hash, const std::vector<int> &pp); static void setFilesPriority(const QString &hash, const std::vector<int> &pp);
static void setFilesPath(const QString &hash, const QStringList &path_list); static void setFilesPath(const QString &hash, const QStringList &path_list);
static void setSavePath(const QString &hash, const QString &save_path); static void setSavePath(const QString &hash, const QString &save_path);
static void setLabel(const QString &hash, const QString &label); static void setLabel(const QString &hash, const QString &label);
static void setSequential(const QString &hash, const bool &sequential); static void setSequential(const QString &hash, const bool &sequential);
static bool isSequential(const QString &hash); static bool isSequential(const QString &hash);
static void setSeedingMode(const QString &hash, const bool &seed); static void setSeedingMode(const QString &hash, const bool &seed);
static bool isSeedingMode(const QString &hash); static bool isSeedingMode(const QString &hash);
static QString getSavePath(const QString &hash); static QString getSavePath(const QString &hash);
static QStringList getFilesPath(const QString &hash); static QStringList getFilesPath(const QString &hash);
static QString getLabel(const QString &hash); static QString getLabel(const QString &hash);
static void getFilesPriority(const QString &hash, std::vector<int> &fp); static void getFilesPriority(const QString &hash, std::vector<int> &fp);
static bool isMoveInProgress(const QString &hash); static bool isMoveInProgress(const QString &hash);
static void enqueueMove(const QString &hash, const QString &queuedPath); static void enqueueMove(const QString &hash, const QString &queuedPath);
static void startMove(const QString &hash, const QString &oldPath, const QString& newPath); static void startMove(const QString &hash, const QString &oldPath, const QString& newPath);
static void finishMove(const QString &hash); static void finishMove(const QString &hash);
static QString getOldPath(const QString &hash); static QString getOldPath(const QString &hash);
static QString getNewPath(const QString &hash); static QString getNewPath(const QString &hash);
static QString getQueuedPath(const QString &hash); static QString getQueuedPath(const QString &hash);
static void setAddPaused(const QString &hash, const bool &paused); static void setAddPaused(const QString &hash, const bool &paused);
static bool isAddPaused(const QString &hash); static bool isAddPaused(const QString &hash);
private: private:
struct TorrentData { struct TorrentData
TorrentData(): sequential(false), seed(false), add_paused(Preferences::instance()->addTorrentsInPause()) {} {
std::vector<int> files_priority; TorrentData();
QStringList path_list; std::vector<int> files_priority;
QString save_path; QStringList path_list;
QString label; QString save_path;
bool sequential; QString label;
bool seed; bool sequential;
bool add_paused; bool seed;
}; bool add_paused;
};
struct TorrentMoveState { struct TorrentMoveState
TorrentMoveState(QString oldPath, QString newPath) {
: oldPath(oldPath) TorrentMoveState(QString oldPath, QString newPath);
, newPath(newPath) // the moving occurs from oldPath to newPath
{} // queuedPath is where files should be moved to, when current moving is completed
QString oldPath;
QString newPath;
QString queuedPath;
};
// the moving occurs from oldPath to newPath static QHash<QString, TorrentData> data;
// queuedPath is where files should be moved to, when current moving is completed static QHash<QString, TorrentMoveState> torrentMoveStates;
QString oldPath;
QString newPath;
QString queuedPath;
};
static QHash<QString, TorrentData> data;
static QHash<QString, TorrentMoveState> torrentMoveStates;
}; };
class HiddenData { class HiddenData
{
public: public:
static void addData(const QString &hash); static void addData(const QString &hash);
static bool hasData(const QString &hash); static bool hasData(const QString &hash);
static void deleteData(const QString &hash); static void deleteData(const QString &hash);
static int getSize(); static int getSize();
static int getDownloadingSize(); static int getDownloadingSize();
static void gotMetadata(const QString &hash); static void gotMetadata(const QString &hash);
private: private:
static QHash<QString, bool> data; static QHash<QString, bool> data;
static unsigned int metadata_counter; static unsigned int metadata_counter;
}; };
class TorrentPersistentData : QObject { class TorrentPersistentData: QObject
Q_OBJECT {
Q_DISABLE_COPY(TorrentPersistentData) Q_OBJECT
// This class stores strings w/o modifying separators Q_DISABLE_COPY(TorrentPersistentData)
// This class stores strings w/o modifying separators
public: public:
enum RatioLimit { enum RatioLimit
USE_GLOBAL_RATIO = -2, {
NO_RATIO_LIMIT = -1 USE_GLOBAL_RATIO = -2,
}; NO_RATIO_LIMIT = -1
};
public: public:
TorrentPersistentData(); TorrentPersistentData();
~TorrentPersistentData(); ~TorrentPersistentData();
void save(); void save();
public: public:
static TorrentPersistentData& instance(); static TorrentPersistentData& instance();
bool isKnownTorrent(QString hash); bool isKnownTorrent(QString hash);
QStringList knownTorrents(); QStringList knownTorrents();
void setRatioLimit(const QString &hash, const qreal &ratio); void setRatioLimit(const QString &hash, const qreal &ratio);
qreal getRatioLimit(const QString &hash); qreal getRatioLimit(const QString &hash);
bool hasPerTorrentRatioLimit() ; bool hasPerTorrentRatioLimit();
void setAddedDate(const QString &hash, const QDateTime &time); void setAddedDate(const QString &hash, const QDateTime &time);
QDateTime getAddedDate(const QString &hash); QDateTime getAddedDate(const QString &hash);
void setErrorState(const QString &hash, const bool has_error); void setErrorState(const QString &hash, const bool has_error);
bool hasError(const QString &hash); bool hasError(const QString &hash);
QDateTime getSeedDate(const QString &hash); QDateTime getSeedDate(const QString &hash);
void deletePersistentData(const QString &hash); void deletePersistentData(const QString &hash);
void saveTorrentPersistentData(const QTorrentHandle &h, const QString &save_path = QString::null, const bool is_magnet = false); void saveTorrentPersistentData(const QTorrentHandle &h, const QString &save_path = QString::null, const bool is_magnet = false);
// Setters // Setters
void saveSavePath(const QString &hash, const QString &save_path); void saveSavePath(const QString &hash, const QString &save_path);
void saveLabel(const QString &hash, const QString &label); void saveLabel(const QString &hash, const QString &label);
void saveName(const QString &hash, const QString &name); void saveName(const QString &hash, const QString &name);
void savePriority(const QTorrentHandle &h); void savePriority(const QTorrentHandle &h);
void savePriority(const QString &hash, const int &queue_pos); void savePriority(const QString &hash, const int &queue_pos);
void saveSeedStatus(const QTorrentHandle &h); void saveSeedStatus(const QTorrentHandle &h);
void saveSeedStatus(const QString &hash, const bool seedStatus); void saveSeedStatus(const QString &hash, const bool seedStatus);
// Getters // Getters
QString getSavePath(const QString &hash); QString getSavePath(const QString &hash);
QString getLabel(const QString &hash); QString getLabel(const QString &hash);
QString getName(const QString &hash); QString getName(const QString &hash);
int getPriority(const QString &hash); int getPriority(const QString &hash);
bool isSeed(const QString &hash); bool isSeed(const QString &hash);
bool isMagnet(const QString &hash); bool isMagnet(const QString &hash);
QString getMagnetUri(const QString &hash); QString getMagnetUri(const QString &hash);
private: private:
void markDirty(); void markDirty();
private slots: private slots:
void saveImpl(); void saveImpl();
private: private:
QHash<QString, QVariant> all_data; QHash<QString, QVariant> all_data;
bool dirty; bool dirty;
QTimer timer; QTimer timer;
static TorrentPersistentData* m_instance; static TorrentPersistentData* m_instance;
}; };
#endif // TORRENTPERSISTENTDATA_H #endif // TORRENTPERSISTENTDATA_H