1 files changed, 98 insertions, 0 deletions
diff --git a/accessible/android/RootAccessibleWrap.cpp b/accessible/android/RootAccessibleWrap.cpp
new file mode 100644
index 0000000000..6a9430da25
--- /dev/null
+++ b/accessible/android/RootAccessibleWrap.cpp
@@ -0,0 +1,98 @@
+/* -*- Mode: c++; c-basic-offset: 2; tab-width: 20; indent-tabs-mode: nil; -*-
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "RootAccessibleWrap.h"
+
+#include "Accessible-inl.h"
+#include "AccessibleOrProxy.h"
+#include "DocAccessibleParent.h"
+#include "DocAccessible-inl.h"
+#include "ProxyAccessibleWrap.h"
+#include "SessionAccessibility.h"
+#include "mozilla/PresShell.h"
+
+using namespace mozilla;
+using namespace mozilla::a11y;
+
+RootAccessibleWrap::RootAccessibleWrap(dom::Document* aDoc,
+                                       PresShell* aPresShell)
+    : RootAccessible(aDoc, aPresShell) {}
+
+RootAccessibleWrap::~RootAccessibleWrap() {}
+
+AccessibleWrap* RootAccessibleWrap::GetContentAccessible() {
+  if (ProxyAccessible* proxy = GetPrimaryRemoteTopLevelContentDoc()) {
+    return WrapperFor(proxy);
+  }
+
+  // Find first document that is not defunct or hidden.
+  // This is exclusively for Fennec which has a deck of browser elements.
+  // Otherwise, standard GeckoView will only have one browser element.
+  for (size_t i = 0; i < ChildDocumentCount(); i++) {
+    DocAccessible* childDoc = GetChildDocumentAt(i);
+    if (childDoc && !childDoc->IsDefunct() && !childDoc->IsHidden()) {
+      return childDoc;
+    }
+  }
+
+  return nullptr;
+}
+
+AccessibleWrap* RootAccessibleWrap::FindAccessibleById(int32_t aID) {
+  AccessibleWrap* contentAcc = GetContentAccessible();
+
+  if (!contentAcc) {
+    return nullptr;
+  }
+
+  if (aID == AccessibleWrap::kNoID) {
+    return contentAcc;
+  }
+
+  if (contentAcc->IsProxy()) {
+    return FindAccessibleById(static_cast<DocProxyAccessibleWrap*>(contentAcc),
+                              aID);
+  }
+
+  return FindAccessibleById(
+      static_cast<DocAccessibleWrap*>(contentAcc->AsDoc()), aID);
+}
+
+AccessibleWrap* RootAccessibleWrap::FindAccessibleById(
+    DocProxyAccessibleWrap* aDoc, int32_t aID) {
+  AccessibleWrap* acc = aDoc->GetAccessibleByID(aID);
+  uint32_t index = 0;
+  while (!acc) {
+    auto child =
+        static_cast<DocProxyAccessibleWrap*>(aDoc->GetChildDocumentAt(index++));
+    if (!child) {
+      break;
+    }
+    // A child document's id is not in its parent document's hash table.
+    if (child->VirtualViewID() == aID) {
+      acc = child;
+    } else {
+      acc = FindAccessibleById(child, aID);
+    }
+  }
+
+  return acc;
+}
+
+AccessibleWrap* RootAccessibleWrap::FindAccessibleById(DocAccessibleWrap* aDoc,
+                                                       int32_t aID) {
+  AccessibleWrap* acc = aDoc->GetAccessibleByID(aID);
+  uint32_t index = 0;
+  while (!acc) {
+    auto child =
+        static_cast<DocAccessibleWrap*>(aDoc->GetChildDocumentAt(index++));
+    if (!child) {
+      break;
+    }
+    acc = FindAccessibleById(child, aID);
+  }
+
+  return acc;
+}