headless: Allow per-context protocol handlers

headless/lib/browser/headless_browser_context_impl.h

 // 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.
 
 #ifndef HEADLESS_LIB_BROWSER_HEADLESS_BROWSER_CONTEXT_IMPL_H_
 #define HEADLESS_LIB_BROWSER_HEADLESS_BROWSER_CONTEXT_IMPL_H_
 
 #include <memory>
 
 #include "base/files/file_path.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/browser/resource_context.h"
 #include "headless/lib/browser/headless_url_request_context_getter.h"
 #include "headless/public/headless_browser.h"
 #include "headless/public/headless_browser_context.h"
 
 namespace headless {
 class HeadlessResourceContext;
 
 class HeadlessBrowserContextImpl : public HeadlessBrowserContext,
                                    public content::BrowserContext {
  public:
-  explicit HeadlessBrowserContextImpl(HeadlessBrowser::Options* options);
+  explicit HeadlessBrowserContextImpl(ProtocolHandlerMap protocol_handlers,
+                                      HeadlessBrowser::Options* options);
   ~HeadlessBrowserContextImpl() override;
 
   static HeadlessBrowserContextImpl* From(
       HeadlessBrowserContext* browser_context);
 
   // BrowserContext implementation:
   std::unique_ptr<content::ZoomLevelDelegate> CreateZoomLevelDelegate(
       const base::FilePath& partition_path) override;
   base::FilePath GetPath() const override;
   bool IsOffTheRecord() const override;
@@ skipping 19 matching lines @@
       bool in_memory) override;
 
   HeadlessBrowser::Options* options() const { return options_; }
   void SetOptionsForTesting(HeadlessBrowser::Options* options);
 
  private:
   // Performs initialization of the HeadlessBrowserContextImpl while IO is still
   // allowed on the current thread.
   void InitWhileIOAllowed();
 
+  ProtocolHandlerMap protocol_handlers_;
+  HeadlessBrowser::Options* options_;  // Not owned.
+  std::unique_ptr<HeadlessResourceContext> resource_context_;
   base::FilePath path_;
-  std::unique_ptr<HeadlessResourceContext> resource_context_;
-  HeadlessBrowser::Options* options_;  // Not owned.
 
   DISALLOW_COPY_AND_ASSIGN(HeadlessBrowserContextImpl);
 };
 
 }  // namespace headless
 
 #endif  // HEADLESS_LIB_BROWSER_HEADLESS_BROWSER_CONTEXT_IMPL_H_