whitespace.js 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.list = exports.nodes = void 0;
  6. function t() {
  7. var data = _interopRequireWildcard(require("@babel/types"));
  8. t = function t() {
  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 crawl(node, state) {
  15. if (state === void 0) {
  16. state = {};
  17. }
  18. if (t().isMemberExpression(node)) {
  19. crawl(node.object, state);
  20. if (node.computed) crawl(node.property, state);
  21. } else if (t().isBinary(node) || t().isAssignmentExpression(node)) {
  22. crawl(node.left, state);
  23. crawl(node.right, state);
  24. } else if (t().isCallExpression(node)) {
  25. state.hasCall = true;
  26. crawl(node.callee, state);
  27. } else if (t().isFunction(node)) {
  28. state.hasFunction = true;
  29. } else if (t().isIdentifier(node)) {
  30. state.hasHelper = state.hasHelper || isHelper(node.callee);
  31. }
  32. return state;
  33. }
  34. function isHelper(node) {
  35. if (t().isMemberExpression(node)) {
  36. return isHelper(node.object) || isHelper(node.property);
  37. } else if (t().isIdentifier(node)) {
  38. return node.name === "require" || node.name[0] === "_";
  39. } else if (t().isCallExpression(node)) {
  40. return isHelper(node.callee);
  41. } else if (t().isBinary(node) || t().isAssignmentExpression(node)) {
  42. return t().isIdentifier(node.left) && isHelper(node.left) || isHelper(node.right);
  43. } else {
  44. return false;
  45. }
  46. }
  47. function isType(node) {
  48. return t().isLiteral(node) || t().isObjectExpression(node) || t().isArrayExpression(node) || t().isIdentifier(node) || t().isMemberExpression(node);
  49. }
  50. var nodes = {
  51. AssignmentExpression: function AssignmentExpression(node) {
  52. var state = crawl(node.right);
  53. if (state.hasCall && state.hasHelper || state.hasFunction) {
  54. return {
  55. before: state.hasFunction,
  56. after: true
  57. };
  58. }
  59. },
  60. SwitchCase: function SwitchCase(node, parent) {
  61. return {
  62. before: node.consequent.length || parent.cases[0] === node,
  63. after: !node.consequent.length && parent.cases[parent.cases.length - 1] === node
  64. };
  65. },
  66. LogicalExpression: function LogicalExpression(node) {
  67. if (t().isFunction(node.left) || t().isFunction(node.right)) {
  68. return {
  69. after: true
  70. };
  71. }
  72. },
  73. Literal: function Literal(node) {
  74. if (node.value === "use strict") {
  75. return {
  76. after: true
  77. };
  78. }
  79. },
  80. CallExpression: function CallExpression(node) {
  81. if (t().isFunction(node.callee) || isHelper(node)) {
  82. return {
  83. before: true,
  84. after: true
  85. };
  86. }
  87. },
  88. VariableDeclaration: function VariableDeclaration(node) {
  89. for (var i = 0; i < node.declarations.length; i++) {
  90. var declar = node.declarations[i];
  91. var enabled = isHelper(declar.id) && !isType(declar.init);
  92. if (!enabled) {
  93. var state = crawl(declar.init);
  94. enabled = isHelper(declar.init) && state.hasCall || state.hasFunction;
  95. }
  96. if (enabled) {
  97. return {
  98. before: true,
  99. after: true
  100. };
  101. }
  102. }
  103. },
  104. IfStatement: function IfStatement(node) {
  105. if (t().isBlockStatement(node.consequent)) {
  106. return {
  107. before: true,
  108. after: true
  109. };
  110. }
  111. }
  112. };
  113. exports.nodes = nodes;
  114. nodes.ObjectProperty = nodes.ObjectTypeProperty = nodes.ObjectMethod = function (node, parent) {
  115. if (parent.properties[0] === node) {
  116. return {
  117. before: true
  118. };
  119. }
  120. };
  121. nodes.ObjectTypeCallProperty = function (node, parent) {
  122. if (parent.callProperties[0] === node && (!parent.properties || !parent.properties.length)) {
  123. return {
  124. before: true
  125. };
  126. }
  127. };
  128. nodes.ObjectTypeIndexer = function (node, parent) {
  129. if (parent.indexers[0] === node && (!parent.properties || !parent.properties.length) && (!parent.callProperties || !parent.callProperties.length)) {
  130. return {
  131. before: true
  132. };
  133. }
  134. };
  135. var list = {
  136. VariableDeclaration: function VariableDeclaration(node) {
  137. return node.declarations.map(function (decl) {
  138. return decl.init;
  139. });
  140. },
  141. ArrayExpression: function ArrayExpression(node) {
  142. return node.elements;
  143. },
  144. ObjectExpression: function ObjectExpression(node) {
  145. return node.properties;
  146. }
  147. };
  148. exports.list = list;
  149. [["Function", true], ["Class", true], ["Loop", true], ["LabeledStatement", true], ["SwitchStatement", true], ["TryStatement", true]].forEach(function (_ref) {
  150. var type = _ref[0],
  151. amounts = _ref[1];
  152. if (typeof amounts === "boolean") {
  153. amounts = {
  154. after: amounts,
  155. before: amounts
  156. };
  157. }
  158. [type].concat(t().FLIPPED_ALIAS_KEYS[type] || []).forEach(function (type) {
  159. nodes[type] = function () {
  160. return amounts;
  161. };
  162. });
  163. });