index.js 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.buildNamespaceInitStatements = buildNamespaceInitStatements;
  6. exports.ensureStatementsHoisted = ensureStatementsHoisted;
  7. Object.defineProperty(exports, "getDynamicImportSource", {
  8. enumerable: true,
  9. get: function () {
  10. return _dynamicImport.getDynamicImportSource;
  11. }
  12. });
  13. Object.defineProperty(exports, "getModuleName", {
  14. enumerable: true,
  15. get: function () {
  16. return _getModuleName.default;
  17. }
  18. });
  19. Object.defineProperty(exports, "hasExports", {
  20. enumerable: true,
  21. get: function () {
  22. return _normalizeAndLoadMetadata.hasExports;
  23. }
  24. });
  25. Object.defineProperty(exports, "isModule", {
  26. enumerable: true,
  27. get: function () {
  28. return _helperModuleImports.isModule;
  29. }
  30. });
  31. Object.defineProperty(exports, "isSideEffectImport", {
  32. enumerable: true,
  33. get: function () {
  34. return _normalizeAndLoadMetadata.isSideEffectImport;
  35. }
  36. });
  37. exports.rewriteModuleStatementsAndPrepareHeader = rewriteModuleStatementsAndPrepareHeader;
  38. Object.defineProperty(exports, "rewriteThis", {
  39. enumerable: true,
  40. get: function () {
  41. return _rewriteThis.default;
  42. }
  43. });
  44. exports.wrapInterop = wrapInterop;
  45. var _assert = require("assert");
  46. var _t = require("@babel/types");
  47. var _template = require("@babel/template");
  48. var _helperModuleImports = require("@babel/helper-module-imports");
  49. var _rewriteThis = require("./rewrite-this");
  50. var _rewriteLiveReferences = require("./rewrite-live-references");
  51. var _normalizeAndLoadMetadata = require("./normalize-and-load-metadata");
  52. var _dynamicImport = require("./dynamic-import");
  53. var _getModuleName = require("./get-module-name");
  54. const {
  55. booleanLiteral,
  56. callExpression,
  57. cloneNode,
  58. directive,
  59. directiveLiteral,
  60. expressionStatement,
  61. identifier,
  62. isIdentifier,
  63. memberExpression,
  64. stringLiteral,
  65. valueToNode,
  66. variableDeclaration,
  67. variableDeclarator
  68. } = _t;
  69. function rewriteModuleStatementsAndPrepareHeader(path, {
  70. loose,
  71. exportName,
  72. strict,
  73. allowTopLevelThis,
  74. strictMode,
  75. noInterop,
  76. importInterop = noInterop ? "none" : "babel",
  77. lazy,
  78. esNamespaceOnly,
  79. filename,
  80. constantReexports = loose,
  81. enumerableModuleMeta = loose,
  82. noIncompleteNsImportDetection
  83. }) {
  84. (0, _normalizeAndLoadMetadata.validateImportInteropOption)(importInterop);
  85. _assert((0, _helperModuleImports.isModule)(path), "Cannot process module statements in a script");
  86. path.node.sourceType = "script";
  87. const meta = (0, _normalizeAndLoadMetadata.default)(path, exportName, {
  88. importInterop,
  89. initializeReexports: constantReexports,
  90. lazy,
  91. esNamespaceOnly,
  92. filename
  93. });
  94. if (!allowTopLevelThis) {
  95. (0, _rewriteThis.default)(path);
  96. }
  97. (0, _rewriteLiveReferences.default)(path, meta);
  98. if (strictMode !== false) {
  99. const hasStrict = path.node.directives.some(directive => {
  100. return directive.value.value === "use strict";
  101. });
  102. if (!hasStrict) {
  103. path.unshiftContainer("directives", directive(directiveLiteral("use strict")));
  104. }
  105. }
  106. const headers = [];
  107. if ((0, _normalizeAndLoadMetadata.hasExports)(meta) && !strict) {
  108. headers.push(buildESModuleHeader(meta, enumerableModuleMeta));
  109. }
  110. const nameList = buildExportNameListDeclaration(path, meta);
  111. if (nameList) {
  112. meta.exportNameListName = nameList.name;
  113. headers.push(nameList.statement);
  114. }
  115. headers.push(...buildExportInitializationStatements(path, meta, constantReexports, noIncompleteNsImportDetection));
  116. return {
  117. meta,
  118. headers
  119. };
  120. }
  121. function ensureStatementsHoisted(statements) {
  122. statements.forEach(header => {
  123. header._blockHoist = 3;
  124. });
  125. }
  126. function wrapInterop(programPath, expr, type) {
  127. if (type === "none") {
  128. return null;
  129. }
  130. if (type === "node-namespace") {
  131. return callExpression(programPath.hub.addHelper("interopRequireWildcard"), [expr, booleanLiteral(true)]);
  132. } else if (type === "node-default") {
  133. return null;
  134. }
  135. let helper;
  136. if (type === "default") {
  137. helper = "interopRequireDefault";
  138. } else if (type === "namespace") {
  139. helper = "interopRequireWildcard";
  140. } else {
  141. throw new Error(`Unknown interop: ${type}`);
  142. }
  143. return callExpression(programPath.hub.addHelper(helper), [expr]);
  144. }
  145. function buildNamespaceInitStatements(metadata, sourceMetadata, constantReexports = false) {
  146. const statements = [];
  147. let srcNamespace = identifier(sourceMetadata.name);
  148. if (sourceMetadata.lazy) srcNamespace = callExpression(srcNamespace, []);
  149. for (const localName of sourceMetadata.importsNamespace) {
  150. if (localName === sourceMetadata.name) continue;
  151. statements.push(_template.default.statement`var NAME = SOURCE;`({
  152. NAME: localName,
  153. SOURCE: cloneNode(srcNamespace)
  154. }));
  155. }
  156. if (constantReexports) {
  157. statements.push(...buildReexportsFromMeta(metadata, sourceMetadata, true));
  158. }
  159. for (const exportName of sourceMetadata.reexportNamespace) {
  160. statements.push((sourceMetadata.lazy ? _template.default.statement`
  161. Object.defineProperty(EXPORTS, "NAME", {
  162. enumerable: true,
  163. get: function() {
  164. return NAMESPACE;
  165. }
  166. });
  167. ` : _template.default.statement`EXPORTS.NAME = NAMESPACE;`)({
  168. EXPORTS: metadata.exportName,
  169. NAME: exportName,
  170. NAMESPACE: cloneNode(srcNamespace)
  171. }));
  172. }
  173. if (sourceMetadata.reexportAll) {
  174. const statement = buildNamespaceReexport(metadata, cloneNode(srcNamespace), constantReexports);
  175. statement.loc = sourceMetadata.reexportAll.loc;
  176. statements.push(statement);
  177. }
  178. return statements;
  179. }
  180. const ReexportTemplate = {
  181. constant: _template.default.statement`EXPORTS.EXPORT_NAME = NAMESPACE_IMPORT;`,
  182. constantComputed: _template.default.statement`EXPORTS["EXPORT_NAME"] = NAMESPACE_IMPORT;`,
  183. spec: _template.default.statement`
  184. Object.defineProperty(EXPORTS, "EXPORT_NAME", {
  185. enumerable: true,
  186. get: function() {
  187. return NAMESPACE_IMPORT;
  188. },
  189. });
  190. `
  191. };
  192. function buildReexportsFromMeta(meta, metadata, constantReexports) {
  193. const namespace = metadata.lazy ? callExpression(identifier(metadata.name), []) : identifier(metadata.name);
  194. const {
  195. stringSpecifiers
  196. } = meta;
  197. return Array.from(metadata.reexports, ([exportName, importName]) => {
  198. let NAMESPACE_IMPORT = cloneNode(namespace);
  199. if (importName === "default" && metadata.interop === "node-default") {
  200. } else if (stringSpecifiers.has(importName)) {
  201. NAMESPACE_IMPORT = memberExpression(NAMESPACE_IMPORT, stringLiteral(importName), true);
  202. } else {
  203. NAMESPACE_IMPORT = memberExpression(NAMESPACE_IMPORT, identifier(importName));
  204. }
  205. const astNodes = {
  206. EXPORTS: meta.exportName,
  207. EXPORT_NAME: exportName,
  208. NAMESPACE_IMPORT
  209. };
  210. if (constantReexports || isIdentifier(NAMESPACE_IMPORT)) {
  211. if (stringSpecifiers.has(exportName)) {
  212. return ReexportTemplate.constantComputed(astNodes);
  213. } else {
  214. return ReexportTemplate.constant(astNodes);
  215. }
  216. } else {
  217. return ReexportTemplate.spec(astNodes);
  218. }
  219. });
  220. }
  221. function buildESModuleHeader(metadata, enumerableModuleMeta = false) {
  222. return (enumerableModuleMeta ? _template.default.statement`
  223. EXPORTS.__esModule = true;
  224. ` : _template.default.statement`
  225. Object.defineProperty(EXPORTS, "__esModule", {
  226. value: true,
  227. });
  228. `)({
  229. EXPORTS: metadata.exportName
  230. });
  231. }
  232. function buildNamespaceReexport(metadata, namespace, constantReexports) {
  233. return (constantReexports ? _template.default.statement`
  234. Object.keys(NAMESPACE).forEach(function(key) {
  235. if (key === "default" || key === "__esModule") return;
  236. VERIFY_NAME_LIST;
  237. if (key in EXPORTS && EXPORTS[key] === NAMESPACE[key]) return;
  238. EXPORTS[key] = NAMESPACE[key];
  239. });
  240. ` :
  241. _template.default.statement`
  242. Object.keys(NAMESPACE).forEach(function(key) {
  243. if (key === "default" || key === "__esModule") return;
  244. VERIFY_NAME_LIST;
  245. if (key in EXPORTS && EXPORTS[key] === NAMESPACE[key]) return;
  246. Object.defineProperty(EXPORTS, key, {
  247. enumerable: true,
  248. get: function() {
  249. return NAMESPACE[key];
  250. },
  251. });
  252. });
  253. `)({
  254. NAMESPACE: namespace,
  255. EXPORTS: metadata.exportName,
  256. VERIFY_NAME_LIST: metadata.exportNameListName ? (0, _template.default)`
  257. if (Object.prototype.hasOwnProperty.call(EXPORTS_LIST, key)) return;
  258. `({
  259. EXPORTS_LIST: metadata.exportNameListName
  260. }) : null
  261. });
  262. }
  263. function buildExportNameListDeclaration(programPath, metadata) {
  264. const exportedVars = Object.create(null);
  265. for (const data of metadata.local.values()) {
  266. for (const name of data.names) {
  267. exportedVars[name] = true;
  268. }
  269. }
  270. let hasReexport = false;
  271. for (const data of metadata.source.values()) {
  272. for (const exportName of data.reexports.keys()) {
  273. exportedVars[exportName] = true;
  274. }
  275. for (const exportName of data.reexportNamespace) {
  276. exportedVars[exportName] = true;
  277. }
  278. hasReexport = hasReexport || !!data.reexportAll;
  279. }
  280. if (!hasReexport || Object.keys(exportedVars).length === 0) return null;
  281. const name = programPath.scope.generateUidIdentifier("exportNames");
  282. delete exportedVars.default;
  283. return {
  284. name: name.name,
  285. statement: variableDeclaration("var", [variableDeclarator(name, valueToNode(exportedVars))])
  286. };
  287. }
  288. function buildExportInitializationStatements(programPath, metadata, constantReexports = false, noIncompleteNsImportDetection = false) {
  289. const initStatements = [];
  290. for (const [localName, data] of metadata.local) {
  291. if (data.kind === "import") {
  292. } else if (data.kind === "hoisted") {
  293. initStatements.push([
  294. data.names[0], buildInitStatement(metadata, data.names, identifier(localName))]);
  295. } else if (!noIncompleteNsImportDetection) {
  296. for (const exportName of data.names) {
  297. initStatements.push([exportName, null]);
  298. }
  299. }
  300. }
  301. for (const data of metadata.source.values()) {
  302. if (!constantReexports) {
  303. const reexportsStatements = buildReexportsFromMeta(metadata, data, false);
  304. const reexports = [...data.reexports.keys()];
  305. for (let i = 0; i < reexportsStatements.length; i++) {
  306. initStatements.push([reexports[i], reexportsStatements[i]]);
  307. }
  308. }
  309. if (!noIncompleteNsImportDetection) {
  310. for (const exportName of data.reexportNamespace) {
  311. initStatements.push([exportName, null]);
  312. }
  313. }
  314. }
  315. initStatements.sort(([a], [b]) => {
  316. if (a < b) return -1;
  317. if (b < a) return 1;
  318. return 0;
  319. });
  320. const results = [];
  321. if (noIncompleteNsImportDetection) {
  322. for (const [, initStatement] of initStatements) {
  323. results.push(initStatement);
  324. }
  325. } else {
  326. const chunkSize = 100;
  327. for (let i = 0; i < initStatements.length; i += chunkSize) {
  328. let uninitializedExportNames = [];
  329. for (let j = 0; j < chunkSize && i + j < initStatements.length; j++) {
  330. const [exportName, initStatement] = initStatements[i + j];
  331. if (initStatement !== null) {
  332. if (uninitializedExportNames.length > 0) {
  333. results.push(buildInitStatement(metadata, uninitializedExportNames, programPath.scope.buildUndefinedNode()));
  334. uninitializedExportNames = [];
  335. }
  336. results.push(initStatement);
  337. } else {
  338. uninitializedExportNames.push(exportName);
  339. }
  340. }
  341. if (uninitializedExportNames.length > 0) {
  342. results.push(buildInitStatement(metadata, uninitializedExportNames, programPath.scope.buildUndefinedNode()));
  343. }
  344. }
  345. }
  346. return results;
  347. }
  348. const InitTemplate = {
  349. computed: _template.default.expression`EXPORTS["NAME"] = VALUE`,
  350. default: _template.default.expression`EXPORTS.NAME = VALUE`
  351. };
  352. function buildInitStatement(metadata, exportNames, initExpr) {
  353. const {
  354. stringSpecifiers,
  355. exportName: EXPORTS
  356. } = metadata;
  357. return expressionStatement(exportNames.reduce((acc, exportName) => {
  358. const params = {
  359. EXPORTS,
  360. NAME: exportName,
  361. VALUE: acc
  362. };
  363. if (stringSpecifiers.has(exportName)) {
  364. return InitTemplate.computed(params);
  365. } else {
  366. return InitTemplate.default(params);
  367. }
  368. }, initExpr));
  369. }
  370. //# sourceMappingURL=index.js.map