FileUploadExtension.php

<?php

/*
 * Copyright (C) 2015 IRSTEA
 * All rights reserved.
 */

namespace Irstea\FileUploadBundle\Twig;

use Irstea\FileUploadBundle\Model\UploadedFileInterface;
use Irstea\FileUploadBundle\Utils\MimeTypeIcon;
use \NumberFormatter;
use Symfony\Component\Translation\TranslatorInterface;
use Twig_Environment;
use Twig_Extension;
use Twig_SimpleFilter;
use Twig_SimpleFunction;

/**
 * Description of FileUploadExtension
 *
 * @author Guillaume Perréal <guillaume.perreal@irstea.fr>
 */
class FileUploadExtension extends Twig_Extension
{
    /**
     *
     * @var TranslatorInterface
     */
    private $translator;

    /**
     *
     * @param TranslatorInterface $translator
     */
    public function __construct(TranslatorInterface $translator)
    {
        $this->translator = $translator;
    }

    public function getFunctions()
    {
        return [
            new Twig_SimpleFunction(
                'irstea_uploaded_file',
                [$this, 'formatUploadedFile'],
                [
                    'needs_environment' => true,
                    'is_safe' => ['html']
                ]
            ),
            new Twig_SimpleFunction(
                'irstea_file_icon',
                [$this, 'formatFileIcon'],
                ['is_safe' => ['html']]
            ),
        ];
    }

    public function getFilters()
    {
        return [
            new Twig_SimpleFilter(
                'irstea_file_size',
                [$this, 'formatFileSize'],
                ['needs_context' => true]
            ),
        ];
    }

    public function getName()
    {
        return "irstea_file_upload";
    }

    public function formatUploadedFile(Twig_Environment $env, $file = null)
    {
        if(null === $file) {
            return '';
        } elseif($file instanceof \Irstea\FileUploadBundle\Entity\UploadedFile) {
            $file = $file->toArray();
        } elseif(!is_array($file)) {
            throw new \InvalidArgumentException("irstea_uploaded_file expects an UploadedFile instance, array or null");
        }

        return $env->render('IrsteaFileUploadBundle:Extension:uploaded_file.html.twig', ['file' => $file]);
    }

    public function formatFileIcon($mimeType)
    {
        $icon = MimeTypeIcon::getMimeTypeIcon($mimeType);
        return sprintf('<i class="icon fa fa-file%s-o"></i>', ($icon && $icon !== 'file') ? ('-'.$icon) : '');
    }

    public function formatFileSize($context, $size, $precision = null, $locale = null)
    {
        if(!is_numeric($size)) {
            return '';
        }

        if($size > 1000000000) {
            $size /= 1000000000.0;
            $defaultPrecision = 2;
            $unit = 'Gi';
        } elseif($size > 1000000) {
            $size /= 1000000.0;
            $defaultPrecision = 1;
            $unit = 'Mi';
        } elseif($size > 1000) {
            $size /= 1000.0;
            $defaultPrecision = 1;
            $unit = 'Ki';
        } else {
            $unit = '';
            $defaultPrecision = 0;
    }

        if(null === $locale) {
            $locale = $context['app']->getRequest()->getLocale();
        }

        $formatter = NumberFormatter::create($locale, NumberFormatter::DECIMAL);
        $formatter->setAttribute(NumberFormatter::MIN_FRACTION_DIGITS, null !== $precision ? $precision : 0);
        $formatter->setAttribute(NumberFormatter::MAX_FRACTION_DIGITS, null !== $precision ? $precision : $defaultPrecision);

        return $this->translator->trans(
            'file_size(%size%,%unit%)',
            [
                '%size%' => $formatter->format($size),
                '%unit%' => $unit,
            ],
            'file_upload',
            $locale
        );
    }
}