Dashboard sipadu mbip
選択できるのは25トピックまでです。 トピックは、先頭が英数字で、英数字とダッシュ('-')を使用した35文字以内のものにしてください。

inferers.js 6.9KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221
  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; } var cache = _getRequireWildcardCache(); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; if (obj != null) { 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. const id = this.get("id");
  41. if (!id.isIdentifier()) return;
  42. const init = this.get("init");
  43. let type = init.getTypeAnnotation();
  44. if (type && type.type === "AnyTypeAnnotation") {
  45. if (init.isCallExpression() && init.get("callee").isIdentifier({
  46. name: "Array"
  47. }) && !init.scope.hasBinding("Array", true)) {
  48. type = ArrayExpression();
  49. }
  50. }
  51. return type;
  52. }
  53. function TypeCastExpression(node) {
  54. return node.typeAnnotation;
  55. }
  56. TypeCastExpression.validParent = true;
  57. function NewExpression(node) {
  58. if (this.get("callee").isIdentifier()) {
  59. return t.genericTypeAnnotation(node.callee);
  60. }
  61. }
  62. function TemplateLiteral() {
  63. return t.stringTypeAnnotation();
  64. }
  65. function UnaryExpression(node) {
  66. const operator = node.operator;
  67. if (operator === "void") {
  68. return t.voidTypeAnnotation();
  69. } else if (t.NUMBER_UNARY_OPERATORS.indexOf(operator) >= 0) {
  70. return t.numberTypeAnnotation();
  71. } else if (t.STRING_UNARY_OPERATORS.indexOf(operator) >= 0) {
  72. return t.stringTypeAnnotation();
  73. } else if (t.BOOLEAN_UNARY_OPERATORS.indexOf(operator) >= 0) {
  74. return t.booleanTypeAnnotation();
  75. }
  76. }
  77. function BinaryExpression(node) {
  78. const operator = node.operator;
  79. if (t.NUMBER_BINARY_OPERATORS.indexOf(operator) >= 0) {
  80. return t.numberTypeAnnotation();
  81. } else if (t.BOOLEAN_BINARY_OPERATORS.indexOf(operator) >= 0) {
  82. return t.booleanTypeAnnotation();
  83. } else if (operator === "+") {
  84. const right = this.get("right");
  85. const left = this.get("left");
  86. if (left.isBaseType("number") && right.isBaseType("number")) {
  87. return t.numberTypeAnnotation();
  88. } else if (left.isBaseType("string") || right.isBaseType("string")) {
  89. return t.stringTypeAnnotation();
  90. }
  91. return t.unionTypeAnnotation([t.stringTypeAnnotation(), t.numberTypeAnnotation()]);
  92. }
  93. }
  94. function LogicalExpression() {
  95. return t.createUnionTypeAnnotation([this.get("left").getTypeAnnotation(), this.get("right").getTypeAnnotation()]);
  96. }
  97. function ConditionalExpression() {
  98. return t.createUnionTypeAnnotation([this.get("consequent").getTypeAnnotation(), this.get("alternate").getTypeAnnotation()]);
  99. }
  100. function SequenceExpression() {
  101. return this.get("expressions").pop().getTypeAnnotation();
  102. }
  103. function ParenthesizedExpression() {
  104. return this.get("expression").getTypeAnnotation();
  105. }
  106. function AssignmentExpression() {
  107. return this.get("right").getTypeAnnotation();
  108. }
  109. function UpdateExpression(node) {
  110. const operator = node.operator;
  111. if (operator === "++" || operator === "--") {
  112. return t.numberTypeAnnotation();
  113. }
  114. }
  115. function StringLiteral() {
  116. return t.stringTypeAnnotation();
  117. }
  118. function NumericLiteral() {
  119. return t.numberTypeAnnotation();
  120. }
  121. function BooleanLiteral() {
  122. return t.booleanTypeAnnotation();
  123. }
  124. function NullLiteral() {
  125. return t.nullLiteralTypeAnnotation();
  126. }
  127. function RegExpLiteral() {
  128. return t.genericTypeAnnotation(t.identifier("RegExp"));
  129. }
  130. function ObjectExpression() {
  131. return t.genericTypeAnnotation(t.identifier("Object"));
  132. }
  133. function ArrayExpression() {
  134. return t.genericTypeAnnotation(t.identifier("Array"));
  135. }
  136. function RestElement() {
  137. return ArrayExpression();
  138. }
  139. RestElement.validParent = true;
  140. function Func() {
  141. return t.genericTypeAnnotation(t.identifier("Function"));
  142. }
  143. const isArrayFrom = t.buildMatchMemberExpression("Array.from");
  144. const isObjectKeys = t.buildMatchMemberExpression("Object.keys");
  145. const isObjectValues = t.buildMatchMemberExpression("Object.values");
  146. const isObjectEntries = t.buildMatchMemberExpression("Object.entries");
  147. function CallExpression() {
  148. const {
  149. callee
  150. } = this.node;
  151. if (isObjectKeys(callee)) {
  152. return t.arrayTypeAnnotation(t.stringTypeAnnotation());
  153. } else if (isArrayFrom(callee) || isObjectValues(callee)) {
  154. return t.arrayTypeAnnotation(t.anyTypeAnnotation());
  155. } else if (isObjectEntries(callee)) {
  156. return t.arrayTypeAnnotation(t.tupleTypeAnnotation([t.stringTypeAnnotation(), t.anyTypeAnnotation()]));
  157. }
  158. return resolveCall(this.get("callee"));
  159. }
  160. function TaggedTemplateExpression() {
  161. return resolveCall(this.get("tag"));
  162. }
  163. function resolveCall(callee) {
  164. callee = callee.resolve();
  165. if (callee.isFunction()) {
  166. if (callee.is("async")) {
  167. if (callee.is("generator")) {
  168. return t.genericTypeAnnotation(t.identifier("AsyncIterator"));
  169. } else {
  170. return t.genericTypeAnnotation(t.identifier("Promise"));
  171. }
  172. } else {
  173. if (callee.node.returnType) {
  174. return callee.node.returnType;
  175. } else {}
  176. }
  177. }
  178. }