Add HeadlessTabSocket

headless/lib/browser/headless_browser_context_impl.cc

 // 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.
 
 #include "headless/lib/browser/headless_browser_context_impl.h"
 
 #include <memory>
 #include <string>
 #include <utility>
 #include <vector>
 
 #include "base/guid.h"
 #include "base/memory/ptr_util.h"
 #include "base/path_service.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/resource_context.h"
 #include "content/public/browser/storage_partition.h"
+#include "headless/grit/headless_lib_resources.h"
 #include "headless/lib/browser/headless_browser_context_options.h"
 #include "headless/lib/browser/headless_browser_impl.h"
 #include "headless/lib/browser/headless_permission_manager.h"
 #include "headless/lib/browser/headless_url_request_context_getter.h"
 #include "headless/public/util/black_hole_protocol_handler.h"
 #include "headless/public/util/in_memory_protocol_handler.h"
 #include "net/url_request/url_request_context.h"
+#include "ui/base/resource/resource_bundle.h"
 
 namespace headless {
 
 namespace {
 const char kHeadlessMojomProtocol[] = "headless-mojom";
 }
 
 // Contains net::URLRequestContextGetter required for resource loading.
 // Must be destructed on the IO thread as per content::ResourceContext
 // requirements.
@@ skipping 324 matching lines @@
 
 HeadlessBrowserContext::Builder&
 HeadlessBrowserContext::Builder::AddJsMojoBindings(
     const std::string& mojom_name,
     const std::string& js_bindings) {
   mojo_bindings_.emplace_back(mojom_name, js_bindings);
   return *this;
 }
 
 HeadlessBrowserContext::Builder&
+HeadlessBrowserContext::Builder::AddTabSocketMojoBindings() {
+  std::string js_bindings =
+      ui::ResourceBundle::GetSharedInstance()
+          .GetRawDataResource(IDR_HEADLESS_TAB_SOCKET_MOJOM_JS)
+          .as_string();
+  mojo_bindings_.emplace_back("headless/lib/tab_socket.mojom", js_bindings);
+  return *this;
+}
+
+HeadlessBrowserContext::Builder&
 HeadlessBrowserContext::Builder::EnableUnsafeNetworkAccessWithMojoBindings(
     bool enable_http_and_https_if_mojo_used) {
   enable_http_and_https_if_mojo_used_ = enable_http_and_https_if_mojo_used;
   return *this;
 }
 
 HeadlessBrowserContext::Builder&
 HeadlessBrowserContext::Builder::SetOverrideWebPreferencesCallback(
     base::Callback<void(WebPreferences*)> callback) {
   options_->override_web_preferences_callback_ = std::move(callback);
@@ skipping 31 matching lines @@
 HeadlessBrowserContext::Builder::MojoBindings::MojoBindings() {}
 
 HeadlessBrowserContext::Builder::MojoBindings::MojoBindings(
     const std::string& mojom_name,
     const std::string& js_bindings)
     : mojom_name(mojom_name), js_bindings(js_bindings) {}
 
 HeadlessBrowserContext::Builder::MojoBindings::~MojoBindings() {}
 
 }  // namespace headless