geekhub-php / CheTheatre

src/AppBundle/Command/RemoveBrokenMediaObjectsCommand.php

<?php

namespace AppBundle\Command;

use Doctrine\ORM\EntityManager;
use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\PropertyAccess\PropertyAccessor;
use Sonata\MediaBundle\Entity\BaseMedia as SonataEntityMedia;

class RemoveBrokenMediaObjectsCommand extends ContainerAwareCommand

The class Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand has been deprecated with message: since Symfony 4.2, use {@see Command} instead.

{
    protected function configure()
    {
        $this
            ->setName('app:remove-image')
            ->setDescription('Remove image that not relation to media')
        ;
    }

    /**
     * @param  InputInterface  $input
     * @param  OutputInterface $output
     * @return int|null|void
     */
    protected function execute(InputInterface $input, OutputInterface $output)
    {
        $mediaManager = $this->getContainer()->get('sonata.media.manager.media');

        $associatedMediaMappings = $this->getAssociationMediaMappings();

        $associatedMediaObjects = $this->getAssociatedMediaObjects($associatedMediaMappings);
        $associatedMediaObjectsIds = array_map(function (SonataEntityMedia $media) {
            return $media->getId();
        }, $associatedMediaObjects);

        foreach ($mediaManager->findAll() as $media) {
            if (false == in_array($media->getId(), $associatedMediaObjectsIds)) {

It seems like you are loosely comparing two booleans. Considering using the strict comparison === instead.

                $output->writeln(sprintf('Removed media with id "%s"', $media->getId()));
                $mediaManager->delete($media);
            }
        }

        $output->writeln('Deleted media without reference object was successful');
    }

    /**
     * @param  array $associationMappings
     * @return array
     */
    protected function getAssociatedMediaObjects(array $associationMappings)
    {
        /** @var EntityManager $em */
        $em = $this->getContainer()->get('doctrine')->getManager();

        if ($em->getFilters()->isEnabled('softdeleteable')) {
            $em->getFilters()->disable('softdeleteable');
        }

        $accessor = new PropertyAccessor();

        $mediaObjects = [];

        foreach ($associationMappings as $repositoryName => $properties) {
            foreach ($properties as $property) {
                $objects = $em->getRepository($repositoryName)->findAll();

                foreach ($objects as $object) {
                    if (!is_null($accessor->getValue($object, $property))) {
                        $mediaObjects[] = $accessor->getValue($object, $property);
                    }
                }
            }
        }

        return $mediaObjects;
    }

    /**
     * @return array associationMappings[className][propertyName][metadata]
     */
    protected function getAssociationMediaMappings()
    {
        $appClassMetadata = $this
            ->getContainer()
            ->get('sonata.media.manager.media')
            ->getEntityManager()
            ->getMetadataFactory()
            ->getAllMetadata()
        ;

        $mediaAssociationMappings = [];

        foreach ($appClassMetadata as $classMetadata) {
            foreach ($classMetadata->associationMappings as $propertyName => $associationMapping) {
                if ('App\Entity\Media' == $associationMapping['targetEntity']) {
                    $mediaAssociationMappings[$classMetadata->name][] = $propertyName;
                }
            }
        }

        return $mediaAssociationMappings;
    }
}