Add DataAttribute utils and allow JSON serialization on it

wavevision · Mar 13, 2020 · 40d2f1f · 40d2f1f
1 parent 7caa617
commit 40d2f1f
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 8 deletions.
diff --git a/src/Utils/DOM/DataAttribute.php b/src/Utils/DOM/DataAttribute.php
@@ -2,11 +2,12 @@
 
 namespace Wavevision\Utils\DOM;
 
+use JsonSerializable;
 use Nette\SmartObject;
 use Nette\Utils\Html;
 use Wavevision\Utils\Strings;
 
-final class DataAttribute
+final class DataAttribute implements JsonSerializable
 {
 
 	use SmartObject;
@@ -26,6 +27,14 @@ public function __toString(): string
 		return $this->asString();
 	}
 
+	/**
+	 * @return array<string, string>
+	 */
+	public function jsonSerialize(): array
+	{
+		return $this->asArray();
+	}
+
 	/**
 	 * @param mixed $value
 	 * @return array<string, string>
@@ -37,11 +46,12 @@ public function asArray($value = null): array
 
 	/**
 	 * @param Html<mixed> $element
+	 * @param mixed $value
 	 * @return Html<mixed>
 	 */
-	public function assign(Html $element): Html
+	public function assign(Html $element, $value = null): Html
 	{
-		$element->setAttribute($this->currentName, $this->currentValue);
+		$element->setAttribute($this->currentName, $this->value($value));
 		return $element;
 	}
 
@@ -53,6 +63,22 @@ public function asString($value = null): string
 		return sprintf('%s="%s"', $this->currentName, $this->value($value));
 	}
 
+	/**
+	 * @param Html<mixed> $element
+	 */
+	public function get(Html $element): ?string
+	{
+		return $element->getAttribute($this->currentName);
+	}
+
+	/**
+	 * @param Html<mixed> $element
+	 */
+	public function has(Html $element): bool
+	{
+		return $this->get($element) !== null;
+	}
+
 	public function name(): string
 	{
 		return $this->currentName;

diff --git a/tests/UtilsTests/DOM/DataAttributeTest.php b/tests/UtilsTests/DOM/DataAttributeTest.php
@@ -3,12 +3,18 @@
 namespace Wavevision\UtilsTests\DOM;
 
 use Nette\Utils\Html;
+use Nette\Utils\Json;
 use PHPUnit\Framework\TestCase;
 use Wavevision\Utils\DOM\DataAttribute;
 
 class DataAttributeTest extends TestCase
 {
 
+	public function testJsonSerialize(): void
+	{
+		$this->assertEquals('{"data-test":""}', Json::encode($this->createDataAttribute()));
+	}
+
 	public function testAsArray(): void
 	{
 		$this->assertEquals(['data-test' => ''], $this->createDataAttribute()->asArray());
@@ -22,9 +28,18 @@ public function testAsString(): void
 
 	public function testAssign(): void
 	{
-		$element = Html::el();
-		$this->createDataAttribute()->assign($element);
-		$this->assertSame('', $element->getAttribute('data-test'));
+		$attribute = $this->createDataAttribute();
+		$this->assertSame('', $attribute->get($attribute->assign(Html::el())));
+	}
+
+	public function testGet(): void
+	{
+		$this->assertNull($this->createDataAttribute()->get(Html::el()));
+	}
+
+	public function testHas(): void
+	{
+		$this->assertFalse($this->createDataAttribute()->has(Html::el()));
 	}
 
 	public function testName(): void
@@ -36,8 +51,8 @@ public function testRemove(): void
 	{
 		$attribute = $this->createDataAttribute();
 		$element = Html::el();
-		$this->assertTrue($attribute->assign($element)->getAttribute($attribute->name()) === '');
-		$this->assertNull($attribute->remove($element)->getAttribute($attribute->name()));
+		$this->assertEquals('', $attribute->get($attribute->assign($element)));
+		$this->assertNull($attribute->get($attribute->remove($element)));
 	}
 
 	public function testValue(): void