Add the LoggedIn Predictor, to detect which websites a user is likely

chrome/browser/prerender/prerender_tab_helper.cc

Index: chrome/browser/prerender/prerender_tab_helper.cc
===================================================================
--- chrome/browser/prerender/prerender_tab_helper.cc	(revision 194439)
+++ chrome/browser/prerender/prerender_tab_helper.cc	(working copy)
@@ -7,13 +7,18 @@
 #include "base/metrics/histogram.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/time.h"
+#include "chrome/browser/predictors/logged_in_predictor_table.h"
 #include "chrome/browser/prerender/prerender_histograms.h"
 #include "chrome/browser/prerender/prerender_manager.h"
 #include "chrome/browser/prerender/prerender_manager_factory.h"
 #include "chrome/browser/profiles/profile.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/browser/navigation_details.h"
+#include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_view.h"
+#include "content/public/common/frame_navigate_params.h"
 #include "skia/ext/platform_canvas.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "ui/gfx/rect.h"
@@ -128,7 +133,8 @@
 };
 
 PrerenderTabHelper::PrerenderTabHelper(content::WebContents* web_contents)
-    : content::WebContentsObserver(web_contents) {
+    : content::WebContentsObserver(web_contents),
+      weak_factory_(this) {
 }
 
 PrerenderTabHelper::~PrerenderTabHelper() {
@@ -138,6 +144,8 @@
     const GURL& url,
     content::RenderViewHost* render_view_host) {
   url_ = url;
+  RecordEvent(EVENT_MAINFRAME_CHANGE);
+  RecordEventIfLoggedInURL(EVENT_MAINFRAME_CHANGE_DOMAIN_LOGGED_IN, url);
   PrerenderManager* prerender_manager = MaybeGetPrerenderManager();
   if (!prerender_manager)
     return;
@@ -154,6 +162,9 @@
     content::RenderViewHost* render_view_host) {
   if (!is_main_frame)
     return;
+  RecordEvent(EVENT_MAINFRAME_COMMIT);
+  RecordEventIfLoggedInURL(EVENT_MAINFRAME_COMMIT_DOMAIN_LOGGED_IN,
+                           validated_url);
   url_ = validated_url;
   PrerenderManager* prerender_manager = MaybeGetPrerenderManager();
   if (!prerender_manager)
@@ -210,11 +221,40 @@
   }
 }
 
+void PrerenderTabHelper::DidNavigateAnyFrame(
+      const content::LoadCommittedDetails& details,
+      const content::FrameNavigateParams& params) {
+  scoped_refptr<predictors::LoggedInPredictorTable> logged_in_table =
+      MaybeGetLoggedInTable();
+  if (params.password_form.origin.is_valid() && logged_in_table.get()) {
+    content::BrowserThread::PostTask(
+        content::BrowserThread::DB, FROM_HERE,
+        base::Bind(&predictors::LoggedInPredictorTable::Add,
+                   logged_in_table,
+                   params.url));
+    RecordEvent(EVENT_LOGIN_ACTION_ADDED);
+  }
+}
+
 PrerenderManager* PrerenderTabHelper::MaybeGetPrerenderManager() const {
   return PrerenderManagerFactory::GetForProfile(
       Profile::FromBrowserContext(web_contents()->GetBrowserContext()));
 }
 
+scoped_refptr<predictors::LoggedInPredictorTable>
+PrerenderTabHelper::MaybeGetLoggedInTable() const {
+  RecordEvent(EVENT_LOGGED_IN_TABLE_REQUESTED);
+  PrerenderManager* prerender_manager = MaybeGetPrerenderManager();
+  if (prerender_manager) {
+    predictors::LoggedInPredictorTable* result =
+        prerender_manager->logged_in_predictor_table();
+    if (result)
+      RecordEvent(EVENT_LOGGED_IN_TABLE_PRESENT);
+    return result;
+  }
+  return NULL;
+}
+
 bool PrerenderTabHelper::IsPrerendering() {
   PrerenderManager* prerender_manager = MaybeGetPrerenderManager();
   if (!prerender_manager)
@@ -249,4 +289,41 @@
   }
 }
 
+void PrerenderTabHelper::RecordEvent(PrerenderTabHelper::Event event) const {
+  UMA_HISTOGRAM_ENUMERATION("Prerender.TabHelperEvent",
+                            event, PrerenderTabHelper::EVENT_MAX_VALUE);
+}
+
+void PrerenderTabHelper::RecordEventIfLoggedInURL(
+    PrerenderTabHelper::Event event, const GURL& url) {
+  scoped_refptr<predictors::LoggedInPredictorTable> logged_in_table =
+      MaybeGetLoggedInTable();
+  if (!logged_in_table.get())
+    return;
+  scoped_ptr<bool> is_present(new bool);
+  scoped_ptr<bool> lookup_succeeded(new bool);
+  bool* is_present_ptr = is_present.get();
+  bool* lookup_succeeded_ptr = lookup_succeeded.get();
+  content::BrowserThread::PostTaskAndReply(
+      content::BrowserThread::DB, FROM_HERE,
+      base::Bind(&predictors::LoggedInPredictorTable::HasUserLoggedIn,
+                 logged_in_table,
+                 url,
+                 is_present_ptr,
+                 lookup_succeeded_ptr),
+      base::Bind(&PrerenderTabHelper::RecordEventIfLoggedInURLResult,
+                 weak_factory_.GetWeakPtr(),
+                 event,
+                 base::Passed(&is_present),
+                 base::Passed(&lookup_succeeded)));
+}
+
+void PrerenderTabHelper::RecordEventIfLoggedInURLResult(
+    PrerenderTabHelper::Event event,
+    scoped_ptr<bool> is_present,
+    scoped_ptr<bool> lookup_succeeded) {
+  if (*lookup_succeeded && *is_present)
+    RecordEvent(event);
+}
+
 }  // namespace prerender