Bug 1625953 - P1: Introduce getRelationsByType for mozAccessible. r=morgan

A small refactor to make all of the relations getting easier.

Differential Revision: https://phabricator.services.mozilla.com/D101084

accessible/mac/MacUtils.h

@@ -11,17 +11,18 @@
 #include "nsStringFwd.h"
 
 @class NSString;
+@class mozAccessible;
 
 namespace mozilla {
 namespace a11y {
 namespace utils {
 
 // convert an array of Gecko accessibles to an NSArray of native accessibles
-NSMutableArray* ConvertToNSArray(nsTArray<Accessible*>& aArray);
+NSArray<mozAccessible*>* ConvertToNSArray(nsTArray<Accessible*>& aArray);
 
 // convert an array of Gecko proxy accessibles to an NSArray of native
 // accessibles
-NSMutableArray* ConvertToNSArray(nsTArray<ProxyAccessible*>& aArray);
+NSArray<mozAccessible*>* ConvertToNSArray(nsTArray<ProxyAccessible*>& aArray);
 
 /**
  * Get a localized string from the string bundle.

accessible/mac/MacUtils.mm

@@ -17,7 +17,7 @@ namespace a11y {
 namespace utils {
 
 // convert an array of Gecko accessibles to an NSArray of native accessibles
-NSMutableArray* ConvertToNSArray(nsTArray<Accessible*>& aArray) {
+NSArray<mozAccessible*>* ConvertToNSArray(nsTArray<Accessible*>& aArray) {
   NSMutableArray* nativeArray = [[NSMutableArray alloc] init];
 
   // iterate through the list, and get each native accessible.
@@ -34,7 +34,7 @@ NSMutableArray* ConvertToNSArray(nsTArray<Accessible*>& aArray) {
 
 // convert an array of Gecko proxy accessibles to an NSArray of native
 // accessibles
-NSMutableArray* ConvertToNSArray(nsTArray<ProxyAccessible*>& aArray) {
+NSArray<mozAccessible*>* ConvertToNSArray(nsTArray<ProxyAccessible*>& aArray) {
   NSMutableArray* nativeArray = [[NSMutableArray alloc] init];
 
   // iterate through the list, and get each native accessible.

accessible/mac/mozAccessible.h

@@ -114,6 +114,10 @@ inline mozAccessible* GetNativeFromGeckoAccessible(
 // Get ARIA role
 - (nsStaticAtom*)ARIARole;
 
+// Get array of related mozAccessibles
+- (NSArray<mozAccessible*>*)getRelationsByType:
+    (mozilla::a11y::RelationType)relationType;
+
 #pragma mark - mozAccessible protocol / widget
 
 // override

accessible/mac/mozAccessible.mm

@@ -598,25 +598,17 @@ struct RoleDescrComparator {
     if (flag == eNameFromSubtree) {
       return nil;
     }
-
-    if (![self providesLabelNotTitle]) {
-      Relation rel = acc->RelationByType(RelationType::LABELLED_BY);
-      if (rel.Next() && !rel.Next()) {
-        return nil;
-      }
-    }
   } else if (proxy) {
     uint32_t flag = proxy->Name(name);
     if (flag == eNameFromSubtree) {
       return nil;
     }
+  }
 
-    if (![self providesLabelNotTitle]) {
-      nsTArray<ProxyAccessible*> rels =
-          proxy->RelationByType(RelationType::LABELLED_BY);
-      if (rels.Length() == 1) {
-        return nil;
-      }
+  if (![self providesLabelNotTitle]) {
+    NSArray* relations = [self getRelationsByType:RelationType::LABELLED_BY];
+    if ([relations count] == 1) {
+      return nil;
     }
   }
 
@@ -748,24 +740,9 @@ struct RoleDescrComparator {
 - (id)moxTitleUIElement {
   MOZ_ASSERT(!mGeckoAccessible.IsNull());
 
-  if (Accessible* acc = mGeckoAccessible.AsAccessible()) {
-    Relation rel = acc->RelationByType(RelationType::LABELLED_BY);
-    Accessible* tempAcc = rel.Next();
-    if (tempAcc && !rel.Next()) {
-      mozAccessible* label = GetNativeFromGeckoAccessible(tempAcc);
-      return [label isAccessibilityElement] ? label : nil;
-    }
-
-    return nil;
-  }
-
-  ProxyAccessible* proxy = mGeckoAccessible.AsProxy();
-  nsTArray<ProxyAccessible*> rel =
-      proxy->RelationByType(RelationType::LABELLED_BY);
-  ProxyAccessible* tempProxy = rel.SafeElementAt(0);
-  if (tempProxy && rel.Length() <= 1) {
-    mozAccessible* label = GetNativeFromGeckoAccessible(tempProxy);
-    return [label isAccessibilityElement] ? label : nil;
+  NSArray* relations = [self getRelationsByType:RelationType::LABELLED_BY];
+  if ([relations count] == 1) {
+    return [relations firstObject];
   }
 
   return nil;
@@ -986,6 +963,24 @@ struct RoleDescrComparator {
   }
 }
 
+- (NSArray<mozAccessible*>*)getRelationsByType:(RelationType)relationType {
+  if (Accessible* acc = mGeckoAccessible.AsAccessible()) {
+    NSMutableArray<mozAccessible*>* relations = [[NSMutableArray alloc] init];
+    Relation rel = acc->RelationByType(relationType);
+    while (Accessible* relAcc = rel.Next()) {
+      if (mozAccessible* relNative = GetNativeFromGeckoAccessible(relAcc)) {
+        [relations addObject:relNative];
+      }
+    }
+
+    return relations;
+  }
+
+  ProxyAccessible* proxy = mGeckoAccessible.AsProxy();
+  nsTArray<ProxyAccessible*> rel = proxy->RelationByType(relationType);
+  return utils::ConvertToNSArray(rel);
+}
+
 - (void)handleAccessibleTextChangeEvent:(NSString*)change
                                inserted:(BOOL)isInserted
                             inContainer:(const AccessibleOrProxy&)container

accessible/mac/mozActionElements.mm

@@ -88,21 +88,7 @@ enum CheckboxValue {
   }
 
   if ([attribute isEqualToString:NSAccessibilityLinkedUIElementsAttribute]) {
-    if (HTMLRadioButtonAccessible* radioAcc =
-            (HTMLRadioButtonAccessible*)mGeckoAccessible.AsAccessible()) {
-      NSMutableArray* radioSiblings = [NSMutableArray new];
-      Relation rel = radioAcc->RelationByType(RelationType::MEMBER_OF);
-      Accessible* tempAcc;
-      while ((tempAcc = rel.Next())) {
-        [radioSiblings addObject:GetNativeFromGeckoAccessible(tempAcc)];
-      }
-      return radioSiblings;
-    } else {
-      ProxyAccessible* proxy = mGeckoAccessible.AsProxy();
-      nsTArray<ProxyAccessible*> accs =
-          proxy->RelationByType(RelationType::MEMBER_OF);
-      return utils::ConvertToNSArray(accs);
-    }
+    return [self getRelationsByType:RelationType::MEMBER_OF];
   }
 
   return [super accessibilityAttributeValue:attribute];

accessible/mac/mozTableAccessible.mm

@@ -481,21 +481,11 @@ using namespace mozilla::a11y;
   // all rows are direct children of the outline; they use
   // relations to expose their heirarchy structure.
 
-  mozAccessible* disclosingRow = nil;
   // first we check the relations to see if we're in a xul tree
   // with weird row semantics
-  if (mGeckoAccessible.IsAccessible()) {
-    Relation rel = mGeckoAccessible.AsAccessible()->RelationByType(
-        RelationType::NODE_CHILD_OF);
-    Accessible* maybeParent = rel.Next();
-    disclosingRow =
-        maybeParent ? GetNativeFromGeckoAccessible(maybeParent) : nil;
-  } else {
-    nsTArray<ProxyAccessible*> accs =
-        mGeckoAccessible.AsProxy()->RelationByType(RelationType::NODE_CHILD_OF);
-    disclosingRow =
-        accs.Length() > 0 ? GetNativeFromGeckoAccessible(accs[0]) : nil;
-  }
+  NSArray<mozAccessible*>* disclosingRows =
+      [self getRelationsByType:RelationType::NODE_CHILD_OF];
+  mozAccessible* disclosingRow = [disclosingRows firstObject];
 
   if (disclosingRow) {
     // if we find a row from our relation check,
@@ -507,6 +497,7 @@ using namespace mozilla::a11y;
 
     return disclosingRow;
   }
+
   mozAccessible* parent = (mozAccessible*)[self moxUnignoredParent];
   // otherwise, its likely we're in an aria tree, so we can use
   // these role and subrole checks
@@ -539,24 +530,10 @@ using namespace mozilla::a11y;
   // xul trees so we have to use relations first and then fall-back
   // to the children filter for non-xul outlines.
 
-  NSMutableArray* disclosedRows = [[NSMutableArray alloc] init];
   // first we check the relations to see if we're in a xul tree
   // with weird row semantics
-  if (mGeckoAccessible.IsAccessible()) {
-    Relation rel = mGeckoAccessible.AsAccessible()->RelationByType(
-        RelationType::NODE_PARENT_OF);
-    Accessible* acc = nullptr;
-    while ((acc = rel.Next())) {
-      [disclosedRows addObject:GetNativeFromGeckoAccessible(acc)];
-    }
-  } else {
-    nsTArray<ProxyAccessible*> accs =
-        mGeckoAccessible.AsProxy()->RelationByType(
-            RelationType::NODE_PARENT_OF);
-    disclosedRows = utils::ConvertToNSArray(accs);
-  }
-
-  if (disclosedRows) {
+  if (NSArray* disclosedRows =
+          [self getRelationsByType:RelationType::NODE_PARENT_OF]) {
     // if we find rows from our relation check, return them here
     return disclosedRows;
   }