Permalink
Fetching contributors…
Cannot retrieve contributors at this time
268 lines (229 sloc) 9.6 KB
///////////////////////////////////////////////////////////////////////////////
// LameXP - Audio Encoder Front-End
// Copyright (C) 2004-2016 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
bool ok[4] = {false, false, false, false};
ok[0] = MUtils::Registry::reg_value_write(MUtils::Registry::root_user, QString("Software\\Classes\\%1") .arg(lamexpFileType), QString(), lamexpFileInfo);
ok[1] = MUtils::Registry::reg_value_write(MUtils::Registry::root_user, QString("Software\\Classes\\%1\\shell") .arg(lamexpFileType), QString(), lamexpShellAction);
ok[2] = MUtils::Registry::reg_value_write(MUtils::Registry::root_user, QString("Software\\Classes\\%1\\shell\\%2") .arg(lamexpFileType, lamexpShellAction), QString(), lamexpShellText);
ok[3] = MUtils::Registry::reg_value_write(MUtils::Registry::root_user, QString("Software\\Classes\\%1\\shell\\%2\\command").arg(lamexpFileType, lamexpShellAction), QString(), lamexpShellCommand);
if(!(ok[0] && ok[1] && ok[2] && ok[3]))
{
qWarning("Failed to register the LameXP file type!");
return;
}
//Detect supported file types
QStringList types;
initializeTypes(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++)
{
//Remove LameXP-specific types altogether
if(iter->startsWith('.'))
{
QString currentFileType;
if(MUtils::Registry::reg_value_read(MUtils::Registry::root_user, QString("Software\\Classes\\%1").arg(*iter), QString(), currentFileType))
{
if(currentFileType.compare(lamexpFileType, Qt::CaseInsensitive) == 0)
{
MUtils::Registry::reg_key_delete(MUtils::Registry::root_user, QString("Software\\Classes\\%1").arg(*iter));
continue;
}
}
}
//Remove shell action for non-LameXP types
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::initializeTypes(const QString &lamexpFileType, const QString &lamexpShellAction, QStringList &nativeTypes)
{
nativeTypes.clear();
const QString progId = "Progid";
//Map supported extensions to native types
const QStringList &supportedExts = DecoderRegistry::getSupportedExts();
for(QStringList::ConstIterator iter = supportedExts.constBegin(); iter != supportedExts.constEnd(); iter++)
{
const QString currentExt = (*iter).mid(1).trimmed(); /*remove leading asterisk*/
if(currentExt.isEmpty() || (!currentExt.startsWith('.')))
{
qWarning("Invalid file extension '%s' encountered -> skipping!", currentExt);
continue;
}
bool hasExistingType = false;
QString currentType;
if(MUtils::Registry::reg_key_exists(MUtils::Registry::root_classes, currentExt))
{
if(MUtils::Registry::reg_value_read(MUtils::Registry::root_classes, currentExt, QString(), currentType))
{
currentType = QDir::toNativeSeparators(currentType);
if((currentType.compare(lamexpFileType, Qt::CaseInsensitive) != 0) && (!nativeTypes.contains(currentType, Qt::CaseInsensitive)))
{
nativeTypes.append(currentType);
hasExistingType = true;
}
}
}
if(!hasExistingType)
{
currentType = lamexpFileType;
MUtils::Registry::reg_value_write(MUtils::Registry::root_user, QString("Software\\Classes\\%1").arg(currentExt), QString(), lamexpFileType);
}
const QString userChoiceKey = QString("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\FileExts\\%1\\UserChoice").arg(currentExt);
if(MUtils::Registry::reg_key_exists(MUtils::Registry::root_user, userChoiceKey))
{
if(MUtils::Registry::reg_value_read(MUtils::Registry::root_user, userChoiceKey, 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(currentExt), 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);
}
}
}
}
}