inferers.js 7.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.VariableDeclarator = VariableDeclarator;
  6. exports.TypeCastExpression = TypeCastExpression;
  7. exports.NewExpression = NewExpression;
  8. exports.TemplateLiteral = TemplateLiteral;
  9. exports.UnaryExpression = UnaryExpression;
  10. exports.BinaryExpression = BinaryExpression;
  11. exports.LogicalExpression = LogicalExpression;
  12. exports.ConditionalExpression = ConditionalExpression;
  13. exports.SequenceExpression = SequenceExpression;
  14. exports.ParenthesizedExpression = ParenthesizedExpression;
  15. exports.AssignmentExpression = AssignmentExpression;
  16. exports.UpdateExpression = UpdateExpression;
  17. exports.StringLiteral = StringLiteral;
  18. exports.NumericLiteral = NumericLiteral;
  19. exports.BooleanLiteral = BooleanLiteral;
  20. exports.NullLiteral = NullLiteral;
  21. exports.RegExpLiteral = RegExpLiteral;
  22. exports.ObjectExpression = ObjectExpression;
  23. exports.ArrayExpression = ArrayExpression;
  24. exports.RestElement = RestElement;
  25. exports.ClassDeclaration = exports.ClassExpression = exports.FunctionDeclaration = exports.ArrowFunctionExpression = exports.FunctionExpression = Func;
  26. exports.CallExpression = CallExpression;
  27. exports.TaggedTemplateExpression = TaggedTemplateExpression;
  28. Object.defineProperty(exports, "Identifier", {
  29. enumerable: true,
  30. get: function () {
  31. return _infererReference.default;
  32. }
  33. });
  34. var t = _interopRequireWildcard(require("@babel/types"));
  35. var _infererReference = _interopRequireDefault(require("./inferer-reference"));
  36. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  37. function _getRequireWildcardCache() { if (typeof WeakMap !== "function") return null; var cache = new WeakMap(); _getRequireWildcardCache = function () { return cache; }; return cache; }
  38. 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; }
  39. function VariableDeclarator() {
  40. var _type;
  41. const id = this.get("id");
  42. if (!id.isIdentifier()) return;
  43. const init = this.get("init");
  44. let type = init.getTypeAnnotation();
  45. if (((_type = type) == null ? void 0 : _type.type) === "AnyTypeAnnotation") {
  46. if (init.isCallExpression() && init.get("callee").isIdentifier({
  47. name: "Array"
  48. }) && !init.scope.hasBinding("Array", true)) {
  49. type = ArrayExpression();
  50. }
  51. }
  52. return type;
  53. }
  54. function TypeCastExpression(node) {
  55. return node.typeAnnotation;
  56. }
  57. TypeCastExpression.validParent = true;
  58. function NewExpression(node) {
  59. if (this.get("callee").isIdentifier()) {
  60. return t.genericTypeAnnotation(node.callee);
  61. }
  62. }
  63. function TemplateLiteral() {
  64. return t.stringTypeAnnotation();
  65. }
  66. function UnaryExpression(node) {
  67. const operator = node.operator;
  68. if (operator === "void") {
  69. return t.voidTypeAnnotation();
  70. } else if (t.NUMBER_UNARY_OPERATORS.indexOf(operator) >= 0) {
  71. return t.numberTypeAnnotation();
  72. } else if (t.STRING_UNARY_OPERATORS.indexOf(operator) >= 0) {
  73. return t.stringTypeAnnotation();
  74. } else if (t.BOOLEAN_UNARY_OPERATORS.indexOf(operator) >= 0) {
  75. return t.booleanTypeAnnotation();
  76. }
  77. }
  78. function BinaryExpression(node) {
  79. const operator = node.operator;
  80. if (t.NUMBER_BINARY_OPERATORS.indexOf(operator) >= 0) {
  81. return t.numberTypeAnnotation();
  82. } else if (t.BOOLEAN_BINARY_OPERATORS.indexOf(operator) >= 0) {
  83. return t.booleanTypeAnnotation();
  84. } else if (operator === "+") {
  85. const right = this.get("right");
  86. const left = this.get("left");
  87. if (left.isBaseType("number") && right.isBaseType("number")) {
  88. return t.numberTypeAnnotation();
  89. } else if (left.isBaseType("string") || right.isBaseType("string")) {
  90. return t.stringTypeAnnotation();
  91. }
  92. return t.unionTypeAnnotation([t.stringTypeAnnotation(), t.numberTypeAnnotation()]);
  93. }
  94. }
  95. function LogicalExpression() {
  96. const argumentTypes = [this.get("left").getTypeAnnotation(), this.get("right").getTypeAnnotation()];
  97. if (t.isTSTypeAnnotation(argumentTypes[0]) && t.createTSUnionType) {
  98. return t.createTSUnionType(argumentTypes);
  99. }
  100. if (t.createFlowUnionType) {
  101. return t.createFlowUnionType(argumentTypes);
  102. }
  103. return t.createUnionTypeAnnotation(argumentTypes);
  104. }
  105. function ConditionalExpression() {
  106. const argumentTypes = [this.get("consequent").getTypeAnnotation(), this.get("alternate").getTypeAnnotation()];
  107. if (t.isTSTypeAnnotation(argumentTypes[0]) && t.createTSUnionType) {
  108. return t.createTSUnionType(argumentTypes);
  109. }
  110. if (t.createFlowUnionType) {
  111. return t.createFlowUnionType(argumentTypes);
  112. }
  113. return t.createUnionTypeAnnotation(argumentTypes);
  114. }
  115. function SequenceExpression() {
  116. return this.get("expressions").pop().getTypeAnnotation();
  117. }
  118. function ParenthesizedExpression() {
  119. return this.get("expression").getTypeAnnotation();
  120. }
  121. function AssignmentExpression() {
  122. return this.get("right").getTypeAnnotation();
  123. }
  124. function UpdateExpression(node) {
  125. const operator = node.operator;
  126. if (operator === "++" || operator === "--") {
  127. return t.numberTypeAnnotation();
  128. }
  129. }
  130. function StringLiteral() {
  131. return t.stringTypeAnnotation();
  132. }
  133. function NumericLiteral() {
  134. return t.numberTypeAnnotation();
  135. }
  136. function BooleanLiteral() {
  137. return t.booleanTypeAnnotation();
  138. }
  139. function NullLiteral() {
  140. return t.nullLiteralTypeAnnotation();
  141. }
  142. function RegExpLiteral() {
  143. return t.genericTypeAnnotation(t.identifier("RegExp"));
  144. }
  145. function ObjectExpression() {
  146. return t.genericTypeAnnotation(t.identifier("Object"));
  147. }
  148. function ArrayExpression() {
  149. return t.genericTypeAnnotation(t.identifier("Array"));
  150. }
  151. function RestElement() {
  152. return ArrayExpression();
  153. }
  154. RestElement.validParent = true;
  155. function Func() {
  156. return t.genericTypeAnnotation(t.identifier("Function"));
  157. }
  158. const isArrayFrom = t.buildMatchMemberExpression("Array.from");
  159. const isObjectKeys = t.buildMatchMemberExpression("Object.keys");
  160. const isObjectValues = t.buildMatchMemberExpression("Object.values");
  161. const isObjectEntries = t.buildMatchMemberExpression("Object.entries");
  162. function CallExpression() {
  163. const {
  164. callee
  165. } = this.node;
  166. if (isObjectKeys(callee)) {
  167. return t.arrayTypeAnnotation(t.stringTypeAnnotation());
  168. } else if (isArrayFrom(callee) || isObjectValues(callee)) {
  169. return t.arrayTypeAnnotation(t.anyTypeAnnotation());
  170. } else if (isObjectEntries(callee)) {
  171. return t.arrayTypeAnnotation(t.tupleTypeAnnotation([t.stringTypeAnnotation(), t.anyTypeAnnotation()]));
  172. }
  173. return resolveCall(this.get("callee"));
  174. }
  175. function TaggedTemplateExpression() {
  176. return resolveCall(this.get("tag"));
  177. }
  178. function resolveCall(callee) {
  179. callee = callee.resolve();
  180. if (callee.isFunction()) {
  181. if (callee.is("async")) {
  182. if (callee.is("generator")) {
  183. return t.genericTypeAnnotation(t.identifier("AsyncIterator"));
  184. } else {
  185. return t.genericTypeAnnotation(t.identifier("Promise"));
  186. }
  187. } else {
  188. if (callee.node.returnType) {
  189. return callee.node.returnType;
  190. } else {}
  191. }
  192. }
  193. }