Chris@0: <?php
Chris@0: /**
Chris@0:  * Zend Framework (http://framework.zend.com/)
Chris@0:  *
Chris@0:  * @see       http://github.com/zendframework/zend-diactoros for the canonical source repository
Chris@0:  * @copyright Copyright (c) 2015-2016 Zend Technologies USA Inc. (http://www.zend.com)
Chris@0:  * @license   https://github.com/zendframework/zend-diactoros/blob/master/LICENSE.md New BSD License
Chris@0:  */
Chris@0: 
Chris@0: namespace Zend\Diactoros\Response;
Chris@0: 
Chris@0: use InvalidArgumentException;
Chris@0: use Psr\Http\Message\StreamInterface;
Chris@0: use Zend\Diactoros\Response;
Chris@0: use Zend\Diactoros\Stream;
Chris@0: 
Chris@16: use function get_class;
Chris@16: use function gettype;
Chris@16: use function is_object;
Chris@16: use function is_string;
Chris@16: use function sprintf;
Chris@16: 
Chris@0: /**
Chris@0:  * HTML response.
Chris@0:  *
Chris@0:  * Allows creating a response by passing an HTML string to the constructor;
Chris@0:  * by default, sets a status code of 200 and sets the Content-Type header to
Chris@0:  * text/html.
Chris@0:  */
Chris@0: class HtmlResponse extends Response
Chris@0: {
Chris@0:     use InjectContentTypeTrait;
Chris@0: 
Chris@0:     /**
Chris@0:      * Create an HTML response.
Chris@0:      *
Chris@0:      * Produces an HTML response with a Content-Type of text/html and a default
Chris@0:      * status of 200.
Chris@0:      *
Chris@0:      * @param string|StreamInterface $html HTML or stream for the message body.
Chris@0:      * @param int $status Integer status code for the response; 200 by default.
Chris@0:      * @param array $headers Array of headers to use at initialization.
Chris@0:      * @throws InvalidArgumentException if $html is neither a string or stream.
Chris@0:      */
Chris@0:     public function __construct($html, $status = 200, array $headers = [])
Chris@0:     {
Chris@0:         parent::__construct(
Chris@0:             $this->createBody($html),
Chris@0:             $status,
Chris@0:             $this->injectContentType('text/html; charset=utf-8', $headers)
Chris@0:         );
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * Create the message body.
Chris@0:      *
Chris@0:      * @param string|StreamInterface $html
Chris@0:      * @return StreamInterface
Chris@0:      * @throws InvalidArgumentException if $html is neither a string or stream.
Chris@0:      */
Chris@0:     private function createBody($html)
Chris@0:     {
Chris@0:         if ($html instanceof StreamInterface) {
Chris@0:             return $html;
Chris@0:         }
Chris@0: 
Chris@0:         if (! is_string($html)) {
Chris@0:             throw new InvalidArgumentException(sprintf(
Chris@0:                 'Invalid content (%s) provided to %s',
Chris@0:                 (is_object($html) ? get_class($html) : gettype($html)),
Chris@0:                 __CLASS__
Chris@0:             ));
Chris@0:         }
Chris@0: 
Chris@0:         $body = new Stream('php://temp', 'wb+');
Chris@0:         $body->write($html);
Chris@0:         $body->rewind();
Chris@0:         return $body;
Chris@0:     }
Chris@0: }