App\Services\Upload\ExtendedFileSystemStorage

Complexity

Total Complexity

21

Size/Duplication

Total Lines	78
Duplicated Lines	0 %

Importance

Changes	3
Bugs	0	Features	0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	doResolvePath()	0	9	3
C	resolveUri()	0	38	13
A	__construct()	0	4	1
A	convertWindowsDirectorySeparator()	0	3	1
A	doUpload()	0	5	1
A	doRemove()	0	5	2

<?php
/*
 * Copyright (C) 2020  Jan Böhmer
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published
 * by the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

namespace App\Services\Upload;

use App\Entity\PaymentOrder;
use App\Entity\SEPAExport;
use Symfony\Component\HttpFoundation\File\File;
use Symfony\Component\HttpFoundation\File\UploadedFile;
use Symfony\Component\Routing\Generator\UrlGeneratorInterface;
use Vich\UploaderBundle\Mapping\PropertyMapping;
use Vich\UploaderBundle\Mapping\PropertyMappingFactory;
use Vich\UploaderBundle\Storage\AbstractStorage;
use Vich\UploaderBundle\Storage\FileSystemStorage;

/**
 * We need to extend the original filesystemstorage as we have stored our payment_order files privately and want to use
 * the normal interface to access them via URI...
 * Based on Vich FileSystemStorage class
 */
class ExtendedFileSystemStorage extends AbstractStorage
{
    private $router;

    public function __construct(PropertyMappingFactory $factory, UrlGeneratorInterface $router)
    {
        parent::__construct($factory);
        $this->router = $router;
    }

    public function resolveUri($obj, ?string $fieldName = null, ?string $className = null): ?string
    {

        [$mapping, $name] = $this->getFilename($obj, $fieldName, $className);

        if (empty($name)) {
            return null;
        }

        $uploadDir = $this->convertWindowsDirectorySeparator($mapping->getUploadDir($obj));
        $uploadDir = empty($uploadDir) ? '' : $uploadDir.'/';

        $tmp = \sprintf('%s/%s', $mapping->getUriPrefix(), $uploadDir.$name);

        if (null !== $tmp && $obj instanceof PaymentOrder) {
            if ('printed_form' === $fieldName || 'printed_form_file' === $fieldName) {
                return $this->router->generate('file_payment_order_form', [
                    'id' => $obj->getId(),
                ]);
            }

            if ('references' === $fieldName || 'references_file' === $fieldName) {
                return $this->router->generate('file_payment_order_references', [
                    'id' => $obj->getId(),
                ]);
            }
        }

        if (null !== $tmp && $obj instanceof SEPAExport)
        {
            if ('xml' === $fieldName || 'xml_file' === $fieldName) {
                return $this->router->generate('file_sepa_export_xml', [
                    'id' => $obj->getId(),
                ]);
            }
        }

        return $tmp;
    }

    protected function doUpload(PropertyMapping $mapping, UploadedFile $file, ?string $dir, string $name): ?File
    {
        $uploadDir = $mapping->getUploadDestination().\DIRECTORY_SEPARATOR.$dir;

        return $file->move($uploadDir, $name);
    }

    protected function doRemove(PropertyMapping $mapping, ?string $dir, string $name): ?bool
    {
        $file = $this->doResolvePath($mapping, $dir, $name);

        return \file_exists($file) && \unlink($file);
    }

    protected function doResolvePath(PropertyMapping $mapping, ?string $dir, string $name, ?bool $relative = false): string
    {
        $path = !empty($dir) ? $dir.\DIRECTORY_SEPARATOR.$name : $name;

        if ($relative) {
            return $path;
        }

        return $mapping->getUploadDestination().\DIRECTORY_SEPARATOR.$path;
    }

    private function convertWindowsDirectorySeparator(string $string): string
    {
        return \str_replace('\\', '/', $string);
    }
}