index.js 9.5 KB

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