Integrates the SnapshotController into the PrerenderingLoader to provide logic for determining when…

chrome/browser/android/offline_pages/prerendering_loader.h

 // Copyright 2016 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 CHROME_BROWSER_ANDROID_OFFLINE_PAGES_PRERENDERING_LOADER_H_
 #define CHROME_BROWSER_ANDROID_OFFLINE_PAGES_PRERENDERING_LOADER_H_
 
 #include <memory>
 
 #include "base/callback.h"
 #include "chrome/browser/android/offline_pages/prerender_adapter.h"
 #include "components/offline_pages/background/offliner.h"
+#include "components/offline_pages/snapshot_controller.h"
 
 class GURL;
 
 namespace content {
 class BrowserContext;
 class WebContents;
 class SessionStorageNamespace;
 }  // namespace content
 
 namespace gfx {
 class Size;
 }  // namespace gfx
 
 namespace offline_pages {
 
 // A client-side page loader that integrates with the PrerenderManager to do
 // the page loading in the background. It operates on a single thread and
 // needs to run on BrowserThread::UI to work with the PrerenderManager.
 // It supports a single load request at a time.
-class PrerenderingLoader : public PrerenderAdapter::Observer {
+class PrerenderingLoader : public PrerenderAdapter::Observer,
+                           public SnapshotController::Client {
  public:
   // Reports status of a load page request with loaded contents if available.
   typedef base::Callback<void(Offliner::RequestStatus, content::WebContents*)>
       LoadPageCallback;
 
   explicit PrerenderingLoader(content::BrowserContext* browser_context);
   ~PrerenderingLoader() override;
 
   // Loads a page in the background if possible and returns whether the
   // request was accepted. If so, the LoadPageCallback will be informed
@@ skipping 29 matching lines @@
   // Overrides the prerender stack adapter for unit testing.
   void SetAdapterForTesting(
       std::unique_ptr<PrerenderAdapter> prerender_adapter);
 
   // PrerenderAdapter::Observer implementation:
   void OnPrerenderStart() override;
   void OnPrerenderStopLoading() override;
   void OnPrerenderDomContentLoaded() override;
   void OnPrerenderStop() override;
 
+  // SnapshotController::Client implementation:
+  void StartSnapshot() override;
+
  private:
   // State of the loader (only one request may be active at a time).
   enum class State {
     IDLE,     // No active load request.
     PENDING,  // Load request is pending the start of prerendering.
     LOADING,  // Loading in progress.
     LOADED,   // Loaded and now waiting for requestor to StopLoading().
   };
 
   // Handles some event/signal that the load request has succeeded or failed.
   // It may be due to some asynchronous trigger that occurs after the request
   // has completed for some other reason/event.
   void HandleLoadEvent();
 
   // Handles some event/signal that loading has stopped (whether due to a
   // failure, cancel, or stop request). It may be due to some asynchronous
   // trigger that occurs after the request has stopped for some other reason.
   void HandleLoadingStopped();
 
   // Cancels any current prerender and moves loader to idle state.
   void CancelPrerender();
 
   // Tracks loading state including whether the Loader is idle.
   State state_;
 
+  // Handles determining when to report page is LOADED.
+  std::unique_ptr<SnapshotController> snapshot_controller_;
+
   // Not owned.
   content::BrowserContext* browser_context_;
 
   // Adapter for handling calls to the prerender stack.
   std::unique_ptr<PrerenderAdapter> adapter_;
 
   // A WebContents for the active load request that is used to hold the session
   // storage namespace for rendering. This will NOT have the loaded page.
   std::unique_ptr<content::WebContents> session_contents_;
 
   // Callback to call when the active load request completes, fails, or is
   // canceled.
   LoadPageCallback callback_;
 
   DISALLOW_COPY_AND_ASSIGN(PrerenderingLoader);
 };
 
 }  // namespace offline_pages
 
 #endif  // CHROME_BROWSER_ANDROID_OFFLINE_PAGES_PRERENDERING_LOADER_H_