Chris@0: <?php
Chris@0: 
Chris@0: namespace DrupalCodeGenerator\Twig;
Chris@0: 
Chris@0: use DrupalCodeGenerator\Utils;
Chris@0: use Twig_Environment;
Chris@0: use Twig_LoaderInterface;
Chris@0: use Twig_SimpleFilter;
Chris@0: 
Chris@0: /**
Chris@0:  * Stores the Twig configuration.
Chris@0:  */
Chris@0: class TwigEnvironment extends Twig_Environment {
Chris@0: 
Chris@0:   /**
Chris@0:    * Constructs Twig environment object.
Chris@0:    *
Chris@0:    * @param \Twig_LoaderInterface $loader
Chris@0:    *   The Twig loader.
Chris@0:    */
Chris@0:   public function __construct(Twig_LoaderInterface $loader) {
Chris@0:     parent::__construct($loader);
Chris@0: 
Chris@0:     $this->addFilter(new Twig_SimpleFilter('plural', function ($string) {
Chris@0:       switch (substr($string, -1)) {
Chris@0:         case 'y':
Chris@0:           return substr($string, 0, -1) . 'ies';
Chris@0: 
Chris@0:         case 's':
Chris@0:           return $string . 'es';
Chris@0: 
Chris@0:         default:
Chris@0:           return $string . 's';
Chris@0:       }
Chris@0:     }));
Chris@0: 
Chris@0:     $this->addFilter(new Twig_SimpleFilter('article', function ($string) {
Chris@0:       $article = in_array(strtolower($string[0]), ['a', 'e', 'i', 'o', 'u']) ? 'an' : 'a';
Chris@0:       return $article . ' ' . $string;
Chris@0:     }));
Chris@0: 
Chris@0:     $this->addFilter(new Twig_SimpleFilter('underscore2hyphen', function ($string) {
Chris@0:       // @codeCoverageIgnoreStart
Chris@0:       return str_replace('_', '-', $string);
Chris@0:       // @codeCoverageIgnoreEnd
Chris@0:     }, ['deprecated' => TRUE]));
Chris@0: 
Chris@0:     $this->addFilter(new Twig_SimpleFilter('hyphen2underscore', function ($string) {
Chris@0:       // @codeCoverageIgnoreStart
Chris@0:       return str_replace('-', '_', $string);
Chris@0:       // @codeCoverageIgnoreEnd
Chris@0:     }, ['deprecated' => TRUE]));
Chris@0: 
Chris@0:     $this->addFilter(new Twig_SimpleFilter('u2h', function ($string) {
Chris@0:       return str_replace('_', '-', $string);
Chris@0:     }));
Chris@0: 
Chris@0:     $this->addFilter(new Twig_SimpleFilter('h2u', function ($string) {
Chris@0:       return str_replace('-', '_', $string);
Chris@0:     }));
Chris@0: 
Chris@0:     $this->addFilter(new Twig_SimpleFilter('camelize', function ($string, $upper_mode = TRUE) {
Chris@0:       return Utils::camelize($string, $upper_mode);
Chris@0:     }));
Chris@0: 
Chris@0:     $this->addTokenParser(new TwigSortTokenParser());
Chris@0:   }
Chris@0: 
Chris@0: }