mirror of
https://github.com/rustdesk/rustdesk.git
synced 2024-12-05 21:29:05 +08:00
260 lines
7.3 KiB
C++
260 lines
7.3 KiB
C++
#include "win32_window.h"
|
|
|
|
#include <flutter_windows.h>
|
|
#include <shobjidl_core.h>
|
|
|
|
#include "resource.h"
|
|
|
|
namespace {
|
|
|
|
constexpr const wchar_t kWindowClassName[] = L"FLUTTER_RUNNER_WIN32_WINDOW";
|
|
|
|
// The number of Win32Window objects that currently exist.
|
|
static int g_active_window_count = 0;
|
|
|
|
using EnableNonClientDpiScaling = BOOL __stdcall(HWND hwnd);
|
|
|
|
// Scale helper to convert logical scaler values to physical using passed in
|
|
// scale factor
|
|
int Scale(int source, double scale_factor) {
|
|
return static_cast<int>(source * scale_factor);
|
|
}
|
|
|
|
// Dynamically loads the |EnableNonClientDpiScaling| from the User32 module.
|
|
// This API is only needed for PerMonitor V1 awareness mode.
|
|
void EnableFullDpiSupportIfAvailable(HWND hwnd) {
|
|
HMODULE user32_module = LoadLibraryA("User32.dll");
|
|
if (!user32_module) {
|
|
return;
|
|
}
|
|
auto enable_non_client_dpi_scaling =
|
|
reinterpret_cast<EnableNonClientDpiScaling*>(
|
|
GetProcAddress(user32_module, "EnableNonClientDpiScaling"));
|
|
if (enable_non_client_dpi_scaling != nullptr) {
|
|
enable_non_client_dpi_scaling(hwnd);
|
|
FreeLibrary(user32_module);
|
|
}
|
|
}
|
|
|
|
} // namespace
|
|
|
|
// Manages the Win32Window's window class registration.
|
|
class WindowClassRegistrar {
|
|
public:
|
|
~WindowClassRegistrar() = default;
|
|
|
|
// Returns the singleton registar instance.
|
|
static WindowClassRegistrar* GetInstance() {
|
|
if (!instance_) {
|
|
instance_ = new WindowClassRegistrar();
|
|
}
|
|
return instance_;
|
|
}
|
|
|
|
// Returns the name of the window class, registering the class if it hasn't
|
|
// previously been registered.
|
|
const wchar_t* GetWindowClass();
|
|
|
|
// Unregisters the window class. Should only be called if there are no
|
|
// instances of the window.
|
|
void UnregisterWindowClass();
|
|
|
|
private:
|
|
WindowClassRegistrar() = default;
|
|
|
|
static WindowClassRegistrar* instance_;
|
|
|
|
bool class_registered_ = false;
|
|
};
|
|
|
|
WindowClassRegistrar* WindowClassRegistrar::instance_ = nullptr;
|
|
|
|
const wchar_t* WindowClassRegistrar::GetWindowClass() {
|
|
if (!class_registered_) {
|
|
WNDCLASS window_class{};
|
|
window_class.hCursor = LoadCursor(nullptr, IDC_ARROW);
|
|
window_class.lpszClassName = kWindowClassName;
|
|
window_class.style = CS_HREDRAW | CS_VREDRAW;
|
|
window_class.cbClsExtra = 0;
|
|
window_class.cbWndExtra = 0;
|
|
window_class.hInstance = GetModuleHandle(nullptr);
|
|
window_class.hIcon =
|
|
LoadIcon(window_class.hInstance, MAKEINTRESOURCE(IDI_APP_ICON));
|
|
window_class.hbrBackground = 0;
|
|
window_class.lpszMenuName = nullptr;
|
|
window_class.lpfnWndProc = Win32Window::WndProc;
|
|
RegisterClass(&window_class);
|
|
class_registered_ = true;
|
|
}
|
|
return kWindowClassName;
|
|
}
|
|
|
|
void WindowClassRegistrar::UnregisterWindowClass() {
|
|
UnregisterClass(kWindowClassName, nullptr);
|
|
class_registered_ = false;
|
|
}
|
|
|
|
Win32Window::Win32Window() {
|
|
++g_active_window_count;
|
|
}
|
|
|
|
Win32Window::~Win32Window() {
|
|
--g_active_window_count;
|
|
Destroy();
|
|
}
|
|
|
|
bool Win32Window::CreateAndShow(const std::wstring& title,
|
|
const Point& origin,
|
|
const Size& size, bool showOnTaskBar) {
|
|
Destroy();
|
|
|
|
const wchar_t* window_class =
|
|
WindowClassRegistrar::GetInstance()->GetWindowClass();
|
|
|
|
const POINT target_point = {static_cast<LONG>(origin.x),
|
|
static_cast<LONG>(origin.y)};
|
|
HMONITOR monitor = MonitorFromPoint(target_point, MONITOR_DEFAULTTONEAREST);
|
|
UINT dpi = FlutterDesktopGetDpiForMonitor(monitor);
|
|
double scale_factor = dpi / 96.0;
|
|
|
|
HWND window = CreateWindow(
|
|
window_class, title.c_str(), WS_OVERLAPPEDWINDOW,
|
|
Scale(origin.x, scale_factor), Scale(origin.y, scale_factor),
|
|
Scale(size.width, scale_factor), Scale(size.height, scale_factor),
|
|
nullptr, nullptr, GetModuleHandle(nullptr), this);
|
|
|
|
if (!window) {
|
|
return false;
|
|
}
|
|
|
|
if (!showOnTaskBar) {
|
|
// hide from taskbar
|
|
HRESULT hr;
|
|
ITaskbarList* pTaskbarList;
|
|
hr = CoCreateInstance(CLSID_TaskbarList, NULL, CLSCTX_INPROC_SERVER,IID_ITaskbarList,(void**)&pTaskbarList);
|
|
if (FAILED(hr)) {
|
|
return false;
|
|
}
|
|
hr = pTaskbarList->HrInit();
|
|
hr = pTaskbarList->DeleteTab(window);
|
|
hr = pTaskbarList->Release();
|
|
}
|
|
|
|
return OnCreate();
|
|
}
|
|
|
|
// static
|
|
LRESULT CALLBACK Win32Window::WndProc(HWND const window,
|
|
UINT const message,
|
|
WPARAM const wparam,
|
|
LPARAM const lparam) noexcept {
|
|
if (message == WM_NCCREATE) {
|
|
auto window_struct = reinterpret_cast<CREATESTRUCT*>(lparam);
|
|
SetWindowLongPtr(window, GWLP_USERDATA,
|
|
reinterpret_cast<LONG_PTR>(window_struct->lpCreateParams));
|
|
|
|
auto that = static_cast<Win32Window*>(window_struct->lpCreateParams);
|
|
EnableFullDpiSupportIfAvailable(window);
|
|
that->window_handle_ = window;
|
|
} else if (Win32Window* that = GetThisFromHandle(window)) {
|
|
return that->MessageHandler(window, message, wparam, lparam);
|
|
}
|
|
|
|
return DefWindowProc(window, message, wparam, lparam);
|
|
}
|
|
|
|
LRESULT
|
|
Win32Window::MessageHandler(HWND hwnd,
|
|
UINT const message,
|
|
WPARAM const wparam,
|
|
LPARAM const lparam) noexcept {
|
|
switch (message) {
|
|
case WM_DESTROY:
|
|
window_handle_ = nullptr;
|
|
Destroy();
|
|
if (quit_on_close_) {
|
|
PostQuitMessage(0);
|
|
}
|
|
return 0;
|
|
|
|
case WM_DPICHANGED: {
|
|
auto newRectSize = reinterpret_cast<RECT*>(lparam);
|
|
LONG newWidth = newRectSize->right - newRectSize->left;
|
|
LONG newHeight = newRectSize->bottom - newRectSize->top;
|
|
|
|
SetWindowPos(hwnd, nullptr, newRectSize->left, newRectSize->top, newWidth,
|
|
newHeight, SWP_NOZORDER | SWP_NOACTIVATE);
|
|
|
|
return 0;
|
|
}
|
|
case WM_SIZE: {
|
|
RECT rect = GetClientArea();
|
|
if (child_content_ != nullptr) {
|
|
// Size and position the child window.
|
|
MoveWindow(child_content_, rect.left, rect.top, rect.right - rect.left,
|
|
rect.bottom - rect.top, TRUE);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
case WM_ACTIVATE:
|
|
if (child_content_ != nullptr) {
|
|
SetFocus(child_content_);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
return DefWindowProc(window_handle_, message, wparam, lparam);
|
|
}
|
|
|
|
void Win32Window::Destroy() {
|
|
OnDestroy();
|
|
|
|
if (window_handle_) {
|
|
DestroyWindow(window_handle_);
|
|
window_handle_ = nullptr;
|
|
}
|
|
if (g_active_window_count == 0) {
|
|
WindowClassRegistrar::GetInstance()->UnregisterWindowClass();
|
|
}
|
|
}
|
|
|
|
Win32Window* Win32Window::GetThisFromHandle(HWND const window) noexcept {
|
|
return reinterpret_cast<Win32Window*>(
|
|
GetWindowLongPtr(window, GWLP_USERDATA));
|
|
}
|
|
|
|
void Win32Window::SetChildContent(HWND content) {
|
|
child_content_ = content;
|
|
SetParent(content, window_handle_);
|
|
RECT frame = GetClientArea();
|
|
|
|
MoveWindow(content, frame.left, frame.top, frame.right - frame.left,
|
|
frame.bottom - frame.top, true);
|
|
|
|
SetFocus(child_content_);
|
|
}
|
|
|
|
RECT Win32Window::GetClientArea() {
|
|
RECT frame;
|
|
GetClientRect(window_handle_, &frame);
|
|
return frame;
|
|
}
|
|
|
|
HWND Win32Window::GetHandle() {
|
|
return window_handle_;
|
|
}
|
|
|
|
void Win32Window::SetQuitOnClose(bool quit_on_close) {
|
|
quit_on_close_ = quit_on_close;
|
|
}
|
|
|
|
bool Win32Window::OnCreate() {
|
|
// No-op; provided for subclasses.
|
|
return true;
|
|
}
|
|
|
|
void Win32Window::OnDestroy() {
|
|
// No-op; provided for subclasses.
|
|
}
|