Palmtree\Csv\Downloader

Complexity

Total Complexity

15

Size/Duplication

Total Lines	115
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Test Coverage

Coverage

0%

Importance

Changes	2
Bugs	0	Features	0

9 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	11	2
A	getResponseHeaders()	0	4	1
A	addResponseHeaders()	0	8	2
A	addResponseHeader()	0	4	1
A	removeResponseHeader()	0	4	1
A	getFilename()	0	4	1
A	setFilename()	0	6	1
B	download()	0	22	5
A	getResponseBody()	0	10	1

<?php

namespace Palmtree\Csv;

class Downloader extends Writer
{
    /**
     * Default headers used to tell client the response is
     * a downloadable, non-cacheable file.
     * @var array
     */
    protected $responseHeaders = [
        'Content-Type'              => 'application/octet-stream',
        'Content-Description'       => 'File Transfer',
        'Content-Transfer-Encoding' => 'Binary',
        'Content-Disposition'       => 'attachment; filename="%s"',
        'Expires'                   => '0',
        'Cache-Control'             => 'must-revalidate, post-check=0, pre-check=0',
        'Pragma'                    => 'public',
        'Content-Length'            => '%s',
    ];

    protected $filename;

    public function __construct($filename = '', $responseHeaders = [])
    {
        if (!$filename) {
            $filename = time() . '.csv';
        }

        $this->setFilename($filename);
        $this->addResponseHeaders($responseHeaders);

        parent::__construct('php://temp');
    }

    public function getResponseHeaders()
    {
        return $this->responseHeaders;
    }

    public function addResponseHeaders($userHeaders = [])
    {
        foreach ($userHeaders as $key => $value) {
            $this->addResponseHeader($key, $value);
        }

        return $this;
    }

    public function addResponseHeader($key, $value)
    {
        $this->responseHeaders[$key] = $value;
    }

    public function removeResponseHeader($key)
    {
        unset($this->responseHeaders[$key]);
    }

    /**
     * @return string
     */
    public function getFilename()
    {
        return $this->filename;
    }

    /**
     * @param string $filename
     *
     * @return Downloader
     */
    public function setFilename($filename)
    {
        $this->filename = $filename;

        return $this;
    }

    /**
     * Attempts to send the file as a download to the client.
     *
     * @throws \Exception
     */
    public function download()
    {
        $body = $this->getResponseBody();

        if (!headers_sent()) {
            $headers = $this->getResponseHeaders();

            foreach ($headers as $key => $value) {
                if ($key === 'Content-Length') {
                    $value = sprintf($value, mb_strlen($body));
                } elseif ($key === 'Content-Disposition') {
                    $value = sprintf($value, $this->getFilename());
                }

                header(sprintf('%s: %s', $key, $value));
            }
        }

        print $body;

        $this->closeFileHandle();
    }

    protected function getResponseBody()
    {
        $this->trimTrailingLineEnding();

        rewind($this->getFileHandle());

        $body = stream_get_contents($this->getFileHandle());

        return $body;
    }
}