Vladimir Golovnev (Glassez)
9 years ago
7 changed files with 320 additions and 189 deletions
@ -0,0 +1,209 @@
@@ -0,0 +1,209 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2016 Vladimir Golovnev <glassez@yandex.ru> |
||||
* Copyright (C) 2014 sledgehammer999 <hammered999@gmail.com> |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
#include <QFile> |
||||
#include <QHash> |
||||
#include <QStringList> |
||||
#include <QSettings> |
||||
|
||||
#include "logger.h" |
||||
#include "utils/fs.h" |
||||
#include "settingsstorage.h" |
||||
|
||||
namespace |
||||
{ |
||||
inline QSettings *createSettings(const QString &name) |
||||
{ |
||||
#ifdef Q_OS_WIN |
||||
return new QSettings(QSettings::IniFormat, QSettings::UserScope, "qBittorrent", name); |
||||
#else |
||||
return new QSettings("qBittorrent", name); |
||||
#endif |
||||
} |
||||
|
||||
QString mapKey(const QString &key) |
||||
{ |
||||
static const QHash<QString, QString> keyMapping = {}; |
||||
return keyMapping.value(key, key); |
||||
} |
||||
} |
||||
|
||||
SettingsStorage *SettingsStorage::m_instance = nullptr; |
||||
|
||||
SettingsStorage::SettingsStorage() |
||||
: m_dirty(false) |
||||
, m_lock(QReadWriteLock::Recursive) |
||||
{ |
||||
QSettings *settings; |
||||
QStringList keys; |
||||
|
||||
#ifdef Q_OS_MAC |
||||
settings = createSettings("qBittorrent"); |
||||
#else |
||||
settings = createSettings("qBittorrent_new"); |
||||
QString newPath = settings->fileName(); |
||||
|
||||
// This means that the PC closed either due to power outage
|
||||
// or because the disk was full. In any case the settings weren't transfered
|
||||
// in their final position. So assume that qbittorrent_new.ini/qbittorrent_new.conf
|
||||
// contains the most recent settings.
|
||||
if (!settings->allKeys().isEmpty()) { |
||||
Logger::instance()->addMessage(tr("Detected unclean program exit. Using fallback file to restore settings."), Log::WARNING); |
||||
m_dirty = true; |
||||
} |
||||
else { |
||||
delete settings; |
||||
settings = createSettings("qBittorrent"); |
||||
} |
||||
#endif |
||||
|
||||
keys = settings->allKeys(); |
||||
|
||||
// Copy everything into memory. This means even keys inserted in the file manually
|
||||
// or that we don't touch directly in this code(eg disabled by ifdef). This ensures
|
||||
// that they will be copied over when save our settings to disk.
|
||||
foreach (const QString &key, keys) |
||||
m_data[key] = settings->value(key); |
||||
|
||||
//Ensures sync to disk before we attempt to manipulate the files from save().
|
||||
delete settings; |
||||
|
||||
#ifndef Q_OS_MAC |
||||
Utils::Fs::forceRemove(newPath); |
||||
|
||||
if (m_dirty) |
||||
save(); |
||||
#endif |
||||
|
||||
m_timer.setSingleShot(true); |
||||
m_timer.setInterval(5 * 1000); |
||||
connect(&m_timer, SIGNAL(timeout()), SLOT(save())); |
||||
} |
||||
|
||||
SettingsStorage::~SettingsStorage() |
||||
{ |
||||
save(); |
||||
} |
||||
|
||||
void SettingsStorage::initInstance() |
||||
{ |
||||
if (!m_instance) |
||||
m_instance = new SettingsStorage; |
||||
} |
||||
|
||||
void SettingsStorage::freeInstance() |
||||
{ |
||||
delete m_instance; |
||||
m_instance = nullptr; |
||||
} |
||||
|
||||
SettingsStorage *SettingsStorage::instance() |
||||
{ |
||||
return m_instance; |
||||
} |
||||
|
||||
bool SettingsStorage::save() |
||||
{ |
||||
QWriteLocker locker(&m_lock); |
||||
|
||||
if (!m_dirty) return false; |
||||
|
||||
#ifndef Q_OS_MAC |
||||
// QSettings delete the file before writing it out. This can result in problems
|
||||
// if the disk is full or a power outage occurs. Those events might occur
|
||||
// between deleting the file and recreating it. This is a safety measure.
|
||||
// Write everything to qBittorrent_new.ini/qBittorrent_new.conf and if it succeeds
|
||||
// replace qBittorrent_new.ini/qBittorrent.conf with it.
|
||||
QSettings *settings = createSettings("qBittorrent_new"); |
||||
#else |
||||
QSettings *settings = createSettings("qBittorrent"); |
||||
#endif |
||||
|
||||
foreach (const QString &key, m_data.keys()) |
||||
settings->setValue(key, m_data[key]); |
||||
|
||||
m_dirty = false; |
||||
locker.unlock(); |
||||
|
||||
#ifndef Q_OS_MAC |
||||
settings->sync(); // Important to get error status
|
||||
QString newPath = settings->fileName(); |
||||
QSettings::Status status = settings->status(); |
||||
|
||||
if (status != QSettings::NoError) { |
||||
if (status == QSettings::AccessError) |
||||
Logger::instance()->addMessage(tr("An access error occurred while trying to write the configuration file."), Log::CRITICAL); |
||||
else |
||||
Logger::instance()->addMessage(tr("A format error occurred while trying to write the configuration file."), Log::CRITICAL); |
||||
|
||||
delete settings; |
||||
Utils::Fs::forceRemove(newPath); |
||||
return false; |
||||
} |
||||
|
||||
delete settings; |
||||
QString finalPath = newPath; |
||||
int index = finalPath.lastIndexOf("_new", -1, Qt::CaseInsensitive); |
||||
finalPath.remove(index, 4); |
||||
Utils::Fs::forceRemove(finalPath); |
||||
QFile::rename(newPath, finalPath); |
||||
#else |
||||
delete settings; |
||||
#endif |
||||
|
||||
return true; |
||||
} |
||||
|
||||
QVariant SettingsStorage::loadValue(const QString &key, const QVariant &defaultValue) const |
||||
{ |
||||
QReadLocker locker(&m_lock); |
||||
return m_data.value(mapKey(key), defaultValue); |
||||
} |
||||
|
||||
void SettingsStorage::storeValue(const QString &key, const QVariant &value) |
||||
{ |
||||
QString realKey = mapKey(key); |
||||
QWriteLocker locker(&m_lock); |
||||
if (m_data.value(realKey) != value) { |
||||
m_dirty = true; |
||||
m_timer.start(); |
||||
m_data.insert(realKey, value); |
||||
} |
||||
} |
||||
|
||||
void SettingsStorage::removeValue(const QString &key) |
||||
{ |
||||
QString realKey = mapKey(key); |
||||
QWriteLocker locker(&m_lock); |
||||
if (m_data.contains(realKey)) { |
||||
m_dirty = true; |
||||
m_timer.start(); |
||||
m_data.remove(realKey); |
||||
} |
||||
} |
||||
@ -0,0 +1,65 @@
@@ -0,0 +1,65 @@
|
||||
/*
|
||||
* Bittorrent Client using Qt and libtorrent. |
||||
* Copyright (C) 2016 Vladimir Golovnev <glassez@yandex.ru> |
||||
* Copyright (C) 2014 sledgehammer999 <hammered999@gmail.com> |
||||
* |
||||
* 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. |
||||
*/ |
||||
|
||||
#ifndef SETTINGSSTORAGE_H |
||||
#define SETTINGSSTORAGE_H |
||||
|
||||
#include <QObject> |
||||
#include <QVariantHash> |
||||
#include <QTimer> |
||||
#include <QReadWriteLock> |
||||
|
||||
class SettingsStorage: public QObject |
||||
{ |
||||
Q_OBJECT |
||||
SettingsStorage(); |
||||
~SettingsStorage(); |
||||
|
||||
public: |
||||
static void initInstance(); |
||||
static void freeInstance(); |
||||
static SettingsStorage* instance(); |
||||
|
||||
QVariant loadValue(const QString &key, const QVariant &defaultValue = QVariant()) const; |
||||
void storeValue(const QString &key, const QVariant &value); |
||||
void removeValue(const QString &key); |
||||
|
||||
public slots: |
||||
bool save(); |
||||
|
||||
private: |
||||
static SettingsStorage *m_instance; |
||||
|
||||
QVariantHash m_data; |
||||
bool m_dirty; |
||||
QTimer m_timer; |
||||
mutable QReadWriteLock m_lock; |
||||
}; |
||||
|
||||
#endif // SETTINGSSTORAGE_H
|
||||
Loading…
Reference in new issue