Enable builtin Mojo JS modules in layout tests

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_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/WebURLResponse.h"
 #include "third_party/WebKit/public/web/WebFrame.h"
 #include "third_party/WebKit/public/web/WebScriptSource.h"
 #include "third_party/WebKit/public/web/WebSecurityOrigin.h"
 
 using v8::Context;
 using v8::HandleScope;
 using v8::Isolate;
 using v8::Object;
 using v8::ObjectTemplate;
 using v8::Script;
 
 namespace content {
 
 namespace {
 
 void RunMain(base::WeakPtr<gin::Runner> runner,
              v8::Local<v8::Value> module) {
   v8::Isolate* isolate = runner->GetContextHolder()->isolate();
   v8::Local<v8::Function> start;
   CHECK(gin::ConvertFromV8(isolate, module, &start));
   runner->Call(start, runner->global(), 0, NULL);
 }
 
+using ModuleSourceMap =
+    std::map<std::string, scoped_refptr<base::RefCountedMemory>>;
+
+base::LazyInstance<scoped_ptr<ModuleSourceMap>>::Leaky g_module_sources;
+
+scoped_refptr<base::RefCountedMemory> GetBuiltinModuleData(
+    const std::string& path) {
+  static const struct {
+    const char* path;
+    const int id;
+  } kBuiltinModuleResources[] = {
+    { mojo::kBindingsModuleName, IDR_MOJO_BINDINGS_JS },
+    { mojo::kBufferModuleName, IDR_MOJO_BUFFER_JS },
+    { mojo::kCodecModuleName, IDR_MOJO_CODEC_JS },
+    { mojo::kConnectionModuleName, IDR_MOJO_CONNECTION_JS },
+    { mojo::kConnectorModuleName, IDR_MOJO_CONNECTOR_JS },
+    { mojo::kRouterModuleName, IDR_MOJO_ROUTER_JS },
+    { mojo::kUnicodeModuleName, IDR_MOJO_UNICODE_JS },
+    { mojo::kValidatorModuleName, IDR_MOJO_VALIDATOR_JS },
+  };
+
+  scoped_ptr<ModuleSourceMap>& module_sources = g_module_sources.Get();
+  if (!module_sources) {
+    // Initialize the module source map on first access.
+    module_sources.reset(new ModuleSourceMap);
+    for (size_t i = 0; i < arraysize(kBuiltinModuleResources); ++i) {
+      const auto& resource = kBuiltinModuleResources[i];
+      scoped_refptr<base::RefCountedMemory> data =
+          GetContentClient()->GetDataResourceBytes(resource.id);
+      DCHECK_GT(data->size(), 0u);
+      module_sources->insert(std::make_pair(std::string(resource.path), data));
+    }
+  }
+
+  DCHECK(module_sources);
+  auto source_iter = module_sources->find(path);
+  if (source_iter == module_sources->end())
+    return nullptr;
+  return source_iter->second;
+}
+
 }  // namespace
 
 MojoContextState::MojoContextState(blink::WebFrame* frame,
-                                   v8::Local<v8::Context> context)
+                                   v8::Local<v8::Context> context,
+                                   bool for_layout_tests)
     : frame_(frame),
       module_added_(false),
       module_prefix_(frame_->securityOrigin().toString().utf8() + "/") {
   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);
-  gin::ModuleRegistry::InstallGlobals(context->GetIsolate(), context->Global());
+  v8::Local<v8::Object> install_target;
+  if (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'.
+    install_target = context->Global();
+  }
+  gin::ModuleRegistry::InstallGlobals(context->GetIsolate(), install_target);
   // Warning |frame| may be destroyed.
   // TODO(sky): add test for this.
 }
 
 MojoContextState::~MojoContextState() {
   gin::Runner::Scope scoper(runner_.get());
   gin::ModuleRegistry::From(
       runner_->GetContextHolder()->context())->RemoveObserver(this);
 }
 
 void MojoContextState::Run() {
   gin::ContextHolder* context_holder = runner_->GetContextHolder();
   gin::ModuleRegistry::From(context_holder->context())->LoadModule(
       context_holder->isolate(),
       "main",
       base::Bind(RunMain, runner_->GetWeakPtr()));
 }
 
 void MojoContextState::FetchModules(const std::vector<std::string>& ids) {
   gin::Runner::Scope scoper(runner_.get());
   gin::ContextHolder* context_holder = runner_->GetContextHolder();
   gin::ModuleRegistry* registry = gin::ModuleRegistry::From(
       context_holder->context());
   for (size_t i = 0; i < ids.size(); ++i) {
     if (fetched_modules_.find(ids[i]) == fetched_modules_.end() &&
         registry->available_modules().count(ids[i]) == 0) {
-      FetchModule(ids[i]);
+      scoped_refptr<base::RefCountedMemory> data = GetBuiltinModuleData(ids[i]);
+      if (data)
+        runner_->Run(std::string(data->front_as<char>(), data->size()), ids[i]);
+      else
+        FetchModule(ids[i]);
     }
   }
 }
 
 void MojoContextState::FetchModule(const std::string& id) {
   const GURL url(module_prefix_ + id);
   // TODO(sky): better error checks here?
   DCHECK(url.is_valid() && !url.is_empty());
   DCHECK(fetched_modules_.find(id) == fetched_modules_.end());
   fetched_modules_.insert(id);
@@ skipping 33 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