Implement actions for Automation API.

chrome/browser/extensions/api/automation_internal/automation_internal_api.cc

 // 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.
 
 #include "chrome/browser/extensions/api/automation_internal/automation_internal_api.h"
 
 #include "base/command_line.h"
 #include "base/lazy_instance.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/values.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/extensions/api/automation_internal.h"
 #include "content/public/browser/ax_event_notification_details.h"
 #include "content/public/browser/browser_accessibility_state.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_source.h"
 #include "content/public/browser/notification_types.h"
+#include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/render_widget_host.h"
+#include "content/public/browser/render_widget_host.h"
 #include "content/public/browser/render_widget_host_view.h"
 #include "content/public/browser/web_contents.h"
 #include "extensions/browser/event_router.h"
 #include "extensions/browser/extension_system.h"
 #include "ui/accessibility/ax_enums.h"
 #include "ui/accessibility/ax_node_data.h"
 
 namespace extensions {
 class AutomationWebContentsObserver;
 } // namespace extensions
@@ skipping 10 matching lines @@
 void DispatchEvent(content::BrowserContext* context,
                    const std::string& event_name,
                    scoped_ptr<base::ListValue> args) {
   if (context && extensions::ExtensionSystem::Get(context)->event_router()) {
     scoped_ptr<Event> event(new Event(event_name, args.Pass()));
     event->restrict_to_browser_context = context;
     ExtensionSystem::Get(context)->event_router()->BroadcastEvent(event.Pass());
   }
 }
 
+  // Helper to get an ActionType from a |ListValue|.
+  bool GetActionType(base::ListValue* value,
+                     int position,
+                     api::automation_internal::ActionType* out_type) {
+    std::string action_value;
+    if (!value->GetString(position, &action_value))
+      return false;
+    *out_type = api::automation_internal::ParseActionType(
+        action_value);
+    return true;
+  }
+
 }  // namespace
 
 // Helper class that receives accessibility data from |WebContents|.
 class AutomationWebContentsObserver
     : public content::WebContentsObserver,
       public content::WebContentsUserData<AutomationWebContentsObserver> {
  public:
   virtual ~AutomationWebContentsObserver() {}
 
   // content::WebContentsObserver overrides.
   virtual void AccessibilityEventReceived(
       const std::vector<content::AXEventNotificationDetails>& details)
       OVERRIDE {
     if (!CommandLine::ForCurrentProcess()->HasSwitch(
             switches::kEnableAutomationAPI)) {
       return;
     }
 
     std::vector<content::AXEventNotificationDetails>::const_iterator iter =
         details.begin();
     for (; iter != details.end(); iter++) {
       scoped_ptr<base::ListValue> args(new base::ListValue());
       const content::AXEventNotificationDetails& event = *iter;
+      int process_id = event.process_id;

[aboxhall, 2014/04/01 17:49:51]

We may wish to consider combining these into some kind of unique ID on the
browser side, analogous to/possibly reusing/extending tab id. Out of scope for
this CL though.

[David Tseng, 2014/04/01 21:27:15]

I'm ok either way. Leaving them separate means we can easily address RWH for
actions though without parsing/mapping back.

       int routing_id = event.routing_id;
 
       base::DictionaryValue* axTreeUpdate = new base::DictionaryValue();
       // TODO(dtseng): These strings should be auto-generated by the IDL
       // compiler.
+      axTreeUpdate->Set("process_id",
+                        base::Value::CreateIntegerValue(process_id));
       axTreeUpdate->Set("routing_id",
                         base::Value::CreateIntegerValue(routing_id));
       axTreeUpdate->SetString("event_type", ToString(iter->event_type));
       base::ListValue* nodes = new base::ListValue();
       axTreeUpdate->Set("nodes", nodes);
       for (size_t i = 0; i < event.nodes.size(); i++) {
         const ui::AXNodeData& node = event.nodes[i];
         AddNodeData(node, nodes);
       }
       args->Append(axTreeUpdate);
@@ skipping 121 matching lines @@
   content::RenderWidgetHost* rwh =
       contents->GetRenderWidgetHostView()->GetRenderWidgetHost();
   if (!rwh)
     return false;
 
   AutomationWebContentsObserver::CreateForWebContents(contents);
 
   rwh->EnableTreeOnlyAccessibilityMode();
 
   results_ = api::automation_internal::EnableCurrentTab::Results::Create(
-      rwh->GetRoutingID());
+      rwh->GetProcess()->GetID(), rwh->GetRoutingID());
 
   SendResponse(true);
   return true;
 }
 
+bool AutomationInternalPerformActionFunction::RunImpl() {
+  int process_id;
+  EXTENSION_FUNCTION_VALIDATE(args_->GetInteger(0, &process_id));
+
+  int routing_id;
+  EXTENSION_FUNCTION_VALIDATE(args_->GetInteger(1, &routing_id));
+
+  int automation_id;

[aboxhall, 2014/04/01 17:49:51]

node_id maybe? Or automation_node_id?

[David Tseng, 2014/04/01 21:27:15]

Sure, automation_node_id.

+  EXTENSION_FUNCTION_VALIDATE(args_->GetInteger(2, &automation_id));
+  api::automation_internal::ActionType action_type =

[aboxhall, 2014/04/01 17:49:51]

Nit: newline above this line

[David Tseng, 2014/04/01 21:27:15]

Done.

+      api::automation_internal::ACTION_TYPE_NONE;
+  EXTENSION_FUNCTION_VALIDATE(GetActionType(args_.get(), 3, &action_type));
+  content::RenderWidgetHost* rwh =

[aboxhall, 2014/04/01 17:49:51]

Nit: newline above this line (i.e. group declaring and getting arguments
together, keep this separate).

[David Tseng, 2014/04/01 21:27:15]

Done.

+      content::RenderWidgetHost::FromID(process_id, routing_id);
+
+  switch (action_type) {
+    case api::automation_internal::ACTION_TYPE_DO_DEFAULT:

[aboxhall, 2014/04/01 17:49:51]

Nit: these case statements are all indented, but default is not. IIRC all of
them should be indented.

[David Tseng, 2014/04/01 21:27:15]

Done. (emacs c-mode tab indent failed me here).

+      rwh->AccessibilityDoDefaultAction(automation_id);
+      break;
+    case api::automation_internal::ACTION_TYPE_FOCUS:
+      rwh->AccessibilitySetFocus(automation_id);
+      break;
+    case api::automation_internal::ACTION_TYPE_MAKE_VISIBLE:
+      rwh->AccessibilityScrollToMakeVisible(automation_id, gfx::Rect());
+      break;
+    case api::automation_internal::ACTION_TYPE_SET_SELECTION: {
+      int start_index = 0, end_index = 0;
+      base::ListValue* opt_args;
+      EXTENSION_FUNCTION_VALIDATE(args_->GetList(4, &opt_args));
+      EXTENSION_FUNCTION_VALIDATE(opt_args->GetInteger(0, &start_index));
+      EXTENSION_FUNCTION_VALIDATE(opt_args->GetInteger(1, &end_index));
+      rwh->AccessibilitySetTextSelection(
+          automation_id, start_index, end_index);
+      break;
+    }
+  default:
+      NOTREACHED();
+  }
+  return true;
+}
+
 }  // namespace extensions