PhUml\Parser\Code\Visitors\TraitVisitor

Complexity

Total Complexity

3

Size/Duplication

Total Lines	18
Duplicated Lines	0 %

Test Coverage

Coverage

100%

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\Code\Visitors;

use PhpParser\Node;
use PhpParser\Node\Stmt\Trait_;
use PhpParser\NodeVisitorAbstract;
use PhUml\Code\Codebase;
use PhUml\Parser\Code\Builders\TraitDefinitionBuilder;

class TraitVisitor extends NodeVisitorAbstract
{
    /** @var TraitDefinitionBuilder */
    private $builder;

    /** @var Codebase */
    private $codebase;

    public function __construct(TraitDefinitionBuilder $builder, Codebase $codebase)
    {
        $this->builder = $builder;
        $this->codebase = $codebase;
    }

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