255 lines
8.9 KiB
C++
255 lines
8.9 KiB
C++
///////////////////////////////////////////////////////////////////////////////
|
|
// LameXP - Audio Encoder Front-End
|
|
// Copyright (C) 2004-2015 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>
|
|
#include <MUtils/OSSupport.h>
|
|
#include <MUtils/Registry.h>
|
|
|
|
//Qt
|
|
#include <QString>
|
|
#include <QStringList>
|
|
#include <QRegExp>
|
|
#include <QApplication>
|
|
#include <QFileInfo>
|
|
#include <QDir>
|
|
#include <QMutexLocker>
|
|
#include <QtConcurrentRun>
|
|
|
|
//Const
|
|
static const char *g_lamexpShellAction = "ConvertWithLameXP";
|
|
static const char *g_lamexpFileType = "LameXP.SupportedAudioFile";
|
|
|
|
//Mutex
|
|
QMutex ShellIntegration::m_mutex;
|
|
|
|
//State values
|
|
static const int STATE_ENABLED = 1;
|
|
static const int STATE_UNKNOWN = 0;
|
|
static const int STATE_DISABLD = -1;
|
|
|
|
//State
|
|
volatile int ShellIntegration::m_state = STATE_UNKNOWN;
|
|
|
|
//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);
|
|
|
|
//Checking
|
|
if(m_state == STATE_ENABLED)
|
|
{
|
|
return; /*already enabled, don't enable again!*/
|
|
}
|
|
|
|
//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
|
|
const bool ok1 = MUtils::Registry::reg_value_write(MUtils::Registry::root_user, QString("Software\\Classes\\%1") .arg(lamexpFileType), QString(), lamexpFileInfo);
|
|
const bool ok2 = MUtils::Registry::reg_value_write(MUtils::Registry::root_user, QString("Software\\Classes\\%1\\shell") .arg(lamexpFileType), QString(), lamexpShellAction);
|
|
const bool ok3 = MUtils::Registry::reg_value_write(MUtils::Registry::root_user, QString("Software\\Classes\\%1\\shell\\%2") .arg(lamexpFileType, lamexpShellAction), QString(), lamexpShellText);
|
|
const bool ok4 = MUtils::Registry::reg_value_write(MUtils::Registry::root_user, QString("Software\\Classes\\%1\\shell\\%2\\command").arg(lamexpFileType, lamexpShellAction), QString(), lamexpShellCommand);
|
|
if(!(ok1 && ok2 && ok3 && ok4))
|
|
{
|
|
qWarning("Failed to create LameXP file type!");
|
|
return;
|
|
}
|
|
|
|
//Detect supported file types
|
|
QStringList types;
|
|
detectTypes(lamexpFileType, lamexpShellAction, types);
|
|
|
|
//Add LameXP shell action to all supported file types
|
|
for(QStringList::ConstIterator iter = types.constBegin(); iter != types.constEnd(); iter++)
|
|
{
|
|
MUtils::Registry::reg_value_write(MUtils::Registry::root_user, QString("Software\\Classes\\%1\\shell\\%2") .arg((*iter), lamexpShellAction), QString(), lamexpShellText);
|
|
MUtils::Registry::reg_value_write(MUtils::Registry::root_user, QString("Software\\Classes\\%1\\shell\\%2\\command").arg((*iter), lamexpShellAction), QString(), lamexpShellCommand);
|
|
}
|
|
|
|
//Shell notification
|
|
MUtils::OS::shell_change_notification();
|
|
m_state = STATE_ENABLED;
|
|
}
|
|
|
|
void ShellIntegration::remove(bool async)
|
|
{
|
|
//Remove asynchronously
|
|
if(async)
|
|
{
|
|
QFuture<void>(QtConcurrent::run(remove, false));
|
|
return;
|
|
}
|
|
|
|
//Serialize
|
|
QMutexLocker lock(&m_mutex);
|
|
|
|
//Checking
|
|
if(m_state == STATE_DISABLD)
|
|
{
|
|
return; /*already enabled, don't enable again!*/
|
|
}
|
|
|
|
//Init some consts
|
|
const QString lamexpFileType(g_lamexpFileType);
|
|
const QString lamexpShellAction(g_lamexpShellAction);
|
|
|
|
//Initialization
|
|
QStringList fileTypes;
|
|
|
|
//Find all registered file types
|
|
if(!MUtils::Registry::reg_enum_subkeys(MUtils::Registry::root_user, "Software\\Classes", fileTypes))
|
|
{
|
|
qWarning("Failed to enumerate file types!");
|
|
return;
|
|
}
|
|
|
|
//Remove shell action from all file types
|
|
for(QStringList::ConstIterator iter = fileTypes.constBegin(); iter != fileTypes.constEnd(); iter++)
|
|
{
|
|
MUtils::Registry::reg_key_delete(MUtils::Registry::root_user, QString("Software\\Classes\\%1\\shell\\%2").arg((*iter), lamexpShellAction));
|
|
|
|
//Remove from sub-tree too
|
|
QStringList subTypes;
|
|
if(MUtils::Registry::reg_enum_subkeys(MUtils::Registry::root_user, QString("Software\\Classes\\%1").arg(*iter), subTypes))
|
|
{
|
|
for(QStringList::ConstIterator iter2 = subTypes.constBegin(); iter2 != subTypes.constEnd(); iter2++)
|
|
{
|
|
MUtils::Registry::reg_key_delete(MUtils::Registry::root_user, QString("Software\\Classes\\%1\\%2\\shell\\%3").arg((*iter), (*iter2), lamexpShellAction));
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
//Unregister LameXP file type
|
|
MUtils::Registry::reg_key_delete(MUtils::Registry::root_user, QString("Software\\Classes\\%1").arg(lamexpFileType));
|
|
|
|
//Shell notification
|
|
MUtils::OS::shell_change_notification();
|
|
m_state = STATE_DISABLD;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////
|
|
// Private Functions
|
|
////////////////////////////////////////////////////////////
|
|
|
|
void ShellIntegration::detectTypes(const QString &lamexpFileType, const QString &lamexpShellAction, QStringList &nativeTypes)
|
|
{
|
|
nativeTypes.clear();
|
|
const QStringList supportedTypes = DecoderRegistry::getSupportedTypes();
|
|
const QString progId = "Progid";
|
|
|
|
QRegExp regExp1("\\((.+)\\)");
|
|
QRegExp regExp2("(\\.\\w+)");
|
|
|
|
//Find all supported file extensions
|
|
QStringList extensions;
|
|
for(QStringList::ConstIterator iter = supportedTypes.constBegin(); iter != supportedTypes.constEnd(); iter++)
|
|
{
|
|
if(regExp1.lastIndexIn(*iter) > 0)
|
|
{
|
|
int lastIndex = 0;
|
|
while((lastIndex = regExp2.indexIn(regExp1.cap(1), lastIndex) + 1) >= 1)
|
|
{
|
|
if(!extensions.contains(regExp2.cap(1), Qt::CaseInsensitive))
|
|
{
|
|
extensions.append(regExp2.cap(1));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
//Map supported extensions to native types
|
|
for(QStringList::ConstIterator iter = extensions.constBegin(); iter != extensions.constEnd(); iter++)
|
|
{
|
|
QString currentType;
|
|
if(MUtils::Registry::reg_value_read(MUtils::Registry::root_classes, (*iter), QString(), currentType))
|
|
{
|
|
currentType = QDir::toNativeSeparators(currentType);
|
|
if((currentType.compare(lamexpFileType, Qt::CaseInsensitive) != 0) && (!nativeTypes.contains(currentType, Qt::CaseInsensitive)))
|
|
{
|
|
nativeTypes.append(currentType);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
MUtils::Registry::reg_value_write(MUtils::Registry::root_user, (*iter), QString(), lamexpFileType);
|
|
}
|
|
|
|
if(MUtils::Registry::reg_value_read(MUtils::Registry::root_user, QString("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\FileExts\\%1\\UserChoice").arg(*iter), progId, currentType))
|
|
{
|
|
currentType = QDir::toNativeSeparators(currentType);
|
|
if((currentType.compare(lamexpFileType, Qt::CaseInsensitive) != 0) && (!nativeTypes.contains(currentType, Qt::CaseInsensitive)))
|
|
{
|
|
nativeTypes.append(currentType);
|
|
}
|
|
}
|
|
|
|
QStringList progids;
|
|
if(MUtils::Registry::reg_enum_values(MUtils::Registry::root_user, QString("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\FileExts\\%1\\OpenWithProgids").arg(*iter), progids))
|
|
{
|
|
for(QStringList::ConstIterator iter2 = progids.constBegin(); iter2 != progids.constEnd(); iter2++)
|
|
{
|
|
currentType = QDir::toNativeSeparators(currentType);
|
|
if((iter2->compare(lamexpFileType, Qt::CaseInsensitive) != 0) && (!nativeTypes.contains((*iter2), Qt::CaseInsensitive)))
|
|
{
|
|
nativeTypes.append(*iter2);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|