Ivan Sorokin
10 years ago
12 changed files with 510 additions and 405 deletions
@ -0,0 +1,440 @@
@@ -0,0 +1,440 @@
|
||||
/*
|
||||
* 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. |
||||
* |
||||
* In addition, as a special exception, the copyright holders give permission to |
||||
* link this program with the OpenSSL project's "OpenSSL" library (or with |
||||
* modified versions of it that use the same license as the "OpenSSL" library), |
||||
* and distribute the linked executables. You must obey the GNU General Public |
||||
* License in all respects for all of the code used other than "OpenSSL". If you |
||||
* modify file(s), you may extend this exception to your version of the file(s), |
||||
* but you are not obligated to do so. If you do not wish to do so, delete this |
||||
* exception statement from your version. |
||||
* |
||||
* Contact : chris@qbittorrent.org |
||||
*/ |
||||
|
||||
#include "torrentpersistentdata.h" |
||||
|
||||
#include <QDebug> |
||||
#include <QVariant> |
||||
|
||||
#include "qinisettings.h" |
||||
#include "misc.h" |
||||
#include "qtorrenthandle.h" |
||||
|
||||
#include <libtorrent/version.hpp> |
||||
#include <libtorrent/magnet_uri.hpp> |
||||
|
||||
QHash<QString, TorrentTempData::TorrentData> TorrentTempData::data = QHash<QString, TorrentTempData::TorrentData>(); |
||||
QHash<QString, TorrentTempData::TorrentMoveState> TorrentTempData::torrentMoveStates = QHash<QString, TorrentTempData::TorrentMoveState>(); |
||||
QHash<QString, bool> HiddenData::data = QHash<QString, bool>(); |
||||
unsigned int HiddenData::metadata_counter = 0; |
||||
|
||||
bool TorrentTempData::hasTempData(const QString &hash) { |
||||
return data.contains(hash); |
||||
} |
||||
|
||||
void TorrentTempData::deleteTempData(const QString &hash) { |
||||
data.remove(hash); |
||||
} |
||||
|
||||
void TorrentTempData::setFilesPriority(const QString &hash, const std::vector<int> &pp) { |
||||
data[hash].files_priority = pp; |
||||
} |
||||
|
||||
void TorrentTempData::setFilesPath(const QString &hash, const QStringList &path_list) { |
||||
data[hash].path_list = path_list; |
||||
} |
||||
|
||||
void TorrentTempData::setSavePath(const QString &hash, const QString &save_path) { |
||||
data[hash].save_path = save_path; |
||||
} |
||||
|
||||
void TorrentTempData::setLabel(const QString &hash, const QString &label) { |
||||
data[hash].label = label; |
||||
} |
||||
|
||||
void TorrentTempData::setSequential(const QString &hash, const bool &sequential) { |
||||
data[hash].sequential = sequential; |
||||
} |
||||
|
||||
bool TorrentTempData::isSequential(const QString &hash) { |
||||
return data.value(hash).sequential; |
||||
} |
||||
|
||||
void TorrentTempData::setSeedingMode(const QString &hash, const bool &seed) { |
||||
data[hash].seed = seed; |
||||
} |
||||
|
||||
bool TorrentTempData::isSeedingMode(const QString &hash) { |
||||
return data.value(hash).seed; |
||||
} |
||||
|
||||
QString TorrentTempData::getSavePath(const QString &hash) { |
||||
return data.value(hash).save_path; |
||||
} |
||||
|
||||
QStringList TorrentTempData::getFilesPath(const QString &hash) { |
||||
return data.value(hash).path_list; |
||||
} |
||||
|
||||
QString TorrentTempData::getLabel(const QString &hash) { |
||||
return data.value(hash).label; |
||||
} |
||||
|
||||
void TorrentTempData::getFilesPriority(const QString &hash, std::vector<int> &fp) { |
||||
fp = data.value(hash).files_priority; |
||||
} |
||||
|
||||
bool TorrentTempData::isMoveInProgress(const QString &hash) { |
||||
return torrentMoveStates.find(hash) != torrentMoveStates.end(); |
||||
} |
||||
|
||||
void TorrentTempData::enqueueMove(const QString &hash, const QString &queuedPath) { |
||||
QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash); |
||||
if (i == torrentMoveStates.end()) { |
||||
Q_ASSERT(false); |
||||
return; |
||||
} |
||||
i->queuedPath = queuedPath; |
||||
} |
||||
|
||||
void TorrentTempData::startMove(const QString &hash, const QString &oldPath, const QString& newPath) { |
||||
QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash); |
||||
if (i != torrentMoveStates.end()) { |
||||
Q_ASSERT(false); |
||||
return; |
||||
} |
||||
|
||||
torrentMoveStates.insert(hash, TorrentMoveState(oldPath, newPath)); |
||||
} |
||||
|
||||
void TorrentTempData::finishMove(const QString &hash) { |
||||
QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash); |
||||
if (i == torrentMoveStates.end()) { |
||||
Q_ASSERT(false); |
||||
return; |
||||
} |
||||
torrentMoveStates.erase(i); |
||||
} |
||||
|
||||
QString TorrentTempData::getOldPath(const QString &hash) { |
||||
QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash); |
||||
if (i == torrentMoveStates.end()) { |
||||
Q_ASSERT(false); |
||||
return QString(); |
||||
} |
||||
return i->oldPath; |
||||
} |
||||
|
||||
QString TorrentTempData::getNewPath(const QString &hash) { |
||||
QHash<QString, TorrentMoveState>::iterator i = torrentMoveStates.find(hash); |
||||
if (i == torrentMoveStates.end()) { |
||||
Q_ASSERT(false); |
||||
return QString(); |
||||
} |
||||
return i->newPath; |
||||
} |
||||
|
||||
QString TorrentTempData::getQueuedPath(const QString &hash) { |
||||
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) { |
||||
data[hash] = false; |
||||
} |
||||
|
||||
bool HiddenData::hasData(const QString &hash) { |
||||
return data.contains(hash); |
||||
} |
||||
|
||||
void HiddenData::deleteData(const QString &hash) { |
||||
if (data.value(hash, false)) |
||||
metadata_counter--; |
||||
data.remove(hash); |
||||
} |
||||
|
||||
int HiddenData::getSize() { |
||||
return data.size(); |
||||
} |
||||
|
||||
int HiddenData::getDownloadingSize() { |
||||
return data.size() - metadata_counter; |
||||
} |
||||
|
||||
void HiddenData::gotMetadata(const QString &hash) { |
||||
if (!data.contains(hash)) |
||||
return; |
||||
data[hash] = true; |
||||
metadata_counter++; |
||||
} |
||||
|
||||
bool TorrentPersistentData::isKnownTorrent(QString hash) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
return all_data.contains(hash); |
||||
} |
||||
|
||||
QStringList TorrentPersistentData::knownTorrents() { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
return all_data.keys(); |
||||
} |
||||
|
||||
void TorrentPersistentData::setRatioLimit(const QString &hash, const qreal &ratio) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
data["max_ratio"] = ratio; |
||||
all_data[hash] = data; |
||||
settings.setValue("torrents", all_data); |
||||
} |
||||
|
||||
qreal TorrentPersistentData::getRatioLimit(const QString &hash) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
return data.value("max_ratio", USE_GLOBAL_RATIO).toReal(); |
||||
} |
||||
|
||||
bool TorrentPersistentData::hasPerTorrentRatioLimit() { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
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) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
if (!data.contains("add_date")) { |
||||
data["add_date"] = time; |
||||
all_data[hash] = data; |
||||
settings.setValue("torrents", all_data); |
||||
} |
||||
} |
||||
|
||||
QDateTime TorrentPersistentData::getAddedDate(const QString &hash) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
QDateTime dt = data.value("add_date").toDateTime(); |
||||
if (!dt.isValid()) { |
||||
setAddedDate(hash); |
||||
dt = QDateTime::currentDateTime(); |
||||
} |
||||
return dt; |
||||
} |
||||
|
||||
void TorrentPersistentData::setErrorState(const QString &hash, const bool has_error) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
data["has_error"] = has_error; |
||||
all_data[hash] = data; |
||||
settings.setValue("torrents", all_data); |
||||
} |
||||
|
||||
bool TorrentPersistentData::hasError(const QString &hash) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
return data.value("has_error", false).toBool(); |
||||
} |
||||
|
||||
QDateTime TorrentPersistentData::getSeedDate(const QString &hash) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
return data.value("seed_date").toDateTime(); |
||||
} |
||||
|
||||
void TorrentPersistentData::deletePersistentData(const QString &hash) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
if (all_data.contains(hash)) { |
||||
all_data.remove(hash); |
||||
settings.setValue("torrents", all_data); |
||||
} |
||||
} |
||||
|
||||
void TorrentPersistentData::saveTorrentPersistentData(const QTorrentHandle &h, const QString &save_path, const bool is_magnet) { |
||||
Q_ASSERT(h.is_valid()); |
||||
qDebug("Saving persistent data for %s", qPrintable(h.hash())); |
||||
// Save persistent data
|
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
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; |
||||
settings.setValue("torrents", all_data); |
||||
qDebug("TorrentPersistentData: Saving save_path %s, hash: %s", qPrintable(h.save_path()), qPrintable(h.hash())); |
||||
// Set Added date
|
||||
setAddedDate(h.hash()); |
||||
// Finally, remove temp data
|
||||
TorrentTempData::deleteTempData(h.hash()); |
||||
} |
||||
|
||||
void TorrentPersistentData::saveSavePath(const QString &hash, const QString &save_path) { |
||||
Q_ASSERT(!hash.isEmpty()); |
||||
qDebug("TorrentPersistentData::saveSavePath(%s)", qPrintable(save_path)); |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
data["save_path"] = save_path; |
||||
all_data[hash] = data; |
||||
settings.setValue("torrents", all_data); |
||||
qDebug("TorrentPersistentData: Saving save_path: %s, hash: %s", qPrintable(save_path), qPrintable(hash)); |
||||
} |
||||
|
||||
void TorrentPersistentData::saveLabel(const QString &hash, const QString &label) { |
||||
Q_ASSERT(!hash.isEmpty()); |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
data["label"] = label; |
||||
all_data[hash] = data; |
||||
settings.setValue("torrents", all_data); |
||||
} |
||||
|
||||
void TorrentPersistentData::saveName(const QString &hash, const QString &name) { |
||||
Q_ASSERT(!hash.isEmpty()); |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
data["name"] = name; |
||||
all_data[hash] = data; |
||||
settings.setValue("torrents", all_data); |
||||
} |
||||
|
||||
void TorrentPersistentData::savePriority(const QTorrentHandle &h) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
QHash<QString, QVariant> data = all_data[h.hash()].toHash(); |
||||
data["priority"] = h.queue_position(); |
||||
all_data[h.hash()] = data; |
||||
settings.setValue("torrents", all_data); |
||||
} |
||||
|
||||
void TorrentPersistentData::savePriority(const QString &hash, const int &queue_pos) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
QHash<QString, QVariant> data = all_data[hash].toHash(); |
||||
data["priority"] = queue_pos; |
||||
all_data[hash] = data; |
||||
settings.setValue("torrents", all_data); |
||||
} |
||||
|
||||
void TorrentPersistentData::saveSeedStatus(const QTorrentHandle &h) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
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; |
||||
settings.setValue("torrents", all_data); |
||||
} |
||||
} |
||||
|
||||
void TorrentPersistentData::saveSeedStatus(const QString &hash, const bool seedStatus) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
QHash<QString, QVariant> data = all_data[hash].toHash(); |
||||
data["seed"] = seedStatus; |
||||
all_data[hash] = data; |
||||
settings.setValue("torrents", all_data); |
||||
} |
||||
|
||||
QString TorrentPersistentData::getSavePath(const QString &hash) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
//qDebug("TorrentPersistentData: getSavePath %s", data["save_path"].toString().toLocal8Bit().data());
|
||||
return data.value("save_path").toString(); |
||||
} |
||||
|
||||
QString TorrentPersistentData::getLabel(const QString &hash) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
return data.value("label", "").toString(); |
||||
} |
||||
|
||||
QString TorrentPersistentData::getName(const QString &hash) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
return data.value("name", "").toString(); |
||||
} |
||||
|
||||
int TorrentPersistentData::getPriority(const QString &hash) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
return data.value("priority", -1).toInt(); |
||||
} |
||||
|
||||
bool TorrentPersistentData::isSeed(const QString &hash) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
return data.value("seed", false).toBool(); |
||||
} |
||||
|
||||
bool TorrentPersistentData::isMagnet(const QString &hash) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
return data.value("is_magnet", false).toBool(); |
||||
} |
||||
|
||||
QString TorrentPersistentData::getMagnetUri(const QString &hash) { |
||||
QIniSettings settings(QString::fromUtf8("qBittorrent"), QString::fromUtf8("qBittorrent-resume")); |
||||
const QHash<QString, QVariant> all_data = settings.value("torrents").toHash(); |
||||
const QHash<QString, QVariant> data = all_data.value(hash).toHash(); |
||||
Q_ASSERT(data.value("is_magnet", false).toBool()); |
||||
return data.value("magnet_uri").toString(); |
||||
} |
Loading…
Reference in new issue