index.js 2.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.default = splitExportDeclaration;
  6. function t() {
  7. const data = _interopRequireWildcard(require("@babel/types"));
  8. t = function () {
  9. return data;
  10. };
  11. return data;
  12. }
  13. 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; } }
  14. function splitExportDeclaration(exportDeclaration) {
  15. if (!exportDeclaration.isExportDeclaration()) {
  16. throw new Error("Only export declarations can be splitted.");
  17. }
  18. const isDefault = exportDeclaration.isExportDefaultDeclaration();
  19. const declaration = exportDeclaration.get("declaration");
  20. const isClassDeclaration = declaration.isClassDeclaration();
  21. if (isDefault) {
  22. const standaloneDeclaration = declaration.isFunctionDeclaration() || isClassDeclaration;
  23. const scope = declaration.isScope() ? declaration.scope.parent : declaration.scope;
  24. let id = declaration.node.id;
  25. let needBindingRegistration = false;
  26. if (!id) {
  27. needBindingRegistration = true;
  28. id = scope.generateUidIdentifier("default");
  29. if (standaloneDeclaration || declaration.isFunctionExpression() || declaration.isClassExpression()) {
  30. declaration.node.id = t().cloneNode(id);
  31. }
  32. }
  33. const updatedDeclaration = standaloneDeclaration ? declaration : t().variableDeclaration("var", [t().variableDeclarator(t().cloneNode(id), declaration.node)]);
  34. const updatedExportDeclaration = t().exportNamedDeclaration(null, [t().exportSpecifier(t().cloneNode(id), t().identifier("default"))]);
  35. exportDeclaration.insertAfter(updatedExportDeclaration);
  36. exportDeclaration.replaceWith(updatedDeclaration);
  37. if (needBindingRegistration) {
  38. scope.registerDeclaration(exportDeclaration);
  39. }
  40. return exportDeclaration;
  41. }
  42. if (exportDeclaration.get("specifiers").length > 0) {
  43. throw new Error("It doesn't make sense to split exported specifiers.");
  44. }
  45. const bindingIdentifiers = declaration.getOuterBindingIdentifiers();
  46. const specifiers = Object.keys(bindingIdentifiers).map(name => {
  47. return t().exportSpecifier(t().identifier(name), t().identifier(name));
  48. });
  49. const aliasDeclar = t().exportNamedDeclaration(null, specifiers);
  50. exportDeclaration.insertAfter(aliasDeclar);
  51. exportDeclaration.replaceWith(declaration.node);
  52. return exportDeclaration;
  53. }