PhUml\Parser\Raw\Visitors\ClassVisitor

Complexity

Total Complexity

3

Size/Duplication

Total Lines	18
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	leaveNode()	0	4	2
A	__construct()	0	4	1

<?php
/**
 * PHP version 7.1
 *
 * This source file is subject to the license that is bundled with this package in the file LICENSE.
 */

namespace PhUml\Parser\Raw\Visitors;

use PhpParser\Node;
use PhpParser\Node\Stmt\Class_;
use PhpParser\NodeVisitorAbstract;
use PhUml\Parser\Raw\Builders\ClassBuilder;
use PhUml\Parser\Raw\RawDefinitions;

/**
 * It extracts the `RawDefinition` of a class and adds it to the collection of `RawDefinitions`
 */
class ClassVisitor extends NodeVisitorAbstract
{
    /** @var RawDefinitions */
    private $definitions;

    /** @var ClassBuilder */
    private $builder;

    public function __construct(RawDefinitions $definitions, ClassBuilder $builder = null)
    {
        $this->definitions = $definitions;
        $this->builder = $builder ?? new ClassBuilder();
    }

    public function leaveNode(Node $node)
    {
        if ($node instanceof Class_) {
            $this->definitions->add($this->builder->build($node));
        }
    }
}