OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "headless/lib/renderer/headless_content_renderer_client.h" | 5 #include "headless/lib/renderer/headless_content_renderer_client.h" |
6 | 6 |
| 7 #include "base/strings/utf_string_conversions.h" |
| 8 #include "content/public/renderer/render_frame.h" |
| 9 |
7 namespace headless { | 10 namespace headless { |
8 | 11 |
9 HeadlessContentRendererClient::HeadlessContentRendererClient() {} | 12 HeadlessContentRendererClient::HeadlessContentRendererClient() {} |
10 | 13 |
11 HeadlessContentRendererClient::~HeadlessContentRendererClient() {} | 14 HeadlessContentRendererClient::~HeadlessContentRendererClient() {} |
12 | 15 |
| 16 namespace { |
| 17 const char kMojoBindingsShim[] = |
| 18 "// Shim to let code use define() instead of mojo.define() \n" |
| 19 "window.define = (function() { \n" |
| 20 " let moduleCache = new Map(); \n" |
| 21 " return function(name, deps, factory) { \n" |
| 22 " let promise = moduleCache.get(name); \n" |
| 23 " if (promise === undefined) { \n" |
| 24 " // This promise must be cached as mojo.define will only call the \n" |
| 25 " // factory function the first time the module is defined. \n" |
| 26 " promise = new Promise(resolve => { \n" |
| 27 " mojo.define(name, deps, (...modules) => { \n" |
| 28 " let result = factory(...modules); \n" |
| 29 " return result; \n" |
| 30 " }); \n" |
| 31 " }); \n" |
| 32 " moduleCache.set(name, promise); \n" |
| 33 " } \n" |
| 34 " return promise; \n" |
| 35 " } \n" |
| 36 "})(); \n" |
| 37 " \n" |
| 38 "// This code is run before the browser has sent us the mojo bindings so \n" |
| 39 "// we need to get fancy and use nested proxy classes to define a promise\n" |
| 40 "// to an arbitary window.mojo.services.myModule.myInterface. \n" |
| 41 "if (window.hasOwnProperty('mojo')) { \n" |
| 42 " let mojoBindings = new Map(); \n" |
| 43 " let resolvePending = true; \n" |
| 44 " \n" |
| 45 " // Factory for returning the second level dynamic property which \n" |
| 46 " // resolves to a promise to the corresponding mojo interface... \n" |
| 47 " let interfaceProxyFactory = function(serviceName) { \n" |
| 48 " return new Proxy({}, { \n" |
| 49 " get: function(target, interfaceName) { \n" |
| 50 " let name = serviceName + '::' + interfaceName; \n" |
| 51 " let binding = mojoBindings.get(name); \n" |
| 52 " if (binding === undefined) { \n" |
| 53 " binding = {}; \n" |
| 54 " binding.promise = new Promise(function(resolve, reject) { \n" |
| 55 " binding.resolve = resolve; \n" |
| 56 " binding.reject = reject; \n" |
| 57 " if (!resolvePending) \n" |
| 58 " reject(); \n" |
| 59 " }); \n" |
| 60 " mojoBindings.set(name, binding); \n" |
| 61 " } \n" |
| 62 " return binding.promise; \n" |
| 63 " }, \n" |
| 64 " set: function(target, name, value) { \n" |
| 65 " throw new Error( \n" |
| 66 " 'Assignment to the mojo services proxy is not allowed'); \n" |
| 67 " } \n" |
| 68 " }); \n" |
| 69 " }; \n" |
| 70 " \n" |
| 71 " // Top level dynamic property. \n" |
| 72 " window.mojo.services = new Proxy({}, { \n" |
| 73 " get: function(target, serviceName) { \n" |
| 74 " if (!(serviceName in target)) { \n" |
| 75 " let interfaceProxy = interfaceProxyFactory(serviceName); \n" |
| 76 " target[serviceName] = interfaceProxy; \n" |
| 77 " return interfaceProxy; \n" |
| 78 " } \n" |
| 79 " return target[serviceName]; \n" |
| 80 " }, \n" |
| 81 " set: function(target, name, value) { \n" |
| 82 " throw new Error( \n" |
| 83 " 'Assignment to the mojo services proxy is not allowed'); \n" |
| 84 " } \n" |
| 85 " }); \n" |
| 86 " \n" |
| 87 " // Resolve promises for the listed |serviceNames|. \n" |
| 88 " window.mojo.resolvePromisesForServices_ = function(serviceNames) { \n" |
| 89 " let numServices = serviceNames.length; \n" |
| 90 " for (let i = 0; i < numServices; ++i) { \n" |
| 91 " let serviceName = serviceNames[i]; \n" |
| 92 " // Use mojo to obtain the module binding. \n" |
| 93 " define([ \n" |
| 94 " serviceName, \n" |
| 95 " 'mojo/public/js/core', \n" |
| 96 " 'mojo/public/js/router', \n" |
| 97 " 'content/public/renderer/frame_service_registry', \n" |
| 98 " ], function(serviceMojom, mojoCore, routerModule, \n" |
| 99 " serviceProvider) { \n" |
| 100 " // A mojom binding may contain bindings for a number of \n" |
| 101 " // interfaces. Iterate through all of them and resolve any \n" |
| 102 " // promises. \n" |
| 103 " for (let m in serviceMojom) { \n" |
| 104 " if (typeof serviceMojom[m] == 'object') { \n" |
| 105 " let service = serviceMojom[m]; \n" |
| 106 " let binding = mojoBindings.get(service.name); \n" |
| 107 " let interface = new service.proxyClass( \n" |
| 108 " new routerModule.Router( \n" |
| 109 " serviceProvider.connectToService(service.name))); \n" |
| 110 " if (binding === undefined) { \n" |
| 111 " // Store resolved promise in case binding is requested. \n" |
| 112 " mojoBindings.set( \n" |
| 113 " service.name, \n" |
| 114 " {'promise': Promise.resolve(interface)}); \n" |
| 115 " } else { \n" |
| 116 " binding.resolve(interface); \n" |
| 117 " // Since this promise has been resolved we don't want to \n" |
| 118 " // try and reject it below! \n" |
| 119 " delete binding.reject; \n" |
| 120 " } \n" |
| 121 " } \n" |
| 122 " } \n" |
| 123 " }); \n" |
| 124 " } \n" |
| 125 " // Reject any remaining promises that didn't get resolved. \n" |
| 126 " for (let [key, value] of mojoBindings) { \n" |
| 127 " if ('reject' in value) { \n" |
| 128 " value.reject(); \n" |
| 129 " } \n" |
| 130 " } \n" |
| 131 " // Reject any subsequent unknown properties. \n" |
| 132 " resolvePending = false; \n" |
| 133 " }; \n" |
| 134 "}"; |
| 135 } // namespace |
| 136 |
| 137 void HeadlessContentRendererClient::RunScriptsAtDocumentStart( |
| 138 content::RenderFrame* render_frame) { |
| 139 render_frame->ExecuteJavaScript(base::UTF8ToUTF16(kMojoBindingsShim)); |
| 140 } |
| 141 |
13 } // namespace headless | 142 } // namespace headless |
OLD | NEW |