FileUploadExtension.php 3.64 KiB
<?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]
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137
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 ); } }