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

chrome/browser/automation/automation_provider_observers.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/automation_provider_observers.h"
 
 #include <deque>
 
 #include "base/basictypes.h"
 #include "base/callback.h"
 #include "base/file_util.h"
 #include "base/json/json_writer.h"
 #include "base/scoped_ptr.h"
 #include "base/string_util.h"
 #include "base/stringprintf.h"
 #include "base/thread_restrictions.h"
 #include "base/values.h"
 #include "chrome/app/chrome_command_ids.h"
 #include "chrome/browser/automation/automation_provider.h"
 #include "chrome/browser/automation/automation_provider_json.h"
 #include "chrome/browser/bookmarks/bookmark_model.h"
 #include "chrome/browser/browser.h"
 #include "chrome/browser/browser_list.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/dom_operation_notification_details.h"
+#include "chrome/browser/dom_ui/most_visited_handler.h"
+#include "chrome/browser/dom_ui/new_tab_ui.h"
 #include "chrome/browser/download/download_item.h"
 #include "chrome/browser/download/save_package.h"
 #include "chrome/browser/extensions/extension_host.h"
 #include "chrome/browser/extensions/extension_process_manager.h"
 #include "chrome/browser/extensions/extension_updater.h"
+#include "chrome/browser/history/top_sites.h"
 #include "chrome/browser/login_prompt.h"
 #include "chrome/browser/metrics/metric_event_duration_details.h"
 #include "chrome/browser/notifications/balloon.h"
 #include "chrome/browser/notifications/balloon_collection.h"
 #include "chrome/browser/printing/print_job.h"
 #include "chrome/browser/profile.h"
 #include "chrome/browser/renderer_host/render_view_host.h"
 #include "chrome/browser/search_engines/template_url_model.h"
+#include "chrome/browser/sessions/tab_restore_service.h"
 #include "chrome/browser/tab_contents/navigation_controller.h"
 #include "chrome/browser/tab_contents/tab_contents.h"
 #include "chrome/browser/tab_contents/thumbnail_generator.h"
 #include "chrome/browser/translate/page_translated_details.h"
 #include "chrome/browser/translate/translate_infobar_delegate.h"
 #include "chrome/common/automation_constants.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/notification_service.h"
 #include "gfx/codec/png_codec.h"
 #include "gfx/rect.h"
+#include "googleurl/src/gurl.h"
 
 #if defined(OS_CHROMEOS)
 #include "chrome/browser/chromeos/login/authentication_notification_details.h"
 #endif
 
 // Holds onto start and stop timestamps for a particular tab
 class InitialLoadObserver::TabTime {
  public:
   explicit TabTime(base::TimeTicks started)
       : load_start_time_(started) {
@@ skipping 1337 matching lines @@
   render_view_->ExecuteJavascriptInWebFrame(L"", javascript);
 }
 
 void PageSnapshotTaker::SendMessage(bool success) {
   AutomationMsg_CaptureEntirePageAsPNG::WriteReplyParams(reply_message_,
                                                          success);
   automation_->Send(reply_message_);
   delete this;
 }
 
+NTPInfoObserver::NTPInfoObserver(
+    AutomationProvider* automation,
+    IPC::Message* reply_message,
+    CancelableRequestConsumer* consumer)
+        : automation_(automation),
+          reply_message_(reply_message),
+          consumer_(consumer),
+          request_(NULL),
+          ntp_info_(new DictionaryValue) {
+  top_sites_ = automation_->profile()->GetTopSites();
+  if (!top_sites_) {
+    AutomationJSONReply(automation_, reply_message_)
+        .SendError("Profile does not have service for querying the top sites.");
+    return;
+  }
+  TabRestoreService* service = automation_->profile()->GetTabRestoreService();
+  if (!service) {
+    AutomationJSONReply(automation_, reply_message_)
+        .SendError("No TabRestoreService.");
+    return;
+  }
+
+  // Get the info that would be displayed in the recently closed section.
+  ListValue* recently_closed_list = new ListValue;
+  NewTabUI::AddRecentlyClosedEntries(service->entries(),
+                                     recently_closed_list);
+  ntp_info_->Set("recently_closed", recently_closed_list);
+
+  // Add default site URLs.
+  ListValue* default_sites_list = new ListValue;
+  std::vector<GURL> urls = MostVisitedHandler::GetPrePopulatedUrls();
+  for (size_t i = 0; i < urls.size(); ++i) {
+    default_sites_list->Append(Value::CreateStringValue(
+        urls[i].possibly_invalid_spec()));
+  }
+  ntp_info_->Set("default_sites", default_sites_list);
+
+  registrar_.Add(this, NotificationType::TOP_SITES_UPDATED,
+                 Source<history::TopSites>(top_sites_));
+  if (top_sites_->loaded()) {
+    OnTopSitesLoaded();
+  } else {
+    registrar_.Add(this, NotificationType::TOP_SITES_LOADED,
+                   Source<Profile>(automation_->profile()));
+  }
+}
+
+void NTPInfoObserver::Observe(NotificationType type,
+                              const NotificationSource& source,
+                              const NotificationDetails& details) {
+  if (type == NotificationType::TOP_SITES_LOADED) {
+    OnTopSitesLoaded();
+  } else if (type == NotificationType::TOP_SITES_UPDATED) {
+    Details<CancelableRequestProvider::Handle> request_details(details);
+    if (request_ == *request_details.ptr()) {
+      top_sites_->GetMostVisitedURLs(
+          consumer_,
+          NewCallback(this, &NTPInfoObserver::OnTopSitesReceived));
+    }
+  }
+}
+
+void NTPInfoObserver::OnTopSitesLoaded() {
+  request_ = top_sites_->StartQueryForMostVisited();
+}
+
+void NTPInfoObserver::OnTopSitesReceived(
+    const history::MostVisitedURLList& visited_list) {
+  ListValue* list_value = new ListValue;
+  for (size_t i = 0; i < visited_list.size(); ++i) {
+    const history::MostVisitedURL& visited = visited_list[i];
+    if (visited.url.spec().empty())
+      break;  // This is the signal that there are no more real visited sites.
+    DictionaryValue* dict = new DictionaryValue;
+    dict->SetString("url", visited.url.spec());
+    dict->SetString("title", visited.title);
+    dict->SetBoolean("is_pinned", top_sites_->IsURLPinned(visited.url));
+    list_value->Append(dict);
+  }
+  ntp_info_->Set("most_visited", list_value);
+  AutomationJSONReply(automation_, reply_message_).SendSuccess(ntp_info_.get());
+  delete this;
+}
+
 AutocompleteEditFocusedObserver::AutocompleteEditFocusedObserver(
     AutomationProvider* automation,
     AutocompleteEditModel* autocomplete_edit,
     IPC::Message* reply_message)
     : automation_(automation),
       reply_message_(reply_message),
       autocomplete_edit_model_(autocomplete_edit) {
   Source<AutocompleteEditModel> source(autocomplete_edit);
   registrar_.Add(this, NotificationType::AUTOCOMPLETE_EDIT_FOCUSED, source);
 }
@@ skipping 21 matching lines @@
       this, &OnNotificationBalloonCountObserver::OnBalloonCollectionChanged));
 }
 
 void OnNotificationBalloonCountObserver::OnBalloonCollectionChanged() {
   if (static_cast<int>(collection_->GetActiveBalloons().size()) == count_) {
     collection_->set_on_collection_changed_callback(NULL);
     reply_.SendSuccess(NULL);
     delete this;
   }
 }