Add pyauto hook for getting and manipulating the data underneath the NTP.

chrome/browser/automation/testing_automation_provider.cc

 // Copyright (c) 2010 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/automation/testing_automation_provider.h"
 
 #include "app/message_box_flags.h"
 #include "base/command_line.h"
 #include "base/json/json_reader.h"
 #include "base/json/json_writer.h"
@@ skipping 22 matching lines @@
 #include "chrome/browser/blocked_content_container.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/browser_shutdown.h"
 #include "chrome/browser/browser_window.h"
 #include "chrome/browser/debugger/devtools_manager.h"
 #include "chrome/browser/download/download_prefs.h"
 #include "chrome/browser/download/download_shelf.h"
 #include "chrome/browser/extensions/extension_host.h"
 #include "chrome/browser/extensions/extensions_service.h"
 #include "chrome/browser/find_bar.h"
+#include "chrome/browser/history/top_sites.h"
 #include "chrome/browser/location_bar.h"
 #include "chrome/browser/login_prompt.h"
 #include "chrome/browser/native_app_modal_dialog.h"
 #include "chrome/browser/profile.h"
 #include "chrome/browser/platform_util.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profile_manager.h"
 #include "chrome/browser/renderer_host/render_process_host.h"
 #include "chrome/browser/renderer_host/render_view_host.h"
 #include "chrome/browser/search_engines/keyword_editor_controller.h"
@@ skipping 2081 matching lines @@
 
   handler_map["SignInToSync"] = &TestingAutomationProvider::SignInToSync;
   handler_map["GetSyncInfo"] = &TestingAutomationProvider::GetSyncInfo;
   handler_map["AwaitSyncCycleCompletion"] =
       &TestingAutomationProvider::AwaitSyncCycleCompletion;
   handler_map["EnableSyncForDatatypes"] =
       &TestingAutomationProvider::EnableSyncForDatatypes;
   handler_map["DisableSyncForDatatypes"] =
       &TestingAutomationProvider::DisableSyncForDatatypes;
 
+  handler_map["GetNTPInfo"] =
+      &TestingAutomationProvider::GetNTPInfo;
+  handler_map["MoveNTPMostVisitedThumbnail"] =
+      &TestingAutomationProvider::MoveNTPMostVisitedThumbnail;
+  handler_map["RemoveNTPMostVisitedThumbnail"] =
+      &TestingAutomationProvider::RemoveNTPMostVisitedThumbnail;
+  handler_map["UnpinNTPMostVisitedThumbnail"] =
+      &TestingAutomationProvider::UnpinNTPMostVisitedThumbnail;
+  handler_map["RestoreAllNTPMostVisitedThumbnails"] =
+      &TestingAutomationProvider::RestoreAllNTPMostVisitedThumbnails;
+
   if (handler_map.find(std::string(command)) != handler_map.end()) {
     (this->*handler_map[command])(browser, dict_value, reply_message);
   } else {
     std::string error_string = "Unknown command. Options: ";
     for (std::map<std::string, JsonHandler>::const_iterator it =
          handler_map.begin(); it != handler_map.end(); ++it) {
       error_string += it->first + ", ";
     }
     AutomationJSONReply(this, reply_message).SendError(error_string);
   }
@@ skipping 2144 matching lines @@
   const BalloonCollection::Balloons& balloons = collection->GetActiveBalloons();
   if (static_cast<int>(balloons.size()) == count) {
     AutomationJSONReply(this, reply_message).SendSuccess(NULL);
     return;
   }
   // This will delete itself when finished.
   new OnNotificationBalloonCountObserver(
       this, reply_message, collection, count);
 }
 
+// Sample JSON input: { "command": "GetNTPInfo" }
+// For output, refer to chrome/test/pyautolib/ntp_model.py.
+void TestingAutomationProvider::GetNTPInfo(
+    Browser* browser,
+    DictionaryValue* args,
+    IPC::Message* reply_message) {
+  // This observer will delete itself.
+  new NTPInfoObserver(this, reply_message, &consumer_);
+}
+
+void TestingAutomationProvider::MoveNTPMostVisitedThumbnail(
+    Browser* browser,
+    DictionaryValue* args,
+    IPC::Message* reply_message) {
+  AutomationJSONReply reply(this, reply_message);
+  std::string url, error;
+  int index, old_index;
+  if (!args->GetString("url", &url)) {
+    reply.SendError("Missing or invalid 'url' key.");
+    return;
+  }
+  if (!args->GetInteger("index", &index)) {
+    reply.SendError("Missing or invalid 'index' key.");
+    return;
+  }
+  if (!args->GetInteger("old_index", &old_index)) {
+    reply.SendError("Missing or invalid 'old_index' key");
+    return;
+  }
+  history::TopSites* top_sites = browser->profile()->GetTopSites();
+  if (!top_sites) {
+    reply.SendError("TopSites service is not initialized.");
+    return;
+  }
+  GURL swapped;
+  // If thumbnail A is switching positions with a pinned thumbnail B, B
+  // should be pinned in the old index of A.
+  if (top_sites->GetPinnedURLAtIndex(index, &swapped)) {
+    top_sites->AddPinnedURL(swapped, old_index);
+  }
+  top_sites->AddPinnedURL(GURL(url), index);
+  reply.SendSuccess(NULL);
+}
+
+void TestingAutomationProvider::RemoveNTPMostVisitedThumbnail(
+    Browser* browser,
+    DictionaryValue* args,
+    IPC::Message* reply_message) {
+  AutomationJSONReply reply(this, reply_message);
+  std::string url;
+  if (!args->GetString("url", &url)) {
+    reply.SendError("Missing or invalid 'url' key.");
+    return;
+  }
+  history::TopSites* top_sites = browser->profile()->GetTopSites();
+  if (!top_sites) {
+    reply.SendError("TopSites service is not initialized.");
+    return;
+  }
+  top_sites->AddBlacklistedURL(GURL(url));
+  reply.SendSuccess(NULL);
+}
+
+void TestingAutomationProvider::UnpinNTPMostVisitedThumbnail(
+    Browser* browser,
+    DictionaryValue* args,
+    IPC::Message* reply_message) {
+  AutomationJSONReply reply(this, reply_message);
+  std::string url;
+  if (!args->GetString("url", &url)) {
+    reply.SendError("Missing or invalid 'url' key.");
+    return;
+  }
+  history::TopSites* top_sites = browser->profile()->GetTopSites();
+  if (!top_sites) {
+    reply.SendError("TopSites service is not initialized.");
+    return;
+  }
+  top_sites->RemovePinnedURL(GURL(url));
+  reply.SendSuccess(NULL);
+}
+
+void TestingAutomationProvider::RestoreAllNTPMostVisitedThumbnails(
+    Browser* browser,
+    DictionaryValue* args,
+    IPC::Message* reply_message) {
+  AutomationJSONReply reply(this, reply_message);
+  history::TopSites* top_sites = browser->profile()->GetTopSites();
+  if (!top_sites) {
+    reply.SendError("TopSites service is not initialized.");
+    return;
+  }
+  top_sites->ClearBlacklistedURLs();
+  reply.SendSuccess(NULL);
+}
+
 void TestingAutomationProvider::WaitForTabCountToBecome(
     int browser_handle,
     int target_tab_count,
     IPC::Message* reply_message) {
   if (!browser_tracker_->ContainsHandle(browser_handle)) {
     AutomationMsg_WaitForTabCountToBecome::WriteReplyParams(reply_message,
                                                             false);
     Send(reply_message);
     return;
   }
@@ skipping 135 matching lines @@
   DCHECK(type == NotificationType::SESSION_END);
   // OnBrowserRemoved does a ReleaseLater. When session end is received we exit
   // before the task runs resulting in this object not being deleted. This
   // Release balance out the Release scheduled by OnBrowserRemoved.
   Release();
 }
 
 void TestingAutomationProvider::OnRemoveProvider() {
   AutomationProviderList::GetInstance()->RemoveProvider(this);
 }