103 lines
2.8 KiB
C++
103 lines
2.8 KiB
C++
#include "WindowsShell.h"
|
|
#include "resource.h"
|
|
#include "ApplicationData.h"
|
|
#include "Util.h"
|
|
|
|
#include <Windows.h>
|
|
#include <shellapi.h>
|
|
#include <commctrl.h>
|
|
#include <strsafe.h>
|
|
|
|
#include <iostream>
|
|
|
|
const UINT TRAY_ID = 420;
|
|
const UINT WMAPP_NOTIFYCALLBACK = WM_APP + 1;
|
|
|
|
extern DrawData* gDrawData;
|
|
extern ApplicationData* gAppData;
|
|
extern bool justDocked;
|
|
extern bool isHidden;
|
|
|
|
void initShell(DrawData& drawData)
|
|
{
|
|
// Set window icon
|
|
HINSTANCE instance = GetModuleHandle(NULL);
|
|
LPWSTR iconId = MAKEINTRESOURCE(IDI_ICON1);
|
|
HANDLE iconLarge = LoadImageW(instance, iconId, IMAGE_ICON, 64, 64, 0);
|
|
HANDLE iconSmall = LoadImageW(instance, iconId, IMAGE_ICON, 32, 32, 0);
|
|
SendMessage(drawData.window_handle, WM_SETICON, ICON_BIG, (LPARAM)iconLarge);
|
|
SendMessage(drawData.window_handle, WM_SETICON, ICON_SMALL, (LPARAM)iconSmall);
|
|
SendMessage(drawData.window_handle, WM_SETICON, ICON_SMALL2, (LPARAM)iconSmall);
|
|
|
|
// Set tray icon
|
|
NOTIFYICONDATA nid = { sizeof(nid) };
|
|
nid.hWnd = drawData.window_handle;
|
|
nid.uFlags = NIF_ICON | NIF_TIP | NIF_MESSAGE | NIF_SHOWTIP;
|
|
nid.uID = TRAY_ID;
|
|
nid.uCallbackMessage = WMAPP_NOTIFYCALLBACK;
|
|
|
|
HRESULT iconResult;
|
|
iconResult = LoadIconMetric(instance, iconId, LIM_SMALL, &nid.hIcon);
|
|
isError(iconResult, "Failed to load tray icon image: ");
|
|
|
|
iconResult = LoadString(instance, IDS_STRING_TOOLTIP, nid.szTip, ARRAYSIZE(nid.szTip));
|
|
isError(iconResult, "Failed to load tray icon text: ");
|
|
|
|
Shell_NotifyIcon(NIM_ADD, &nid);
|
|
|
|
nid.uVersion = NOTIFYICON_VERSION_4;
|
|
Shell_NotifyIcon(NIM_SETVERSION, &nid);
|
|
|
|
if (!SetWindowsHookEx(WH_CALLWNDPROC, trayIconEventHandler, instance, GetCurrentThreadId()))
|
|
{
|
|
std::cout << "Failed to hook tray icon events: " << std::hex << GetLastError() << std::endl;
|
|
}
|
|
|
|
// Set window minimize behavior
|
|
glfwSetWindowIconifyCallback(drawData.window, [](GLFWwindow* window, int isIconified) {
|
|
if (isIconified && gAppData->settings.docked)
|
|
{
|
|
glfwHideWindow(window);
|
|
}
|
|
isHidden = isIconified;
|
|
});
|
|
|
|
glfwSetWindowFocusCallback(drawData.window, [](GLFWwindow* window, int isFocused) {
|
|
if (!isFocused && gAppData->settings.docked && !justDocked)
|
|
{
|
|
glfwIconifyWindow(window);
|
|
}
|
|
});
|
|
}
|
|
|
|
void cleanupShell(DrawData& drawData)
|
|
{
|
|
// Remove tray icon
|
|
NOTIFYICONDATA nid = { sizeof(nid) };
|
|
nid.hWnd = drawData.window_handle;
|
|
nid.uID = TRAY_ID;
|
|
Shell_NotifyIcon(NIM_DELETE, &nid);
|
|
}
|
|
|
|
LRESULT CALLBACK trayIconEventHandler(int code, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
if (code >= 0)
|
|
{
|
|
CWPSTRUCT* data = (CWPSTRUCT*)lParam;
|
|
|
|
if (data->message == WMAPP_NOTIFYCALLBACK)
|
|
{
|
|
auto id = HIWORD(data->lParam);
|
|
auto trayEvent = LOWORD(data->lParam);
|
|
|
|
if (id == TRAY_ID && (trayEvent == WM_LBUTTONUP || trayEvent == WM_RBUTTONUP))
|
|
{
|
|
glfwShowWindow(gDrawData->window);
|
|
glfwRestoreWindow(gDrawData->window);
|
|
}
|
|
}
|
|
}
|
|
|
|
return CallNextHookEx(NULL, code, wParam, lParam);
|
|
}
|