IsAnyClassName

Author: Bush2021

src/uia.cc

@@ -3,6 +3,7 @@
 #include <uiautomation.h>
 #include <wrl/client.h>
 
+#include <initializer_list>
 #include <optional>
 #include <string_view>
 
@@ -170,20 +171,24 @@ ComPtr<IUIAutomationCondition> CreateClassCondition(
 }
 
 bool IsClassName(const ComPtr<IUIAutomationElement>& element,
-                 std::wstring_view class_name_target) {
-  if (!element) {
+                 std::wstring_view expected_class_name) {
+  const auto actual_class_name =
+      GetCurrentStringProperty(element, UIA_ClassNamePropertyId);
+  if (!actual_class_name.has_value()) {
     return false;
   }
+  return actual_class_name.value() == expected_class_name;
+}
 
-  BSTR class_name = nullptr;
-  if (FAILED(element->get_CurrentClassName(&class_name)) || !class_name) {
+bool IsAnyClassName(
+    const ComPtr<IUIAutomationElement>& element,
+    std::initializer_list<std::wstring_view> expected_class_names) {
+  const auto actual_class_name =
+      GetCurrentStringProperty(element, UIA_ClassNamePropertyId);
+  if (!actual_class_name.has_value()) {
     return false;
   }
-
-  const std::wstring_view class_name_view(class_name);
-  const bool is_match = (class_name_view == class_name_target);
-  SysFreeString(class_name);
-  return is_match;
+  return std::ranges::contains(expected_class_names, actual_class_name.value());
 }
 
 ComPtr<IUIAutomationElement> FindFirstInSubtree(
@@ -365,11 +370,10 @@ bool IsOnTabBar(POINT pt) {
   }
 
   // Fast path to avoid `GetParentElement` calls for common tab bar elements.
-  if (IsClassName(pointed, L"HorizontalTabStripRegionView") ||
-      IsClassName(pointed, L"TabStrip::TabDragContextImpl") ||
-      IsClassName(pointed, L"TabStripControlButton") ||
-      IsClassName(pointed, L"FrameGrabHandle") ||
-      IsClassName(pointed, L"ScrollView")) {
+  if (IsAnyClassName(
+          pointed,
+          {L"HorizontalTabStripRegionView", L"TabStrip::TabDragContextImpl",
+           L"TabStripControlButton", L"FrameGrabHandle", L"ScrollView"})) {
     return true;
   }
   return FindParentByClass(automation, pointed,
@@ -401,9 +405,7 @@ bool IsOnBookmark(POINT pt) {
   if (is_blocked_scheme || !looks_like_url) {
     return false;
   }
-
-  return IsClassName(pointed, L"BookmarkButton") ||
-         IsClassName(pointed, L"MenuItemView");
+  return IsAnyClassName(pointed, {L"BookmarkButton", L"MenuItemView"});
 }
 
 bool IsOmniboxFocused() {
@@ -416,6 +418,5 @@ bool IsOmniboxFocused() {
   if (!focused) {
     return false;
   }
-  return IsClassName(focused, L"OmniboxViewViews") ||
-         IsClassName(focused, L"OmniboxResultView");
+  return IsAnyClassName(focused, {L"OmniboxViewViews", L"OmniboxResultView"});
 }