Allow chromedriver to install an extension and get all installed extension IDs.

chrome/test/webdriver/commands/chrome_commands.cc

 // Copyright (c) 2011 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/test/webdriver/commands/chrome_commands.h"
 
 #include <string>
 #include <vector>
 
 #include "base/file_path.h"
+#include "base/stringprintf.h"
 #include "chrome/test/automation/value_conversion_util.h"
 #include "chrome/test/webdriver/commands/response.h"
 #include "chrome/test/webdriver/webdriver_error.h"
 #include "chrome/test/webdriver/webdriver_session.h"
 
 namespace webdriver {
 
 ExtensionsCommand::ExtensionsCommand(
     const std::vector<std::string>& path_segments,
     const base::DictionaryValue* const parameters)
@@ skipping 32 matching lines @@
   std::string extension_id;
   Error* error = session_->InstallExtension(
       FilePath(path_string), &extension_id);
   if (error) {
     response->SetError(error);
     return;
   }
   response->SetValue(CreateValueFrom(extension_id));
 }
 
+ViewsCommand::ViewsCommand(
+    const std::vector<std::string>& path_segments,
+    const base::DictionaryValue* const parameters)
+    : WebDriverCommand(path_segments, parameters) {}
+
+ViewsCommand::~ViewsCommand() {}
+
+bool ViewsCommand::DoesGet() {
+  return true;
+}
+
+bool ViewsCommand::DoesPost() {
+  return true;
+}
+
+void ViewsCommand::ExecuteGet(Response* const response) {
+  std::vector<WebViewInfo> views;
+  Error* error = session_->GetViews(&views);
+  if (error) {
+    response->SetError(error);
+    return;
+  }
+  base::ListValue* views_list = new base::ListValue();
+  for (size_t i = 0; i < views.size(); ++i) {
+    base::DictionaryValue* dict = new base::DictionaryValue();
+    AutomationId& id = views[i].view_id.id.view_id;
+    dict->SetString("handle", base::StringPrintf(
+        "%d|%s", id.type(), id.id().c_str()));
+    dict->SetInteger("type", id.type());
+    if (!views[i].extension_id.empty())
+      dict->SetString("extension_id", views[i].extension_id);
+    views_list->Append(dict);
+  }
+  response->SetValue(views_list);
+}
+
+void ViewsCommand::ExecutePost(Response* const response) {
+/*
+  Error* error = session_->SwitchToView(views, false / restrict_to_tabs /);
+  if (error) {
+    response->SetError(error);
+    return;
+  }
+*/
+}
+
 }  // namespace webdriver