mirror of
https://github.com/LegacyUpdate/LegacyUpdate.git
synced 2025-01-22 14:12:07 -05:00
102 lines
2.5 KiB
C
102 lines
2.5 KiB
C
// Based on https://nsis.sourceforge.io/TaskbarProgress_plug-in - zlib licensed
|
|
// Cleaned up and refactored into C by Legacy Update
|
|
#include <windows.h>
|
|
#include <nsis/pluginapi.h>
|
|
#include <commctrl.h>
|
|
#include <objbase.h>
|
|
#include <shobjidl.h>
|
|
#include "main.h"
|
|
|
|
static ITaskbarList3 *g_taskbarList;
|
|
static UINT g_totalRange;
|
|
static WNDPROC g_origWndProc;
|
|
|
|
LRESULT CALLBACK ProgressBarWndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam) {
|
|
if (g_origWndProc == NULL) {
|
|
return 0;
|
|
}
|
|
|
|
switch (uMsg) {
|
|
case PBM_SETRANGE:
|
|
g_totalRange = LOWORD(lParam) + HIWORD(lParam);
|
|
break;
|
|
|
|
case PBM_SETRANGE32:
|
|
g_totalRange = wParam + lParam;
|
|
break;
|
|
|
|
case PBM_SETPOS:
|
|
if (g_taskbarList != NULL) {
|
|
ITaskbarList3_SetProgressValue(g_taskbarList, g_hwndParent, wParam, g_totalRange);
|
|
}
|
|
break;
|
|
|
|
case WM_DESTROY:
|
|
SetWindowLongPtr(hwnd, GWLP_WNDPROC, (LONG_PTR)g_origWndProc);
|
|
|
|
if (g_taskbarList != NULL) {
|
|
ITaskbarList3_SetProgressState(g_taskbarList, g_hwndParent, TBPF_NOPROGRESS);
|
|
ITaskbarList3_Release(g_taskbarList);
|
|
g_taskbarList = NULL;
|
|
}
|
|
|
|
g_origWndProc = NULL;
|
|
break;
|
|
}
|
|
|
|
return CallWindowProc(g_origWndProc, hwnd, uMsg, wParam, lParam);
|
|
}
|
|
|
|
static UINT_PTR NSISPluginCallback(enum NSPIM event) {
|
|
// Does nothing, but keeping a callback registered prevents NSIS from unloading the plugin
|
|
return 0;
|
|
}
|
|
|
|
PLUGIN_METHOD(InitTaskbarProgress) {
|
|
PLUGIN_INIT();
|
|
|
|
extra->RegisterPluginCallback(g_hInstance, NSISPluginCallback);
|
|
|
|
if (g_taskbarList != NULL && g_origWndProc != NULL) {
|
|
// Already initialised
|
|
return;
|
|
}
|
|
|
|
HWND innerWindow = FindWindowEx(g_hwndParent, NULL, L"#32770", NULL);
|
|
HWND progressBar = FindWindowEx(innerWindow, NULL, L"msctls_progress32", NULL);
|
|
PBRANGE range;
|
|
HRESULT hr;
|
|
|
|
if (progressBar == NULL) {
|
|
goto fail;
|
|
}
|
|
|
|
hr = CoCreateInstance(&CLSID_TaskbarList, NULL, CLSCTX_INPROC_SERVER, &IID_ITaskbarList3, (void**)&g_taskbarList);
|
|
if (!SUCCEEDED(hr)) {
|
|
goto fail;
|
|
}
|
|
|
|
hr = ITaskbarList3_HrInit(g_taskbarList);
|
|
if (!SUCCEEDED(hr)) {
|
|
goto fail;
|
|
}
|
|
|
|
// Get the initial progress bar range
|
|
SendMessage(progressBar, PBM_GETRANGE, 0, (LPARAM)&range);
|
|
g_totalRange = range.iLow + range.iHigh;
|
|
|
|
// Add our own window procedure so we can respond to progress bar updates
|
|
g_origWndProc = (WNDPROC)SetWindowLongPtr(progressBar, GWLP_WNDPROC, (LONG_PTR)ProgressBarWndProc);
|
|
if (g_origWndProc == NULL) {
|
|
goto fail;
|
|
}
|
|
return;
|
|
|
|
fail:
|
|
if (g_taskbarList != NULL) {
|
|
ITaskbarList3_Release(g_taskbarList);
|
|
g_taskbarList = NULL;
|
|
}
|
|
|
|
g_origWndProc = NULL;
|
|
}
|