296 lines
9.5 KiB
C++
296 lines
9.5 KiB
C++
///////////////////////////////////////////////////////////////////////////////
|
|
// LameXP - Audio Encoder Front-End
|
|
// Copyright (C) 2004-2014 LoRd_MuldeR <MuldeR2@GMX.de>
|
|
//
|
|
// 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, but always including the *additional*
|
|
// restrictions defined in the "License.txt" file.
|
|
//
|
|
// 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.
|
|
//
|
|
// http://www.gnu.org/licenses/gpl-2.0.txt
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "ShellIntegration.h"
|
|
|
|
//Internal
|
|
#include "Global.h"
|
|
#include "Registry_Decoder.h"
|
|
|
|
//MUtils
|
|
#include <MUtils/Global.h>
|
|
#include <MUtils/Exception.h>
|
|
|
|
//Qt
|
|
#include <QString>
|
|
#include <QStringList>
|
|
#include <QRegExp>
|
|
#include <QApplication>
|
|
#include <QFileInfo>
|
|
#include <QDir>
|
|
#include <QMutexLocker>
|
|
#include <QtConcurrentRun>
|
|
|
|
//Win32 API
|
|
#include <Shlobj.h>
|
|
#include <Shlwapi.h>
|
|
|
|
//Const
|
|
static const char *g_lamexpShellAction = "ConvertWithLameXP";
|
|
static const char *g_lamexpFileType = "LameXP.SupportedAudioFile";
|
|
|
|
//Mutex
|
|
QMutex ShellIntegration::m_mutex;
|
|
|
|
//Macros
|
|
#define REG_WRITE_STRING(KEY, STR) RegSetValueEx(key, NULL, NULL, REG_SZ, reinterpret_cast<const BYTE*>(STR.utf16()), (STR.size() + 1) * sizeof(wchar_t))
|
|
|
|
////////////////////////////////////////////////////////////
|
|
// Constructor
|
|
////////////////////////////////////////////////////////////
|
|
|
|
ShellIntegration::ShellIntegration(void)
|
|
{
|
|
MUTILS_THROW("Cannot create instance of this class, sorry!");
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////
|
|
// Public Functions
|
|
////////////////////////////////////////////////////////////
|
|
|
|
void ShellIntegration::install(bool async)
|
|
{
|
|
//Install asynchronously
|
|
if(async)
|
|
{
|
|
QFuture<void>(QtConcurrent::run(install, false));
|
|
return;
|
|
}
|
|
|
|
//Serialize
|
|
QMutexLocker lock(&m_mutex);
|
|
|
|
//Registry key
|
|
HKEY key = NULL;
|
|
|
|
//Init some consts
|
|
const QString lamexpFileType(g_lamexpFileType);
|
|
const QString lamexpFileInfo(tr("Audio File supported by LameXP"));
|
|
const QString lamexpShellText(tr("Convert this file with LameXP v%1").arg(QString().sprintf("%d.%02d", lamexp_version_major(), lamexp_version_minor())));
|
|
const QString lamexpShellCommand = QString("\"%1\" --add \"%2\"").arg(QDir::toNativeSeparators(QFileInfo(QApplication::applicationFilePath()).canonicalFilePath()), "%1");
|
|
const QString lamexpShellAction(g_lamexpShellAction);
|
|
|
|
//Register the LameXP file type
|
|
if(RegCreateKeyEx(HKEY_CURRENT_USER, MUTILS_WCHR(QString("Software\\Classes\\%1").arg(lamexpFileType)), NULL, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &key, NULL) == ERROR_SUCCESS)
|
|
{
|
|
REG_WRITE_STRING(key, lamexpFileInfo);
|
|
RegCloseKey(key);
|
|
}
|
|
if(RegCreateKeyEx(HKEY_CURRENT_USER, MUTILS_WCHR(QString("Software\\Classes\\%1\\shell").arg(lamexpFileType)), NULL, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &key, NULL) == ERROR_SUCCESS)
|
|
{
|
|
REG_WRITE_STRING(key, lamexpShellAction);
|
|
RegCloseKey(key);
|
|
}
|
|
if(RegCreateKeyEx(HKEY_CURRENT_USER, MUTILS_WCHR(QString("Software\\Classes\\%1\\shell\\%2").arg(lamexpFileType, lamexpShellAction)), NULL, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &key, NULL) == ERROR_SUCCESS)
|
|
{
|
|
REG_WRITE_STRING(key, lamexpShellText);
|
|
RegCloseKey(key);
|
|
}
|
|
if(RegCreateKeyEx(HKEY_CURRENT_USER, MUTILS_WCHR(QString("Software\\Classes\\%1\\shell\\%2\\command").arg(lamexpFileType, lamexpShellAction)), NULL, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &key, NULL) == ERROR_SUCCESS)
|
|
{
|
|
REG_WRITE_STRING(key, lamexpShellCommand);
|
|
RegCloseKey(key);
|
|
}
|
|
|
|
//Detect supported file types
|
|
QStringList *types = detectTypes(lamexpFileType, lamexpShellAction);
|
|
|
|
//Add LameXP shell action to all supported file types
|
|
while(types && (!types->isEmpty()))
|
|
{
|
|
QString currentType = types->takeFirst();
|
|
|
|
if(RegCreateKeyEx(HKEY_CURRENT_USER, MUTILS_WCHR(QString("Software\\Classes\\%1\\shell\\%2").arg(currentType, lamexpShellAction)), NULL, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &key, NULL) == ERROR_SUCCESS)
|
|
{
|
|
REG_WRITE_STRING(key, lamexpShellText);
|
|
RegCloseKey(key);
|
|
}
|
|
|
|
if(RegCreateKeyEx(HKEY_CURRENT_USER, MUTILS_WCHR(QString("Software\\Classes\\%1\\shell\\%2\\command").arg(currentType, lamexpShellAction)), NULL, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &key, NULL) == ERROR_SUCCESS)
|
|
{
|
|
REG_WRITE_STRING(key, lamexpShellCommand);
|
|
RegCloseKey(key);
|
|
}
|
|
}
|
|
|
|
//Shell notification
|
|
SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, NULL, NULL);
|
|
|
|
//Free
|
|
delete types;
|
|
}
|
|
|
|
void ShellIntegration::remove(bool async)
|
|
{
|
|
//Remove asynchronously
|
|
if(async)
|
|
{
|
|
QFuture<void>(QtConcurrent::run(remove, false));
|
|
return;
|
|
}
|
|
|
|
//Serialize
|
|
QMutexLocker lock(&m_mutex);
|
|
|
|
//Init some consts
|
|
const QString lamexpFileType(g_lamexpFileType);
|
|
const QString lamexpShellAction(g_lamexpShellAction);
|
|
|
|
//Initialization
|
|
HKEY key = NULL;
|
|
QStringList fileTypes;
|
|
|
|
//Find all registered file types
|
|
if(RegOpenKeyEx(HKEY_CURRENT_USER, L"Software\\Classes", NULL, KEY_ENUMERATE_SUB_KEYS ,&key) == ERROR_SUCCESS)
|
|
{
|
|
wchar_t name[256];
|
|
|
|
for(DWORD i = 0; true; i++)
|
|
{
|
|
DWORD size = 256;
|
|
if(RegEnumKeyEx(key, i, name, &size, NULL, NULL, NULL, NULL) == ERROR_SUCCESS)
|
|
{
|
|
fileTypes << QString::fromUtf16(reinterpret_cast<const unsigned short*>(name));
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
//Remove shell action from all file types
|
|
while(!fileTypes.isEmpty())
|
|
{
|
|
SHDeleteKey(HKEY_CURRENT_USER, MUTILS_WCHR(QString("Software\\Classes\\%1\\shell\\%2").arg(fileTypes.takeFirst(), lamexpShellAction)));
|
|
}
|
|
|
|
//Unregister LameXP file type
|
|
SHDeleteKey(HKEY_CURRENT_USER, MUTILS_WCHR(QString("Software\\Classes\\%1").arg(lamexpFileType)));
|
|
|
|
//Shell notification
|
|
SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, NULL, NULL);
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////
|
|
// Private Functions
|
|
////////////////////////////////////////////////////////////
|
|
|
|
QStringList *ShellIntegration::detectTypes(const QString &lamexpFileType, const QString &lamexpShellAction)
|
|
{
|
|
HKEY key = NULL;
|
|
|
|
QStringList *nativeTypes = new QStringList();
|
|
QStringList supportedTypes = DecoderRegistry::getSupportedTypes();
|
|
QStringList extensions;
|
|
|
|
QRegExp regExp1("\\((.+)\\)");
|
|
QRegExp regExp2("(\\.\\w+)");
|
|
|
|
//Find all supported file extensions
|
|
while(!supportedTypes.isEmpty())
|
|
{
|
|
if(regExp1.lastIndexIn(supportedTypes.takeFirst()) > 0)
|
|
{
|
|
int lastIndex = 0;
|
|
while((lastIndex = regExp2.indexIn(regExp1.cap(1), lastIndex) + 1) >= 1)
|
|
{
|
|
extensions.append(regExp2.cap(1));
|
|
}
|
|
}
|
|
}
|
|
|
|
//Map supported extensions to native types
|
|
while(!extensions.isEmpty())
|
|
{
|
|
QString currentExt = extensions.takeFirst();
|
|
|
|
if(RegOpenKeyEx(HKEY_CLASSES_ROOT, MUTILS_WCHR(currentExt), NULL, KEY_QUERY_VALUE, &key) == ERROR_SUCCESS)
|
|
{
|
|
wchar_t data[256];
|
|
DWORD dataLen = 256 * sizeof(wchar_t);
|
|
DWORD type = NULL;
|
|
|
|
if(RegQueryValueEx(key, NULL, NULL, &type, reinterpret_cast<BYTE*>(data), &dataLen) == ERROR_SUCCESS)
|
|
{
|
|
if((type == REG_SZ) || (type == REG_EXPAND_SZ))
|
|
{
|
|
QString currentType = QString::fromUtf16(reinterpret_cast<unsigned short*>(data));
|
|
if((currentType.compare(lamexpFileType, Qt::CaseInsensitive) != 0) && !nativeTypes->contains(currentType, Qt::CaseInsensitive))
|
|
{
|
|
nativeTypes->append(currentType);
|
|
}
|
|
}
|
|
}
|
|
RegCloseKey(key);
|
|
}
|
|
else
|
|
{
|
|
if(RegCreateKeyEx(HKEY_CURRENT_USER, MUTILS_WCHR(QString("Software\\Classes\\%1").arg(currentExt)), NULL, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE, NULL, &key, NULL) == ERROR_SUCCESS)
|
|
{
|
|
REG_WRITE_STRING(key, lamexpFileType);
|
|
RegCloseKey(key);
|
|
}
|
|
}
|
|
|
|
if(RegOpenKeyEx(HKEY_CURRENT_USER, MUTILS_WCHR(QString("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\FileExts\\%1\\UserChoice").arg(currentExt)), NULL, KEY_QUERY_VALUE, &key) == ERROR_SUCCESS)
|
|
{
|
|
wchar_t data[256];
|
|
DWORD dataLen = 256 * sizeof(wchar_t);
|
|
DWORD type = NULL;
|
|
|
|
if(RegQueryValueEx(key, L"Progid", NULL, &type, reinterpret_cast<BYTE*>(data), &dataLen) == ERROR_SUCCESS)
|
|
{
|
|
if((type == REG_SZ) || (type == REG_EXPAND_SZ))
|
|
{
|
|
QString currentType = QString::fromUtf16(reinterpret_cast<unsigned short*>(data));
|
|
if((currentType.compare(lamexpFileType, Qt::CaseInsensitive) != 0) && !nativeTypes->contains(currentType, Qt::CaseInsensitive))
|
|
{
|
|
nativeTypes->append(currentType);
|
|
}
|
|
}
|
|
}
|
|
RegCloseKey(key);
|
|
}
|
|
|
|
if(RegOpenKeyEx(HKEY_CURRENT_USER, MUTILS_WCHR(QString("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\FileExts\\%1\\OpenWithProgids").arg(currentExt)), NULL, KEY_QUERY_VALUE, &key) == ERROR_SUCCESS)
|
|
{
|
|
wchar_t name[256];
|
|
for(DWORD i = 0; true; i++)
|
|
{
|
|
DWORD size = 256;
|
|
if(RegEnumValue(key, i, name, &size, NULL, NULL, NULL, NULL) == ERROR_SUCCESS)
|
|
{
|
|
QString currentType = QString::fromUtf16(reinterpret_cast<unsigned short*>(name));
|
|
if((currentType.compare(lamexpFileType, Qt::CaseInsensitive) != 0) && !nativeTypes->contains(currentType, Qt::CaseInsensitive))
|
|
{
|
|
nativeTypes->append(currentType);
|
|
}
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
RegCloseKey(key);
|
|
}
|
|
}
|
|
|
|
return nativeTypes;
|
|
}
|