<?php
namespace Application\Job;
 
use Concrete\Core\Entity\File\File;
use Concrete\Core\File\FileList;
use Concrete\Core\Entity\File\StorageLocation\StorageLocation;
use Concrete\Core\Job\QueueableJob;
use Doctrine\ORM\EntityManagerInterface;
use ZendQueue\Message;
use ZendQueue\Queue;
 
class S3StoragePush extends QueueableJob
{
    /**
     * @var EntityManagerInterface
     */
    protected $entityManager;
 
    /**
     * @var \Concrete\Core\Entity\StorageLocation\StorageLocation
     */
    private $fromLocation;
 
    /**
     * @var \Concrete\Core\Entity\StorageLocation\StorageLocation
     */
    private $toLocation;
 
    /**
     * Initialize the instance.
     *
     * @param EntityManagerInterface $entityManager
     */
    public function __construct(EntityManagerInterface $entityManager)
    {
        parent::__construct();
        $this->entityManager = $entityManager;
        $this->fromLocation = $this->entityManager->find(StorageLocation::class, 1);
        $this->toLocation = $this->entityManager->find(StorageLocation::class, 2);
    }
 
    /**
     * {@inheritdoc}
     *
     * @see \Concrete\Core\Job\Job::getJobName()
     */
    public function getJobName()
    {
        return t('Migrate to S3');
    }
 
    /**
     * {@inheritdoc}
     *
     * @see \Concrete\Core\Job\Job::getJobDescription()
     */
    public function getJobDescription()
    {
        return t('Moves all local files to S3.');
    }
 
    /**
     * {@inheritdoc}
     *
     * @see \Concrete\Core\Job\QueueableJob::start()
     */
    public function start(Queue $queue)
    {
        $list = new FileList();
        $list->filterByStorageLocation($this->fromLocation);
        $q = $list->deliverQueryObject()->execute();
        while (false !== ($row = $q->fetch())) {
            $queue->send((int) $row['fID']);
        }
    }
 
    /**
     * {@inheritdoc}
     *
     * @see \Concrete\Core\Job\QueueableJob::processQueueItem()
     */
    public function processQueueItem(Message $msg)
    {
        $fID = (int) $msg->body;
        if ($fID > 0) {
            $file = $this->entityManager->find(File::class, $fID);
            if ($file !== null) {
                $file->setFileStorageLocation($this->toLocation);
            }
        }
    }
 
    /**
     * {@inheritdoc}
     *
     * @see \Concrete\Core\Job\QueueableJob::finish()
     */
    public function finish(Queue $q)
    {
        return t('All files have been processed.');
    }
 
}