123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185 |
- "use strict";
- exports.__esModule = true;
- var _getIterator2 = require("babel-runtime/core-js/get-iterator");
- var _getIterator3 = _interopRequireDefault(_getIterator2);
- exports.default = function (node) {
- if (!this.isReferenced()) return;
- var binding = this.scope.getBinding(node.name);
- if (binding) {
- if (binding.identifier.typeAnnotation) {
- return binding.identifier.typeAnnotation;
- } else {
- return getTypeAnnotationBindingConstantViolations(this, node.name);
- }
- }
- if (node.name === "undefined") {
- return t.voidTypeAnnotation();
- } else if (node.name === "NaN" || node.name === "Infinity") {
- return t.numberTypeAnnotation();
- } else if (node.name === "arguments") {}
- };
- var _babelTypes = require("babel-types");
- var t = _interopRequireWildcard(_babelTypes);
- function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) newObj[key] = obj[key]; } } newObj.default = obj; return newObj; } }
- function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
- function getTypeAnnotationBindingConstantViolations(path, name) {
- var binding = path.scope.getBinding(name);
- var types = [];
- path.typeAnnotation = t.unionTypeAnnotation(types);
- var functionConstantViolations = [];
- var constantViolations = getConstantViolationsBefore(binding, path, functionConstantViolations);
- var testType = getConditionalAnnotation(path, name);
- if (testType) {
- var testConstantViolations = getConstantViolationsBefore(binding, testType.ifStatement);
- constantViolations = constantViolations.filter(function (path) {
- return testConstantViolations.indexOf(path) < 0;
- });
- types.push(testType.typeAnnotation);
- }
- if (constantViolations.length) {
- constantViolations = constantViolations.concat(functionConstantViolations);
- for (var _iterator = constantViolations, _isArray = Array.isArray(_iterator), _i = 0, _iterator = _isArray ? _iterator : (0, _getIterator3.default)(_iterator);;) {
- var _ref;
- if (_isArray) {
- if (_i >= _iterator.length) break;
- _ref = _iterator[_i++];
- } else {
- _i = _iterator.next();
- if (_i.done) break;
- _ref = _i.value;
- }
- var violation = _ref;
- types.push(violation.getTypeAnnotation());
- }
- }
- if (types.length) {
- return t.createUnionTypeAnnotation(types);
- }
- }
- function getConstantViolationsBefore(binding, path, functions) {
- var violations = binding.constantViolations.slice();
- violations.unshift(binding.path);
- return violations.filter(function (violation) {
- violation = violation.resolve();
- var status = violation._guessExecutionStatusRelativeTo(path);
- if (functions && status === "function") functions.push(violation);
- return status === "before";
- });
- }
- function inferAnnotationFromBinaryExpression(name, path) {
- var operator = path.node.operator;
- var right = path.get("right").resolve();
- var left = path.get("left").resolve();
- var target = void 0;
- if (left.isIdentifier({ name: name })) {
- target = right;
- } else if (right.isIdentifier({ name: name })) {
- target = left;
- }
- if (target) {
- if (operator === "===") {
- return target.getTypeAnnotation();
- } else if (t.BOOLEAN_NUMBER_BINARY_OPERATORS.indexOf(operator) >= 0) {
- return t.numberTypeAnnotation();
- } else {
- return;
- }
- } else {
- if (operator !== "===") return;
- }
- var typeofPath = void 0;
- var typePath = void 0;
- if (left.isUnaryExpression({ operator: "typeof" })) {
- typeofPath = left;
- typePath = right;
- } else if (right.isUnaryExpression({ operator: "typeof" })) {
- typeofPath = right;
- typePath = left;
- }
- if (!typePath && !typeofPath) return;
- typePath = typePath.resolve();
- if (!typePath.isLiteral()) return;
- var typeValue = typePath.node.value;
- if (typeof typeValue !== "string") return;
- if (!typeofPath.get("argument").isIdentifier({ name: name })) return;
- return t.createTypeAnnotationBasedOnTypeof(typePath.node.value);
- }
- function getParentConditionalPath(path) {
- var parentPath = void 0;
- while (parentPath = path.parentPath) {
- if (parentPath.isIfStatement() || parentPath.isConditionalExpression()) {
- if (path.key === "test") {
- return;
- } else {
- return parentPath;
- }
- } else {
- path = parentPath;
- }
- }
- }
- function getConditionalAnnotation(path, name) {
- var ifStatement = getParentConditionalPath(path);
- if (!ifStatement) return;
- var test = ifStatement.get("test");
- var paths = [test];
- var types = [];
- do {
- var _path = paths.shift().resolve();
- if (_path.isLogicalExpression()) {
- paths.push(_path.get("left"));
- paths.push(_path.get("right"));
- }
- if (_path.isBinaryExpression()) {
- var type = inferAnnotationFromBinaryExpression(name, _path);
- if (type) types.push(type);
- }
- } while (paths.length);
- if (types.length) {
- return {
- typeAnnotation: t.createUnionTypeAnnotation(types),
- ifStatement: ifStatement
- };
- } else {
- return getConditionalAnnotation(ifStatement, name);
- }
- }
- module.exports = exports["default"];
|