roxblnfk\SmartStream\Data\FileBucket

Complexity

Total Complexity

26

Size/Duplication

Total Lines	126
Duplicated Lines	0 %

Test Coverage

Coverage

77.14%

Importance

Changes	2
Bugs	2	Features	1

15 Methods

Rating	Name	Duplication	Size	Complexity
A	contentType()	0	7	4
A	isInline()	0	3	1
A	hasDisposition()	0	3	1
A	withInline()	0	5	1
A	getContentType()	0	3	1
A	isAttachment()	0	3	1
A	withContentType()	0	5	1
A	getFileName()	0	3	1
A	createFromPath()	0	3	1
A	setAttachment()	0	5	1
A	setContentType()	0	4	1
A	withAttachment()	0	5	1
A	setInline()	0	4	1
A	__construct()	0	21	6
A	setDispositionHeader()	0	14	4

<?php

namespace roxblnfk\SmartStream\Data;

use SplFileInfo;
use Yiisoft\Http\Header;

class FileBucket extends DataBucket
{
    public const TYPE_OCTET_STREAM = 'application/octet-stream';

    protected const DISPOSITION_INLINE = 'inline';
    protected const DISPOSITION_ATTACHMENT = 'attachment';

    protected const IS_CONVERTABLE = false;
    protected ?string $contentType = null;
    protected ?string $contentDisposition = null;
    protected ?string $fileName = null;

    /**
     * FileBucket constructor.
     * @param string|resource|SplFileInfo $data
     * @param null|string $contentType
     * @param null|string $filename
     * @throws \Exception
     */
    public function __construct($data, string $contentType = null, string $filename = null)
    {
        switch (true) {
            case $data instanceof SplFileInfo:
                $filename = $filename ?? $data->getFilename();
                $contentType = $contentType ?? $this->contentType($filename);
                parent::__construct($data);
                break;
            case is_string($data):
            case is_resource($data):
                parent::__construct($data);
                break;
            default:
                throw new \Exception('The $data parameter must be a resource, a string or an instance of SplFileInfo');
        }

        if ($contentType !== null) {
            $this->setContentType($contentType);
        }
        if ($filename !== null) {
            $this->setAttachment($filename);
        }
    }

    public static function createFromPath(string $filePath): self
    {
        return new static(new SplFileInfo($filePath));
    }

    public function getContentType(): ?string
    {
        return $this->contentType;
    }
    public function getFileName(): ?string
    {
        return $this->fileName;
    }
    public function hasDisposition(): bool
    {
        return $this->contentDisposition !== null;
    }
    public function isAttachment(): bool
    {
        return $this->contentDisposition === self::DISPOSITION_ATTACHMENT;
    }
    public function isInline(): bool
    {
        return $this->contentDisposition === self::DISPOSITION_INLINE;
    }

    public function withAttachment(string $filename = null): self
    {
        $clone = clone $this;
        $clone->setAttachment($filename);
        return $clone;
    }
    public function withContentType(?string $contentType): self
    {
        $clone = clone $this;
        $clone->setContentType($contentType);
        return $clone;
    }
    public function withInline(): self
    {
        $clone = clone $this;
        $clone->setInline();
        return $clone;
    }

    public function setContentType(?string $contentType): void
    {
        $this->contentType = $contentType;
        $this->setHeader(Header::CONTENT_TYPE, $contentType);
    }
    public function setInline(): void
    {
        $this->contentDisposition = self::DISPOSITION_INLINE;
        $this->setDispositionHeader();
    }
    final protected function setAttachment(?string $filename): void
    {
        $this->contentDisposition = self::DISPOSITION_ATTACHMENT;
        $this->fileName = $filename;
        $this->setDispositionHeader();
    }
    final protected function setDispositionHeader(): void
    {
        if ($this->contentDisposition === null) {
            return;
        }
        $headerBody = ($this->contentDisposition === self::DISPOSITION_ATTACHMENT && $this->fileName !== null)
            ? sprintf(
                '%s; filename="%s"; filename*=UTF-8\'\'%s',
                $this->contentDisposition,
                preg_replace('/[\x00-\x1F\x7F\"]/', ' ', $this->fileName),
                rawurlencode($this->fileName)
            )
            : $this->contentDisposition;
        $this->setHeader(Header::CONTENT_DISPOSITION, $headerBody);
    }
    private function contentType(?string $filename): ?string
    {
        if ($filename === null) {
            return null;
        }
        $result = function_exists('mime_content_type') ? mime_content_type($filename) : null;
        return is_string($result) ? $result : null;
    }
}