public function leaveNode(Node $node) { if (!$node instanceof Expr) { return; } $type = null; if ($node instanceof Expr\Assign || $node instanceof Expr\AssignRef) { $type = $node->expr->getAttribute('type'); } elseif ($node instanceof Expr\Ternary) { $exprType1 = $node->if === null ? $node->cond->getAttribute('type') : $node->if->getAttribute('type'); $exprType2 = $node->else->getAttribute('type'); $type = Type::alternatives([$exprType1, $exprType2]); } elseif ($node instanceof Expr\BinaryOp\Coalesce) { $exprType1 = $node->left->getAttribute('type'); $exprType2 = $node->right->getAttribute('type'); $type = Type::alternatives([$exprType1, $exprType2]); } elseif ($node instanceof Expr\ErrorSuppress) { $type = $node->expr->getAttribute('type'); } elseif ($node instanceof Expr\Eval_) { $type = Type::mixed_(); } elseif ($node instanceof Expr\Exit_) { $type = Type::null_(); } if ($type !== null || !$node->hasAttribute('type')) { if ($type === null) { $type = Type::mixed_(); } $node->setAttribute('type', $type); } }
public function test_MethodCall_alternatives() { $method = (new Method())->setDocReturnType(Type::int_())->setStatic(false); $refl = $this->getMockBuilder(Reflection::class)->disableOriginalConstructor()->getMock(); $refl->expects($this->once())->method('findMethod')->with($this->equalTo('\\C'), $this->equalTo('f'))->willReturn($method); $var1 = new Expr\Variable('a', ['type' => Type::alternatives([Type::bool_(), Type::object_('\\C')])]); $expr = new Expr\MethodCall($var1, 'f'); $this->infer([$expr], $refl); $this->assertTrue($expr->getAttribute('type')->equals(Type::int_())); }
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); } }
/** * @param Method[] $methods * @param Method $method */ protected function mergeMethod(array &$methods, Method $method) { // TODO: Do a better job here? $method->setDocReturnType(Type::alternatives([$method->getDocReturnType(), $method->getReturnType()])); $params = []; foreach ($method->getParams() as $param) { $param->setDocType(Type::alternatives([$param->getDocType(), $param->getTypeHint()])); $params[$param->getName()] = $param; } if (isset($methods[strtolower($method->getName())])) { $baseMethod = $methods[strtolower($method->getName())]; $method->setDocReturnType(Type::alternatives([$method->getDocReturnType(), $baseMethod->getDocReturnType()])); foreach ($baseMethod->getParams() as $baseParam) { if (array_key_exists($baseParam->getName(), $params)) { $param = $params[$baseParam->getName()]; $param->setDocType(Type::alternatives([$param->getDocType(), $baseParam->getDocType()])); } } } $methods[strtolower($method->getName())] = $method; }