vitiko/IphpFileStoreBundle

<?php

namespace Iphp\FileStoreBundle\Command;

use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand;
use Symfony\Component\HttpFoundation\File\UploadedFile;
use Sensio\Bundle\GeneratorBundle\Command\GenerateDoctrineCommand;
use Sensio\Bundle\GeneratorBundle\Generator\DoctrineEntityGenerator;
use Sensio\Bundle\GeneratorBundle\Command\Helper\DialogHelper;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Output\OutputInterface;


class RepairFileDataCommand extends ContainerAwareCommand
{

    protected function configure()
    {
        $this->setName('iphp:filestore:repair')
            ->addOption('entity', null, InputOption::VALUE_REQUIRED, 'The entity class name (shortcut notation)')
            ->addOption('field', null, InputOption::VALUE_REQUIRED, 'The field with file data')
            ->addOption('maxresults', null, InputOption::VALUE_OPTIONAL, 'Max results limitation')
            ->addOption('webdir', null, InputOption::VALUE_OPTIONAL, 'Web dir for searching file')
            ->addOption('force', null, InputOption::VALUE_OPTIONAL, 'Force reupload and rename files');

    }


    protected function getPropertyMappingFactory()
    {
        return $this->getContainer()->get('iphp.filestore.mapping.factory');
    }


    protected function getWebDir(InputInterface $input)
    {
        $webDir = $input->getOption('webdir');

        if (!$webDir && $this->getContainer()->has('iphp.web_dir'))
            $webDir = $this->getContainer()->getParameter('iphp.web_dir');

        if (!$webDir) $webDir = str_replace('\\', '/', realpath($this->getContainer()->getParameter('kernel.root_dir') . '/../web/'));

        if (!$webDir) throw new \InvalidArgumentException ('
         For resolving IphpFileStoreBundle uploaded files need to set --webdir option');

        return $webDir;
    }


    function getMaxResults(InputInterface $input)
    {
        $maxResults = $input->getOption('maxresults');
        if (!$maxResults || !is_numeric($maxResults)) $maxResults = 1000;

        return $maxResults;
    }


    function getEntityManager()
    {
        return $this->getContainer()->get('doctrine')->getManager();
    }

    function getMappingFromField($entity, $field)
    {
        return $this->getPropertyMappingFactory()->getMappingFromField($entity, new \ReflectionClass($entity), $field);
    }


    function  getRepository($entityFullName)
    {
        return $this->getEntityManager()->getRepository($entityFullName);
    }

    function getEntityIds($entityFullName, $maxResults)
    {
        return $this->getEntityManager()->createQuery(
            "SELECT e.id FROM " . $entityFullName . " e ORDER BY e.id ASC"
        )->setMaxResults($maxResults)->getArrayResult();
    }

    protected function execute(InputInterface $input, OutputInterface $output)
    {
        $entityFullName = $input->getOption('entity');
        $field = $input->getOption('field');
        $force = $input->getOption('force') ? true : false;
        $webDir = $this->getWebDir($input);


        foreach ($this->getEntityIds($entityFullName, $this->getMaxResults($input)) as $pos => $e) {

            $toFlush = false;
            $entity = $this->getRepository($entityFullName)->findOneById($e['id']);
            $fileData = $entity->{'get' . ucfirst($field)}();

            if (!$fileData) continue;

            $fileNameByWebPath = $webDir . $fileData['path'];
            $fileNameByWebPathExists = file_exists($fileNameByWebPath);

            $resolvedFileName = $this->getMappingFromField($entity, $field)->resolveFileName($fileData['fileName']);
            $resolvedFileNameExists = file_exists($resolvedFileName) ? 'exists' : 'NO';

            if (!$fileNameByWebPathExists && !$resolvedFileNameExists) {
                $output->writeln("can't find file ");
                continue;
            }

            if ($fileNameByWebPathExists && $resolvedFileNameExists && !$force) continue;

            //uploadedFile because need to move to new destination (not copy)
            $file = new UploadedFile ($fileNameByWebPathExists ? $fileNameByWebPath : $resolvedFileNameExists,
                $fileData['originalName'], $fileData['mimeType'], null, null, true);

            $entity->{'set' . ucfirst($field)} ($file);

            $this->getEntityManager()->persist($entity);

            $toFlush = true;
            if ($pos % 20 == 0 && $toFlush) $this->getEntityManager()->flush();
            if ($pos % 100 == 0) $this->getEntityManager()->clear();
        }

        $this->getEntityManager()->flush();
    }
}