phug-php/phug

<?php

namespace Phug\Compiler\NodeCompiler;

use Phug\Compiler\AbstractNodeCompiler;
use Phug\Formatter\Element\AssignmentElement;
use Phug\Formatter\ElementInterface;
use Phug\Parser\Node\AssignmentNode;
use Phug\Parser\Node\AttributeNode;
use Phug\Parser\NodeInterface;
use Phug\Util\OrderableInterface;
use SplObjectStorage;

class AssignmentNodeCompiler extends AbstractNodeCompiler
{
    public function compileNode(NodeInterface $node, ElementInterface $parent = null)
    {
        $this->getCompiler()->assert(
            $node instanceof AssignmentNode,
            'Unexpected '.get_class($node).' given to assignment compiler.',
            $node
        );

        /**
         * @var AssignmentNode $node
         */
        $name = $node->getName();
        $attributes = new SplObjectStorage();

        foreach ($node->getAttributes() as $attribute) {
            /* @var AttributeNode $attribute */
            $attributeElement = $this->getCompiler()->compileNode($attribute, $parent);

            if ($attribute instanceof OrderableInterface && $attributeElement instanceof OrderableInterface) {
                $attributeElement->setOrder($attribute->getOrder());
            }

            $attributes->attach($attributeElement);
        }

        return new AssignmentElement($name, $attributes, null, $node);
    }
}