custom/plugins/SwagPlatformSecurity/src/Fixes/NEXT24667/SecurityExtension.php line 120

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Swag\Security\Fixes\NEXT24667;
  5. use Twig\Extension\AbstractExtension;
  6. use Twig\TwigFilter;
  8. /**
  9.  * @internal
  10.  */
  11. class SecurityExtension extends AbstractExtension
  12. {
  13.     /**
  14.      * @var array<string>
  15.      */
  16.     private $allowedPHPFunctions;
  18.     /**
  19.      * @param array<string> $allowedPHPFunctions
  20.      */
  21.     public function __construct(array $allowedPHPFunctions)
  22.     {
  23.         $this->allowedPHPFunctions = $allowedPHPFunctions;
  24.     }
  26.     /**
  27.      * @return TwigFilter[]
  28.      */
  29.     public function getFilters(): array
  30.     {
  31.         return [
  32.             new TwigFilter('map', [$this, 'map']),
  33.             new TwigFilter('reduce', [$this, 'reduce']),
  34.             new TwigFilter('filter', [$this, 'filter']),
  35.             new TwigFilter('sort', [$this, 'sort']),
  36.         ];
  37.     }
  39.     /**
  40.      * @param iterable<mixed> $array
  41.      * @param string|callable|\Closure $function
  42.      *
  43.      * @return array<mixed>
  44.      */
  45.     public function map(iterable $array, $function): array
  46.     {
  47.         if (\is_array($function)) {
  48.             $function = implode('::', $function);
  49.         }
  51.         if (\is_string($function) && !\in_array($function, $this->allowedPHPFunctions, true)) {
  52.             throw new \RuntimeException(sprintf('Function "%s" is not allowed', $function));
  53.         }
  55.         $result = [];
  56.         foreach ($array as $key => $value) {
  57.             // @phpstan-ignore-next-line
  58.             $result[$key] = $function($value);
  59.         }
  61.         return $result;
  62.     }
  64.     /**
  65.      * @param iterable<mixed> $array
  66.      * @param string|callable|\Closure $function
  67.      * @param mixed $initial
  68.      *
  69.      * @return mixed
  70.      */
  71.     public function reduce(iterable $array, $function, $initial = null)
  72.     {
  73.         if (\is_array($function)) {
  74.             $function = implode('::', $function);
  75.         }
  77.         if (\is_string($function) && !\in_array($function, $this->allowedPHPFunctions, true)) {
  78.             throw new \RuntimeException(sprintf('Function "%s" is not allowed', $function));
  79.         }
  81.         if (!\is_array($array)) {
  82.             $array = iterator_to_array($array);
  83.         }
  85.         // @phpstan-ignore-next-line
  86.         return array_reduce($array, $function, $initial);
  87.     }
  89.     /**
  90.      * @param iterable<mixed> $array
  91.      * @param string|callable|\Closure $arrow
  92.      *
  93.      * @return iterable<mixed>
  94.      */
  95.     public function filter(iterable $array, $arrow): iterable
  96.     {
  97.         if (\is_array($arrow)) {
  98.             $arrow = implode('::', $arrow);
  99.         }
  101.         if (\is_string($arrow) && !\in_array($arrow, $this->allowedPHPFunctions, true)) {
  102.             throw new \RuntimeException(sprintf('Function "%s" is not allowed', $arrow));
  103.         }
  105.         if (\is_array($array)) {
  106.             // @phpstan-ignore-next-line
  107.             return array_filter($array, $arrow, \ARRAY_FILTER_USE_BOTH);
  108.         }
  110.         // @phpstan-ignore-next-line
  111.         return new \CallbackFilterIterator(new \IteratorIterator($array), $arrow);
  112.     }
  114.     /**
  115.      * @param iterable<mixed> $array
  116.      * @param string|callable|\Closure|null $arrow
  117.      *
  118.      * @return array<mixed>
  119.      */
  120.     public function sort(iterable $array, $arrow = null): array
  121.     {
  122.         if (\is_array($arrow)) {
  123.             $arrow = implode('::', $arrow);
  124.         }
  126.         if (\is_string($arrow) && !\in_array($arrow, $this->allowedPHPFunctions, true)) {
  127.             throw new \RuntimeException(sprintf('Function "%s" is not allowed', $arrow));
  128.         }
  130.         if ($array instanceof \Traversable) {
  131.             $array = iterator_to_array($array);
  132.         }
  134.         if ($arrow !== null) {
  135.             // @phpstan-ignore-next-line
  136.             uasort($array, $arrow);
  137.         } else {
  138.             asort($array);
  139.         }
  141.         return $array;
  142.     }
  143. }