index.js 12 KB

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