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.
		
		
		
		
		
			
		
			
				
					
					
						
							164 lines
						
					
					
						
							4.5 KiB
						
					
					
				
			
		
		
	
	
							164 lines
						
					
					
						
							4.5 KiB
						
					
					
				| "use strict";
 | |
| 
 | |
| Object.defineProperty(exports, "__esModule", {
 | |
|   value: true
 | |
| });
 | |
| exports.default = _default;
 | |
| var _t = require("@babel/types");
 | |
| var _util = require("./util");
 | |
| const {
 | |
|   BOOLEAN_NUMBER_BINARY_OPERATORS,
 | |
|   createTypeAnnotationBasedOnTypeof,
 | |
|   numberTypeAnnotation,
 | |
|   voidTypeAnnotation
 | |
| } = _t;
 | |
| function _default(node) {
 | |
|   if (!this.isReferenced()) return;
 | |
| 
 | |
|   const binding = this.scope.getBinding(node.name);
 | |
|   if (binding) {
 | |
|     if (binding.identifier.typeAnnotation) {
 | |
|       return binding.identifier.typeAnnotation;
 | |
|     } else {
 | |
|       return getTypeAnnotationBindingConstantViolations(binding, this, node.name);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   if (node.name === "undefined") {
 | |
|     return voidTypeAnnotation();
 | |
|   } else if (node.name === "NaN" || node.name === "Infinity") {
 | |
|     return numberTypeAnnotation();
 | |
|   } else if (node.name === "arguments") {
 | |
|   }
 | |
| }
 | |
| 
 | |
| function getTypeAnnotationBindingConstantViolations(binding, path, name) {
 | |
|   const types = [];
 | |
|   const functionConstantViolations = [];
 | |
|   let constantViolations = getConstantViolationsBefore(binding, path, functionConstantViolations);
 | |
|   const testType = getConditionalAnnotation(binding, path, name);
 | |
|   if (testType) {
 | |
|     const testConstantViolations = getConstantViolationsBefore(binding, testType.ifStatement);
 | |
| 
 | |
|     constantViolations = constantViolations.filter(path => testConstantViolations.indexOf(path) < 0);
 | |
| 
 | |
|     types.push(testType.typeAnnotation);
 | |
|   }
 | |
|   if (constantViolations.length) {
 | |
| 
 | |
|     constantViolations.push(...functionConstantViolations);
 | |
| 
 | |
|     for (const violation of constantViolations) {
 | |
|       types.push(violation.getTypeAnnotation());
 | |
|     }
 | |
|   }
 | |
|   if (!types.length) {
 | |
|     return;
 | |
|   }
 | |
|   return (0, _util.createUnionType)(types);
 | |
| }
 | |
| function getConstantViolationsBefore(binding, path, functions) {
 | |
|   const violations = binding.constantViolations.slice();
 | |
|   violations.unshift(binding.path);
 | |
|   return violations.filter(violation => {
 | |
|     violation = violation.resolve();
 | |
|     const status = violation._guessExecutionStatusRelativeTo(path);
 | |
|     if (functions && status === "unknown") functions.push(violation);
 | |
|     return status === "before";
 | |
|   });
 | |
| }
 | |
| function inferAnnotationFromBinaryExpression(name, path) {
 | |
|   const operator = path.node.operator;
 | |
|   const right = path.get("right").resolve();
 | |
|   const left = path.get("left").resolve();
 | |
|   let target;
 | |
|   if (left.isIdentifier({
 | |
|     name
 | |
|   })) {
 | |
|     target = right;
 | |
|   } else if (right.isIdentifier({
 | |
|     name
 | |
|   })) {
 | |
|     target = left;
 | |
|   }
 | |
|   if (target) {
 | |
|     if (operator === "===") {
 | |
|       return target.getTypeAnnotation();
 | |
|     }
 | |
|     if (BOOLEAN_NUMBER_BINARY_OPERATORS.indexOf(operator) >= 0) {
 | |
|       return numberTypeAnnotation();
 | |
|     }
 | |
|     return;
 | |
|   }
 | |
|   if (operator !== "===" && operator !== "==") return;
 | |
| 
 | |
|   let typeofPath;
 | |
|   let typePath;
 | |
|   if (left.isUnaryExpression({
 | |
|     operator: "typeof"
 | |
|   })) {
 | |
|     typeofPath = left;
 | |
|     typePath = right;
 | |
|   } else if (right.isUnaryExpression({
 | |
|     operator: "typeof"
 | |
|   })) {
 | |
|     typeofPath = right;
 | |
|     typePath = left;
 | |
|   }
 | |
|   if (!typeofPath) return;
 | |
|   if (!typeofPath.get("argument").isIdentifier({
 | |
|     name
 | |
|   })) return;
 | |
| 
 | |
|   typePath = typePath.resolve();
 | |
|   if (!typePath.isLiteral()) return;
 | |
| 
 | |
|   const typeValue = typePath.node.value;
 | |
|   if (typeof typeValue !== "string") return;
 | |
| 
 | |
|   return createTypeAnnotationBasedOnTypeof(typeValue);
 | |
| }
 | |
| function getParentConditionalPath(binding, path, name) {
 | |
|   let parentPath;
 | |
|   while (parentPath = path.parentPath) {
 | |
|     if (parentPath.isIfStatement() || parentPath.isConditionalExpression()) {
 | |
|       if (path.key === "test") {
 | |
|         return;
 | |
|       }
 | |
|       return parentPath;
 | |
|     }
 | |
|     if (parentPath.isFunction()) {
 | |
|       if (parentPath.parentPath.scope.getBinding(name) !== binding) return;
 | |
|     }
 | |
|     path = parentPath;
 | |
|   }
 | |
| }
 | |
| function getConditionalAnnotation(binding, path, name) {
 | |
|   const ifStatement = getParentConditionalPath(binding, path, name);
 | |
|   if (!ifStatement) return;
 | |
|   const test = ifStatement.get("test");
 | |
|   const paths = [test];
 | |
|   const types = [];
 | |
|   for (let i = 0; i < paths.length; i++) {
 | |
|     const path = paths[i];
 | |
|     if (path.isLogicalExpression()) {
 | |
|       if (path.node.operator === "&&") {
 | |
|         paths.push(path.get("left"));
 | |
|         paths.push(path.get("right"));
 | |
|       }
 | |
|     } else if (path.isBinaryExpression()) {
 | |
|       const type = inferAnnotationFromBinaryExpression(name, path);
 | |
|       if (type) types.push(type);
 | |
|     }
 | |
|   }
 | |
|   if (types.length) {
 | |
|     return {
 | |
|       typeAnnotation: (0, _util.createUnionType)(types),
 | |
|       ifStatement
 | |
|     };
 | |
|   }
 | |
|   return getConditionalAnnotation(binding, ifStatement, name);
 | |
| }
 | |
| 
 | |
| //# sourceMappingURL=inferer-reference.js.map
 |