add UIA properties

microsoft · Oct 3, 2022 · 9fd73c7 · 9fd73c7
1 parent bf6f0ff
commit 9fd73c7
Show file tree

Hide file tree

Showing 2 changed files with 54 additions and 0 deletions.
diff --git a/src/types/TermControlUiaProvider.cpp b/src/types/TermControlUiaProvider.cpp
@@ -19,6 +19,56 @@ HRESULT TermControlUiaProvider::RuntimeClassInitialize(_In_ ::Microsoft::Console
     return S_OK;
 }
 
+// Implementation of IRawElementProviderSimple::get_PropertyValue.
+// Gets custom properties.
+IFACEMETHODIMP TermControlUiaProvider::GetPropertyValue(_In_ PROPERTYID propertyId,
+                                                        _Out_ VARIANT* pVariant) noexcept
+{
+    pVariant->vt = VT_EMPTY;
+
+    // Returning the default will leave the property as the default
+    // so we only really need to touch it for the properties we want to implement
+    if (propertyId == UIA_ClassNamePropertyId)
+    {
+        pVariant->bstrVal = SysAllocString(L"TermControl");
+        if (pVariant->bstrVal != nullptr)
+        {
+            pVariant->vt = VT_BSTR;
+        }
+    }
+    else if (propertyId == UIA_ControlTypePropertyId)
+    {
+        pVariant->vt = VT_I4;
+        pVariant->lVal = UIA_TextControlTypeId;
+    }
+    else if (propertyId == UIA_LocalizedControlTypePropertyId)
+    {
+        // TODO: we should use RS_(L"TerminalControl_ControlType"),
+        // but that's exposed/defined in the TermControl project
+        pVariant->bstrVal = SysAllocString(L"terminal");
+        if (pVariant->bstrVal != nullptr)
+        {
+            pVariant->vt = VT_BSTR;
+        }
+    }
+    else if (propertyId == UIA_OrientationPropertyId)
+    {
+        pVariant->vt = VT_I4;
+        pVariant->lVal = OrientationType_Vertical;
+    }
+    else if (propertyId == UIA_LiveSettingPropertyId)
+    {
+        pVariant->vt = VT_I4;
+        pVariant->lVal = LiveSetting::Polite;
+    }
+    else
+    {
+        // fall back to the shared implementation
+        ScreenInfoUiaProviderBase::GetPropertyValue(propertyId, pVariant);
+    }
+    return S_OK;
+}
+
 IFACEMETHODIMP TermControlUiaProvider::Navigate(_In_ NavigateDirection direction,
                                                 _COM_Outptr_result_maybenull_ IRawElementProviderFragment** ppProvider) noexcept
 {

diff --git a/src/types/TermControlUiaProvider.hpp b/src/types/TermControlUiaProvider.hpp
@@ -32,6 +32,10 @@ namespace Microsoft::Terminal
         HRESULT RuntimeClassInitialize(_In_ ::Microsoft::Console::Types::IUiaData* const uiaData,
                                        _In_ ::Microsoft::Console::Types::IControlAccessibilityInfo* controlInfo) noexcept;
 
+        // IRawElementProviderSimple methods
+        IFACEMETHODIMP GetPropertyValue(_In_ PROPERTYID idProp,
+                                        _Out_ VARIANT* pVariant) noexcept override;
+
         // IRawElementProviderFragment methods
         IFACEMETHODIMP Navigate(_In_ NavigateDirection direction,
                                 _COM_Outptr_result_maybenull_ IRawElementProviderFragment** ppProvider) noexcept override;