Merge pull request #13 from drupol/AttributeNode

Add an AttributeNode node type.

Author: drupol

spec/drupol/phptree/Node/AttributeNodeSpec.php

@@ -0,0 +1,34 @@
+<?php
+
+declare(strict_types = 1);
+
+namespace spec\drupol\phptree\Node;
+
+use drupol\phptree\Node\AttributeNode;
+use PhpSpec\ObjectBehavior;
+
+class AttributeNodeSpec extends ObjectBehavior
+{
+    public function it_can_set_and_get_the_attributes()
+    {
+        $attributes = [
+            'foo' => 'bar',
+        ];
+
+        $attributes = new \ArrayObject($attributes);
+
+        $this
+            ->setAttributes($attributes)
+            ->getAttributes()
+            ->shouldReturn($attributes);
+
+        $this
+            ->setAttribute('bar', 'foo')
+            ->getAttribute('bar')
+            ->shouldReturn('foo');
+    }
+    public function it_is_initializable()
+    {
+        $this->shouldHaveType(AttributeNode::class);
+    }
+}

src/Node/AttributeNode.php

@@ -0,0 +1,68 @@
+<?php
+
+declare(strict_types = 1);
+
+namespace drupol\phptree\Node;
+
+use drupol\phptree\Traverser\TraverserInterface;
+
+/**
+ * Class AttributeNode.
+ */
+class AttributeNode extends NaryNode implements AttributeNodeInterface
+{
+    /**
+     * ValueNode constructor.
+     *
+     * @param array $attributes
+     * @param null|int $capacity
+     * @param null|\drupol\phptree\Traverser\TraverserInterface $traverser
+     * @param null|\drupol\phptree\Node\NodeInterface $parent
+     */
+    public function __construct(
+        array $attributes = [],
+        ?int $capacity = 0,
+        ?TraverserInterface $traverser = null,
+        ?NodeInterface $parent = null
+    ) {
+        parent::__construct($capacity, $traverser, $parent);
+
+        $this->storage()->set('attributes', new \ArrayObject($attributes));
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function getAttribute($key)
+    {
+        return $this->getAttributes()[$key] ?? null;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function getAttributes(): \Traversable
+    {
+        return $this->storage()->get('attributes');
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function setAttribute(string $key, $value): AttributeNodeInterface
+    {
+        $this->getAttributes()[$key] = $value;
+
+        return $this;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function setAttributes(\Traversable $attributes): AttributeNodeInterface
+    {
+        $this->storage()->set('attributes', $attributes);
+
+        return $this;
+    }
+}

src/Node/AttributeNodeInterface.php

@@ -0,0 +1,30 @@
+<?php
+
+declare(strict_types = 1);
+
+namespace drupol\phptree\Node;
+
+/**
+ * Interface AttributeNodeInterface.
+ */
+interface AttributeNodeInterface extends NaryNodeInterface
+{
+    /**
+     * {@inheritdoc}
+     */
+    public function getAttribute($key);
+    /**
+     * {@inheritdoc}
+     */
+    public function getAttributes(): \Traversable;
+
+    /**
+     * {@inheritdoc}
+     */
+    public function setAttribute(string $key, $value): AttributeNodeInterface;
+
+    /**
+     * {@inheritdoc}
+     */
+    public function setAttributes(\Traversable $attributes): AttributeNodeInterface;
+}