Add APIs to cache navigation info and send via callback

chrome/android/java/src/org/chromium/chrome/browser/customtabs/CustomTabObserver.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.app.Application;
+import android.graphics.Bitmap;
+import android.graphics.Rect;
 import android.os.SystemClock;
 import android.support.customtabs.CustomTabsCallback;
 import android.support.customtabs.CustomTabsSessionToken;
 
 import org.chromium.base.metrics.RecordHistogram;
+import org.chromium.chrome.R;
+import org.chromium.chrome.browser.prerender.ExternalPrerenderHandler;
 import org.chromium.chrome.browser.tab.EmptyTabObserver;
 import org.chromium.chrome.browser.tab.Tab;
 import org.chromium.chrome.browser.tab.TabObserver;
 import org.chromium.components.security_state.ConnectionSecurityLevel;
+import org.chromium.content_public.browser.ContentBitmapCallback;
 import org.chromium.content_public.browser.LoadUrlParams;
 
 import java.util.concurrent.TimeUnit;
 
 /**
  * A {@link TabObserver} that also handles custom tabs specific logging and messaging.
  */
 class CustomTabObserver extends EmptyTabObserver {
     private final CustomTabsConnection mCustomTabsConnection;
     private final CustomTabsSessionToken mSession;
     private final boolean mOpenedByChrome;
+    private float mScaleForNavigationInfo = 1f;
 
     private long mIntentReceivedTimestamp;
     private long mPageLoadStartedTimestamp;
 
     private static final int STATE_RESET = 0;
     private static final int STATE_WAITING_LOAD_START = 1;
     private static final int STATE_WAITING_LOAD_FINISH = 2;
     private int mCurrentState;
 
     public CustomTabObserver(
             Application application, CustomTabsSessionToken session, boolean openedByChrome) {
         if (openedByChrome) {
             mCustomTabsConnection = null;
         } else {
             mCustomTabsConnection = CustomTabsConnection.getInstance(application);
         }
         mSession = session;
+        if (!openedByChrome && mCustomTabsConnection.shouldSendNavigationInfoForSession(mSession)) {
+            float desiredWidth = application.getResources().getDimensionPixelSize(
+                    R.dimen.custom_tabs_screenshot_width);
+            float desiredHeight = application.getResources().getDimensionPixelSize(
+                    R.dimen.custom_tabs_screenshot_height);
+            Rect bounds = ExternalPrerenderHandler.estimateContentSize(application, false);
+            mScaleForNavigationInfo = (bounds.width() == 0 || bounds.height() == 0) ? 1f :
+                    Math.min(desiredWidth / bounds.width(), desiredHeight / bounds.height());
+        }
         mOpenedByChrome = openedByChrome;
         resetPageLoadTracking();
     }
 
     /**
      * Tracks the next page load, with timestamp as the origin of time.
      */
     public void trackNextPageLoadFromTimestamp(long timestamp) {
         mIntentReceivedTimestamp = timestamp;
         mCurrentState = STATE_WAITING_LOAD_START;
@@ skipping 51 matching lines @@
             // Same bounds and bucket count as "Startup.FirstCommitNavigationTime"
             RecordHistogram.recordCustomTimesHistogram(
                     histogramPrefix + ".IntentToFirstCommitNavigationTime", timeToPageLoadStartedMs,
                     1, TimeUnit.MINUTES.toMillis(1), TimeUnit.MILLISECONDS, 225);
             // Same bounds and bucket count as PLT histograms.
             RecordHistogram.recordCustomTimesHistogram(histogramPrefix + ".IntentToPageLoadedTime",
                     timeToPageLoadFinishedMs, 10, TimeUnit.MINUTES.toMillis(10),
                     TimeUnit.MILLISECONDS, 100);
         }
         resetPageLoadTracking();
+        captureNavigationInfo(tab);
     }
 
     @Override
     public void onDidAttachInterstitialPage(Tab tab) {
         if (tab.getSecurityLevel() != ConnectionSecurityLevel.SECURITY_ERROR) return;
         resetPageLoadTracking();
         if (mCustomTabsConnection != null) {
             mCustomTabsConnection.notifyNavigationEvent(
                     mSession, CustomTabsCallback.NAVIGATION_FAILED);
         }
     }
 
     @Override
     public void onPageLoadFailed(Tab tab, int errorCode) {
         resetPageLoadTracking();
         if (mCustomTabsConnection != null) {
             mCustomTabsConnection.notifyNavigationEvent(
                     mSession, CustomTabsCallback.NAVIGATION_FAILED);
         }
     }
 
     private void resetPageLoadTracking() {
         mCurrentState = STATE_RESET;
         mIntentReceivedTimestamp = -1;
     }
+
+    private void captureNavigationInfo(final Tab tab) {
+        if (mCustomTabsConnection == null) return;
+        if (!mCustomTabsConnection.shouldSendNavigationInfoForSession(mSession)) return;
+
+        ContentBitmapCallback callback = new ContentBitmapCallback() {
+            @Override
+            public void onFinishGetBitmap(Bitmap bitmap, int response) {
+                mCustomTabsConnection.sendNavigationInfo(
+                        mSession, tab.getUrl(), tab.getTitle(), bitmap);
+            }
+        };
+        tab.getWebContents().getContentBitmapAsync(
+                Bitmap.Config.ARGB_8888, mScaleForNavigationInfo, new Rect(), callback);
+    }
 }