BaseArrow.php

<?php
/*
 * Copyright (C) 2016-2017 IRSTEA
 * All rights reserved.
 */

namespace Irstea\PlantUmlBundle\Model\Arrow;

use Irstea\PlantUmlBundle\Model\ArrowInterface;
use Irstea\PlantUmlBundle\Model\NodeInterface;
use Irstea\PlantUmlBundle\Writer\WriterInterface;

/**
 * Description of Arrow.
 */
class BaseArrow implements ArrowInterface
{
    /**
     * @var NodeInterface
     */
    private $source;

    /**
     * @var NodeInterface
     */
    private $target;

    /**
     * @var string
     */
    private $link;

    /**
     * @var string
     */
    private $linkSource;

    /**
     * @var string
     */
    private $linkTarget;

    /**
     * @var string
     */
    private $sourceCardinality;

    /**
     * @var string
     */
    private $targetCardinality;

    /**
     * @var int
     */
    private static $splitCount = 0;

    public function __construct(
        NodeInterface $source,
        NodeInterface $target,
        $link = '--',
        $label = null,
        $linkSource = '',
        $linkTarget = '',
        $sourceCardinality = '',
        $targetCardinality = ''
    ) {
        $this->source = $source;
        $this->target = $target;
        $this->link = $link;
        $this->label = $label;
        $this->linkSource = $linkSource;
        $this->linkTarget = $linkTarget;
        $this->sourceCardinality = $sourceCardinality;
        $this->targetCardinality = $targetCardinality;
    }

    public function writeTo(WriterInterface $writer)
    {
        $this->source->writeAliasTo($writer);
        $this->writeLinkTo($writer);
        $this->target->writeAliasTo($writer);
        $this->writeLabelTo($writer);
        $writer->write("\n");

        return $this;
    }

    protected function writeLabelTo(WriterInterface $writer)
    {
        if ($this->label) {
            $writer->writeFormatted(' : %s', $this->label);
        }

        return $this;
    }

    protected function writeLinkTo(WriterInterface $writer, $linkSource = null, $linkTarget = null)
    {
        $this->writeSourceCardinalityTo($writer);
        $writer->writeFormatted(' %s%s%s ', $linkSource ?: $this->linkSource, $this->link, $linkTarget ?: $this->linkTarget);
        $this->writeTargetCardinalityTo($writer);

        return $this;
    }

    protected function writeSourceCardinalityTo(WriterInterface $writer)
    {
        if ($this->sourceCardinality !== '') {
            $writer->writeFormatted(' "%s"', $this->sourceCardinality);
        }

        return $this;
    }

    protected function writeTargetCardinalityTo(WriterInterface $writer)
    {
        if ($this->targetCardinality !== '') {
            $writer->writeFormatted('"%s" ', $this->targetCardinality);
        }

        return $this;
    }
}