1 files changed, 53 insertions, 0 deletions
diff --git a/src/platform_win/window_manager.cpp b/src/platform_win/window_manager.cpp
new file mode 100644
index 00000000..62a73499
--- /dev/null
+++ b/src/platform_win/window_manager.cpp
@@ -0,0 +1,53 @@
+#include "window_manager.hpp"
+
+#include "cru/platform/win/win_application.hpp"
+#include "cru/platform/win/win_native_window.hpp"
+
+#include <assert.h>
+
+namespace cru::platform::win {
+LRESULT __stdcall GeneralWndProc(HWND hWnd, UINT Msg, WPARAM wParam,
+                                 LPARAM lParam) {
+  auto window = WinApplication::GetInstance()->GetWindowManager()->FromHandle(hWnd);
+
+  LRESULT result;
+  if (window != nullptr &&
+      window->HandleNativeWindowMessage(hWnd, Msg, wParam, lParam, &result))
+    return result;
+
+  return DefWindowProc(hWnd, Msg, wParam, lParam);
+}
+
+WindowManager::WindowManager(WinApplication* application) {
+  application_ = application;
+  general_window_class_ = std::make_shared<WindowClass>(
+      L"CruUIWindowClass", GeneralWndProc,
+      application->GetInstanceHandle());
+}
+
+void WindowManager::RegisterWindow(HWND hwnd, WinNativeWindow* window) {
+  assert(window_map_.count(hwnd) == 0);  // The hwnd is already in the map.
+  window_map_.emplace(hwnd, window);
+}
+
+void WindowManager::UnregisterWindow(HWND hwnd) {
+  const auto find_result = window_map_.find(hwnd);
+  assert(find_result != window_map_.end());  // The hwnd is not in the map.
+  window_map_.erase(find_result);
+  if (window_map_.empty()) application_->Quit(0);
+}
+
+WinNativeWindow* WindowManager::FromHandle(HWND hwnd) {
+  const auto find_result = window_map_.find(hwnd);
+  if (find_result == window_map_.end())
+    return nullptr;
+  else
+    return find_result->second;
+}
+
+std::vector<WinNativeWindow*> WindowManager::GetAllWindows() const {
+  std::vector<WinNativeWindow*> windows;
+  for (auto [key, value] : window_map_) windows.push_back(value);
+  return windows;
+}
+}