Go\ParserReflection\NodeVisitor\GeneratorDetector

Complexity

Total Complexity

5

Size/Duplication

Total Lines	31
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Test Coverage

Coverage

100%

2 Methods

Rating	Name	Duplication	Size	Complexity
A	isGenerator()	0	4	1
A	enterNode()	0	13	4

<?php
/**
 * Parser Reflection API
 *
 * @copyright Copyright 2015, Lisachenko Alexander <lisachenko.it@gmail.com>
 *
 * This source file is subject to the license that is bundled
 * with this source code in the file LICENSE.
 */

namespace Go\ParserReflection\NodeVisitor;

use PhpParser\Node;
use PhpParser\NodeTraverser;
use PhpParser\NodeVisitorAbstract;

/**
 * Visitor to check if the method body
 */
class GeneratorDetector extends NodeVisitorAbstract
{
    private $isGenerator = false;

    /**
     * {@inheritDoc}
     */
    public function enterNode(Node $node)
    {
        // There may be internal generators in closures, we do not need to look at them
        if ($node instanceof Node\Expr\Closure) {
            return NodeTraverser::DONT_TRAVERSE_CHILDREN;
        }

        if ($node instanceof Node\Expr\Yield_ || $node instanceof Node\Expr\YieldFrom) {
            $this->isGenerator = true;
        }

        return null;
    }

    /**
     * Returns the
     *
     * @return bool
     */
    public function isGenerator()
    {
        return $this->isGenerator;
    }
}