69 lines
2.2 KiB
C
69 lines
2.2 KiB
C
//go:build windows
|
|
|
|
#include <windows.h>
|
|
#include <shobjidl.h>
|
|
#include <initguid.h>
|
|
#include <stdio.h>
|
|
#include "taskbar_buttons.h"
|
|
|
|
DEFINE_GUID(IID_ITaskbarList3,
|
|
0xEA1AFB91, 0x9E28, 0x4B86, 0x90, 0xE9, 0x9E, 0x9F, 0x8A, 0x5E, 0xEF, 0xAF);
|
|
|
|
static ITaskbarList3 *g_taskbar = NULL;
|
|
static ThumbnailCallback g_callback = NULL;
|
|
static HWND g_mainHWnd = NULL;
|
|
static WNDPROC g_originalProc = NULL;
|
|
|
|
THUMBBUTTON g_thumbButtons[3];
|
|
|
|
LRESULT CALLBACK OverrideWndProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam) {
|
|
if (msg == WM_COMMAND) {
|
|
int buttonId = LOWORD(wParam);
|
|
if (g_callback) {
|
|
g_callback(buttonId);
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
return CallWindowProc(g_originalProc, hwnd, msg, wParam, lParam);
|
|
}
|
|
|
|
void hook_window_proc(HWND hwnd) {
|
|
g_originalProc = (WNDPROC)SetWindowLongPtr(hwnd, GWLP_WNDPROC, (LONG_PTR)OverrideWndProc);
|
|
}
|
|
|
|
int initialize_taskbar_buttons(void *hwndPtr, ThumbnailCallback cb) {
|
|
g_callback = cb;
|
|
g_mainHWnd = (HWND)hwndPtr;
|
|
hook_window_proc(g_mainHWnd);
|
|
|
|
CoInitialize(NULL);
|
|
CoCreateInstance(&CLSID_TaskbarList, NULL, CLSCTX_INPROC_SERVER, &IID_ITaskbarList3, (void**)&g_taskbar);
|
|
|
|
if (g_taskbar) {
|
|
g_taskbar->lpVtbl->HrInit(g_taskbar);
|
|
}
|
|
|
|
if (!g_taskbar || !g_mainHWnd) return -1;
|
|
|
|
ZeroMemory(g_thumbButtons, sizeof(g_thumbButtons));
|
|
|
|
g_thumbButtons[0].dwMask = THB_FLAGS | THB_TOOLTIP;
|
|
g_thumbButtons[0].iId = 1;
|
|
g_thumbButtons[0].dwFlags = THBF_ENABLED;
|
|
wcscpy_s(g_thumbButtons[0].szTip, ARRAYSIZE(g_thumbButtons[0].szTip), L"Previous");
|
|
|
|
g_thumbButtons[1].dwMask = THB_FLAGS | THB_TOOLTIP;
|
|
g_thumbButtons[1].iId = 2;
|
|
g_thumbButtons[1].dwFlags = THBF_ENABLED;
|
|
wcscpy_s(g_thumbButtons[1].szTip, ARRAYSIZE(g_thumbButtons[1].szTip), L"Play");
|
|
|
|
g_thumbButtons[2].dwMask = THB_FLAGS | THB_TOOLTIP;
|
|
g_thumbButtons[2].iId = 3;
|
|
g_thumbButtons[2].dwFlags = THBF_ENABLED;
|
|
wcscpy_s(g_thumbButtons[2].szTip, ARRAYSIZE(g_thumbButtons[2].szTip), L"Next");
|
|
|
|
g_taskbar->lpVtbl->ThumbBarAddButtons(g_taskbar, g_mainHWnd, ARRAYSIZE(g_thumbButtons), g_thumbButtons);
|
|
|
|
return 0;
|
|
} |