PhUml\Graphviz\Builders\NodeLabelBuilder

Complexity

Total Complexity

6

Size/Duplication

Total Lines	42
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	removeNewLinesFrom()	0	3	1
A	forClass()	0	5	1
A	forInterface()	0	5	1
A	buildLabel()	0	6	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\Graphviz\Builders;

use PhUml\Code\ClassDefinition;
use PhUml\Code\InterfaceDefinition;
use Twig_Environment as TemplateEngine;
use Twig_Error_Loader as LoaderError;
use Twig_Error_Runtime as RuntimeError;
use Twig_Error_Syntax as SyntaxError;

/**
 * It creates an HTML table out of either a class or an interface.
 *
 * The table is used as a label for the nodes in the digraph use to create the class diagram
 */
class NodeLabelBuilder
{
    /** @var TemplateEngine */
    private $engine;

    /** @var HtmlLabelStyle */
    private $style;

    public function __construct(TemplateEngine $engine, HtmlLabelStyle $style)
    {
        $this->engine = $engine;
        $this->style = $style;
    }

    public function forClass(ClassDefinition $class): string
    {
        return $this->buildLabel('class.html.twig', [
            'class' => $class,
            'style' => $this->style,
        ]);
    }

    public function forInterface(InterfaceDefinition $interface): string
    {
        return $this->buildLabel('interface.html.twig', [
            'interface' => $interface,
            'style' => $this->style,
        ]);
    }

    private function buildLabel(string $template, array $options): string
    {
        try {
            return "<{$this->removeNewLinesFrom($this->engine->render($template, $options))}>";
        } catch (LoaderError | RuntimeError | SyntaxError $e) {
            throw new NodeLabelError($e);
        }
    }

    private function removeNewLinesFrom(string $label): string
    {
        return trim(preg_replace('/\s\s+|\n/', '', $label));
    }
}