Added writeJsonObject method for write JsonSerializable object (#10)

composer.json

@@ -34,6 +34,7 @@
   },
   "autoload-dev": {
     "psr-4": {
+      "Tests\\Fixtures\\": "tests/fixtures"
     }
   },
   "scripts": {

src/Extra/ExtraResponseTrait.php

@@ -4,6 +4,7 @@
 
 use Contributte\Psr7\Psr7Response;
 use Contributte\Psr7\Psr7Stream;
+use JsonSerializable;
 
 /**
  * @method Psr7Stream getBody()
@@ -54,6 +55,17 @@ public function writeJsonBody(array $data)
 			->withHeader('Content-Type', 'application/json');
 	}
 
+	/**
+	 * @param JsonSerializable $object
+	 * @return static|Psr7Response
+	 */
+	public function writeJsonObject(JsonSerializable $object)
+	{
+		return $this
+			->writeBody(json_encode($object))
+			->withHeader('Content-Type', 'application/json');
+	}
+
 	/**
 	 * @param bool $assoc
 	 * @return mixed

tests/cases/Psr7Response.phpt

@@ -12,6 +12,7 @@ use Nette\Application\Responses\TextResponse;
 use Nette\Http\IResponse;
 use Nette\Http\Response;
 use Tester\Assert;
+use Tests\Fixtures\JsonObject;
 
 require_once __DIR__ . '/../bootstrap.php';
 
@@ -55,6 +56,14 @@ test(function () {
 	Assert::equal(['foo' => 'bar'], $response->getJsonBody());
 });
 
+// writeJsonObject
+test(function () {
+	$response = Psr7ResponseFactory::fromGlobal();
+	$jsonObject = new JsonObject('bar');
+	$response = $response->writeJsonObject($jsonObject);
+	Assert::equal(['foo' => 'bar'], $response->getJsonBody());
+});
+
 // appendBody
 test(function () {
 	$response = Psr7ResponseFactory::fromGlobal();

tests/fixtures/JsonObject.php

@@ -0,0 +1,35 @@
+<?php
+
+namespace Tests\Fixtures;
+
+use JsonSerializable;
+
+class JsonObject implements JsonSerializable
+{
+
+	/** @var string */
+	private $foo;
+
+	/**
+	 * JsonObject constructor.
+	 *
+	 * @param string $foo
+	 */
+	public function __construct($foo)
+	{
+		$this->foo = $foo;
+	}
+
+	/**
+	 * Specify data which should be serialized to JSON
+	 *
+	 * @link http://php.net/manual/en/jsonserializable.jsonserialize.php
+	 * @return mixed data which can be serialized by <b>json_encode</b>, which is a value of any type other than a resource.
+	 * @since 5.4.0
+	 */
+	public function jsonSerialize()
+	{
+		return ['foo' => $this->foo];
+	}
+
+}