Adding Navigation Throttles to DevTools

content/browser/devtools/protocol/page_handler.h

 // Copyright 2014 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_DEVTOOLS_PROTOCOL_PAGE_HANDLER_H_
 #define CONTENT_BROWSER_DEVTOOLS_PROTOCOL_PAGE_HANDLER_H_
 
 #include <stddef.h>
 
 #include "base/compiler_specific.h"
 #include "base/macros.h"
 #include "base/memory/weak_ptr.h"
 #include "base/time/time.h"
 #include "cc/output/compositor_frame_metadata.h"
 #include "content/browser/devtools/protocol/devtools_protocol_dispatcher.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 #include "content/public/browser/readback_types.h"
 
 class SkBitmap;
 
 namespace content {
 
+class NavigationHandle;
 class RenderFrameHostImpl;
 class WebContentsImpl;
 
 namespace devtools {
+class PageNavigationThrottle;
+
 namespace page {
 
 class ColorPicker;
 
 class PageHandler : public NotificationObserver {
  public:
   typedef DevToolsProtocolClient::Response Response;
 
   PageHandler();
   ~PageHandler() override;
@@ skipping 34 matching lines @@
   Response ScreencastFrameAck(int session_id);
 
   Response HandleJavaScriptDialog(bool accept, const std::string* prompt_text);
 
   Response QueryUsageAndQuota(DevToolsCommandId command_id,
                               const std::string& security_origin);
 
   Response SetColorPickerEnabled(bool enabled);
   Response RequestAppBanner();
 
+  Response SetControlNavigations(bool enabled);
+  Response ProcessNavigation(const std::string& response, int navigation_id);
+
+  std::unique_ptr<PageNavigationThrottle> CreateThrottleForNavigation(
+      NavigationHandle* navigation_handle);
+
+  void OnPageNavigationThrottleDisposed(int navigation_id);
+  void NavigationRequested(const PageNavigationThrottle* throttle);
+
  private:
   WebContentsImpl* GetWebContents();
   void NotifyScreencastVisibility(bool visible);
   void InnerSwapCompositorFrame();
   void ScreencastFrameCaptured(cc::CompositorFrameMetadata metadata,
                                const SkBitmap& bitmap,
                                ReadbackResponse response);
   void ScreencastFrameEncoded(cc::CompositorFrameMetadata metadata,
                               const base::Time& timestamp,
                               const std::string& data);
@@ skipping 21 matching lines @@
   int capture_retry_count_;
   bool has_compositor_frame_metadata_;
   cc::CompositorFrameMetadata next_compositor_frame_metadata_;
   cc::CompositorFrameMetadata last_compositor_frame_metadata_;
   int session_id_;
   int frame_counter_;
   int frames_in_flight_;
 
   std::unique_ptr<ColorPicker> color_picker_;
 
+  bool navigation_throttle_enabled_;
+  int next_navigation_id_;
+  std::map<int, PageNavigationThrottle*> navigation_throttles_;
+
   RenderFrameHostImpl* host_;
   std::unique_ptr<Client> client_;
   NotificationRegistrar registrar_;
   base::WeakPtrFactory<PageHandler> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(PageHandler);
 };
 
 }  // namespace page
 }  // namespace devtools
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_DEVTOOLS_PROTOCOL_PAGE_HANDLER_H_