src/chrome/browser/android/omnibox/omnibox_prerender.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "omnibox_prerender.h"
   6
   7 #include "base/android/jni_string.h"
   8 #include "base/logging.h"
   9 #include "chrome/browser/android/tab_android.h"
  10 #include "chrome/browser/predictors/autocomplete_action_predictor.h"
  11 #include "chrome/browser/predictors/autocomplete_action_predictor_factory.h"
  12 #include "chrome/browser/profiles/profile.h"
  13 #include "chrome/browser/profiles/profile_android.h"
  14 #include "chrome/browser/ui/search/instant_search_prerenderer.h"
  15 #include "components/omnibox/autocomplete_match.h"
  16 #include "components/omnibox/autocomplete_result.h"
  17 #include "content/public/browser/web_contents.h"
  18 #include "jni/OmniboxPrerender_jni.h"
  19 #include "url/gurl.h"
  20
  21 using predictors::AutocompleteActionPredictor;
  22 using predictors::AutocompleteActionPredictorFactory;
  23
  24 OmniboxPrerender::OmniboxPrerender(JNIEnv* env, jobject obj)
  25     : weak_java_omnibox_(env, obj) {
  26 }
  27
  28 OmniboxPrerender::~OmniboxPrerender() {
  29 }
  30
  31 static jlong Init(JNIEnv* env, jobject obj) {
  32   OmniboxPrerender* omnibox = new OmniboxPrerender(env, obj);
  33   return reinterpret_cast<intptr_t>(omnibox);
  34 }
  35
  36 bool RegisterOmniboxPrerender(JNIEnv* env) {
  37   return RegisterNativesImpl(env);
  38 }
  39
  40 void OmniboxPrerender::Clear(JNIEnv* env,
  41                              jobject obj,
  42                              jobject j_profile_android) {
  43   Profile* profile = ProfileAndroid::FromProfileAndroid(j_profile_android);
  44   DCHECK(profile);
  45   if (!profile)
  46     return;
  47   AutocompleteActionPredictor* action_predictor =
  48       AutocompleteActionPredictorFactory::GetForProfile(profile);
  49   action_predictor->ClearTransitionalMatches();
  50   action_predictor->CancelPrerender();
  51 }
  52
  53 void OmniboxPrerender::InitializeForProfile(
  54     JNIEnv* env,
  55     jobject obj,
  56     jobject j_profile_android) {
  57   Profile* profile = ProfileAndroid::FromProfileAndroid(j_profile_android);
  58   // Initialize the AutocompleteActionPredictor for this profile.
  59   // It needs to register for notifications as part of its initialization.
  60   AutocompleteActionPredictorFactory::GetForProfile(profile);
  61 }
  62
  63 void OmniboxPrerender::PrerenderMaybe(JNIEnv* env,
  64                                       jobject obj,
  65                                       jstring j_url,
  66                                       jstring j_current_url,
  67                                       jlong jsource_match,
  68                                       jobject j_profile_android,
  69                                       jobject j_tab) {
  70   AutocompleteResult* autocomplete_result =
  71       reinterpret_cast<AutocompleteResult*>(jsource_match);
  72   Profile* profile = ProfileAndroid::FromProfileAndroid(j_profile_android);
  73   base::string16 url_string =
  74       base::android::ConvertJavaStringToUTF16(env, j_url);
  75   base::string16 current_url_string =
  76       base::android::ConvertJavaStringToUTF16(env, j_current_url);
  77   content::WebContents* web_contents =
  78       TabAndroid::GetNativeTab(env, j_tab)->web_contents();
  79   // TODO(apiccion) Use a delegate for communicating with web_contents.
  80   // This can happen in OmniboxTests since the results are generated
  81   // in Java only.
  82   if (!autocomplete_result)
  83     return;
  84   if (!profile)
  85     return;
  86
  87   const AutocompleteResult::const_iterator default_match(
  88       autocomplete_result->default_match());
  89   if (default_match == autocomplete_result->end())
  90     return;
  91
  92   AutocompleteActionPredictor::Action recommended_action =
  93       AutocompleteActionPredictor::ACTION_NONE;
  94   InstantSearchPrerenderer* prerenderer =
  95       InstantSearchPrerenderer::GetForProfile(profile);
  96   if (prerenderer &&
  97       prerenderer->IsAllowed(*default_match, web_contents)) {
  98     recommended_action = AutocompleteActionPredictor::ACTION_PRERENDER;
  99   } else {
 100     AutocompleteActionPredictor* action_predictor =
 101         AutocompleteActionPredictorFactory::GetForProfile(profile);
 102     if (!action_predictor)
 103       return;
 104
 105     if (action_predictor) {
 106       action_predictor->
 107           RegisterTransitionalMatches(url_string, *autocomplete_result);
 108       recommended_action =
 109           action_predictor->RecommendAction(url_string, *default_match);
 110     }
 111   }
 112
 113   GURL current_url = GURL(current_url_string);
 114   switch (recommended_action) {
 115     case AutocompleteActionPredictor::ACTION_PRERENDER:
 116       // Ask for prerendering if the destination URL is different than the
 117       // current URL.
 118       if (default_match->destination_url != current_url) {
 119         DoPrerender(
 120             *default_match,
 121             profile,
 122             web_contents);
 123       }
 124       break;
 125     case AutocompleteActionPredictor::ACTION_PRECONNECT:
 126       // TODO (apiccion) add preconnect logic
 127       break;
 128     case AutocompleteActionPredictor::ACTION_NONE:
 129       break;
 130     default:
 131       NOTREACHED();
 132       break;
 133   }
 134 }
 135
 136 void OmniboxPrerender::DoPrerender(const AutocompleteMatch& match,
 137                                    Profile* profile,
 138                                    content::WebContents* web_contents) {
 139   DCHECK(profile);
 140   if (!profile)
 141     return;
 142   DCHECK(web_contents);
 143   if (!web_contents)
 144     return;
 145   gfx::Rect container_bounds = web_contents->GetContainerBounds();
 146   InstantSearchPrerenderer* prerenderer =
 147       InstantSearchPrerenderer::GetForProfile(profile);
 148   if (prerenderer && prerenderer->IsAllowed(match, web_contents)) {
 149     prerenderer->Init(
 150         web_contents->GetController().GetSessionStorageNamespaceMap(),
 151         container_bounds.size());
 152     return;
 153   }
 154   predictors::AutocompleteActionPredictorFactory::GetForProfile(profile)->
 155       StartPrerendering(
 156           match.destination_url,
 157           web_contents->GetController().GetSessionStorageNamespaceMap(),
 158           container_bounds.size());
 159 }