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
			| 
											3 years ago
										 | "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
 |