Twig/extra/string-extra/StringExtension.php

<?php

/*
 * This file is part of Twig.
 *
 * (c) Fabien Potencier
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Twig\Extra\String;

use Symfony\Component\String\AbstractUnicodeString;
use Symfony\Component\String\Inflector\EnglishInflector;
use Symfony\Component\String\Inflector\FrenchInflector;
use Symfony\Component\String\Inflector\InflectorInterface;
use Symfony\Component\String\Slugger\AsciiSlugger;
use Symfony\Component\String\Slugger\SluggerInterface;
use Symfony\Component\String\UnicodeString;
use Twig\Extension\AbstractExtension;
use Twig\TwigFilter;

final class StringExtension extends AbstractExtension
{
    private $slugger;
    private $frenchInflector;
    private $englishInflector;

    public function __construct(?SluggerInterface $slugger = null)
    {
        $this->slugger = $slugger ?: new AsciiSlugger();
    }

    public function getFilters()
    {
        return [
            new TwigFilter('u', [$this, 'createUnicodeString']),
            new TwigFilter('slug', [$this, 'createSlug']),
            new TwigFilter('pluralize', [$this, 'pluralize']),
            new TwigFilter('singularize', [$this, 'singularize']),
        ];
    }

    public function createUnicodeString(?string $text): UnicodeString
    {
        return new UnicodeString($text ?? '');
    }

    public function createSlug(string $string, string $separator = '-', ?string $locale = null): AbstractUnicodeString
    {
        return $this->slugger->slug($string, $separator, $locale);
    }

    /**
     * @return array|string
     */
    public function pluralize(string $value, string $lang, bool $singleResult = true)
    {
        switch (true) {
            case $singleResult:
                return $this->getInflector($lang)->pluralize($value)[0];
            default:
                return $this->getInflector($lang)->pluralize($value);
        }
    }

    /**
     * @return array|string
     */
    public function singularize(string $value, string $lang, bool $singleResult = true)
    {
        switch (true) {
            case $singleResult:
                return $this->getInflector($lang)->singularize($value)[0];
            default:
                return $this->getInflector($lang)->singularize($value);
        }
    }

    private function getInflector(string $lang): InflectorInterface
    {
        switch ($lang) {
            case 'fr':
                return $this->frenchInflector ?? $this->frenchInflector = new FrenchInflector();
            case 'en':
                return $this->englishInflector ?? $this->englishInflector = new EnglishInflector();
            default:
                throw new \InvalidArgumentException(sprintf('Language "%s" is not supported.', $lang));
        }
    }
}