Initial support for native accessibility in ARC

chrome/browser/chromeos/arc/accessibility/arc_accessibility_helper_bridge.cc

 // Copyright 2017 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/chromeos/arc/accessibility/arc_accessibility_helper_bridge.h"
 
+#include "base/command_line.h"
 #include "chrome/browser/chromeos/accessibility/accessibility_manager.h"
+#include "chrome/browser/chromeos/arc/accessibility/ax_tree_source_arc.h"
+#include "chromeos/chromeos_switches.h"
 #include "components/arc/arc_bridge_service.h"
+#include "components/exo/shell_surface.h"
+#include "components/exo/surface.h"
 #include "components/exo/wm_helper.h"
+#include "ui/aura/client/aura_constants.h"
 #include "ui/aura/window.h"
 #include "ui/gfx/geometry/rect.h"
+#include "ui/views/view.h"
+#include "ui/views/widget/widget.h"
+
+namespace {
+
+exo::Surface* GetArcSurface(const aura::Window* window) {
+  if (!window)
+    return nullptr;
+
+  exo::Surface* arc_surface = exo::Surface::AsSurface(window);
+  if (!arc_surface)
+    arc_surface = exo::ShellSurface::GetMainSurface(window);
+  return arc_surface;
+}
+
+void DispatchFocusChange(arc::mojom::AccessibilityNodeInfoData* node_data) {
+  chromeos::AccessibilityManager* accessibility_manager =
+      chromeos::AccessibilityManager::Get();
+  if (!accessibility_manager)
+    return;
+
+  exo::WMHelper* wmHelper = exo::WMHelper::GetInstance();
+  aura::Window* focused_window = wmHelper->GetFocusedWindow();
+  if (!focused_window)
+    return;
+
+  aura::Window* toplevel_window = focused_window->GetToplevelWindow();
+
+  gfx::Rect bounds_in_screen = gfx::ScaleToEnclosingRect(
+      node_data->boundsInScreen,
+      1.0f / toplevel_window->layer()->device_scale_factor());
+
+  accessibility_manager->OnViewFocusedInArc(bounds_in_screen);
+}
+
+}  // namespace
 
 namespace arc {
 
 ArcAccessibilityHelperBridge::ArcAccessibilityHelperBridge(
     ArcBridgeService* bridge_service)
     : ArcService(bridge_service), binding_(this) {
   arc_bridge_service()->accessibility_helper()->AddObserver(this);
 }
 
 ArcAccessibilityHelperBridge::~ArcAccessibilityHelperBridge() {
   arc_bridge_service()->accessibility_helper()->RemoveObserver(this);
 }
 
 void ArcAccessibilityHelperBridge::OnInstanceReady() {
   auto* instance = ARC_GET_INSTANCE_FOR_METHOD(
       arc_bridge_service()->accessibility_helper(), Init);
   DCHECK(instance);
   instance->Init(binding_.CreateInterfacePtrAndBind());
+
+  chromeos::AccessibilityManager* accessibility_manager =
+      chromeos::AccessibilityManager::Get();
+  if (base::CommandLine::ForCurrentProcess()->HasSwitch(
+          chromeos::switches::kEnableChromeVoxArcSupport)) {
+    instance->SetFilter(arc::mojom::AccessibilityFilterType::ALL);
+    if (!tree_source_) {
+      tree_source_.reset(new AXTreeSourceArc());
+      exo::WMHelper::GetInstance()->AddActivationObserver(this);
+    }
+  } else if (accessibility_manager &&
+             accessibility_manager->IsFocusHighlightEnabled()) {
+    instance->SetFilter(arc::mojom::AccessibilityFilterType::FOCUS);
+  }
+}
+
+void ArcAccessibilityHelperBridge::OnAccessibilityEventDeprecated(
+    mojom::AccessibilityEventType event_type,
+    mojom::AccessibilityNodeInfoDataPtr event_source) {
+  if (event_type == arc::mojom::AccessibilityEventType::VIEW_FOCUSED)
+    DispatchFocusChange(event_source.get());
 }
 
 void ArcAccessibilityHelperBridge::OnAccessibilityEvent(
-    mojom::AccessibilityEventType event_type,
-    mojom::AccessibilityNodeInfoDataPtr event_source) {
-  if (event_type != mojom::AccessibilityEventType::VIEW_FOCUSED ||
-      event_source.is_null()) {
+    mojom::AccessibilityEventDataPtr event_data) {
+  if (tree_source_) {
+    tree_source_->NotifyAccessibilityEvent(event_data.get());
     return;
   }
 
-  chromeos::AccessibilityManager* accessibility_manager =
-      chromeos::AccessibilityManager::Get();
+  if (event_data->eventType != arc::mojom::AccessibilityEventType::VIEW_FOCUSED)
+    return;
 
-  if (!accessibility_manager ||
-      !accessibility_manager->IsFocusHighlightEnabled()) {
+  CHECK_EQ(1U, event_data.get()->nodeData.size());
+  DispatchFocusChange(event_data.get()->nodeData[0].get());
+}
+
+void ArcAccessibilityHelperBridge::OnWindowActivated(
+    aura::Window* gained_active,
+    aura::Window* lost_active) {
+  if (gained_active == lost_active)
     return;
+
+  exo::Surface* active_surface = GetArcSurface(gained_active);
+  exo::Surface* inactive_surface = GetArcSurface(lost_active);
+
+  // Detach the accessibility tree from an inactive ShellSurface so that any
+  // client walking the desktop tree gets non-duplicated linearization.
+  if (inactive_surface) {
+    views::Widget* widget = views::Widget::GetWidgetForNativeView(lost_active);
+    if (widget && widget->GetContentsView()) {
+      widget->GetNativeWindow()->SetProperty(
+          aura::client::kAccessibleChildTreeIdKey, 0);
+      views::View* view = widget->GetContentsView();
+      view->SetFocusBehavior(views::View::FocusBehavior::NEVER);
+      view->NotifyAccessibilityEvent(ui::AX_EVENT_CHILDREN_CHANGED, false);
+    }
   }
 
-  exo::WMHelper* wmHelper = exo::WMHelper::GetInstance();
-
-  aura::Window* focused_window = wmHelper->GetFocusedWindow();
-  if (!focused_window)
+  if (!active_surface)
     return;
 
-  aura::Window* toplevel_window = focused_window->GetToplevelWindow();
-
-  gfx::Rect bounds_in_screen = gfx::ScaleToEnclosingRect(
-      event_source.get()->boundsInScreen,
-      1.0f / toplevel_window->layer()->device_scale_factor());
-
-  accessibility_manager->OnViewFocusedInArc(bounds_in_screen);
+  views::Widget* widget = views::Widget::GetWidgetForNativeView(gained_active);
+  if (widget && widget->GetContentsView()) {
+    widget->GetNativeWindow()->SetProperty(
+        aura::client::kAccessibleChildTreeIdKey, tree_source_->tree_id());
+    views::View* view = widget->GetContentsView();
+    view->SetFocusBehavior(views::View::FocusBehavior::ALWAYS);
+    view->RequestFocus();
+    view->NotifyAccessibilityEvent(ui::AX_EVENT_CHILDREN_CHANGED, false);
+  }
 }
 
 }  // namespace arc