Browse Source
* Use correct pointer type in NSIS scripts * Only remove qbt specific registry keys when uninstalling or disassociating * Set .torrent Content Type when associating .torrent format * Move ".torrent association" functions to Utils::OS class PR #19709.adaptive-webui-19844
Chocobo1
1 year ago
committed by
GitHub
10 changed files with 276 additions and 185 deletions
@ -0,0 +1,194 @@ |
|||||||
|
/*
|
||||||
|
* Bittorrent Client using Qt and libtorrent. |
||||||
|
* Copyright (C) 2023 Mike Tzou (Chocobo1) |
||||||
|
* 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 "os.h" |
||||||
|
|
||||||
|
#ifdef Q_OS_MACOS |
||||||
|
#include <CoreServices/CoreServices.h> |
||||||
|
#endif // Q_OS_MACOS
|
||||||
|
|
||||||
|
#ifdef Q_OS_WIN |
||||||
|
#include <shlobj.h> |
||||||
|
#endif // Q_OS_WIN
|
||||||
|
|
||||||
|
#include <QString> |
||||||
|
|
||||||
|
#ifdef Q_OS_WIN |
||||||
|
#include <QCoreApplication> |
||||||
|
#include <QRegularExpression> |
||||||
|
#include <QSettings> |
||||||
|
#endif // Q_OS_WIN
|
||||||
|
|
||||||
|
#include "base/global.h" |
||||||
|
#include "base/path.h" |
||||||
|
|
||||||
|
#ifdef Q_OS_MACOS |
||||||
|
namespace |
||||||
|
{ |
||||||
|
const CFStringRef torrentExtension = CFSTR("torrent"); |
||||||
|
const CFStringRef magnetUrlScheme = CFSTR("magnet"); |
||||||
|
} |
||||||
|
|
||||||
|
bool Utils::OS::isTorrentFileAssocSet() |
||||||
|
{ |
||||||
|
bool isSet = false; |
||||||
|
const CFStringRef torrentId = ::UTTypeCreatePreferredIdentifierForTag(kUTTagClassFilenameExtension, torrentExtension, NULL); |
||||||
|
if (torrentId != NULL) |
||||||
|
{ |
||||||
|
const CFStringRef defaultHandlerId = ::LSCopyDefaultRoleHandlerForContentType(torrentId, kLSRolesViewer); |
||||||
|
if (defaultHandlerId != NULL) |
||||||
|
{ |
||||||
|
const CFStringRef myBundleId = ::CFBundleGetIdentifier(::CFBundleGetMainBundle()); |
||||||
|
if (myBundleId != NULL) |
||||||
|
isSet = ::CFStringCompare(myBundleId, defaultHandlerId, 0) == kCFCompareEqualTo; |
||||||
|
::CFRelease(defaultHandlerId); |
||||||
|
} |
||||||
|
::CFRelease(torrentId); |
||||||
|
} |
||||||
|
return isSet; |
||||||
|
} |
||||||
|
|
||||||
|
void Utils::OS::setTorrentFileAssoc() |
||||||
|
{ |
||||||
|
if (isTorrentFileAssocSet()) |
||||||
|
return; |
||||||
|
|
||||||
|
const CFStringRef torrentId = ::UTTypeCreatePreferredIdentifierForTag(kUTTagClassFilenameExtension, torrentExtension, NULL); |
||||||
|
if (torrentId != NULL) |
||||||
|
{ |
||||||
|
const CFStringRef myBundleId = ::CFBundleGetIdentifier(::CFBundleGetMainBundle()); |
||||||
|
if (myBundleId != NULL) |
||||||
|
::LSSetDefaultRoleHandlerForContentType(torrentId, kLSRolesViewer, myBundleId); |
||||||
|
::CFRelease(torrentId); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
bool Utils::OS::isMagnetLinkAssocSet() |
||||||
|
{ |
||||||
|
bool isSet = false; |
||||||
|
const CFStringRef defaultHandlerId = ::LSCopyDefaultHandlerForURLScheme(magnetUrlScheme); |
||||||
|
if (defaultHandlerId != NULL) |
||||||
|
{ |
||||||
|
const CFStringRef myBundleId = ::CFBundleGetIdentifier(::CFBundleGetMainBundle()); |
||||||
|
if (myBundleId != NULL) |
||||||
|
isSet = ::CFStringCompare(myBundleId, defaultHandlerId, 0) == kCFCompareEqualTo; |
||||||
|
::CFRelease(defaultHandlerId); |
||||||
|
} |
||||||
|
return isSet; |
||||||
|
} |
||||||
|
|
||||||
|
void Utils::OS::setMagnetLinkAssoc() |
||||||
|
{ |
||||||
|
if (isMagnetLinkAssocSet()) |
||||||
|
return; |
||||||
|
|
||||||
|
const CFStringRef myBundleId = ::CFBundleGetIdentifier(::CFBundleGetMainBundle()); |
||||||
|
if (myBundleId != NULL) |
||||||
|
::LSSetDefaultHandlerForURLScheme(magnetUrlScheme, myBundleId); |
||||||
|
} |
||||||
|
#endif // Q_OS_MACOS
|
||||||
|
|
||||||
|
#ifdef Q_OS_WIN |
||||||
|
bool Utils::OS::isTorrentFileAssocSet() |
||||||
|
{ |
||||||
|
const QSettings settings(u"HKEY_CURRENT_USER\\Software\\Classes"_s, QSettings::NativeFormat); |
||||||
|
return settings.value(u".torrent/Default"_s).toString() == u"qBittorrent"; |
||||||
|
} |
||||||
|
|
||||||
|
void Utils::OS::setTorrentFileAssoc(const bool set) |
||||||
|
{ |
||||||
|
if (set == isTorrentFileAssocSet()) |
||||||
|
return; |
||||||
|
|
||||||
|
QSettings settings(u"HKEY_CURRENT_USER\\Software\\Classes"_s, QSettings::NativeFormat); |
||||||
|
|
||||||
|
if (set) |
||||||
|
{ |
||||||
|
const QString oldProgId = settings.value(u".torrent/Default"_s).toString(); |
||||||
|
if (!oldProgId.isEmpty() && (oldProgId != u"qBittorrent")) |
||||||
|
settings.setValue((u".torrent/OpenWithProgids/" + oldProgId), QString()); |
||||||
|
|
||||||
|
settings.setValue(u".torrent/Default"_s, u"qBittorrent"_s); |
||||||
|
settings.setValue(u".torrent/Content Type"_s, u"application/x-bittorrent"_s); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
settings.setValue(u".torrent/Default"_s, QString()); |
||||||
|
} |
||||||
|
|
||||||
|
::SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, nullptr, nullptr); |
||||||
|
} |
||||||
|
|
||||||
|
bool Utils::OS::isMagnetLinkAssocSet() |
||||||
|
{ |
||||||
|
const QSettings settings(u"HKEY_CURRENT_USER\\Software\\Classes"_s, QSettings::NativeFormat); |
||||||
|
const QString shellCommand = settings.value(u"magnet/shell/open/command/Default"_s).toString(); |
||||||
|
|
||||||
|
const QRegularExpressionMatch exeRegMatch = QRegularExpression(u"\"([^\"]+)\".*"_s).match(shellCommand); |
||||||
|
if (!exeRegMatch.hasMatch()) |
||||||
|
return false; |
||||||
|
|
||||||
|
const Path assocExe {exeRegMatch.captured(1)}; |
||||||
|
if (assocExe != Path(qApp->applicationFilePath())) |
||||||
|
return false; |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
void Utils::OS::setMagnetLinkAssoc(const bool set) |
||||||
|
{ |
||||||
|
if (set == isMagnetLinkAssocSet()) |
||||||
|
return; |
||||||
|
|
||||||
|
QSettings settings(u"HKEY_CURRENT_USER\\Software\\Classes"_s, QSettings::NativeFormat); |
||||||
|
|
||||||
|
if (set) |
||||||
|
{ |
||||||
|
const QString applicationFilePath = Path(qApp->applicationFilePath()).toString(); |
||||||
|
const QString commandStr = u'"' + applicationFilePath + u"\" \"%1\""; |
||||||
|
const QString iconStr = u'"' + applicationFilePath + u"\",1"; |
||||||
|
|
||||||
|
settings.setValue(u"magnet/Default"_s, u"URL:Magnet link"_s); |
||||||
|
settings.setValue(u"magnet/Content Type"_s, u"application/x-magnet"_s); |
||||||
|
settings.setValue(u"magnet/DefaultIcon/Default"_s, iconStr); |
||||||
|
settings.setValue(u"magnet/shell/Default"_s, u"open"_s); |
||||||
|
settings.setValue(u"magnet/shell/open/command/Default"_s, commandStr); |
||||||
|
settings.setValue(u"magnet/URL Protocol"_s, QString()); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
// only wipe values that are specific to qbt
|
||||||
|
settings.setValue(u"magnet/DefaultIcon/Default"_s, QString()); |
||||||
|
settings.setValue(u"magnet/shell/open/command/Default"_s, QString()); |
||||||
|
} |
||||||
|
|
||||||
|
::SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, nullptr, nullptr); |
||||||
|
} |
||||||
|
#endif // Q_OS_WIN
|
@ -0,0 +1,50 @@ |
|||||||
|
/*
|
||||||
|
* Bittorrent Client using Qt and libtorrent. |
||||||
|
* Copyright (C) 2023 Mike Tzou (Chocobo1) |
||||||
|
* 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. |
||||||
|
*/ |
||||||
|
|
||||||
|
#pragma once |
||||||
|
|
||||||
|
#include <QtSystemDetection> |
||||||
|
|
||||||
|
namespace Utils::OS |
||||||
|
{ |
||||||
|
#ifdef Q_OS_MACOS |
||||||
|
bool isTorrentFileAssocSet(); |
||||||
|
void setTorrentFileAssoc(); |
||||||
|
bool isMagnetLinkAssocSet(); |
||||||
|
void setMagnetLinkAssoc(); |
||||||
|
#endif // Q_OS_MACOS
|
||||||
|
|
||||||
|
#ifdef Q_OS_WIN |
||||||
|
bool isTorrentFileAssocSet(); |
||||||
|
void setTorrentFileAssoc(bool set); |
||||||
|
bool isMagnetLinkAssocSet(); |
||||||
|
void setMagnetLinkAssoc(bool set); |
||||||
|
#endif // Q_OS_WIN
|
||||||
|
} |
Loading…
Reference in new issue