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.
206 lines
5.0 KiB
206 lines
5.0 KiB
"use strict"; |
|
|
|
Object.defineProperty(exports, "__esModule", { |
|
value: true |
|
}); |
|
exports.default = _default; |
|
|
|
var _t = require("@babel/types"); |
|
|
|
const { |
|
BOOLEAN_NUMBER_BINARY_OPERATORS, |
|
createFlowUnionType, |
|
createTSUnionType, |
|
createTypeAnnotationBasedOnTypeof, |
|
createUnionTypeAnnotation, |
|
isTSTypeAnnotation, |
|
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; |
|
} |
|
|
|
if (isTSTypeAnnotation(types[0]) && createTSUnionType) { |
|
return createTSUnionType(types); |
|
} |
|
|
|
if (createFlowUnionType) { |
|
return createFlowUnionType(types); |
|
} |
|
|
|
return createUnionTypeAnnotation(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) { |
|
if (isTSTypeAnnotation(types[0]) && createTSUnionType) { |
|
return { |
|
typeAnnotation: createTSUnionType(types), |
|
ifStatement |
|
}; |
|
} |
|
|
|
if (createFlowUnionType) { |
|
return { |
|
typeAnnotation: createFlowUnionType(types), |
|
ifStatement |
|
}; |
|
} |
|
|
|
return { |
|
typeAnnotation: createUnionTypeAnnotation(types), |
|
ifStatement |
|
}; |
|
} |
|
|
|
return getConditionalAnnotation(binding, ifStatement, name); |
|
} |