Some work about adaptive color scheme for Web UI (PR #19901) http://[316:c51a:62a3:8b9::4]/d4708/qBittorrent/src/branch/adaptive-webui
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

1622 lines
43 KiB

/*
* Bittorrent Client using Qt and libtorrent.
* Copyright (C) 2014 sledgehammer999 <hammered999@gmail.com>
* Copyright (C) 2006 Christophe Dumez <chris@qbittorrent.org>
*
* 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 "preferences.h"
#include <QCryptographicHash>
#include <QDir>
#include <QLocale>
#include <QMutableListIterator>
#include <QSettings>
#ifndef DISABLE_GUI
#include <QApplication>
#else
#include <QCoreApplication>
#endif
#ifdef Q_OS_WIN
#include <shlobj.h>
#include <winreg.h>
#include <QRegularExpression>
#endif
#ifdef Q_OS_MAC
#include <CoreServices/CoreServices.h>
#endif
#include "logger.h"
#include "settingsstorage.h"
#include "utils/fs.h"
#include "utils/misc.h"
Preferences *Preferences::m_instance = nullptr;
Preferences::Preferences() = default;
Preferences *Preferences::instance()
{
return m_instance;
}
void Preferences::initInstance()
{
if (!m_instance)
m_instance = new Preferences;
}
void Preferences::freeInstance()
{
if (m_instance) {
delete m_instance;
m_instance = nullptr;
}
}
const QVariant Preferences::value(const QString &key, const QVariant &defaultValue) const
{
return SettingsStorage::instance()->loadValue(key, defaultValue);
}
void Preferences::setValue(const QString &key, const QVariant &value)
{
SettingsStorage::instance()->storeValue(key, value);
}
// General options
QString Preferences::getLocale() const
{
return value("Preferences/General/Locale", QLocale::system().name()).toString();
}
void Preferences::setLocale(const QString &locale)
{
setValue("Preferences/General/Locale", locale);
}
bool Preferences::deleteTorrentFilesAsDefault() const
{
return value("Preferences/General/DeleteTorrentsFilesAsDefault", false).toBool();
}
void Preferences::setDeleteTorrentFilesAsDefault(bool del)
{
setValue("Preferences/General/DeleteTorrentsFilesAsDefault", del);
}
bool Preferences::confirmOnExit() const
{
return value("Preferences/General/ExitConfirm", true).toBool();
}
void Preferences::setConfirmOnExit(bool confirm)
{
setValue("Preferences/General/ExitConfirm", confirm);
}
bool Preferences::speedInTitleBar() const
{
return value("Preferences/General/SpeedInTitleBar", false).toBool();
}
void Preferences::showSpeedInTitleBar(bool show)
{
setValue("Preferences/General/SpeedInTitleBar", show);
}
bool Preferences::useAlternatingRowColors() const
{
return value("Preferences/General/AlternatingRowColors", true).toBool();
}
void Preferences::setAlternatingRowColors(bool b)
{
setValue("Preferences/General/AlternatingRowColors", b);
}
bool Preferences::getHideZeroValues() const
{
return value("Preferences/General/HideZeroValues", false).toBool();
}
void Preferences::setHideZeroValues(bool b)
{
setValue("Preferences/General/HideZeroValues", b);
}
int Preferences::getHideZeroComboValues() const
{
return value("Preferences/General/HideZeroComboValues", 0).toInt();
}
void Preferences::setHideZeroComboValues(int n)
{
setValue("Preferences/General/HideZeroComboValues", n);
}
// In Mac OS X the dock is sufficient for our needs so we disable the sys tray functionality.
// See extensive discussion in https://github.com/qbittorrent/qBittorrent/pull/3018
#ifndef Q_OS_MAC
bool Preferences::systrayIntegration() const
{
return value("Preferences/General/SystrayEnabled", true).toBool();
}
void Preferences::setSystrayIntegration(bool enabled)
{
setValue("Preferences/General/SystrayEnabled", enabled);
}
bool Preferences::minimizeToTray() const
{
return value("Preferences/General/MinimizeToTray", false).toBool();
}
void Preferences::setMinimizeToTray(bool b)
{
setValue("Preferences/General/MinimizeToTray", b);
}
bool Preferences::minimizeToTrayNotified() const
{
return value("Preferences/General/MinimizeToTrayNotified", false).toBool();
}
void Preferences::setMinimizeToTrayNotified(bool b)
{
setValue("Preferences/General/MinimizeToTrayNotified", b);
}
bool Preferences::closeToTray() const
{
return value("Preferences/General/CloseToTray", true).toBool();
}
void Preferences::setCloseToTray(bool b)
{
setValue("Preferences/General/CloseToTray", b);
}
bool Preferences::closeToTrayNotified() const
{
return value("Preferences/General/CloseToTrayNotified", false).toBool();
}
void Preferences::setCloseToTrayNotified(bool b)
{
setValue("Preferences/General/CloseToTrayNotified", b);
}
#endif
bool Preferences::isToolbarDisplayed() const
{
return value("Preferences/General/ToolbarDisplayed", true).toBool();
}
void Preferences::setToolbarDisplayed(bool displayed)
{
setValue("Preferences/General/ToolbarDisplayed", displayed);
}
bool Preferences::isStatusbarDisplayed() const
{
return value("Preferences/General/StatusbarDisplayed", true).toBool();
}
void Preferences::setStatusbarDisplayed(bool displayed)
{
setValue("Preferences/General/StatusbarDisplayed", displayed);
}
bool Preferences::startMinimized() const
{
return value("Preferences/General/StartMinimized", false).toBool();
}
void Preferences::setStartMinimized(bool b)
{
setValue("Preferences/General/StartMinimized", b);
}
bool Preferences::isSplashScreenDisabled() const
{
return value("Preferences/General/NoSplashScreen", true).toBool();
}
void Preferences::setSplashScreenDisabled(bool b)
{
setValue("Preferences/General/NoSplashScreen", b);
}
// Preventing from system suspend while active torrents are presented.
bool Preferences::preventFromSuspendWhenDownloading() const
{
return value("Preferences/General/PreventFromSuspendWhenDownloading", false).toBool();
}
void Preferences::setPreventFromSuspendWhenDownloading(bool b)
{
setValue("Preferences/General/PreventFromSuspendWhenDownloading", b);
}
bool Preferences::preventFromSuspendWhenSeeding() const
{
return value("Preferences/General/PreventFromSuspendWhenSeeding", false).toBool();
}
void Preferences::setPreventFromSuspendWhenSeeding(bool b)
{
setValue("Preferences/General/PreventFromSuspendWhenSeeding", b);
}
#ifdef Q_OS_WIN
bool Preferences::WinStartup() const
{
QSettings settings("HKEY_CURRENT_USER\\Software\\Microsoft\\Windows\\CurrentVersion\\Run", QSettings::NativeFormat);
return settings.contains("qBittorrent");
}
void Preferences::setWinStartup(bool b)
{
QSettings settings("HKEY_CURRENT_USER\\Software\\Microsoft\\Windows\\CurrentVersion\\Run", QSettings::NativeFormat);
if (b) {
const QString binPath = '"' + Utils::Fs::toNativePath(qApp->applicationFilePath()) + '"';
settings.setValue("qBittorrent", binPath);
}
else {
settings.remove("qBittorrent");
}
}
#endif
// Downloads
QString Preferences::lastLocationPath() const
{
return Utils::Fs::fromNativePath(value("Preferences/Downloads/LastLocationPath").toString());
}
void Preferences::setLastLocationPath(const QString &path)
{
setValue("Preferences/Downloads/LastLocationPath", Utils::Fs::fromNativePath(path));
}
QVariantHash Preferences::getScanDirs() const
{
return value("Preferences/Downloads/ScanDirsV2").toHash();
}
// This must be called somewhere with data from the model
void Preferences::setScanDirs(const QVariantHash &dirs)
{
setValue("Preferences/Downloads/ScanDirsV2", dirs);
}
QString Preferences::getScanDirsLastPath() const
{
return Utils::Fs::fromNativePath(value("Preferences/Downloads/ScanDirsLastPath").toString());
}
void Preferences::setScanDirsLastPath(const QString &path)
{
setValue("Preferences/Downloads/ScanDirsLastPath", Utils::Fs::fromNativePath(path));
}
bool Preferences::isMailNotificationEnabled() const
{
return value("Preferences/MailNotification/enabled", false).toBool();
}
void Preferences::setMailNotificationEnabled(bool enabled)
{
setValue("Preferences/MailNotification/enabled", enabled);
}
QString Preferences::getMailNotificationSender() const
{
return value("Preferences/MailNotification/sender", "qBittorrent_notification@example.com").toString();
}
void Preferences::setMailNotificationSender(const QString &mail)
{
setValue("Preferences/MailNotification/sender", mail);
}
QString Preferences::getMailNotificationEmail() const
{
return value("Preferences/MailNotification/email").toString();
}
void Preferences::setMailNotificationEmail(const QString &mail)
{
setValue("Preferences/MailNotification/email", mail);
}
QString Preferences::getMailNotificationSMTP() const
{
return value("Preferences/MailNotification/smtp_server", "smtp.changeme.com").toString();
}
void Preferences::setMailNotificationSMTP(const QString &smtp_server)
{
setValue("Preferences/MailNotification/smtp_server", smtp_server);
}
bool Preferences::getMailNotificationSMTPSSL() const
{
return value("Preferences/MailNotification/req_ssl", false).toBool();
}
void Preferences::setMailNotificationSMTPSSL(bool use)
{
setValue("Preferences/MailNotification/req_ssl", use);
}
bool Preferences::getMailNotificationSMTPAuth() const
{
return value("Preferences/MailNotification/req_auth", false).toBool();
}
void Preferences::setMailNotificationSMTPAuth(bool use)
{
setValue("Preferences/MailNotification/req_auth", use);
}
QString Preferences::getMailNotificationSMTPUsername() const
{
return value("Preferences/MailNotification/username").toString();
}
void Preferences::setMailNotificationSMTPUsername(const QString &username)
{
setValue("Preferences/MailNotification/username", username);
}
QString Preferences::getMailNotificationSMTPPassword() const
{
return value("Preferences/MailNotification/password").toString();
}
void Preferences::setMailNotificationSMTPPassword(const QString &password)
{
setValue("Preferences/MailNotification/password", password);
}
int Preferences::getActionOnDblClOnTorrentDl() const
{
return value("Preferences/Downloads/DblClOnTorDl", 0).toInt();
}
void Preferences::setActionOnDblClOnTorrentDl(int act)
{
setValue("Preferences/Downloads/DblClOnTorDl", act);
}
int Preferences::getActionOnDblClOnTorrentFn() const
{
return value("Preferences/Downloads/DblClOnTorFn", 1).toInt();
}
void Preferences::setActionOnDblClOnTorrentFn(int act)
{
setValue("Preferences/Downloads/DblClOnTorFn", act);
}
QTime Preferences::getSchedulerStartTime() const
{
return value("Preferences/Scheduler/start_time", QTime(8,0)).toTime();
}
void Preferences::setSchedulerStartTime(const QTime &time)
{
setValue("Preferences/Scheduler/start_time", time);
}
QTime Preferences::getSchedulerEndTime() const
{
return value("Preferences/Scheduler/end_time", QTime(20,0)).toTime();
}
void Preferences::setSchedulerEndTime(const QTime &time)
{
setValue("Preferences/Scheduler/end_time", time);
}
SchedulerDays Preferences::getSchedulerDays() const
{
return static_cast<SchedulerDays>(value("Preferences/Scheduler/days", EVERY_DAY).toInt());
}
void Preferences::setSchedulerDays(SchedulerDays days)
{
setValue("Preferences/Scheduler/days", static_cast<int>(days));
}
// Search
bool Preferences::isSearchEnabled() const
{
return value("Preferences/Search/SearchEnabled", false).toBool();
}
void Preferences::setSearchEnabled(bool enabled)
{
setValue("Preferences/Search/SearchEnabled", enabled);
}
bool Preferences::isWebUiEnabled() const
{
#ifdef DISABLE_GUI
return true;
#else
return value("Preferences/WebUI/Enabled", false).toBool();
#endif
}
void Preferences::setWebUiEnabled(bool enabled)
{
setValue("Preferences/WebUI/Enabled", enabled);
}
bool Preferences::isWebUiLocalAuthEnabled() const
{
return value("Preferences/WebUI/LocalHostAuth", true).toBool();
}
void Preferences::setWebUiLocalAuthEnabled(bool enabled)
{
setValue("Preferences/WebUI/LocalHostAuth", enabled);
}
bool Preferences::isWebUiAuthSubnetWhitelistEnabled() const
{
return value("Preferences/WebUI/AuthSubnetWhitelistEnabled", false).toBool();
}
void Preferences::setWebUiAuthSubnetWhitelistEnabled(bool enabled)
{
setValue("Preferences/WebUI/AuthSubnetWhitelistEnabled", enabled);
}
QList<Utils::Net::Subnet> Preferences::getWebUiAuthSubnetWhitelist() const
{
QList<Utils::Net::Subnet> subnets;
foreach (const QString &rawSubnet, value("Preferences/WebUI/AuthSubnetWhitelist").toStringList()) {
bool ok = false;
const Utils::Net::Subnet subnet = Utils::Net::parseSubnet(rawSubnet.trimmed(), &ok);
if (ok)
subnets.append(subnet);
}
return subnets;
}
void Preferences::setWebUiAuthSubnetWhitelist(QStringList subnets)
{
QMutableListIterator<QString> i(subnets);
while (i.hasNext()) {
bool ok = false;
const Utils::Net::Subnet subnet = Utils::Net::parseSubnet(i.next().trimmed(), &ok);
if (!ok)
i.remove();
}
setValue("Preferences/WebUI/AuthSubnetWhitelist", subnets);
}
QString Preferences::getServerDomains() const
{
return value("Preferences/WebUI/ServerDomains", '*').toString();
}
void Preferences::setServerDomains(const QString &str)
{
setValue("Preferences/WebUI/ServerDomains", str);
}
QString Preferences::getWebUiAddress() const
{
return value("Preferences/WebUI/Address", '*').toString().trimmed();
}
void Preferences::setWebUiAddress(const QString &addr)
{
setValue("Preferences/WebUI/Address", addr.trimmed());
}
quint16 Preferences::getWebUiPort() const
{
return value("Preferences/WebUI/Port", 8080).toInt();
}
void Preferences::setWebUiPort(quint16 port)
{
setValue("Preferences/WebUI/Port", port);
}
bool Preferences::useUPnPForWebUIPort() const
{
#ifdef DISABLE_GUI
return value("Preferences/WebUI/UseUPnP", true).toBool();
#else
return value("Preferences/WebUI/UseUPnP", false).toBool();
#endif
}
void Preferences::setUPnPForWebUIPort(bool enabled)
{
setValue("Preferences/WebUI/UseUPnP", enabled);
}
QString Preferences::getWebUiUsername() const
{
return value("Preferences/WebUI/Username", "admin").toString();
}
void Preferences::setWebUiUsername(const QString &username)
{
setValue("Preferences/WebUI/Username", username);
}
QString Preferences::getWebUiPassword() const
{
QString passHa1 = value("Preferences/WebUI/Password_ha1").toString();
if (passHa1.isEmpty()) {
QCryptographicHash md5(QCryptographicHash::Md5);
md5.addData("adminadmin");
passHa1 = md5.result().toHex();
}
return passHa1;
}
void Preferences::setWebUiPassword(const QString &newPassword)
{
// Do not overwrite current password with its hash
if (newPassword == getWebUiPassword())
return;
// Encode to md5 and save
QCryptographicHash md5(QCryptographicHash::Md5);
md5.addData(newPassword.toLocal8Bit());
setValue("Preferences/WebUI/Password_ha1", md5.result().toHex());
}
bool Preferences::isWebUiClickjackingProtectionEnabled() const
{
return value("Preferences/WebUI/ClickjackingProtection", true).toBool();
}
void Preferences::setWebUiClickjackingProtectionEnabled(bool enabled)
{
setValue("Preferences/WebUI/ClickjackingProtection", enabled);
}
bool Preferences::isWebUiCSRFProtectionEnabled() const
{
return value("Preferences/WebUI/CSRFProtection", true).toBool();
}
void Preferences::setWebUiCSRFProtectionEnabled(bool enabled)
{
setValue("Preferences/WebUI/CSRFProtection", enabled);
}
bool Preferences::isWebUiHttpsEnabled() const
{
return value("Preferences/WebUI/HTTPS/Enabled", false).toBool();
}
void Preferences::setWebUiHttpsEnabled(bool enabled)
{
setValue("Preferences/WebUI/HTTPS/Enabled", enabled);
}
QByteArray Preferences::getWebUiHttpsCertificate() const
{
return value("Preferences/WebUI/HTTPS/Certificate").toByteArray();
}
void Preferences::setWebUiHttpsCertificate(const QByteArray &data)
{
setValue("Preferences/WebUI/HTTPS/Certificate", data);
}
QByteArray Preferences::getWebUiHttpsKey() const
{
return value("Preferences/WebUI/HTTPS/Key").toByteArray();
}
void Preferences::setWebUiHttpsKey(const QByteArray &data)
{
setValue("Preferences/WebUI/HTTPS/Key", data);
}
bool Preferences::isAltWebUiEnabled() const
{
return value("Preferences/WebUI/AlternativeUIEnabled", false).toBool();
}
void Preferences::setAltWebUiEnabled(bool enabled)
{
setValue("Preferences/WebUI/AlternativeUIEnabled", enabled);
}
QString Preferences::getWebUiRootFolder() const
{
return value("Preferences/WebUI/RootFolder").toString();
}
void Preferences::setWebUiRootFolder(const QString &path)
{
setValue("Preferences/WebUI/RootFolder", path);
}
bool Preferences::isDynDNSEnabled() const
{
return value("Preferences/DynDNS/Enabled", false).toBool();
}
void Preferences::setDynDNSEnabled(bool enabled)
{
setValue("Preferences/DynDNS/Enabled", enabled);
}
DNS::Service Preferences::getDynDNSService() const
{
return DNS::Service(value("Preferences/DynDNS/Service", DNS::DYNDNS).toInt());
}
void Preferences::setDynDNSService(int service)
{
setValue("Preferences/DynDNS/Service", service);
}
QString Preferences::getDynDomainName() const
{
return value("Preferences/DynDNS/DomainName", "changeme.dyndns.org").toString();
}
void Preferences::setDynDomainName(const QString &name)
{
setValue("Preferences/DynDNS/DomainName", name);
}
QString Preferences::getDynDNSUsername() const
{
return value("Preferences/DynDNS/Username").toString();
}
void Preferences::setDynDNSUsername(const QString &username)
{
setValue("Preferences/DynDNS/Username", username);
}
QString Preferences::getDynDNSPassword() const
{
return value("Preferences/DynDNS/Password").toString();
}
void Preferences::setDynDNSPassword(const QString &password)
{
setValue("Preferences/DynDNS/Password", password);
}
// Advanced settings
void Preferences::clearUILockPassword()
{
setValue("Locking/password", QString());
}
QString Preferences::getUILockPasswordMD5() const
{
return value("Locking/password").toString();
}
void Preferences::setUILockPassword(const QString &clearPassword)
{
QCryptographicHash md5(QCryptographicHash::Md5);
md5.addData(clearPassword.toLocal8Bit());
QString md5Password = md5.result().toHex();
setValue("Locking/password", md5Password);
}
bool Preferences::isUILocked() const
{
return value("Locking/locked", false).toBool();
}
void Preferences::setUILocked(bool locked)
{
return setValue("Locking/locked", locked);
}
bool Preferences::isAutoRunEnabled() const
{
return value("AutoRun/enabled", false).toBool();
}
void Preferences::setAutoRunEnabled(bool enabled)
{
return setValue("AutoRun/enabled", enabled);
}
QString Preferences::getAutoRunProgram() const
{
return value("AutoRun/program").toString();
}
void Preferences::setAutoRunProgram(const QString &program)
{
setValue("AutoRun/program", program);
}
bool Preferences::shutdownWhenDownloadsComplete() const
{
return value("Preferences/Downloads/AutoShutDownOnCompletion", false).toBool();
}
void Preferences::setShutdownWhenDownloadsComplete(bool shutdown)
{
setValue("Preferences/Downloads/AutoShutDownOnCompletion", shutdown);
}
bool Preferences::suspendWhenDownloadsComplete() const
{
return value("Preferences/Downloads/AutoSuspendOnCompletion", false).toBool();
}
void Preferences::setSuspendWhenDownloadsComplete(bool suspend)
{
setValue("Preferences/Downloads/AutoSuspendOnCompletion", suspend);
}
bool Preferences::hibernateWhenDownloadsComplete() const
{
return value("Preferences/Downloads/AutoHibernateOnCompletion", false).toBool();
}
void Preferences::setHibernateWhenDownloadsComplete(bool hibernate)
{
setValue("Preferences/Downloads/AutoHibernateOnCompletion", hibernate);
}
bool Preferences::shutdownqBTWhenDownloadsComplete() const
{
return value("Preferences/Downloads/AutoShutDownqBTOnCompletion", false).toBool();
}
void Preferences::setShutdownqBTWhenDownloadsComplete(bool shutdown)
{
setValue("Preferences/Downloads/AutoShutDownqBTOnCompletion", shutdown);
}
bool Preferences::dontConfirmAutoExit() const
{
return value("ShutdownConfirmDlg/DontConfirmAutoExit", false).toBool();
}
void Preferences::setDontConfirmAutoExit(bool dontConfirmAutoExit)
{
setValue("ShutdownConfirmDlg/DontConfirmAutoExit", dontConfirmAutoExit);
}
bool Preferences::recheckTorrentsOnCompletion() const
{
return value("Preferences/Advanced/RecheckOnCompletion", false).toBool();
}
void Preferences::recheckTorrentsOnCompletion(bool recheck)
{
setValue("Preferences/Advanced/RecheckOnCompletion", recheck);
}
bool Preferences::resolvePeerCountries() const
{
return value("Preferences/Connection/ResolvePeerCountries", true).toBool();
}
void Preferences::resolvePeerCountries(bool resolve)
{
setValue("Preferences/Connection/ResolvePeerCountries", resolve);
}
bool Preferences::resolvePeerHostNames() const
{
return value("Preferences/Connection/ResolvePeerHostNames", false).toBool();
}
void Preferences::resolvePeerHostNames(bool resolve)
{
setValue("Preferences/Connection/ResolvePeerHostNames", resolve);
}
#if (defined(Q_OS_UNIX) && !defined(Q_OS_MAC))
bool Preferences::useSystemIconTheme() const
{
return value("Preferences/Advanced/useSystemIconTheme", true).toBool();
}
void Preferences::useSystemIconTheme(bool enabled)
{
setValue("Preferences/Advanced/useSystemIconTheme", enabled);
}
#endif
bool Preferences::recursiveDownloadDisabled() const
{
return value("Preferences/Advanced/DisableRecursiveDownload", false).toBool();
}
void Preferences::disableRecursiveDownload(bool disable)
{
setValue("Preferences/Advanced/DisableRecursiveDownload", disable);
}
#ifdef Q_OS_WIN
namespace
{
enum REG_SEARCH_TYPE
{
USER,
SYSTEM_32BIT,
SYSTEM_64BIT
};
QStringList getRegSubkeys(HKEY handle)
{
QStringList keys;
DWORD cSubKeys = 0;
DWORD cMaxSubKeyLen = 0;
LONG res = ::RegQueryInfoKeyW(handle, NULL, NULL, NULL, &cSubKeys, &cMaxSubKeyLen, NULL, NULL, NULL, NULL, NULL, NULL);
if (res == ERROR_SUCCESS) {
++cMaxSubKeyLen; // For null character
LPWSTR lpName = new WCHAR[cMaxSubKeyLen];
DWORD cName;
for (DWORD i = 0; i < cSubKeys; ++i) {
cName = cMaxSubKeyLen;
res = ::RegEnumKeyExW(handle, i, lpName, &cName, NULL, NULL, NULL, NULL);
if (res == ERROR_SUCCESS)
keys.push_back(QString::fromWCharArray(lpName));
}
delete[] lpName;
}
return keys;
}
QString getRegValue(HKEY handle, const QString &name = QString())
{
QString result;
DWORD type = 0;
DWORD cbData = 0;
LPWSTR lpValueName = NULL;
if (!name.isEmpty()) {
lpValueName = new WCHAR[name.size() + 1];
name.toWCharArray(lpValueName);
lpValueName[name.size()] = 0;
}
// Discover the size of the value
::RegQueryValueExW(handle, lpValueName, NULL, &type, NULL, &cbData);
DWORD cBuffer = (cbData / sizeof(WCHAR)) + 1;
LPWSTR lpData = new WCHAR[cBuffer];
LONG res = ::RegQueryValueExW(handle, lpValueName, NULL, &type, (LPBYTE)lpData, &cbData);
if (lpValueName)
delete[] lpValueName;
if (res == ERROR_SUCCESS) {
lpData[cBuffer - 1] = 0;
result = QString::fromWCharArray(lpData);
}
delete[] lpData;
return result;
}
QString pythonSearchReg(const REG_SEARCH_TYPE type)
{
HKEY hkRoot;
if (type == USER)
hkRoot = HKEY_CURRENT_USER;
else
hkRoot = HKEY_LOCAL_MACHINE;
REGSAM samDesired = KEY_READ;
if (type == SYSTEM_32BIT)
samDesired |= KEY_WOW64_32KEY;
else if (type == SYSTEM_64BIT)
samDesired |= KEY_WOW64_64KEY;
QString path;
LONG res = 0;
HKEY hkPythonCore;
res = ::RegOpenKeyExW(hkRoot, L"SOFTWARE\\Python\\PythonCore", 0, samDesired, &hkPythonCore);
if (res == ERROR_SUCCESS) {
QStringList versions = getRegSubkeys(hkPythonCore);
qDebug("Python versions nb: %d", versions.size());
versions.sort();
bool found = false;
while (!found && !versions.empty()) {
const QString version = versions.takeLast() + "\\InstallPath";
LPWSTR lpSubkey = new WCHAR[version.size() + 1];
version.toWCharArray(lpSubkey);
lpSubkey[version.size()] = 0;
HKEY hkInstallPath;
res = ::RegOpenKeyExW(hkPythonCore, lpSubkey, 0, samDesired, &hkInstallPath);
delete[] lpSubkey;
if (res == ERROR_SUCCESS) {
qDebug("Detected possible Python v%s location", qUtf8Printable(version));
path = getRegValue(hkInstallPath);
::RegCloseKey(hkInstallPath);
if (!path.isEmpty() && QDir(path).exists("python.exe")) {
qDebug("Found python.exe at %s", qUtf8Printable(path));
found = true;
}
}
}
if (!found)
path = QString();
::RegCloseKey(hkPythonCore);
}
return path;
}
}
QString Preferences::getPythonPath()
{
QString path = pythonSearchReg(USER);
if (!path.isEmpty())
return path;
path = pythonSearchReg(SYSTEM_32BIT);
if (!path.isEmpty())
return path;
path = pythonSearchReg(SYSTEM_64BIT);
if (!path.isEmpty())
return path;
// Fallback: Detect python from default locations
const QStringList dirs = QDir("C:/").entryList(QStringList("Python*"), QDir::Dirs, QDir::Name | QDir::Reversed);
foreach (const QString &dir, dirs) {
const QString path("C:/" + dir + '/');
if (QFile::exists(path + "python.exe"))
return path;
}
return QString();
}
bool Preferences::neverCheckFileAssoc() const
{
return value("Preferences/Win32/NeverCheckFileAssocation", false).toBool();
}
void Preferences::setNeverCheckFileAssoc(bool check)
{
setValue("Preferences/Win32/NeverCheckFileAssocation", check);
}
bool Preferences::isTorrentFileAssocSet()
{
QSettings settings("HKEY_CURRENT_USER\\Software\\Classes", QSettings::NativeFormat);
if (settings.value(".torrent/Default").toString() != "qBittorrent") {
qDebug(".torrent != qBittorrent");
return false;
}
return true;
}
bool Preferences::isMagnetLinkAssocSet()
{
QSettings settings("HKEY_CURRENT_USER\\Software\\Classes", QSettings::NativeFormat);
// Check magnet link assoc
const QString shellCommand = Utils::Fs::toNativePath(settings.value("magnet/shell/open/command/Default", "").toString());
const QRegularExpressionMatch exeRegMatch = QRegularExpression("\"([^\"]+)\".*").match(shellCommand);
if (!exeRegMatch.hasMatch())
return false;
const QString assocExe = exeRegMatch.captured(1);
if (assocExe.compare(Utils::Fs::toNativePath(qApp->applicationFilePath()), Qt::CaseInsensitive) != 0)
return false;
return true;
}
void Preferences::setTorrentFileAssoc(bool set)
{
QSettings settings("HKEY_CURRENT_USER\\Software\\Classes", QSettings::NativeFormat);
// .Torrent association
if (set) {
QString oldProgId = settings.value(".torrent/Default").toString();
if (!oldProgId.isEmpty() && (oldProgId != "qBittorrent"))
settings.setValue(".torrent/OpenWithProgids/" + oldProgId, "");
settings.setValue(".torrent/Default", "qBittorrent");
}
else if (isTorrentFileAssocSet()) {
settings.setValue(".torrent/Default", "");
}
SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, 0, 0);
}
void Preferences::setMagnetLinkAssoc(bool set)
{
QSettings settings("HKEY_CURRENT_USER\\Software\\Classes", QSettings::NativeFormat);
// Magnet association
if (set) {
const QString commandStr = '"' + qApp->applicationFilePath() + "\" \"%1\"";
const QString iconStr = '"' + qApp->applicationFilePath() + "\",1";
settings.setValue("magnet/Default", "URL:Magnet link");
settings.setValue("magnet/Content Type", "application/x-magnet");
settings.setValue("magnet/URL Protocol", "");
settings.setValue("magnet/DefaultIcon/Default", Utils::Fs::toNativePath(iconStr));
settings.setValue("magnet/shell/Default", "open");
settings.setValue("magnet/shell/open/command/Default", Utils::Fs::toNativePath(commandStr));
}
else if (isMagnetLinkAssocSet()) {
settings.remove("magnet");
}
SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, 0, 0);
}
#endif
#ifdef Q_OS_MAC
namespace
{
CFStringRef torrentExtension = CFSTR("torrent");
CFStringRef magnetUrlScheme = CFSTR("magnet");
}
bool Preferences::isTorrentFileAssocSet()
{
bool isSet = false;
CFStringRef torrentId = UTTypeCreatePreferredIdentifierForTag(kUTTagClassFilenameExtension, torrentExtension, NULL);
if (torrentId != NULL) {
CFStringRef defaultHandlerId = LSCopyDefaultRoleHandlerForContentType(torrentId, kLSRolesViewer);
if (defaultHandlerId != NULL) {
CFStringRef myBundleId = CFBundleGetIdentifier(CFBundleGetMainBundle());
isSet = CFStringCompare(myBundleId, defaultHandlerId, 0) == kCFCompareEqualTo;
CFRelease(defaultHandlerId);
}
CFRelease(torrentId);
}
return isSet;
}
bool Preferences::isMagnetLinkAssocSet()
{
bool isSet = false;
CFStringRef defaultHandlerId = LSCopyDefaultHandlerForURLScheme(magnetUrlScheme);
if (defaultHandlerId != NULL) {
CFStringRef myBundleId = CFBundleGetIdentifier(CFBundleGetMainBundle());
isSet = CFStringCompare(myBundleId, defaultHandlerId, 0) == kCFCompareEqualTo;
CFRelease(defaultHandlerId);
}
return isSet;
}
void Preferences::setTorrentFileAssoc()
{
if (isTorrentFileAssocSet())
return;
CFStringRef torrentId = UTTypeCreatePreferredIdentifierForTag(kUTTagClassFilenameExtension, torrentExtension, NULL);
if (torrentId != NULL) {
CFStringRef myBundleId = CFBundleGetIdentifier(CFBundleGetMainBundle());
LSSetDefaultRoleHandlerForContentType(torrentId, kLSRolesViewer, myBundleId);
CFRelease(torrentId);
}
}
void Preferences::setMagnetLinkAssoc()
{
if (isMagnetLinkAssocSet())
return;
CFStringRef myBundleId = CFBundleGetIdentifier(CFBundleGetMainBundle());
LSSetDefaultHandlerForURLScheme(magnetUrlScheme, myBundleId);
}
#endif
int Preferences::getTrackerPort() const
{
return value("Preferences/Advanced/trackerPort", 9000).toInt();
}
void Preferences::setTrackerPort(int port)
{
setValue("Preferences/Advanced/trackerPort", port);
}
#if defined(Q_OS_WIN) || defined(Q_OS_MAC)
bool Preferences::isUpdateCheckEnabled() const
{
return value("Preferences/Advanced/updateCheck", true).toBool();
}
void Preferences::setUpdateCheckEnabled(bool enabled)
{
setValue("Preferences/Advanced/updateCheck", enabled);
}
#endif
bool Preferences::confirmTorrentDeletion() const
{
return value("Preferences/Advanced/confirmTorrentDeletion", true).toBool();
}
void Preferences::setConfirmTorrentDeletion(bool enabled)
{
setValue("Preferences/Advanced/confirmTorrentDeletion", enabled);
}
bool Preferences::confirmTorrentRecheck() const
{
return value("Preferences/Advanced/confirmTorrentRecheck", true).toBool();
}
void Preferences::setConfirmTorrentRecheck(bool enabled)
{
setValue("Preferences/Advanced/confirmTorrentRecheck", enabled);
}
bool Preferences::confirmRemoveAllTags() const
{
return value("Preferences/Advanced/confirmRemoveAllTags", true).toBool();
}
void Preferences::setConfirmRemoveAllTags(bool enabled)
{
setValue("Preferences/Advanced/confirmRemoveAllTags", enabled);
}
#ifndef Q_OS_MAC
TrayIcon::Style Preferences::trayIconStyle() const
{
return TrayIcon::Style(value("Preferences/Advanced/TrayIconStyle", TrayIcon::NORMAL).toInt());
}
void Preferences::setTrayIconStyle(TrayIcon::Style style)
{
setValue("Preferences/Advanced/TrayIconStyle", style);
}
#endif
// Stuff that don't appear in the Options GUI but are saved
// in the same file.
QDateTime Preferences::getDNSLastUpd() const
{
return value("DNSUpdater/lastUpdateTime").toDateTime();
}
void Preferences::setDNSLastUpd(const QDateTime &date)
{
setValue("DNSUpdater/lastUpdateTime", date);
}
QString Preferences::getDNSLastIP() const
{
return value("DNSUpdater/lastIP").toString();
}
void Preferences::setDNSLastIP(const QString &ip)
{
setValue("DNSUpdater/lastIP", ip);
}
bool Preferences::getAcceptedLegal() const
{
return value("LegalNotice/Accepted", false).toBool();
}
void Preferences::setAcceptedLegal(const bool accepted)
{
setValue("LegalNotice/Accepted", accepted);
}
QByteArray Preferences::getMainGeometry() const
{
return value("MainWindow/geometry").toByteArray();
}
void Preferences::setMainGeometry(const QByteArray &geometry)
{
setValue("MainWindow/geometry", geometry);
}
QByteArray Preferences::getMainVSplitterState() const
{
return value("MainWindow/qt5/vsplitterState").toByteArray();
}
void Preferences::setMainVSplitterState(const QByteArray &state)
{
setValue("MainWindow/qt5/vsplitterState", state);
}
QString Preferences::getMainLastDir() const
{
return value("MainWindowLastDir", QDir::homePath()).toString();
}
void Preferences::setMainLastDir(const QString &path)
{
setValue("MainWindowLastDir", path);
}
QSize Preferences::getPrefSize() const
{
return value("Preferences/State/size").toSize();
}
void Preferences::setPrefSize(const QSize &size)
{
setValue("Preferences/State/size", size);
}
QStringList Preferences::getPrefHSplitterSizes() const
{
return value("Preferences/State/hSplitterSizes").toStringList();
}
void Preferences::setPrefHSplitterSizes(const QStringList &sizes)
{
setValue("Preferences/State/hSplitterSizes", sizes);
}
QByteArray Preferences::getPeerListState() const
{
return value("TorrentProperties/Peers/qt5/PeerListState").toByteArray();
}
void Preferences::setPeerListState(const QByteArray &state)
{
setValue("TorrentProperties/Peers/qt5/PeerListState", state);
}
QString Preferences::getPropSplitterSizes() const
{
return value("TorrentProperties/SplitterSizes").toString();
}
void Preferences::setPropSplitterSizes(const QString &sizes)
{
setValue("TorrentProperties/SplitterSizes", sizes);
}
QByteArray Preferences::getPropFileListState() const
{
return value("TorrentProperties/qt5/FilesListState").toByteArray();
}
void Preferences::setPropFileListState(const QByteArray &state)
{
setValue("TorrentProperties/qt5/FilesListState", state);
}
int Preferences::getPropCurTab() const
{
return value("TorrentProperties/CurrentTab", -1).toInt();
}
void Preferences::setPropCurTab(const int &tab)
{
setValue("TorrentProperties/CurrentTab", tab);
}
bool Preferences::getPropVisible() const
{
return value("TorrentProperties/Visible", false).toBool();
}
void Preferences::setPropVisible(const bool visible)
{
setValue("TorrentProperties/Visible", visible);
}
QByteArray Preferences::getPropTrackerListState() const
{
return value("TorrentProperties/Trackers/qt5/TrackerListState").toByteArray();
}
void Preferences::setPropTrackerListState(const QByteArray &state)
{
setValue("TorrentProperties/Trackers/qt5/TrackerListState", state);
}
QSize Preferences::getRssGeometrySize() const
{
return value("RssFeedDownloader/geometrySize").toSize();
}
void Preferences::setRssGeometrySize(const QSize &geometry)
{
setValue("RssFeedDownloader/geometrySize", geometry);
}
QByteArray Preferences::getRssHSplitterSizes() const
{
return value("RssFeedDownloader/qt5/hsplitterSizes").toByteArray();
}
void Preferences::setRssHSplitterSizes(const QByteArray &sizes)
{
setValue("RssFeedDownloader/qt5/hsplitterSizes", sizes);
}
QStringList Preferences::getRssOpenFolders() const
{
return value("GUI/RSSWidget/OpenedFolders").toStringList();
}
void Preferences::setRssOpenFolders(const QStringList &folders)
{
setValue("GUI/RSSWidget/OpenedFolders", folders);
}
QByteArray Preferences::getRssSideSplitterState() const
{
return value("GUI/RSSWidget/qt5/splitter_h").toByteArray();
}
void Preferences::setRssSideSplitterState(const QByteArray &state)
{
setValue("GUI/RSSWidget/qt5/splitter_h", state);
}
QByteArray Preferences::getRssMainSplitterState() const
{
return value("GUI/RSSWidget/qt5/splitterMain").toByteArray();
}
void Preferences::setRssMainSplitterState(const QByteArray &state)
{
setValue("GUI/RSSWidget/qt5/splitterMain", state);
}
QByteArray Preferences::getSearchTabHeaderState() const
{
return value("SearchTab/qt5/HeaderState").toByteArray();
}
void Preferences::setSearchTabHeaderState(const QByteArray &state)
{
setValue("SearchTab/qt5/HeaderState", state);
}
QStringList Preferences::getSearchEngDisabled() const
{
return value("SearchEngines/disabledEngines").toStringList();
}
void Preferences::setSearchEngDisabled(const QStringList &engines)
{
setValue("SearchEngines/disabledEngines", engines);
}
QString Preferences::getTorImportLastContentDir() const
{
return value("TorrentImport/LastContentDir", QDir::homePath()).toString();
}
void Preferences::setTorImportLastContentDir(const QString &path)
{
setValue("TorrentImport/LastContentDir", path);
}
QByteArray Preferences::getTorImportGeometry() const
{
return value("TorrentImportDlg/dimensions").toByteArray();
}
void Preferences::setTorImportGeometry(const QByteArray &geometry)
{
setValue("TorrentImportDlg/dimensions", geometry);
}
bool Preferences::getStatusFilterState() const
{
return value("TransferListFilters/statusFilterState", true).toBool();
}
void Preferences::setStatusFilterState(const bool checked)
{
setValue("TransferListFilters/statusFilterState", checked);
}
bool Preferences::getCategoryFilterState() const
{
return value("TransferListFilters/CategoryFilterState", true).toBool();
}
void Preferences::setCategoryFilterState(const bool checked)
{
setValue("TransferListFilters/CategoryFilterState", checked);
}
bool Preferences::getTagFilterState() const
{
return value("TransferListFilters/TagFilterState", true).toBool();
}
void Preferences::setTagFilterState(const bool checked)
{
setValue("TransferListFilters/TagFilterState", checked);
}
bool Preferences::getTrackerFilterState() const
{
return value("TransferListFilters/trackerFilterState", true).toBool();
}
void Preferences::setTrackerFilterState(const bool checked)
{
setValue("TransferListFilters/trackerFilterState", checked);
}
int Preferences::getTransSelFilter() const
{
return value("TransferListFilters/selectedFilterIndex", 0).toInt();
}
void Preferences::setTransSelFilter(const int &index)
{
setValue("TransferListFilters/selectedFilterIndex", index);
}
QByteArray Preferences::getTransHeaderState() const
{
return value("TransferList/qt5/HeaderState").toByteArray();
}
void Preferences::setTransHeaderState(const QByteArray &state)
{
setValue("TransferList/qt5/HeaderState", state);
}
bool Preferences::getRegexAsFilteringPattern() const
{
return value("TransferList/UseRegexAsFilteringPattern", false).toBool();
}
void Preferences::setRegexAsFilteringPattern(const bool checked)
{
setValue("TransferList/UseRegexAsFilteringPattern", checked);
}
// From old RssSettings class
bool Preferences::isRSSWidgetEnabled() const
{
return value("GUI/RSSWidget/Enabled", false).toBool();
}
void Preferences::setRSSWidgetVisible(const bool enabled)
{
setValue("GUI/RSSWidget/Enabled", enabled);
}
int Preferences::getToolbarTextPosition() const
{
return value("Toolbar/textPosition", -1).toInt();
}
void Preferences::setToolbarTextPosition(const int position)
{
setValue("Toolbar/textPosition", position);
}
QList<QNetworkCookie> Preferences::getNetworkCookies() const
{
QList<QNetworkCookie> cookies;
QStringList rawCookies = value("Network/Cookies").toStringList();
foreach (const QString &rawCookie, rawCookies)
cookies << QNetworkCookie::parseCookies(rawCookie.toUtf8());
return cookies;
}
void Preferences::setNetworkCookies(const QList<QNetworkCookie> &cookies)
{
QStringList rawCookies;
foreach (const QNetworkCookie &cookie, cookies)
rawCookies << cookie.toRawForm();
setValue("Network/Cookies", rawCookies);
}
int Preferences::getSpeedWidgetPeriod() const
{
return value("SpeedWidget/period", 1).toInt();
}
void Preferences::setSpeedWidgetPeriod(const int period)
{
setValue("SpeedWidget/period", period);
}
bool Preferences::getSpeedWidgetGraphEnable(int id) const
{
// UP and DOWN graphs enabled by default
return value("SpeedWidget/graph_enable_" + QString::number(id), (id == 0 || id == 1)).toBool();
}
void Preferences::setSpeedWidgetGraphEnable(int id, const bool enable)
{
setValue("SpeedWidget/graph_enable_" + QString::number(id), enable);
}
void Preferences::upgrade()
{
SettingsStorage *settingsStorage = SettingsStorage::instance();
QStringList labels = value("TransferListFilters/customLabels").toStringList();
if (!labels.isEmpty()) {
QVariantMap categories = value("BitTorrent/Session/Categories").toMap();
foreach (const QString &label, labels) {
if (!categories.contains(label))
categories[label] = "";
}
setValue("BitTorrent/Session/Categories", categories);
settingsStorage->removeValue("TransferListFilters/customLabels");
}
settingsStorage->removeValue("Preferences/Downloads/AppendLabel");
// Inhibit sleep based on running downloads/available seeds rather than network activity.
if (value("Preferences/General/PreventFromSuspend", false).toBool()) {
setPreventFromSuspendWhenDownloading(true);
setPreventFromSuspendWhenSeeding(true);
}
settingsStorage->removeValue("Preferences/General/PreventFromSuspend");
}
void Preferences::apply()
{
if (SettingsStorage::instance()->save())
emit changed();
}