| Index: headless/lib/renderer/headless_content_renderer_client.cc
|
| diff --git a/headless/lib/renderer/headless_content_renderer_client.cc b/headless/lib/renderer/headless_content_renderer_client.cc
|
| index 5347d6092228090a42332c32e4416f31e25080a8..807b6570018fc56ffba948f17141914144959a61 100644
|
| --- a/headless/lib/renderer/headless_content_renderer_client.cc
|
| +++ b/headless/lib/renderer/headless_content_renderer_client.cc
|
| @@ -4,10 +4,139 @@
|
|
|
| #include "headless/lib/renderer/headless_content_renderer_client.h"
|
|
|
| +#include "base/strings/utf_string_conversions.h"
|
| +#include "content/public/renderer/render_frame.h"
|
| +
|
| namespace headless {
|
|
|
| HeadlessContentRendererClient::HeadlessContentRendererClient() {}
|
|
|
| HeadlessContentRendererClient::~HeadlessContentRendererClient() {}
|
|
|
| +namespace {
|
| +const char kMojoBindingsShim[] =
|
| + "// Shim to let code use define() instead of mojo.define() \n"
|
| + "window.define = (function() { \n"
|
| + " let moduleCache = new Map(); \n"
|
| + " return function(name, deps, factory) { \n"
|
| + " let promise = moduleCache.get(name); \n"
|
| + " if (promise === undefined) { \n"
|
| + " // This promise must be cached as mojo.define will only call the \n"
|
| + " // factory function the first time the module is defined. \n"
|
| + " promise = new Promise(resolve => { \n"
|
| + " mojo.define(name, deps, (...modules) => { \n"
|
| + " let result = factory(...modules); \n"
|
| + " return result; \n"
|
| + " }); \n"
|
| + " }); \n"
|
| + " moduleCache.set(name, promise); \n"
|
| + " } \n"
|
| + " return promise; \n"
|
| + " } \n"
|
| + "})(); \n"
|
| + " \n"
|
| + "// This code is run before the browser has sent us the mojo bindings so \n"
|
| + "// we need to get fancy and use nested proxy classes to define a promise\n"
|
| + "// to an arbitary window.mojo.services.myModule.myInterface. \n"
|
| + "if (window.hasOwnProperty('mojo')) { \n"
|
| + " let mojoBindings = new Map(); \n"
|
| + " let resolvePending = true; \n"
|
| + " \n"
|
| + " // Factory for returning the second level dynamic property which \n"
|
| + " // resolves to a promise to the corresponding mojo interface... \n"
|
| + " let interfaceProxyFactory = function(serviceName) { \n"
|
| + " return new Proxy({}, { \n"
|
| + " get: function(target, interfaceName) { \n"
|
| + " let name = serviceName + '::' + interfaceName; \n"
|
| + " let binding = mojoBindings.get(name); \n"
|
| + " if (binding === undefined) { \n"
|
| + " binding = {}; \n"
|
| + " binding.promise = new Promise(function(resolve, reject) { \n"
|
| + " binding.resolve = resolve; \n"
|
| + " binding.reject = reject; \n"
|
| + " if (!resolvePending) \n"
|
| + " reject(); \n"
|
| + " }); \n"
|
| + " mojoBindings.set(name, binding); \n"
|
| + " } \n"
|
| + " return binding.promise; \n"
|
| + " }, \n"
|
| + " set: function(target, name, value) { \n"
|
| + " throw new Error( \n"
|
| + " 'Assignment to the mojo services proxy is not allowed'); \n"
|
| + " } \n"
|
| + " }); \n"
|
| + " }; \n"
|
| + " \n"
|
| + " // Top level dynamic property. \n"
|
| + " window.mojo.services = new Proxy({}, { \n"
|
| + " get: function(target, serviceName) { \n"
|
| + " if (!(serviceName in target)) { \n"
|
| + " let interfaceProxy = interfaceProxyFactory(serviceName); \n"
|
| + " target[serviceName] = interfaceProxy; \n"
|
| + " return interfaceProxy; \n"
|
| + " } \n"
|
| + " return target[serviceName]; \n"
|
| + " }, \n"
|
| + " set: function(target, name, value) { \n"
|
| + " throw new Error( \n"
|
| + " 'Assignment to the mojo services proxy is not allowed'); \n"
|
| + " } \n"
|
| + " }); \n"
|
| + " \n"
|
| + " // Resolve promises for the listed |serviceNames|. \n"
|
| + " window.mojo.resolvePromisesForServices_ = function(serviceNames) { \n"
|
| + " let numServices = serviceNames.length; \n"
|
| + " for (let i = 0; i < numServices; ++i) { \n"
|
| + " let serviceName = serviceNames[i]; \n"
|
| + " // Use mojo to obtain the module binding. \n"
|
| + " define([ \n"
|
| + " serviceName, \n"
|
| + " 'mojo/public/js/core', \n"
|
| + " 'mojo/public/js/router', \n"
|
| + " 'content/public/renderer/frame_service_registry', \n"
|
| + " ], function(serviceMojom, mojoCore, routerModule, \n"
|
| + " serviceProvider) { \n"
|
| + " // A mojom binding may contain bindings for a number of \n"
|
| + " // interfaces. Iterate through all of them and resolve any \n"
|
| + " // promises. \n"
|
| + " for (let m in serviceMojom) { \n"
|
| + " if (typeof serviceMojom[m] == 'object') { \n"
|
| + " let service = serviceMojom[m]; \n"
|
| + " let binding = mojoBindings.get(service.name); \n"
|
| + " let interface = new service.proxyClass( \n"
|
| + " new routerModule.Router( \n"
|
| + " serviceProvider.connectToService(service.name))); \n"
|
| + " if (binding === undefined) { \n"
|
| + " // Store resolved promise in case binding is requested. \n"
|
| + " mojoBindings.set( \n"
|
| + " service.name, \n"
|
| + " {'promise': Promise.resolve(interface)}); \n"
|
| + " } else { \n"
|
| + " binding.resolve(interface); \n"
|
| + " // Since this promise has been resolved we don't want to \n"
|
| + " // try and reject it below! \n"
|
| + " delete binding.reject; \n"
|
| + " } \n"
|
| + " } \n"
|
| + " } \n"
|
| + " }); \n"
|
| + " } \n"
|
| + " // Reject any remaining promises that didn't get resolved. \n"
|
| + " for (let [key, value] of mojoBindings) { \n"
|
| + " if ('reject' in value) { \n"
|
| + " value.reject(); \n"
|
| + " } \n"
|
| + " } \n"
|
| + " // Reject any subsequent unknown properties. \n"
|
| + " resolvePending = false; \n"
|
| + " }; \n"
|
| + "}";
|
| +} // namespace
|
| +
|
| +void HeadlessContentRendererClient::RunScriptsAtDocumentStart(
|
| + content::RenderFrame* render_frame) {
|
| + render_frame->ExecuteJavaScript(base::UTF8ToUTF16(kMojoBindingsShim));
|
| +}
|
| +
|
| } // namespace headless
|
|
|
Chromium Code Reviews
Issue 2049363003:
Adds support for headless chrome embedder mojo services (Closed)
Base URL: https://chromium.googlesource.com/chromium/src.git@master