open-orchestra/open-orchestra-media-bundle

<?php

namespace OpenOrchestra\Media\Twig;

use OpenOrchestra\Media\DisplayMedia\DisplayMediaManager;
use OpenOrchestra\Media\Model\MediaInterface;
use OpenOrchestra\Media\Repository\MediaRepositoryInterface;
use Symfony\Component\HttpFoundation\RequestStack;

/**
 * Class DisplayMediaExtension
 */
class DisplayMediaExtension extends \Twig_Extension
{
    protected $displayMediaManager;
    protected $mediaRepository;
    protected $requestStack;

    /**
     * @param DisplayMediaManager       $displayMediaManager
     * @param MediaRepositoryInterface  $mediaRepository
     * @param RequestStack              $requestStack
     */
    public function __construct(
        DisplayMediaManager $displayMediaManager,
        MediaRepositoryInterface $mediaRepository,
        RequestStack $requestStack
    ) {
        $this->displayMediaManager = $displayMediaManager;
        $this->mediaRepository = $mediaRepository;
        $this->requestStack = $requestStack;
    }

    /**
     * @return array
     */
    public function getFunctions()
    {
        return array(
            // Render a media or an alternative, using the display strategies
            new \Twig_SimpleFunction(
                'render_media',
                array($this, 'renderMedia'),
                array('is_safe' => array('html'))
            ),

            // Get the url of a media or an alternative
            new \Twig_SimpleFunction('get_media_url', array($this, 'getMediaUrl')),

            // Get the title of a media
            new \Twig_SimpleFunction('get_media_title', array($this, 'getMediaTitle')),
        );
    }

    /**
     * @param string $mediaId
     * @param array  $options
     *
     * @return string
     */
    public function renderMedia($mediaId, array $options = array())
    {
        $media = $this->getMedia($mediaId);

        if ($media) {
            return $this->displayMediaManager->renderMedia($media, $options);
        }

        return '';
    }

    /**
     * @param string $mediaId
     * @param string $format
     *
     * @return string
     */
    public function getMediaUrl($mediaId, $format)
    {
        $media = $this->getMedia($mediaId);

        if ($media) {
            return $this->displayMediaManager->getMediaFormatUrl($media, $format);
        }

        return '';
    }

    /**
     * @param string $mediaId
     *
     * @return string
     */
    public function getMediaTitle($mediaId)
    {
        $media = $this->getMedia($mediaId);

        if ($media) {
            return $media->getTitle($this->getRequestLanguage());
        }

        return '';
    }

    /**
     * Returns the name of the extension.
     *
     * @return string The extension name
     */
    public function getName()
    {
        return 'media';
    }

    /**
     * @param $mediaId
     * @return MediaInterface
     */
    protected function getMedia($mediaId)
    {
        $media = $this->mediaRepository->find($mediaId);

        return $media;
    }

    /**
     * @return string
     */
    protected function getRequestLanguage()
    {
        $defaultLocale = $this->requestStack->getMasterRequest()->getLocale();

        return $this->requestStack->getMasterRequest()->get('language', $defaultLocale);
    }
}