pug-php/ci-pug

<?php

namespace Jade\Nodes;

class Block extends Node
{
    public $isBlock = true;
    public $nodes = array();

    public function __construct($node = null)
    {
        if (null !== $node) {
            $this->push($node);
        }
    }

    public function replace($other)
    {
        $other->nodes = $this->nodes;
    }

    public function push($node)
    {
        return array_push($this->nodes, $node);
    }

    public function isEmpty()
    {
        return 0 == count($this->nodes);
    }

    public function unshift($node)
    {
        return array_unshift($this->nodes, $node);
    }

    public function getYield()
    {
        foreach ($this->nodes as $node) {
            if (isset($node->yield)) {
                return $node;
            } elseif (isset($node->block) && $yield = $node->block->getYield()) {
                return $yield;
            }
        }

        return false;
    }

    public function includeBlock()
    {
        $ret = $this;
        foreach ($this->nodes as $node) {
            if (isset($node->yield)) {
                return $node;
            }

            if (isset($node->block) && $yield = $node->block->getYield()) {
                return $yield;
            }

            if (isset($node->textOnly)) {
                continue;
            }

            if (method_exists($node, 'includeBlock')) {
                $ret = $node->includeBlock();
            } elseif (isset($node->block) && !$node->block->isEmpty()) {
                $ret = $node->block->includeBlock();
            }
            if (isset($ret->yield)) {
                return $ret;
            }
        }

        return $ret;
    }
}