1:   2:   3:   4:   5:   6:   7:   8:   9:  10:  11:  12:  13:  14:  15:  16:  17:  18:  19:  20:  21:  22:  23:  24:  25:  26:  27:  28:  29:  30:  31:  32:  33:  34:  35:  36:  37:  38:  39:  40:  41:  42:  43:  44:  45:  46:  47:  48:  49:  50:  51:  52:  53:  54:  55:  56:  57:  58:  59:  60:  61:  62:  63:  64:  65:  66:  67:  68:  69:  70:  71:  72:  73:  74:  75:  76:  77:  78:  79:  80:  81:  82:  83:  84:  85:  86:  87:  88:  89:  90:  91:  92:  93:  94:  95:  96:  97:  98:  99: 100: 101: 102: 103: 104: 105: 106: 107: 108: 109: 110: 111: 112: 113: 114: 115: 116: 117: 118: 119: 120: 121: 122: 123: 124: 125: 126: 127: 128: 129: 130: 131: 132: 133: 134: 135: 136: 137: 138: 139: 140: 141: 142: 143: 144: 145: 146: 147: 148: 149: 150: 151: 152: 153: 154: 155: 156: 157: 158: 159: 160: 161: 162: 163: 164: 165: 166: 167: 168: 169: 170: 171: 172: 173: 174: 175: 176: 177: 178: 179: 180: 181: 182: 183: 184: 185: 186: 187: 188: 189: 190: 

<?php
/**
 * @author sleibelt
 * @since 2014-04-25
 */

namespace Net\Bazzline\Component\CodeGenerator;

/**
 * Class BlockGenerator
 *
 * @package Net\Bazzline\Component\Locator\LocatorGenerator
 * @todo add helpers like "startWhile", "startIf" and so one
 */
class BlockGenerator extends AbstractContentGenerator implements LineGeneratorDependentInterface
{
    /**
     * @var boolean
     */
    private $addIndented;

    /**
     * @var array|BlockGenerator[]|LineGenerator[]
     */
    private $content = [];

    /**
     * @var LineGenerator
     */
    private $lineGenerator;

    /**
     * @param LineGenerator $lineGenerator
     * @param Indention $indention
     * @param null|string|array|GeneratorInterface $content
     */
    public function __construct(LineGenerator $lineGenerator, Indention $indention, $content = null)
    {
        $this->addIndented = false;
        $this->lineGenerator = $lineGenerator;
        parent::__construct($indention, $content);
    }

    /**
     * @param LineGenerator $generator
     * @return $this
     */
    public function setLineGenerator(LineGenerator $generator)
    {
        $this->lineGenerator = $generator;

        return $this;
    }

    /**
     * @param Indention $indention
     * @return $this
     */
    public function setIndention(Indention $indention)
    {
        parent::setIndention($indention);
        $this->lineGenerator->setIndention($indention);

        return $this;
    }

    /**
     * @param array|GeneratorInterface|string $content
     * @return $this
     * @throws InvalidArgumentException
     * @todo rename to addContent, easy up this method by working on todo in generate method
     */
    public function add($content)
    {
        $indention = ($this->addIndented) ? $this->getIndention()->toString() : '';

        if (is_string($content)) {
            $lineOfContent = $this->getLineGenerator($indention . $content);
            $this->content[] = $lineOfContent;
        } else if (is_array($content)) {
            foreach ($content as $part) {
                $this->add($part);
            }
        } else if ($content instanceof AbstractContentGenerator) {
            $content->setIndention($this->getIndention());
            $this->content[] = $content;
        } else {
            throw new InvalidArgumentException('content has to be string, an array or instance of AbstractContentGenerator');
        }

        return $this;
    }

    /**
     * @return $this
     */
    public function clear()
    {
        $this->content = [];

        return $this;
    }

    /**
     * @return bool
     */
    public function hasContent()
    {
        return (!empty($this->content));
    }

    /**
     * @throws InvalidArgumentException|RuntimeException
     * @return string
     * @todo implement support for string|array beside AbstractContentGenerator in $this->content
     */
    public function generate()
    {
        $string = '';
        $arrayKeys = array_keys($this->content);
        $lastKey = array_pop($arrayKeys);

        foreach ($this->content as $key => $content) {
            if ($content->hasContent()) {
                if ($content instanceof BlockGenerator) {
                    $this->getIndention()->increaseLevel();
                    $string .= $content->generate();
                    $this->getIndention()->decreaseLevel();
                } else {
                    $string .= $content->generate();
                }
                if ($key !== $lastKey) {
                    $string .= PHP_EOL;
                }
            }
        }

        return $string;
    }

    /**
     * @return int
     */
    public function count()
    {
        return (count($this->content));
    }

    /**
     * @todo find better name
     * @return $this
     */
    public function startIndention()
    {
        $this->getIndention()->increaseLevel();
        $this->addIndented = true;

        return $this;
    }

    /**
     * @todo find better name
     * @return $this
     */
    public function stopIndention()
    {
        $this->getIndention()->decreaseLevel();
        if ($this->getIndention()->isSetToInitialLevel()) {
            $this->addIndented = false;
        }

        return $this;
    }

    /**
     * @param null|string $content
     * @return LineGenerator
     */
    final protected function getLineGenerator($content = null)
    {
        $line = clone $this->lineGenerator;

        if (!is_null($content)) {
            $line->add($content);
        }

        return $line;
    }
}