You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							211 lines
						
					
					
						
							6.5 KiB
						
					
					
				
			
		
		
	
	
							211 lines
						
					
					
						
							6.5 KiB
						
					
					
				| "use strict";
 | |
| 
 | |
| Object.defineProperty(exports, "__esModule", {
 | |
|   value: true
 | |
| });
 | |
| exports.ArrayExpression = ArrayExpression;
 | |
| exports.AssignmentExpression = AssignmentExpression;
 | |
| exports.BinaryExpression = BinaryExpression;
 | |
| exports.BooleanLiteral = BooleanLiteral;
 | |
| exports.CallExpression = CallExpression;
 | |
| exports.ConditionalExpression = ConditionalExpression;
 | |
| exports.ClassDeclaration = exports.ClassExpression = exports.FunctionDeclaration = exports.ArrowFunctionExpression = exports.FunctionExpression = Func;
 | |
| Object.defineProperty(exports, "Identifier", {
 | |
|   enumerable: true,
 | |
|   get: function () {
 | |
|     return _infererReference.default;
 | |
|   }
 | |
| });
 | |
| exports.LogicalExpression = LogicalExpression;
 | |
| exports.NewExpression = NewExpression;
 | |
| exports.NullLiteral = NullLiteral;
 | |
| exports.NumericLiteral = NumericLiteral;
 | |
| exports.ObjectExpression = ObjectExpression;
 | |
| exports.ParenthesizedExpression = ParenthesizedExpression;
 | |
| exports.RegExpLiteral = RegExpLiteral;
 | |
| exports.RestElement = RestElement;
 | |
| exports.SequenceExpression = SequenceExpression;
 | |
| exports.StringLiteral = StringLiteral;
 | |
| exports.TSAsExpression = TSAsExpression;
 | |
| exports.TSNonNullExpression = TSNonNullExpression;
 | |
| exports.TaggedTemplateExpression = TaggedTemplateExpression;
 | |
| exports.TemplateLiteral = TemplateLiteral;
 | |
| exports.TypeCastExpression = TypeCastExpression;
 | |
| exports.UnaryExpression = UnaryExpression;
 | |
| exports.UpdateExpression = UpdateExpression;
 | |
| exports.VariableDeclarator = VariableDeclarator;
 | |
| var _t = require("@babel/types");
 | |
| var _infererReference = require("./inferer-reference");
 | |
| var _util = require("./util");
 | |
| const {
 | |
|   BOOLEAN_BINARY_OPERATORS,
 | |
|   BOOLEAN_UNARY_OPERATORS,
 | |
|   NUMBER_BINARY_OPERATORS,
 | |
|   NUMBER_UNARY_OPERATORS,
 | |
|   STRING_UNARY_OPERATORS,
 | |
|   anyTypeAnnotation,
 | |
|   arrayTypeAnnotation,
 | |
|   booleanTypeAnnotation,
 | |
|   buildMatchMemberExpression,
 | |
|   genericTypeAnnotation,
 | |
|   identifier,
 | |
|   nullLiteralTypeAnnotation,
 | |
|   numberTypeAnnotation,
 | |
|   stringTypeAnnotation,
 | |
|   tupleTypeAnnotation,
 | |
|   unionTypeAnnotation,
 | |
|   voidTypeAnnotation,
 | |
|   isIdentifier
 | |
| } = _t;
 | |
| function VariableDeclarator() {
 | |
|   if (!this.get("id").isIdentifier()) return;
 | |
|   return this.get("init").getTypeAnnotation();
 | |
| }
 | |
| function TypeCastExpression(node) {
 | |
|   return node.typeAnnotation;
 | |
| }
 | |
| TypeCastExpression.validParent = true;
 | |
| function TSAsExpression(node) {
 | |
|   return node.typeAnnotation;
 | |
| }
 | |
| TSAsExpression.validParent = true;
 | |
| function TSNonNullExpression() {
 | |
|   return this.get("expression").getTypeAnnotation();
 | |
| }
 | |
| function NewExpression(node) {
 | |
|   if (node.callee.type === "Identifier") {
 | |
|     return genericTypeAnnotation(node.callee);
 | |
|   }
 | |
| }
 | |
| function TemplateLiteral() {
 | |
|   return stringTypeAnnotation();
 | |
| }
 | |
| function UnaryExpression(node) {
 | |
|   const operator = node.operator;
 | |
|   if (operator === "void") {
 | |
|     return voidTypeAnnotation();
 | |
|   } else if (NUMBER_UNARY_OPERATORS.indexOf(operator) >= 0) {
 | |
|     return numberTypeAnnotation();
 | |
|   } else if (STRING_UNARY_OPERATORS.indexOf(operator) >= 0) {
 | |
|     return stringTypeAnnotation();
 | |
|   } else if (BOOLEAN_UNARY_OPERATORS.indexOf(operator) >= 0) {
 | |
|     return booleanTypeAnnotation();
 | |
|   }
 | |
| }
 | |
| function BinaryExpression(node) {
 | |
|   const operator = node.operator;
 | |
|   if (NUMBER_BINARY_OPERATORS.indexOf(operator) >= 0) {
 | |
|     return numberTypeAnnotation();
 | |
|   } else if (BOOLEAN_BINARY_OPERATORS.indexOf(operator) >= 0) {
 | |
|     return booleanTypeAnnotation();
 | |
|   } else if (operator === "+") {
 | |
|     const right = this.get("right");
 | |
|     const left = this.get("left");
 | |
|     if (left.isBaseType("number") && right.isBaseType("number")) {
 | |
|       return numberTypeAnnotation();
 | |
|     } else if (left.isBaseType("string") || right.isBaseType("string")) {
 | |
|       return stringTypeAnnotation();
 | |
|     }
 | |
| 
 | |
|     return unionTypeAnnotation([stringTypeAnnotation(), numberTypeAnnotation()]);
 | |
|   }
 | |
| }
 | |
| function LogicalExpression() {
 | |
|   const argumentTypes = [this.get("left").getTypeAnnotation(), this.get("right").getTypeAnnotation()];
 | |
|   return (0, _util.createUnionType)(argumentTypes);
 | |
| }
 | |
| function ConditionalExpression() {
 | |
|   const argumentTypes = [this.get("consequent").getTypeAnnotation(), this.get("alternate").getTypeAnnotation()];
 | |
|   return (0, _util.createUnionType)(argumentTypes);
 | |
| }
 | |
| function SequenceExpression() {
 | |
|   return this.get("expressions").pop().getTypeAnnotation();
 | |
| }
 | |
| function ParenthesizedExpression() {
 | |
|   return this.get("expression").getTypeAnnotation();
 | |
| }
 | |
| function AssignmentExpression() {
 | |
|   return this.get("right").getTypeAnnotation();
 | |
| }
 | |
| function UpdateExpression(node) {
 | |
|   const operator = node.operator;
 | |
|   if (operator === "++" || operator === "--") {
 | |
|     return numberTypeAnnotation();
 | |
|   }
 | |
| }
 | |
| function StringLiteral() {
 | |
|   return stringTypeAnnotation();
 | |
| }
 | |
| function NumericLiteral() {
 | |
|   return numberTypeAnnotation();
 | |
| }
 | |
| function BooleanLiteral() {
 | |
|   return booleanTypeAnnotation();
 | |
| }
 | |
| function NullLiteral() {
 | |
|   return nullLiteralTypeAnnotation();
 | |
| }
 | |
| function RegExpLiteral() {
 | |
|   return genericTypeAnnotation(identifier("RegExp"));
 | |
| }
 | |
| function ObjectExpression() {
 | |
|   return genericTypeAnnotation(identifier("Object"));
 | |
| }
 | |
| function ArrayExpression() {
 | |
|   return genericTypeAnnotation(identifier("Array"));
 | |
| }
 | |
| function RestElement() {
 | |
|   return ArrayExpression();
 | |
| }
 | |
| RestElement.validParent = true;
 | |
| function Func() {
 | |
|   return genericTypeAnnotation(identifier("Function"));
 | |
| }
 | |
| const isArrayFrom = buildMatchMemberExpression("Array.from");
 | |
| const isObjectKeys = buildMatchMemberExpression("Object.keys");
 | |
| const isObjectValues = buildMatchMemberExpression("Object.values");
 | |
| const isObjectEntries = buildMatchMemberExpression("Object.entries");
 | |
| function CallExpression() {
 | |
|   const {
 | |
|     callee
 | |
|   } = this.node;
 | |
|   if (isObjectKeys(callee)) {
 | |
|     return arrayTypeAnnotation(stringTypeAnnotation());
 | |
|   } else if (isArrayFrom(callee) || isObjectValues(callee) ||
 | |
|   isIdentifier(callee, {
 | |
|     name: "Array"
 | |
|   })) {
 | |
|     return arrayTypeAnnotation(anyTypeAnnotation());
 | |
|   } else if (isObjectEntries(callee)) {
 | |
|     return arrayTypeAnnotation(tupleTypeAnnotation([stringTypeAnnotation(), anyTypeAnnotation()]));
 | |
|   }
 | |
|   return resolveCall(this.get("callee"));
 | |
| }
 | |
| function TaggedTemplateExpression() {
 | |
|   return resolveCall(this.get("tag"));
 | |
| }
 | |
| function resolveCall(callee) {
 | |
|   callee = callee.resolve();
 | |
|   if (callee.isFunction()) {
 | |
|     const {
 | |
|       node
 | |
|     } = callee;
 | |
|     if (node.async) {
 | |
|       if (node.generator) {
 | |
|         return genericTypeAnnotation(identifier("AsyncIterator"));
 | |
|       } else {
 | |
|         return genericTypeAnnotation(identifier("Promise"));
 | |
|       }
 | |
|     } else {
 | |
|       if (node.generator) {
 | |
|         return genericTypeAnnotation(identifier("Iterator"));
 | |
|       } else if (callee.node.returnType) {
 | |
|         return callee.node.returnType;
 | |
|       } else {
 | |
|       }
 | |
|     }
 | |
|   }
 | |
| }
 | |
| 
 | |
| //# sourceMappingURL=inferers.js.map
 |