Adds support for headless chrome embedder mojo services

headless/public/headless_web_contents.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_PUBLIC_HEADLESS_WEB_CONTENTS_H_
 #define HEADLESS_PUBLIC_HEADLESS_WEB_CONTENTS_H_
 
-#include "base/callback.h"
+#include <list>
+
+#include "base/bind.h"
 #include "base/macros.h"
 #include "headless/public/headless_export.h"
+#include "mojo/public/cpp/bindings/interface_request.h"
 #include "ui/gfx/geometry/size.h"
 #include "url/gurl.h"
 
 namespace headless {
 class HeadlessBrowserContext;
 class HeadlessBrowserImpl;
 class HeadlessDevToolsTarget;
 
 // Class representing contents of a browser tab. Should be accessed from browser
 // main thread.
@@ skipping 54 matching lines @@
 
   // Specify the initial window size (default is 800x600).
   Builder& SetWindowSize(const gfx::Size& size);
 
   // Set a browser context for storing session data (e.g., cookies, cache, local
   // storage) for the tab. Several tabs can share the same browser context. If
   // unset, the default browser context will be used. The browser context must
   // outlive this HeadlessWebContents.
   Builder& SetBrowserContext(HeadlessBrowserContext* browser_context);
 
+  // Specify an embedder provided Mojo service to be installed.

[Sami, 2016/06/10 10:27:39]

Please document the arguments.

[alex clarke (OOO till 29th), 2016/06/11 20:51:50]

Done.

+  template <typename Interface>
+  Builder& AddEmbedderMojoService(
+      const base::Callback<void(mojo::InterfaceRequest<Interface>)>&
+          service_factory,
+      const std::string& js_binding) {
+    return AddEmbedderMojoService(
+        Interface::Name_,
+        base::Bind(&Builder::ForwardToServiceFactory<Interface>,
+                   service_factory),
+        js_binding);
+  }
+  Builder& AddEmbedderMojoService(
+      const std::string& service_name,
+      const base::Callback<void(mojo::ScopedMessagePipeHandle)>&
+          service_factory,
+      const std::string& js_binding);
+
   // The returned object is owned by HeadlessBrowser. Call
   // HeadlessWebContents::Close() to dispose it.
   HeadlessWebContents* Build();
 
  private:
   friend class HeadlessBrowserImpl;
   friend class HeadlessWebContentsImpl;
 
+  template <typename Interface>
+  static void ForwardToServiceFactory(
+      const base::Callback<void(mojo::InterfaceRequest<Interface>)>&
+          service_factory,
+      mojo::ScopedMessagePipeHandle handle) {
+    service_factory.Run(mojo::MakeRequest<Interface>(std::move(handle)));
+  }
+
+  struct EmbedderMojoService {
+    EmbedderMojoService();
+    EmbedderMojoService(const std::string& service_name,
+                        const base::Callback<void(
+                            mojo::ScopedMessagePipeHandle)>& service_factory,
+                        const std::string& js_binding);
+    ~EmbedderMojoService();
+
+    std::string service_name;
+    base::Callback<void(mojo::ScopedMessagePipeHandle)> service_factory;
+    std::string js_binding;
+
+    DISALLOW_COPY_AND_ASSIGN(EmbedderMojoService);

[Sami, 2016/06/10 10:27:39]

private:

[alex clarke (OOO till 29th), 2016/06/11 20:51:50]

Done.

+  };
+
   explicit Builder(HeadlessBrowserImpl* browser);
 
   HeadlessBrowserImpl* browser_;
   GURL initial_url_ = GURL("about:blank");
   gfx::Size window_size_ = gfx::Size(800, 600);
   HeadlessBrowserContext* browser_context_;
+  std::list<EmbedderMojoService> embedder_mojo_services_;
 
   DISALLOW_COPY_AND_ASSIGN(Builder);
 };
 
 }  // namespace headless
 
 #endif  // HEADLESS_PUBLIC_HEADLESS_WEB_CONTENTS_H_