overscroll: Take a screenshot at more appropriate times.

content/browser/web_contents/navigation_controller_impl.h

 // Copyright (c) 2012 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.
 
 #ifndef CONTENT_BROWSER_WEB_CONTENTS_NAVIGATION_CONTROLLER_IMPL_H_
 #define CONTENT_BROWSER_WEB_CONTENTS_NAVIGATION_CONTROLLER_IMPL_H_
 
 #include "build/build_config.h"
 #include "base/callback.h"
 #include "base/compiler_specific.h"
 #include "base/gtest_prod_util.h"
 #include "base/memory/linked_ptr.h"
 #include "base/time.h"
 #include "content/browser/ssl/ssl_manager.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_type.h"
 
 struct ViewHostMsg_FrameNavigate_Params;
 
 namespace skia {
 class PlatformBitmap;
 }
 
 namespace content {
 class NavigationEntryImpl;
+class RenderViewHost;
 class WebContentsImpl;
 class SiteInstance;
 struct LoadCommittedDetails;
 
 class CONTENT_EXPORT NavigationControllerImpl
     : public NON_EXPORTED_BASE(NavigationController) {
  public:
   NavigationControllerImpl(
       WebContentsImpl* web_contents,
       BrowserContext* browser_context);
@@ skipping 156 matching lines @@
 
   // Maximum number of entries before we start removing entries from the front.
   static void set_max_entry_count_for_testing(size_t max_entry_count) {
     max_entry_count_for_testing_ = max_entry_count;
   }
   static size_t max_entry_count();
 
   void SetGetTimestampCallbackForTest(
       const base::Callback<base::Time()>& get_timestamp_callback);
 
+  // Takes a screenshot of the page at the current state.
+  void TakeScreenshot();
+
+  void SetTakeScreenshotCallbackForTest(
+      const base::Callback<void(RenderViewHost*)>& take_screenshot_callback);
+
  private:
   friend class RestoreHelper;
   friend class WebContentsImpl;  // For invoking OnReservedPageIDRange.
 
   FRIEND_TEST_ALL_PREFIXES(TimeSmoother, Basic);
   FRIEND_TEST_ALL_PREFIXES(TimeSmoother, SingleDuplicate);
   FRIEND_TEST_ALL_PREFIXES(TimeSmoother, ManyDuplicates);
   FRIEND_TEST_ALL_PREFIXES(TimeSmoother, ClockBackwardsJump);
 
   // Helper class to smooth out runs of duplicate timestamps while still
@@ skipping 93 matching lines @@
   // Inserts up to |max_index| entries from |source| into this. This does NOT
   // adjust any of the members that reference entries_
   // (last_committed_entry_index_, pending_entry_index_ or
   // transient_entry_index_).
   void InsertEntriesFrom(const NavigationControllerImpl& source, int max_index);
 
   // Returns the navigation index that differs from the current entry by the
   // specified |offset|.  The index returned is not guaranteed to be valid.
   int GetIndexForOffset(int offset) const;
 
-  // Takes a screenshot of the page at the current state.
-  void TakeScreenshot();
-
   // The callback invoked when taking the screenshot of the page is complete.
   // This sets the screenshot on the navigation entry.
   void OnScreenshotTaken(int unique_id,
                          skia::PlatformBitmap* bitmap,
                          bool success);
 
   // ---------------------------------------------------------------------------
 
   // The user browser context associated with this controller.
   BrowserContext* browser_context_;
@@ skipping 56 matching lines @@
   // The maximum number of entries that a navigation controller can store.
   static size_t max_entry_count_for_testing_;
 
   // If a repost is pending, its type (RELOAD or RELOAD_IGNORING_CACHE),
   // NO_RELOAD otherwise.
   ReloadType pending_reload_;
 
   // Used to get timestamps for newly-created navigation entries.
   base::Callback<base::Time()> get_timestamp_callback_;
 
+  // A callback that gets called before taking the screenshot of the page. This
+  // is used only for testing.
+  base::Callback<void(RenderViewHost*)> take_screenshot_callback_;
+
   // Used to smooth out timestamps from |get_timestamp_callback_|.
   // Without this, whenever there is a run of redirects or
   // code-generated navigations, those navigations may occur within
   // the timer resolution, leading to things sometimes showing up in
   // the wrong order in the history view.
   TimeSmoother time_smoother_;
 
   DISALLOW_COPY_AND_ASSIGN(NavigationControllerImpl);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_WEB_CONTENTS_NAVIGATION_CONTROLLER_IMPL_H_