public function test_Ternary()
{
$var1 = new Expr\Variable('a', ['type' => Type::array_()]);
$var2 = new Expr\Variable('b', ['type' => Type::int_()]);
$var3 = new Expr\Variable('c', ['type' => Type::float_()]);
$expr = new Expr\Ternary($var1, $var2, $var3);
$this->infer([$expr]);
$this->assertTrue($expr->getAttribute('type')->equals(Type::alternatives([Type::int_(), Type::float_()])));
$expr = new Expr\Ternary($var1, null, $var3);
$this->infer([$expr]);
$this->assertTrue($expr->getAttribute('type')->equals(Type::alternatives([Type::array_(), Type::float_()])));
}
public function leaveNode(Node $node)
{
if ($node instanceof Stmt\Function_ || $node instanceof Stmt\ClassMethod || $node instanceof Expr\Closure) {
array_pop($this->functionScopeStack);
} elseif ($node instanceof Stmt\ClassLike) {
array_pop($this->classStack);
array_pop($this->functionScopeStack);
}
if (!$node instanceof Expr) {
return;
}
$type = null;
$reflections = null;
if ($node instanceof Expr\Variable) {
if (is_string($node->name)) {
if ($node->name === 'this') {
$type = $this->getCurrentClass();
} elseif (array_key_exists('$' . $node->name, $this->getCurrentFunctionScope())) {
$type = $this->getCurrentFunctionScope()['$' . $node->name];
}
}
} elseif ($node instanceof Expr\FuncCall) {
$reflections = [];
if ($node->name instanceof Name) {
$reflections = $this->reflection->findFunction(Type::nameToString($node->name));
} else {
$reflections = $this->findMethods($node->name->getAttribute('type'), '__invoke');
}
$type = $this->functionsReturnType($reflections);
// TODO: ConstFetch
} elseif ($node instanceof Expr\MethodCall) {
$reflections = [];
if (is_string($node->name) || $node->name instanceof Identifier) {
$reflections = $this->findMethods($node->var->getAttribute('type'), (string) $node->name);
}
$type = $this->functionsReturnType($reflections);
} elseif ($node instanceof Expr\StaticCall) {
$reflections = [];
if ($node->class instanceof Name && (is_string($node->name) || $node->name instanceof Identifier)) {
$reflections = $this->findMethods(Type::object_(Type::nameToString($node->class)), (string) $node->name, true);
}
$type = $this->functionsReturnType($reflections);
} elseif ($node instanceof Expr\PropertyFetch) {
$reflections = [];
if (is_string($node->name) || $node->name instanceof Identifier) {
$reflections = $this->findProperties($node->var->getAttribute('type'), '$' . (string) $node->name);
}
$type = $this->variablesType($reflections);
} elseif ($node instanceof Expr\StaticPropertyFetch) {
$reflections = [];
if ($node->class instanceof Name && is_string($node->name)) {
$reflections = $this->findProperties(Type::object_(Type::nameToString($node->class)), '$' . $node->name, true);
}
$type = $this->variablesType($reflections);
} elseif ($node instanceof Expr\ClassConstFetch) {
// TODO ::class
$reflections = [];
if ($node->class instanceof Name) {
$reflections = $this->findClassConsts(Type::object_(Type::nameToString($node->class)), (string) $node->name);
}
$type = $this->constsType($reflections);
} elseif ($node instanceof Expr\ArrayDimFetch) {
$arrayType = $node->var->getAttribute('type');
$altTypes = [$arrayType];
if ($arrayType instanceof AlternativesType) {
$altTypes = $arrayType->getAlternatives();
}
$types = [];
foreach ($altTypes as $altType) {
if ($altType instanceof ArrayType) {
$types[] = $altType->getValueType();
} elseif ($altType instanceof ObjectType) {
$types[] = $this->functionsReturnType($this->findMethods($altType, 'offsetGet'));
// TODO: check for ArrayAccess
}
}
$type = $types !== [] ? Type::alternatives($types) : Type::mixed_();
} elseif ($node instanceof Expr\Array_) {
$type = Type::array_();
} elseif ($node instanceof Expr\New_) {
if ($node->class instanceof Name) {
$type = Type::object_(Type::nameToString($node->class));
} else {
$type = Type::object_();
}
} elseif ($node instanceof Expr\Clone_) {
$type = $node->expr->getAttribute('type');
} elseif ($node instanceof Expr\Closure) {
$type = Type::object_('\\Closure');
} elseif ($node instanceof Expr\Cast\Array_) {
$exprType = $node->expr->getAttribute('type');
$altTypes = [$exprType];
if ($exprType instanceof AlternativesType) {
$altTypes = $exprType->getAlternatives();
}
$types = [];
foreach ($altTypes as $altType) {
if ($altType instanceof ArrayType) {
$types[] = $altType;
} else {
$types[] = Type::array_();
// TODO primitives: (array)int --> int[]
}
}
$type = $types !== [] ? Type::alternatives($types) : Type::mixed_();
} elseif ($node instanceof Expr\Cast\Object_) {
$exprType = $node->expr->getAttribute('type');
$altTypes = [$exprType];
if ($exprType instanceof AlternativesType) {
$altTypes = $exprType->getAlternatives();
}
$types = [];
foreach ($altTypes as $altType) {
if ($altType instanceof ObjectType) {
$types[] = $altType;
} else {
$types[] = Type::object_('\\stdClass');
}
}
$type = $types !== [] ? Type::alternatives($types) : Type::mixed_();
} elseif ($node instanceof Expr\Include_) {
$type = Type::mixed_();
// TODO: Yield_
// TODO: YieldFrom
}
if ($type !== null) {
$node->setAttribute('type', $type);
}
if ($reflections !== null) {
$node->setAttribute('reflections', $reflections);
}
}
