[ImageResizer] Migrate settings from registry to JSON. (#2604)

* Migrate ImageResizer settings from registry to JSON.

* Update last loaded time read/write IO operations.

* Rename registry helper functions.
This commit is contained in:
vldmr11080 2020-05-05 09:00:50 +02:00 committed by GitHub
parent c14c51f551
commit 2b4b23f726
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 140 additions and 69 deletions

View File

@ -37,7 +37,7 @@ HRESULT CContextMenuHandler::Initialize(_In_opt_ PCIDLIST_ABSOLUTE pidlFolder, _
{ {
Uninitialize(); Uninitialize();
if (!CSettings::GetEnabled()) if (!CSettingsInstance().GetEnabled())
{ {
return E_FAIL; return E_FAIL;
} }
@ -62,7 +62,7 @@ HRESULT CContextMenuHandler::QueryContextMenu(_In_ HMENU hmenu, UINT indexMenu,
{ {
return S_OK; return S_OK;
} }
if (!CSettings::GetEnabled()) if (!CSettingsInstance().GetEnabled())
{ {
return E_FAIL; return E_FAIL;
} }
@ -354,7 +354,7 @@ HRESULT __stdcall CContextMenuHandler::GetCanonicalName(GUID* pguidCommandName)
HRESULT __stdcall CContextMenuHandler::GetState(IShellItemArray* psiItemArray, BOOL fOkToBeSlow, EXPCMDSTATE* pCmdState) HRESULT __stdcall CContextMenuHandler::GetState(IShellItemArray* psiItemArray, BOOL fOkToBeSlow, EXPCMDSTATE* pCmdState)
{ {
if (!CSettings::GetEnabled()) if (!CSettingsInstance().GetEnabled())
{ {
*pCmdState = ECS_HIDDEN; *pCmdState = ECS_HIDDEN;
return S_OK; return S_OK;

View File

@ -1,66 +1,116 @@
#include "pch.h" #include "pch.h"
#include <commctrl.h>
#include "Settings.h" #include "Settings.h"
const wchar_t c_rootRegPath[] = L"Software\\Microsoft\\ImageResizer"; #include <common/json.h>
const wchar_t c_enabled[] = L"Enabled"; #include <common/settings_helpers.h>
const bool c_enabledDefault = true; #include <filesystem>
#include <commctrl.h>
bool CSettings::GetEnabled() namespace
{ {
return GetRegBoolValue(c_enabled, c_enabledDefault); const wchar_t c_imageResizerDataFilePath[] = L"\\image-resizer-settings.json";
} const wchar_t c_rootRegPath[] = L"Software\\Microsoft\\ImageResizer";
const wchar_t c_enabled[] = L"Enabled";
bool CSettings::SetEnabled(_In_ bool enabled) unsigned int RegReadInteger(const std::wstring& valueName, unsigned int defaultValue)
{ {
return SetRegBoolValue(c_enabled, enabled);
}
bool CSettings::SetRegBoolValue(_In_ PCWSTR valueName, _In_ bool value)
{
DWORD dwValue = value ? 1 : 0;
return SetRegDWORDValue(valueName, dwValue);
}
bool CSettings::GetRegBoolValue(_In_ PCWSTR valueName, _In_ bool defaultValue)
{
DWORD value = GetRegDWORDValue(valueName, (defaultValue == 0) ? false : true);
return (value == 0) ? false : true;
}
bool CSettings::SetRegDWORDValue(_In_ PCWSTR valueName, _In_ DWORD value)
{
return (SUCCEEDED(HRESULT_FROM_WIN32(SHSetValue(HKEY_CURRENT_USER, c_rootRegPath, valueName, REG_DWORD, &value, sizeof(value)))));
}
DWORD CSettings::GetRegDWORDValue(_In_ PCWSTR valueName, _In_ DWORD defaultValue)
{
DWORD retVal = defaultValue;
DWORD type = REG_DWORD; DWORD type = REG_DWORD;
DWORD dwEnabled = 0; DWORD data = 0;
DWORD cb = sizeof(dwEnabled); DWORD size = sizeof(DWORD);
if (SHGetValue(HKEY_CURRENT_USER, c_rootRegPath, valueName, &type, &dwEnabled, &cb) == ERROR_SUCCESS) if (SHGetValue(HKEY_CURRENT_USER, c_rootRegPath, valueName.c_str(), &type, &data, &size) == ERROR_SUCCESS)
{ {
retVal = dwEnabled; return data;
}
return defaultValue;
} }
return retVal; bool RegReadBoolean(const std::wstring& valueName, bool defaultValue)
}
bool CSettings::SetRegStringValue(_In_ PCWSTR valueName, _In_ PCWSTR value)
{
ULONG cb = (DWORD)((wcslen(value) + 1) * sizeof(*value));
return (SUCCEEDED(HRESULT_FROM_WIN32(SHSetValue(HKEY_CURRENT_USER, c_rootRegPath, valueName, REG_SZ, (const BYTE*)value, cb))));
}
bool CSettings::GetRegStringValue(_In_ PCWSTR valueName, __out_ecount(cchBuf) PWSTR value, DWORD cchBuf)
{
if (cchBuf > 0)
{ {
value[0] = L'\0'; DWORD value = RegReadInteger(valueName.c_str(), defaultValue ? 1 : 0);
return (value == 0) ? false : true;
} }
DWORD type = REG_SZ; bool LastModifiedTime(const std::wstring& filePath, FILETIME* lpFileTime)
ULONG cb = cchBuf * sizeof(*value); {
return (SUCCEEDED(HRESULT_FROM_WIN32(SHGetValue(HKEY_CURRENT_USER, c_rootRegPath, valueName, &type, value, &cb) == ERROR_SUCCESS))); WIN32_FILE_ATTRIBUTE_DATA attr{};
if (GetFileAttributesExW(filePath.c_str(), GetFileExInfoStandard, &attr))
{
*lpFileTime = attr.ftLastWriteTime;
return true;
}
return false;
}
}
CSettings::CSettings()
{
std::wstring result = PTSettingsHelper::get_module_save_folder_location(L"ImageResizer");
jsonFilePath = result + std::wstring(c_imageResizerDataFilePath);
Load();
}
void CSettings::Save()
{
json::JsonObject jsonData;
jsonData.SetNamedValue(c_enabled, json::value(settings.enabled));
json::to_file(jsonFilePath, jsonData);
GetSystemTimeAsFileTime(&lastLoadedTime);
}
void CSettings::Load()
{
if (!std::filesystem::exists(jsonFilePath))
{
MigrateFromRegistry();
Save();
}
else
{
ParseJson();
}
}
void CSettings::Reload()
{
// Load json settings from data file if it is modified in the meantime.
FILETIME lastModifiedTime{};
if (LastModifiedTime(jsonFilePath, &lastModifiedTime) &&
CompareFileTime(&lastModifiedTime, &lastLoadedTime) == 1)
{
Load();
}
}
void CSettings::MigrateFromRegistry()
{
settings.enabled = RegReadBoolean(c_enabled, true);
}
void CSettings::ParseJson()
{
auto json = json::from_file(jsonFilePath);
if (json)
{
const json::JsonObject& jsonSettings = json.value();
try
{
if (json::has(jsonSettings, c_enabled, json::JsonValueType::Boolean))
{
settings.enabled = jsonSettings.GetNamedBoolean(c_enabled);
}
}
catch (const winrt::hresult_error&)
{
}
}
GetSystemTimeAsFileTime(&lastLoadedTime);
}
CSettings& CSettingsInstance()
{
static CSettings instance;
return instance;
} }

View File

@ -3,14 +3,36 @@
class CSettings class CSettings
{ {
public: public:
static bool GetEnabled(); CSettings();
static bool SetEnabled(_In_ bool enabled);
inline bool GetEnabled()
{
Reload();
return settings.enabled;
}
inline void SetEnabled(bool enabled)
{
settings.enabled = enabled;
Save();
}
void Save();
void Load();
private: private:
static bool GetRegBoolValue(_In_ PCWSTR valueName, _In_ bool defaultValue); struct Settings
static bool SetRegBoolValue(_In_ PCWSTR valueName, _In_ bool value); {
static bool SetRegDWORDValue(_In_ PCWSTR valueName, _In_ DWORD value); bool enabled{ true };
static DWORD GetRegDWORDValue(_In_ PCWSTR valueName, _In_ DWORD defaultValue); };
static bool SetRegStringValue(_In_ PCWSTR valueName, _In_ PCWSTR value);
static bool GetRegStringValue(_In_ PCWSTR valueName, __out_ecount(cchBuf) PWSTR value, DWORD cchBuf); void Reload();
void MigrateFromRegistry();
void ParseJson();
Settings settings;
std::wstring jsonFilePath;
FILETIME lastLoadedTime;
}; };
CSettings& CSettingsInstance();

View File

@ -37,7 +37,7 @@ public:
// Constructor // Constructor
ImageResizerModule() ImageResizerModule()
{ {
m_enabled = CSettings::GetEnabled(); m_enabled = CSettingsInstance().GetEnabled();
app_name = GET_RESOURCE_STRING(IDS_IMAGERESIZER); app_name = GET_RESOURCE_STRING(IDS_IMAGERESIZER);
}; };
@ -87,7 +87,7 @@ public:
virtual void enable() virtual void enable()
{ {
m_enabled = true; m_enabled = true;
CSettings::SetEnabled(m_enabled); CSettingsInstance().SetEnabled(m_enabled);
Trace::EnableImageResizer(m_enabled); Trace::EnableImageResizer(m_enabled);
} }
@ -95,7 +95,7 @@ public:
virtual void disable() virtual void disable()
{ {
m_enabled = false; m_enabled = false;
CSettings::SetEnabled(m_enabled); CSettingsInstance().SetEnabled(m_enabled);
Trace::EnableImageResizer(m_enabled); Trace::EnableImageResizer(m_enabled);
} }

View File

@ -441,7 +441,6 @@ void CSettings::Reload()
} }
} }
void CSettings::MigrateFromRegistry() void CSettings::MigrateFromRegistry()
{ {
settings.enabled = GetRegBoolean(c_enabled, true); settings.enabled = GetRegBoolean(c_enabled, true);