WebAssemblyJavascriptGenerator.js 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230
  1. /*
  2. MIT License http://www.opensource.org/licenses/mit-license.php
  3. Author Tobias Koppers @sokra
  4. */
  5. "use strict";
  6. const { RawSource } = require("webpack-sources");
  7. const { UsageState } = require("../ExportsInfo");
  8. const Generator = require("../Generator");
  9. const InitFragment = require("../InitFragment");
  10. const { WEBASSEMBLY_TYPES } = require("../ModuleSourceTypesConstants");
  11. const RuntimeGlobals = require("../RuntimeGlobals");
  12. const Template = require("../Template");
  13. const ModuleDependency = require("../dependencies/ModuleDependency");
  14. const WebAssemblyExportImportedDependency = require("../dependencies/WebAssemblyExportImportedDependency");
  15. const WebAssemblyImportDependency = require("../dependencies/WebAssemblyImportDependency");
  16. /** @typedef {import("webpack-sources").Source} Source */
  17. /** @typedef {import("../Dependency")} Dependency */
  18. /** @typedef {import("../DependencyTemplates")} DependencyTemplates */
  19. /** @typedef {import("../Generator").GenerateContext} GenerateContext */
  20. /** @typedef {import("../Module")} Module */
  21. /** @typedef {import("../Module").SourceTypes} SourceTypes */
  22. /** @typedef {import("../NormalModule")} NormalModule */
  23. /** @typedef {import("../RuntimeTemplate")} RuntimeTemplate */
  24. class WebAssemblyJavascriptGenerator extends Generator {
  25. /**
  26. * @param {NormalModule} module fresh module
  27. * @returns {SourceTypes} available types (do not mutate)
  28. */
  29. getTypes(module) {
  30. return WEBASSEMBLY_TYPES;
  31. }
  32. /**
  33. * @param {NormalModule} module the module
  34. * @param {string=} type source type
  35. * @returns {number} estimate size of the module
  36. */
  37. getSize(module, type) {
  38. return 95 + module.dependencies.length * 5;
  39. }
  40. /**
  41. * @param {NormalModule} module module for which the code should be generated
  42. * @param {GenerateContext} generateContext context for generate
  43. * @returns {Source | null} generated code
  44. */
  45. generate(module, generateContext) {
  46. const {
  47. runtimeTemplate,
  48. moduleGraph,
  49. chunkGraph,
  50. runtimeRequirements,
  51. runtime
  52. } = generateContext;
  53. /** @type {InitFragment<InitFragment<string>>[]} */
  54. const initFragments = [];
  55. const exportsInfo = moduleGraph.getExportsInfo(module);
  56. let needExportsCopy = false;
  57. const importedModules = new Map();
  58. const initParams = [];
  59. let index = 0;
  60. for (const dep of module.dependencies) {
  61. const moduleDep =
  62. dep && dep instanceof ModuleDependency ? dep : undefined;
  63. if (moduleGraph.getModule(dep)) {
  64. let importData = importedModules.get(moduleGraph.getModule(dep));
  65. if (importData === undefined) {
  66. importedModules.set(
  67. moduleGraph.getModule(dep),
  68. (importData = {
  69. importVar: `m${index}`,
  70. index,
  71. request: (moduleDep && moduleDep.userRequest) || undefined,
  72. names: new Set(),
  73. reexports: []
  74. })
  75. );
  76. index++;
  77. }
  78. if (dep instanceof WebAssemblyImportDependency) {
  79. importData.names.add(dep.name);
  80. if (dep.description.type === "GlobalType") {
  81. const exportName = dep.name;
  82. const importedModule = moduleGraph.getModule(dep);
  83. if (importedModule) {
  84. const usedName = moduleGraph
  85. .getExportsInfo(importedModule)
  86. .getUsedName(exportName, runtime);
  87. if (usedName) {
  88. initParams.push(
  89. runtimeTemplate.exportFromImport({
  90. moduleGraph,
  91. chunkGraph,
  92. module: importedModule,
  93. request: dep.request,
  94. importVar: importData.importVar,
  95. originModule: module,
  96. exportName: dep.name,
  97. asiSafe: true,
  98. isCall: false,
  99. callContext: null,
  100. defaultInterop: true,
  101. initFragments,
  102. runtime,
  103. runtimeRequirements
  104. })
  105. );
  106. }
  107. }
  108. }
  109. }
  110. if (dep instanceof WebAssemblyExportImportedDependency) {
  111. importData.names.add(dep.name);
  112. const usedName = moduleGraph
  113. .getExportsInfo(module)
  114. .getUsedName(dep.exportName, runtime);
  115. if (usedName) {
  116. runtimeRequirements.add(RuntimeGlobals.exports);
  117. const exportProp = `${module.exportsArgument}[${JSON.stringify(
  118. usedName
  119. )}]`;
  120. const defineStatement = Template.asString([
  121. `${exportProp} = ${runtimeTemplate.exportFromImport({
  122. moduleGraph,
  123. module: /** @type {Module} */ (moduleGraph.getModule(dep)),
  124. chunkGraph,
  125. request: dep.request,
  126. importVar: importData.importVar,
  127. originModule: module,
  128. exportName: dep.name,
  129. asiSafe: true,
  130. isCall: false,
  131. callContext: null,
  132. defaultInterop: true,
  133. initFragments,
  134. runtime,
  135. runtimeRequirements
  136. })};`,
  137. `if(WebAssembly.Global) ${exportProp} = ` +
  138. `new WebAssembly.Global({ value: ${JSON.stringify(
  139. dep.valueType
  140. )} }, ${exportProp});`
  141. ]);
  142. importData.reexports.push(defineStatement);
  143. needExportsCopy = true;
  144. }
  145. }
  146. }
  147. }
  148. const importsCode = Template.asString(
  149. Array.from(
  150. importedModules,
  151. ([module, { importVar, request, reexports }]) => {
  152. const importStatement = runtimeTemplate.importStatement({
  153. module,
  154. moduleGraph,
  155. chunkGraph,
  156. request,
  157. importVar,
  158. originModule: module,
  159. runtimeRequirements
  160. });
  161. return importStatement[0] + importStatement[1] + reexports.join("\n");
  162. }
  163. )
  164. );
  165. const copyAllExports =
  166. exportsInfo.otherExportsInfo.getUsed(runtime) === UsageState.Unused &&
  167. !needExportsCopy;
  168. // need these globals
  169. runtimeRequirements.add(RuntimeGlobals.module);
  170. runtimeRequirements.add(RuntimeGlobals.moduleId);
  171. runtimeRequirements.add(RuntimeGlobals.wasmInstances);
  172. if (exportsInfo.otherExportsInfo.getUsed(runtime) !== UsageState.Unused) {
  173. runtimeRequirements.add(RuntimeGlobals.makeNamespaceObject);
  174. runtimeRequirements.add(RuntimeGlobals.exports);
  175. }
  176. if (!copyAllExports) {
  177. runtimeRequirements.add(RuntimeGlobals.exports);
  178. }
  179. // create source
  180. const source = new RawSource(
  181. [
  182. '"use strict";',
  183. "// Instantiate WebAssembly module",
  184. `var wasmExports = ${RuntimeGlobals.wasmInstances}[${module.moduleArgument}.id];`,
  185. exportsInfo.otherExportsInfo.getUsed(runtime) !== UsageState.Unused
  186. ? `${RuntimeGlobals.makeNamespaceObject}(${module.exportsArgument});`
  187. : "",
  188. // this must be before import for circular dependencies
  189. "// export exports from WebAssembly module",
  190. copyAllExports
  191. ? `${module.moduleArgument}.exports = wasmExports;`
  192. : "for(var name in wasmExports) " +
  193. "if(name) " +
  194. `${module.exportsArgument}[name] = wasmExports[name];`,
  195. "// exec imports from WebAssembly module (for esm order)",
  196. importsCode,
  197. "",
  198. "// exec wasm module",
  199. `wasmExports[""](${initParams.join(", ")})`
  200. ].join("\n")
  201. );
  202. return InitFragment.addToSource(source, initFragments, generateContext);
  203. }
  204. /**
  205. * @param {Error} error the error
  206. * @param {NormalModule} module module for which the code should be generated
  207. * @param {GenerateContext} generateContext context for generate
  208. * @returns {Source | null} generated code
  209. */
  210. generateError(error, module, generateContext) {
  211. return new RawSource(`throw new Error(${JSON.stringify(error.message)});`);
  212. }
  213. }
  214. module.exports = WebAssemblyJavascriptGenerator;