Adds support for headless chrome embedder mojo services

content/renderer/mojo_context_state.cc

 // 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.
 
 #include "content/renderer/mojo_context_state.h"
 
 #include <stddef.h>
 
 #include <map>
 #include <string>
 
 #include "base/bind.h"
 #include "base/lazy_instance.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/ref_counted_memory.h"
 #include "base/stl_util.h"
 #include "content/grit/content_resources.h"
 #include "content/public/common/content_client.h"
 #include "content/public/renderer/render_frame.h"
 #include "content/public/renderer/resource_fetcher.h"
+#include "content/renderer/mojo_bindings_controller.h"
 #include "content/renderer/mojo_main_runner.h"
 #include "gin/converter.h"
 #include "gin/modules/module_registry.h"
 #include "gin/per_context_data.h"
 #include "gin/public/context_holder.h"
 #include "gin/try_catch.h"
 #include "mojo/public/js/constants.h"
 #include "third_party/WebKit/public/platform/WebSecurityOrigin.h"
 #include "third_party/WebKit/public/platform/WebURLResponse.h"
 #include "third_party/WebKit/public/web/WebFrame.h"
@@ skipping 52 matching lines @@
     }
   }
 
   DCHECK(module_sources);
   auto source_iter = module_sources->find(path);
   if (source_iter == module_sources->end())
     return nullptr;
   return source_iter->second;
 }
 
+std::string GetModulePrefixForBindingsType(MojoBindingsType bindings_type,
+                                           blink::WebFrame* frame) {
+  switch (bindings_type) {
+    case MojoBindingsType::FOR_UI_BINDINGS:
+      return frame->getSecurityOrigin().toString().utf8() + "/";
+    case MojoBindingsType::FOR_LAYOUT_TESTS:
+      return "layout-test-mojom://";
+    case MojoBindingsType::FOR_HEADLESS:
+      return "headless-mojom://";

[jochen (gone - plz use gerrit), 2016/06/27 12:31:57]

this is all a bit of a layering violation, but admittedly already existed before

[alex clarke (OOO till 29th), 2016/07/29 09:06:23]

Acknowledged.

+  }
+  NOTREACHED();
+  return "";
+}
+
 }  // namespace
 
 MojoContextState::MojoContextState(blink::WebFrame* frame,
                                    v8::Local<v8::Context> context,
-                                   bool for_layout_tests)
+                                   MojoBindingsType bindings_type)
     : frame_(frame),
       module_added_(false),
-      module_prefix_(for_layout_tests
-                         ? "layout-test-mojom://"
-                         : frame_->getSecurityOrigin().toString().utf8() +
-                               "/") {
+      module_prefix_(GetModulePrefixForBindingsType(bindings_type, frame)) {
   gin::PerContextData* context_data = gin::PerContextData::From(context);
   gin::ContextHolder* context_holder = context_data->context_holder();
   runner_.reset(new MojoMainRunner(frame_, context_holder));
   gin::Runner::Scope scoper(runner_.get());
   gin::ModuleRegistry::From(context)->AddObserver(this);
   content::RenderFrame::FromWebFrame(frame)
       ->EnsureMojoBuiltinsAreAvailable(context_holder->isolate(), context);
   v8::Local<v8::Object> install_target;
-  if (for_layout_tests) {
+  if (bindings_type == MojoBindingsType::FOR_LAYOUT_TESTS) {
     // In layout tests we install the module system under 'mojo.define'
     // for now to avoid globally exposing something as generic as 'define'.
     //
     // TODO(rockot): Remove this if/when we can integrate gin + ES6 modules.
     install_target = v8::Object::New(context->GetIsolate());
     gin::SetProperty(context->GetIsolate(), context->Global(),
                      gin::StringToSymbol(context->GetIsolate(), "mojo"),
                      install_target);
   } else {
     // Otherwise we're fine installing a global 'define'.
@@ skipping 79 matching lines @@
     const std::vector<std::string>& dependencies) {
   FetchModules(dependencies);
 
   gin::ContextHolder* context_holder = runner_->GetContextHolder();
   gin::ModuleRegistry* registry = gin::ModuleRegistry::From(
       context_holder->context());
   registry->AttemptToLoadMoreModules(context_holder->isolate());
 }
 
 }  // namespace content