CustomTabs: Base version for prerender switch

chrome/android/java/src/org/chromium/chrome/browser/customtabs/ClientManager.java

 // Copyright 2015 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.
 
 package org.chromium.chrome.browser.customtabs;
 
 import android.content.ComponentName;
 import android.content.Context;
 import android.content.Intent;
 import android.content.ServiceConnection;
@@ skipping 113 matching lines @@
         public final PostMessageHandler postMessageHandler;
         public boolean mIgnoreFragments;
         public boolean lowConfidencePrediction;
         public boolean highConfidencePrediction;
         private boolean mShouldHideDomain;
         private boolean mShouldPrerenderOnCellular;
         private boolean mShouldSendNavigationInfo;
         private KeepAliveServiceConnection mKeepAliveConnection;
         private String mPredictedUrl;
         private long mLastMayLaunchUrlTimestamp;
+        private int mSpeculationMode;
 
         public SessionParams(Context context, int uid, DisconnectCallback callback,
                 PostMessageHandler postMessageHandler) {
             this.uid = uid;
             packageName = getPackageName(context, uid);
             disconnectCallback = callback;
             this.postMessageHandler = postMessageHandler;
+            this.mSpeculationMode = CustomTabsConnection.SpeculationParams.PRERENDER;
         }
 
         private static String getPackageName(Context context, int uid) {
             PackageManager packageManager = context.getPackageManager();
             String[] packageList = packageManager.getPackagesForUid(uid);
             if (packageList.length != 1 || TextUtils.isEmpty(packageList[0])) return null;
             return packageList[0];
         }
 
         public KeepAliveServiceConnection getKeepAliveConnection() {
@@ skipping 292 matching lines @@
 
     /**
      * Sets whether prerender should be turned on for mobile networks for given session.
      */
     public synchronized void setPrerenderCellularForSession(
             CustomTabsSessionToken session, boolean prerender) {
         SessionParams params = mSessionParams.get(session);
         if (params != null) params.mShouldPrerenderOnCellular = prerender;
     }
 
+    /**
+     * Sets the speculation mode to be used by default for given session.
+     */
+    public synchronized void setSpeculationModeForSession(
+            CustomTabsSessionToken session, int speculationMode) {
+        SessionParams params = mSessionParams.get(session);
+        if (params != null) params.mSpeculationMode = speculationMode;
+    }
+
+    /**
+     * Get the speculation mode to be used by default for the given session.
+     * If no value has been set will default to PRERENDER mode.
+     */
+    public synchronized int getSpeculationModeForSession(CustomTabsSessionToken session) {
+        SessionParams params = mSessionParams.get(session);
+        return params == null ? CustomTabsConnection.SpeculationParams.PRERENDER
+                              : params.mSpeculationMode;
+    }
+
     /** Tries to bind to a client to keep it alive, and returns true for success. */
     public synchronized boolean keepAliveForSession(CustomTabsSessionToken session, Intent intent) {
         // When an application is bound to a service, its priority is raised to
         // be at least equal to the application's one. This binds to a dummy
         // service (no calls to this service are made).
         if (intent == null || intent.getComponent() == null) return false;
         SessionParams params = mSessionParams.get(session);
         if (params == null) return false;
 
         KeepAliveServiceConnection connection = params.getKeepAliveConnection();
@@ skipping 58 matching lines @@
         SessionParams params = mSessionParams.get(session);
         if (params == null) return;
         mSessionParams.remove(session);
         if (params.postMessageHandler != null) {
             params.postMessageHandler.unbindFromContext(mContext);
         }
         if (params.disconnectCallback != null) params.disconnectCallback.run(session);
         mUidHasCalledWarmup.delete(params.uid);
     }
 }