Fetch info about the state of navigation in a tab

chrome/browser/automation/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/automation_provider.h"
 
 #include <set>
 
 #include "app/message_box_flags.h"
 #include "base/callback.h"
@@ skipping 64 matching lines @@
 #include "chrome/browser/net/url_request_mock_util.h"
 #include "chrome/browser/platform_util.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/printing/print_job.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/ssl/ssl_manager.h"
 #include "chrome/browser/ssl/ssl_blocking_page.h"
 #include "chrome/browser/tab_contents/infobar_delegate.h"
+#include "chrome/browser/tab_contents/navigation_entry.h"
 #include "chrome/browser/tab_contents/tab_contents.h"
 #include "chrome/browser/tab_contents/tab_contents_view.h"
 #include "chrome/browser/translate/translate_infobar_delegate.h"
 #include "chrome/common/automation_constants.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/chrome_version_info.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/json_value_serializer.h"
@@ skipping 631 matching lines @@
       item->SetInteger(
           "pid",
           base::GetProcId(ex_host->render_process_host()->GetHandle()));
       extension_processes->Append(item);
     }
   }
   return_value->Set("extension_processes", extension_processes);
   AutomationJSONReply(this, reply_message).SendSuccess(return_value.get());
 }
 
+// Sample json input: { "command": "GetNavigationInfo" }
+// Refer to GetNavigationInfo() in chrome/test/pyautolib/pyauto.py for
+// sample json output.
+void AutomationProvider::GetNavigationInfo(Browser* browser,
+                                           DictionaryValue* args,
+                                           IPC::Message* reply_message) {
+  AutomationJSONReply reply(this, reply_message);
+  int tab_index;
+  TabContents* tab_contents = NULL;
+  if (!args->GetInteger("tab_index", &tab_index) ||
+      !(tab_contents = browser->GetTabContentsAt(tab_index))) {
+    reply.SendError("tab_index missing or invalid.");
+    return;
+  }
+  scoped_ptr<DictionaryValue> return_value(new DictionaryValue);
+  const NavigationController& controller = tab_contents->controller();
+  NavigationEntry* nav_entry = controller.GetActiveEntry();
+  DCHECK(nav_entry);
+
+  // Security info.
+  DictionaryValue* ssl = new DictionaryValue;
+  std::map<SecurityStyle, std::string> style_to_string;
+  style_to_string[SECURITY_STYLE_UNKNOWN] = "SECURITY_STYLE_UNKNOWN";
+  style_to_string[SECURITY_STYLE_UNAUTHENTICATED] =
+      "SECURITY_STYLE_UNAUTHENTICATED";
+  style_to_string[SECURITY_STYLE_AUTHENTICATION_BROKEN] =
+      "SECURITY_STYLE_AUTHENTICATION_BROKEN";
+  style_to_string[SECURITY_STYLE_AUTHENTICATED] =
+      "SECURITY_STYLE_AUTHENTICATED";
+
+  NavigationEntry::SSLStatus ssl_status = nav_entry->ssl();
+  ssl->SetString("security_style",
+                 style_to_string[ssl_status.security_style()]);
+  ssl->SetBoolean("ran_insecure_content", ssl_status.ran_insecure_content());
+  ssl->SetBoolean("displayed_insecure_content",
+                  ssl_status.displayed_insecure_content());
+  return_value->Set("ssl", ssl);
+
+  // Page type.
+  std::map<NavigationEntry::PageType, std::string> pagetype_to_string;
+  pagetype_to_string[NavigationEntry::NORMAL_PAGE] = "NORMAL_PAGE";
+  pagetype_to_string[NavigationEntry::ERROR_PAGE] = "ERROR_PAGE";
+  pagetype_to_string[NavigationEntry::INTERSTITIAL_PAGE] = "INTERSTITIAL_PAGE";
+  return_value->SetString("page_type",
+                          pagetype_to_string[nav_entry->page_type()]);
+
+  return_value->SetString("favicon_url", nav_entry->favicon().url().spec());
+  reply.SendSuccess(return_value.get());
+}
+
 // Sample json input: { "command": "GetHistoryInfo",
 //                      "search_text": "some text" }
 // Refer chrome/test/pyautolib/history_info.py for sample json output.
 void AutomationProvider::GetHistoryInfo(Browser* browser,
                                         DictionaryValue* args,
                                         IPC::Message* reply_message) {
   consumer_.CancelAllRequests();
 
   string16 search_text;
   args->GetString("search_text", &search_text);
@@ skipping 1325 matching lines @@
   }
   // Ownership remains with "values" variable.
   dict_value = static_cast<DictionaryValue*>(values.get());
   if (!dict_value->GetStringASCII(std::string("command"), &command)) {
     AutomationJSONReply(this, reply_message).SendError(
         "no command key in dict or not a string command");
     return;
   }
 
   // Map json commands to their handlers.
   std::map<std::string, JsonHandler> handler_map;

[John Grabowski, 2010/08/31 05:51:16]

We're starting to get a lot of these.  Perhaps (not in this CL) we can move
construction of this map to a setup routine called by the AutomationProvider? 
(Perhaps in a different CL)

   handler_map["DisablePlugin"] = &AutomationProvider::DisablePlugin;
   handler_map["EnablePlugin"] = &AutomationProvider::EnablePlugin;
   handler_map["GetPluginsInfo"] = &AutomationProvider::GetPluginsInfo;
 
   handler_map["GetBrowserInfo"] = &AutomationProvider::GetBrowserInfo;
 
+  handler_map["GetNavigationInfo"] = &AutomationProvider::GetNavigationInfo;
+
   handler_map["PerformActionOnInfobar"] =
       &AutomationProvider::PerformActionOnInfobar;
 
   handler_map["GetHistoryInfo"] = &AutomationProvider::GetHistoryInfo;
   handler_map["AddHistoryItem"] = &AutomationProvider::AddHistoryItem;
 
   handler_map["GetOmniboxInfo"] = &AutomationProvider::GetOmniboxInfo;
   handler_map["SetOmniboxText"] = &AutomationProvider::SetOmniboxText;
   handler_map["OmniboxAcceptInput"] = &AutomationProvider::OmniboxAcceptInput;
   handler_map["OmniboxMovePopupSelection"] =
@@ skipping 570 matching lines @@
       map->SetContentSetting(HostContentSettingsMap::Pattern(host),
                              content_type, "", setting);
     }
     *success = true;
   }
 }
 
 void AutomationProvider::ResetToDefaultTheme() {
   profile_->ClearTheme();
 }