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/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;
+}
+
+}  // 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_) {
+      tree_.reset(new AXTreeSourceArc());
+      exo::WMHelper::GetInstance()->AddActivationObserver(this);
+    }
+  } else if (accessibility_manager &&
+             accessibility_manager->IsFocusHighlightEnabled()) {
+    instance->SetFilter(arc::mojom::AccessibilityFilterType::FOCUS);
+  }
 }
 
 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_) {
+    tree_->NotifyAccessibilityEvent(event_data.get());
+  } else {

[dmazzoni, 2017/02/09 01:18:09]

nit: maybe just return inside the first if block, to avoid indenting the rest

+    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();
+
+    CHECK_EQ(1, event_data.get()->nodeData.size());
+    gfx::Rect bounds_in_screen = gfx::ScaleToEnclosingRect(
+        event_data.get()->nodeData[0]->boundsInScreen,
+        1.0f / toplevel_window->layer()->device_scale_factor());
+
+    accessibility_manager->OnViewFocusedInArc(bounds_in_screen);
+  }
+}
+
+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);
+  if (inactive_surface) {
+    views::Widget* widget = views::Widget::GetWidgetForNativeView(lost_active);
+    if (widget && widget->GetContentsView()) {
+      widget->GetContentsView()->RemoveChildView(tree_.get());
+      widget->GetContentsView()->NotifyAccessibilityEvent(
+          ui::AX_EVENT_CHILDREN_CHANGED, false);
+    }
   }
 
-  chromeos::AccessibilityManager* accessibility_manager =
-      chromeos::AccessibilityManager::Get();
-
-  if (!accessibility_manager ||
-      !accessibility_manager->IsFocusHighlightEnabled()) {
-    return;
-  }
-
-  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();
+  tree_.reset(new AXTreeSourceArc());
 
-  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->GetContentsView()->AddChildView(tree_.get());
+    widget->GetContentsView()->NotifyAccessibilityEvent(
+        ui::AX_EVENT_CHILDREN_CHANGED, false);
+    tree_->RequestFocus();
+  }
 }
 
 }  // namespace arc