dolphin/Source/Core/WinUpdater/WinUI.cpp

267 lines
6.4 KiB
C++
Raw Normal View History

2018-03-27 22:28:40 +00:00
// Copyright 2018 Dolphin Emulator Project
// Licensed under GPLv2+
// Refer to the license.txt file included.
#include "UpdaterCommon/UI.h"
2018-03-27 22:28:40 +00:00
#include <string>
#include <thread>
#include <Windows.h>
2018-03-27 22:28:40 +00:00
#include <CommCtrl.h>
2018-11-08 10:54:56 +00:00
#include <ShObjIdl.h>
#include <shellapi.h>
2018-03-27 22:28:40 +00:00
#include "Common/StringUtil.h"
namespace
{
HWND window_handle = nullptr;
HWND label_handle = nullptr;
2018-11-15 08:01:29 +00:00
HWND total_progressbar_handle = nullptr;
HWND current_progressbar_handle = nullptr;
2018-11-08 10:54:56 +00:00
ITaskbarList3* taskbar_list = nullptr;
2018-03-27 22:28:40 +00:00
std::thread ui_thread;
2019-03-03 11:56:26 +00:00
int GetWindowHeight(HWND hwnd)
{
RECT rect;
GetWindowRect(hwnd, &rect);
return rect.bottom - rect.top;
}
LRESULT CALLBACK WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
switch (uMsg)
{
case WM_DESTROY:
PostQuitMessage(0);
return 0;
}
return DefWindowProc(hwnd, uMsg, wParam, lParam);
}
2018-03-27 22:28:40 +00:00
}; // namespace
constexpr int PROGRESSBAR_FLAGS = WS_VISIBLE | WS_CHILD | PBS_SMOOTH | PBS_SMOOTHREVERSE;
constexpr int WINDOW_FLAGS = WS_CLIPCHILDREN;
2019-03-03 11:56:26 +00:00
constexpr int PADDING_HEIGHT = 5;
2018-03-27 22:28:40 +00:00
namespace UI
{
bool InitWindow()
2018-03-27 22:28:40 +00:00
{
InitCommonControls();
WNDCLASS wndcl = {};
wndcl.lpfnWndProc = WindowProc;
2018-03-27 22:28:40 +00:00
wndcl.hbrBackground = GetSysColorBrush(COLOR_MENU);
wndcl.lpszClassName = L"UPDATER";
if (!RegisterClass(&wndcl))
return false;
window_handle =
2019-03-03 11:56:26 +00:00
CreateWindow(L"UPDATER", L"Dolphin Updater", WINDOW_FLAGS, CW_USEDEFAULT, CW_USEDEFAULT, 500,
100, nullptr, nullptr, GetModuleHandle(nullptr), 0);
2018-03-27 22:28:40 +00:00
if (!window_handle)
return false;
if (FAILED(CoCreateInstance(CLSID_TaskbarList, NULL, CLSCTX_INPROC_SERVER,
IID_PPV_ARGS(&taskbar_list))))
{
taskbar_list = nullptr;
}
if (taskbar_list && FAILED(taskbar_list->HrInit()))
{
taskbar_list->Release();
taskbar_list = nullptr;
}
2018-11-08 10:54:56 +00:00
2019-03-03 11:56:26 +00:00
int y = PADDING_HEIGHT;
label_handle = CreateWindow(L"STATIC", NULL, WS_VISIBLE | WS_CHILD, 5, y, 500, 25, window_handle,
2018-03-27 22:28:40 +00:00
nullptr, nullptr, 0);
if (!label_handle)
return false;
// Get the default system font
NONCLIENTMETRICS metrics = {};
metrics.cbSize = sizeof(NONCLIENTMETRICS);
if (!SystemParametersInfo(SPI_GETNONCLIENTMETRICS, sizeof(metrics), &metrics, 0))
return false;
SendMessage(label_handle, WM_SETFONT,
reinterpret_cast<WPARAM>(CreateFontIndirect(&metrics.lfMessageFont)), 0);
2019-03-03 11:56:26 +00:00
y += GetWindowHeight(label_handle) + PADDING_HEIGHT;
total_progressbar_handle = CreateWindow(PROGRESS_CLASS, NULL, PROGRESSBAR_FLAGS, 5, y, 470, 25,
2018-11-15 08:01:29 +00:00
window_handle, nullptr, nullptr, 0);
2018-03-27 22:28:40 +00:00
2019-03-03 11:56:26 +00:00
y += GetWindowHeight(total_progressbar_handle) + PADDING_HEIGHT;
2018-11-15 08:01:29 +00:00
if (!total_progressbar_handle)
return false;
2019-03-03 11:56:26 +00:00
current_progressbar_handle = CreateWindow(PROGRESS_CLASS, NULL, PROGRESSBAR_FLAGS, 5, y, 470, 25,
2018-11-15 08:01:29 +00:00
window_handle, nullptr, nullptr, 0);
2019-03-03 11:56:26 +00:00
y += GetWindowHeight(current_progressbar_handle) + PADDING_HEIGHT;
2018-11-15 08:01:29 +00:00
if (!current_progressbar_handle)
2018-03-27 22:28:40 +00:00
return false;
2019-03-03 11:56:26 +00:00
RECT rect;
GetWindowRect(window_handle, &rect);
// Account for the title bar
y += GetSystemMetrics(SM_CYFRAME) + GetSystemMetrics(SM_CYCAPTION) +
GetSystemMetrics(SM_CXPADDEDBORDER);
// ...and window border
y += GetSystemMetrics(SM_CYBORDER);
// Add some padding for good measure
y += PADDING_HEIGHT * 3;
SetWindowPos(window_handle, HWND_TOPMOST, 0, 0, rect.right - rect.left, y, SWP_NOMOVE);
2018-03-27 22:28:40 +00:00
return true;
}
2018-11-15 08:01:29 +00:00
void SetTotalMarquee(bool marquee)
2018-03-27 22:28:40 +00:00
{
2018-11-15 08:01:29 +00:00
SetWindowLong(total_progressbar_handle, GWL_STYLE,
PROGRESSBAR_FLAGS | (marquee ? PBS_MARQUEE : 0));
SendMessage(total_progressbar_handle, PBM_SETMARQUEE, marquee, 0);
if (taskbar_list)
{
taskbar_list->SetProgressState(window_handle, marquee ? TBPF_INDETERMINATE : TBPF_NORMAL);
}
2018-03-27 22:28:40 +00:00
}
2018-11-15 08:01:29 +00:00
void ResetTotalProgress()
{
SendMessage(total_progressbar_handle, PBM_SETPOS, 0, 0);
SetCurrentMarquee(true);
}
void SetTotalProgress(int current, int total)
{
SendMessage(total_progressbar_handle, PBM_SETRANGE32, 0, total);
SendMessage(total_progressbar_handle, PBM_SETPOS, current, 0);
if (taskbar_list)
{
taskbar_list->SetProgressValue(window_handle, current, total);
}
2018-11-15 08:01:29 +00:00
}
void SetCurrentMarquee(bool marquee)
2018-03-27 22:28:40 +00:00
{
2018-11-15 08:01:29 +00:00
SetWindowLong(current_progressbar_handle, GWL_STYLE,
PROGRESSBAR_FLAGS | (marquee ? PBS_MARQUEE : 0));
SendMessage(current_progressbar_handle, PBM_SETMARQUEE, marquee, 0);
}
void ResetCurrentProgress()
{
SendMessage(current_progressbar_handle, PBM_SETPOS, 0, 0);
SetCurrentMarquee(true);
2018-03-27 22:28:40 +00:00
}
2018-11-08 10:54:56 +00:00
void Error(const std::string& text)
2018-03-27 22:28:40 +00:00
{
2018-11-08 10:54:56 +00:00
auto wide_text = UTF8ToUTF16(text);
MessageBox(nullptr,
(L"A fatal error occured and the updater cannot continue:\n " + wide_text).c_str(),
L"Error", MB_ICONERROR);
if (taskbar_list)
{
taskbar_list->SetProgressState(window_handle, TBPF_ERROR);
}
2018-03-27 22:28:40 +00:00
}
2018-11-15 08:01:29 +00:00
void SetCurrentProgress(int current, int total)
2018-03-27 22:28:40 +00:00
{
2018-11-15 08:01:29 +00:00
SendMessage(current_progressbar_handle, PBM_SETRANGE32, 0, total);
SendMessage(current_progressbar_handle, PBM_SETPOS, current, 0);
2018-03-27 22:28:40 +00:00
}
void SetDescription(const std::string& text)
{
SetWindowText(label_handle, UTF8ToUTF16(text).c_str());
}
void MessageLoop()
{
if (!InitWindow())
2018-03-27 22:28:40 +00:00
{
MessageBox(nullptr, L"Window init failed!", L"", MB_ICONERROR);
// Destroying the parent (if exists) destroys all children windows
if (window_handle)
{
DestroyWindow(window_handle);
window_handle = nullptr;
}
return;
2018-03-27 22:28:40 +00:00
}
2018-11-15 08:01:29 +00:00
SetTotalMarquee(true);
SetCurrentMarquee(true);
2018-03-27 22:28:40 +00:00
MSG msg;
while (GetMessage(&msg, nullptr, 0, 0))
2018-03-27 22:28:40 +00:00
{
TranslateMessage(&msg);
DispatchMessage(&msg);
2018-03-27 22:28:40 +00:00
}
}
void Init()
{
ui_thread = std::thread(MessageLoop);
}
2018-03-27 22:28:40 +00:00
void Stop()
{
if (window_handle)
PostMessage(window_handle, WM_CLOSE, 0, 0);
2018-03-27 22:28:40 +00:00
ui_thread.join();
2018-03-27 22:28:40 +00:00
}
void LaunchApplication(std::string path)
{
// Hack: Launching the updater over the explorer ensures that admin priviliges are dropped. Why?
// Ask Microsoft.
ShellExecuteW(nullptr, nullptr, L"explorer.exe", UTF8ToUTF16(path).c_str(), nullptr, SW_SHOW);
}
void Sleep(int sleep)
{
::Sleep(sleep * 1000);
}
void WaitForPID(u32 pid)
{
HANDLE parent_handle = OpenProcess(SYNCHRONIZE, FALSE, static_cast<DWORD>(pid));
if (parent_handle)
{
WaitForSingleObject(parent_handle, INFINITE);
CloseHandle(parent_handle);
}
}
void SetVisible(bool visible)
{
ShowWindow(window_handle, visible ? SW_SHOW : SW_HIDE);
}
2018-03-27 22:28:40 +00:00
}; // namespace UI