index.js 11 KB

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