From d4ca08c4c7a880971f0677761c60807f7c6e68db Mon Sep 17 00:00:00 2001
From: jpark37 <jpark37@users.noreply.github.com>
Date: Wed, 27 Nov 2019 16:42:49 -0800
Subject: [PATCH] win-capture: Add support for finding windows via EnumWindows

The top-level window appears to be preferred for upcoming Windows
Graphics Capture support, at least for UWP apps.
---
 plugins/win-capture/window-helpers.c | 49 ++++++++++++++++++++++++++++
 plugins/win-capture/window-helpers.h |  4 +++
 2 files changed, 53 insertions(+)

diff --git a/plugins/win-capture/window-helpers.c b/plugins/win-capture/window-helpers.c
index 8f19ee48b..8b578e72b 100644
--- a/plugins/win-capture/window-helpers.c
+++ b/plugins/win-capture/window-helpers.c
@@ -424,3 +424,52 @@ HWND find_window(enum window_search_mode mode, enum window_priority priority,
 
 	return best_window;
 }
+
+struct top_level_enum_data {
+	enum window_search_mode mode;
+	enum window_priority priority;
+	const char *class;
+	const char *title;
+	const char *exe;
+	bool uwp_window;
+	HWND best_window;
+	int best_rating;
+};
+
+BOOL CALLBACK enum_windows_proc(HWND window, LPARAM lParam)
+{
+	struct top_level_enum_data *data = (struct top_level_enum_data *)lParam;
+
+	if (!check_window_valid(window, data->mode))
+		return TRUE;
+
+	const int rating = window_rating(window, data->priority, data->class,
+					 data->title, data->exe,
+					 data->uwp_window);
+	if (rating < data->best_rating) {
+		data->best_rating = rating;
+		data->best_window = window;
+	}
+
+	return rating > 0;
+}
+
+HWND find_window_top_level(enum window_search_mode mode,
+			   enum window_priority priority, const char *class,
+			   const char *title, const char *exe)
+{
+	if (!class)
+		return NULL;
+
+	struct top_level_enum_data data;
+	data.mode = mode;
+	data.priority = priority;
+	data.class = class;
+	data.title = title;
+	data.exe = exe;
+	data.uwp_window = strcmp(class, "Windows.UI.Core.CoreWindow") == 0;
+	data.best_window = NULL;
+	data.best_rating = 0x7FFFFFFF;
+	EnumWindows(enum_windows_proc, (LPARAM)&data);
+	return data.best_window;
+}
diff --git a/plugins/win-capture/window-helpers.h b/plugins/win-capture/window-helpers.h
index ffcd7cefc..1537b57a3 100644
--- a/plugins/win-capture/window-helpers.h
+++ b/plugins/win-capture/window-helpers.h
@@ -31,3 +31,7 @@ extern void build_window_strings(const char *str, char **class, char **title,
 extern HWND find_window(enum window_search_mode mode,
 			enum window_priority priority, const char *class,
 			const char *title, const char *exe);
+extern HWND find_window_top_level(enum window_search_mode mode,
+				  enum window_priority priority,
+				  const char *class, const char *title,
+				  const char *exe);