Reimplement automation API on top of C++-backed AXTree.

chrome/renderer/extensions/automation_internal_custom_bindings.h

 // Copyright 2014 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.
 
 #ifndef CHROME_RENDERER_EXTENSIONS_AUTOMATION_INTERNAL_CUSTOM_BINDINGS_H_
 #define CHROME_RENDERER_EXTENSIONS_AUTOMATION_INTERNAL_CUSTOM_BINDINGS_H_
 
 #include "base/compiler_specific.h"
 #include "extensions/renderer/object_backed_native_handler.h"
 #include "ipc/ipc_message.h"
 #include "ui/accessibility/ax_tree.h"
 #include "v8/include/v8.h"
 
 struct ExtensionMsg_AccessibilityEventParams;
 
 namespace extensions {
 
 class AutomationMessageFilter;
 
+struct TreeCache {
+  TreeCache();
+  ~TreeCache();
+
+  int tab_id;
+  int tree_id;
+
+  gfx::Vector2d location_offset;
+  ui::AXTree tree;
+};
+
 // The native component of custom bindings for the chrome.automationInternal
 // API.
-class AutomationInternalCustomBindings : public ObjectBackedNativeHandler {
+class AutomationInternalCustomBindings : public ObjectBackedNativeHandler,
+                                         public ui::AXTreeDelegate {
  public:
   explicit AutomationInternalCustomBindings(ScriptContext* context);
 
   ~AutomationInternalCustomBindings() override;
 
   bool OnMessageReceived(const IPC::Message& message);
 
  private:
   // Returns whether this extension has the "interact" permission set (either
   // explicitly or implicitly after manifest parsing).
   void IsInteractPermitted(const v8::FunctionCallbackInfo<v8::Value>& args);
 
   // Returns an object with bindings that will be added to the
   // chrome.automation namespace.
   void GetSchemaAdditions(const v8::FunctionCallbackInfo<v8::Value>& args);
 
   // Get the routing ID for the extension.
   void GetRoutingID(const v8::FunctionCallbackInfo<v8::Value>& args);
 
+  //
+  void StartCachingAccessibilityTrees(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+
+  //
+  void DestroyAccessibilityTree(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+
+  void GetRootID(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+
+  bool GetNodeHelper(
+      const v8::FunctionCallbackInfo<v8::Value>& args,
+      TreeCache** out_cache,
+      ui::AXNode** out_node);
+  void SetReturnValueFromBaseValue(
+      const v8::FunctionCallbackInfo<v8::Value>& args,
+      base::Value* value);
+
+  void GetParentID(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+  void GetChildCount(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+  void GetChildIDAtIndex(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+  void GetIndexInParent(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+
+  void GetState(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+  void GetRole(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+  void GetLocation(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+
+  bool GetAttributeHelper(
+      const v8::FunctionCallbackInfo<v8::Value>& args,
+      ui::AXNode** out_node,
+      std::string* out_attribute_name);
+
+  void GetStringAttribute(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+  void GetBoolAttribute(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+  void GetIntAttribute(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+  void GetFloatAttribute(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+  void GetIntListAttribute(
+      const v8::FunctionCallbackInfo<v8::Value>& args);
+
   // Handle accessibility events from the browser process.
   void OnAccessibilityEvent(
       const ExtensionMsg_AccessibilityEventParams& params);
 
+  // AXTreeDelegate implementation.
+  void OnNodeWillBeDeleted(ui::AXTree* tree, ui::AXNode* node) override;
+  void OnSubtreeWillBeDeleted(ui::AXTree* tree, ui::AXNode* node) override;
+  void OnNodeCreated(ui::AXTree* tree, ui::AXNode* node) override;
+  void OnNodeChanged(ui::AXTree* tree, ui::AXNode* node) override;
+  void OnAtomicUpdateFinished(ui::AXTree* tree,
+                              bool root_changed,
+                              const std::vector<Change>& changes) override;
+
+  void SendTreeChangeEvent(const char *change_type,
+                           ui::AXTree* tree,
+                           ui::AXNode* node);
+
+  base::hash_map<int, TreeCache*> tree_id_to_tree_cache_map_;
+  base::hash_map<ui::AXTree*, TreeCache*> axtree_to_tree_cache_map_;
   scoped_refptr<AutomationMessageFilter> message_filter_;
 
   DISALLOW_COPY_AND_ASSIGN(AutomationInternalCustomBindings);
 };
 
 }  // namespace extensions
 
 #endif  // CHROME_RENDERER_EXTENSIONS_AUTOMATION_INTERNAL_CUSTOM_BINDINGS_H_