development-x / MediaServiceProvider

src/Media/Provider/MediaProvider.php

<?php

namespace Media\Provider;

use Doctrine\ORM\EntityManager;
use Media\Entity\MediaInterface;
use Media\Provider\MediaProviderInterface;
use Symfony\Component\HttpFoundation\File\UploadedFile;

class MediaProvider implements MediaProviderInterface
{

    /**
     *
     * @var EntityManager
     */
    private $entityManager;

    /**
     *
     * @var string
     */
    private $path;

    /**
     * 
     * @param EntityManager $entityManager
     * @param string $path
     */
    public function __construct(EntityManager $entityManager, $path)

You have injected the EntityManager via parameter $entityManager. This is generally not recommended as it might get closed and become unusable. Instead, it is recommended to inject the ManagerRegistry and retrieve the EntityManager via getManager() each time you need it.

    {
        $this->path = $path;
        $this->entityManager = $entityManager;
    }

    /**
     * 
     * @param UploadedFile $file
     * @param string $directory
     * 
     * @return string
     */
    public function generateName(UploadedFile $file, $directory)
    {
        $extension = $file->guessClientExtension();
        do {
            $filename = md5(uniqid()) . '.' . $extension;
            if (file_exists($directory . DIRECTORY_SEPARATOR . $filename)) {
                $filename = null;
            }
        } while (is_null($filename));

        return $filename;
    }

    public function save(MediaInterface $media, $class, $options)
    {

        if (!$media->getFileContent() instanceof UploadedFile) {
            return $media;
        }

        $directory = $this->path . DIRECTORY_SEPARATOR . $options['context'];
        if (!is_dir($directory) || !is_writeable($directory)) {
            mkdir($directory, '0755', true);
        }

        if ($options['new_on_update']) {
            $entity = new $class;
        } else {
            $entity = clone $media;
        }

        $filename = $this->generateName($media->getFileContent(), $directory);

        $media->getFileContent()->move($directory, $filename);

        $entity->setFileName($filename);
        $entity->setContext($options['context']);
        $entity->setFileType($media->getFileContent()->getClientMimeType());
        $entity->setFileSize($media->getFileContent()->getClientSize());
        $entity->setMetadata([
            'filename' => $media->getFileContent()->getClientOriginalName(),
            'filesize' => $media->getFileContent()->getClientSize(),
            'filetype' => $media->getFileContent()->getClientMimeType(),
            'extension' => $media->getFileContent()->getClientOriginalExtension()
        ]);

        $this->entityManager->persist($entity);
        $this->entityManager->flush();

        return $entity;
    }

}