PhUml\Graphviz\Builders\ClassGraphBuilder

Complexity

Total Complexity

17

Size/Duplication

Total Lines	92
Duplicated Lines	0 %

Importance

Changes

0

8 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
B	extractFrom()	0	22	4
A	isAssociationResolved()	0	3	1
A	addElementsForParameters()	0	7	3
A	addAssociationForVariable()	0	8	2
A	createAssociations()	0	3	1
A	addElementsForAttributes()	0	5	2
A	needAssociation()	0	3	3

<?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\Graphviz\Builders;

use PhUml\Code\ClassDefinition;
use PhUml\Code\Structure;
use PhUml\Code\Variable;
use PhUml\Graphviz\Edge;
use PhUml\Graphviz\Node;

class ClassGraphBuilder
{
    /** @var \PhUml\Graphviz\HasDotRepresentation[] */
    private $dotElements = [];

    /** @var bool[] */
    private $associations = [];

    /** @var bool */
    private $createAssociations;

    /** @var NodeLabelBuilder */
    private $labelBuilder;

    /** @var Structure */
    private $structure;

    public function __construct(NodeLabelBuilder $labelBuilder)
    {
        $this->createAssociations = false;
        $this->labelBuilder = $labelBuilder;
    }

    public function createAssociations(): void
    {
        $this->createAssociations = true;
    }

    /** @return \PhUml\Graphviz\HasDotRepresentation[] */
    public function extractFrom(ClassDefinition $class, Structure $structure): array
    {
        $this->dotElements = [];
        $this->associations = [];
        $this->structure = $structure;

        if ($this->createAssociations) {
            $this->addElementsForAttributes($class);
            $this->addElementsForParameters($class);
        }

        $this->dotElements[] = new Node($class, $this->labelBuilder->forClass($class));

        if ($class->hasParent()) {
            $this->dotElements[] = Edge::inheritance($class->extends, $class);
        }

        foreach ($class->implements as $interface) {
            $this->dotElements[] = Edge::implementation($interface, $class);
        }

        return $this->dotElements;
    }

    /** @return \PhUml\Graphviz\HasDotRepresentation[] */
    private function addElementsForAttributes(ClassDefinition $class): void
    {
        /** @var \PhUml\Code\Attribute $attribute */
        foreach ($class->attributes as $attribute) {
            $this->addAssociationForVariable($class, $attribute);
        }
    }

    /** @return \PhUml\Graphviz\HasDotRepresentation[] */
    private function addElementsForParameters(ClassDefinition $class): void
    {
        if (!$class->hasConstructor()) {
            return;
        }
        foreach ($class->constructorParameters() as $parameter) {
            $this->addAssociationForVariable($class, $parameter);
        }
    }

    private function addAssociationForVariable(ClassDefinition $class, Variable $attribute): void
    {
        if ($this->needAssociation($attribute)) {
            $this->dotElements[] = Edge::association(
                $this->structure->get((string)$attribute->type),
                $class
            );
            $this->associations[strtolower($attribute->type)] = true;
        }
    }

    private function needAssociation(Variable $attribute): bool
    {
        return $attribute->hasType() && !$attribute->isBuiltIn() && !$this->isAssociationResolved($attribute->type);
    }

    private function isAssociationResolved(string $type): bool
    {
        return array_key_exists(strtolower($type), $this->associations);
    }
}