vendor/nzo/url-encryptor-bundle/Annotations/AnnotationResolver.php line 32

  1. <?php
  3. /*
  4.  * This file is part of the NzoUrlEncryptorBundle package.
  5.  *
  6.  * (c) Ala Eddine Khefifi <alakfpro@gmail.com>
  7.  *
  8.  * For the full copyright and license information, please view the LICENSE
  9.  * file that was distributed with this source code.
  10.  */
  12. namespace Nzo\UrlEncryptorBundle\Annotations;
  14. use Doctrine\Common\Annotations\Reader;
  15. use Nzo\UrlEncryptorBundle\Encryptor\Encryptor;
  16. use Symfony\Component\HttpKernel\Event\ControllerEvent;
  18. class AnnotationResolver
  19. {
  20.     private $decryptor;
  21.     private $reader;
  23.     /**
  24.      * @param Reader|null $reader
  25.      */
  26.     public function __construct(Encryptor $decryptor, $reader)
  27.     {
  28.         $this->decryptor = $decryptor;
  29.         $this->reader = $reader;
  30.     }
  32.     public function onKernelController(ControllerEvent $event)
  33.     {
  34.         if (is_array($controller = $event->getController())) {
  35.             $objectController = new \ReflectionObject($controller[0]);
  36.             $method = $objectController->getMethod($controller[1]);
  37.         } elseif (is_object($controller) && method_exists($controller, '__invoke')) {
  38.             $objectController = new \ReflectionObject($controller);
  39.             $method = $objectController->getMethod('__invoke');
  40.         } else {
  41.             return;
  42.         }
  44.         // handle php8 attribute
  45.         if (class_exists('ReflectionAttribute')) {
  46.             if ($this->hasAnnotation($method) && !$this->reader instanceof Reader) {
  47.                 throw new \InvalidArgumentException('NzoEncryptor:  Annotation service not loaded, PHP Attributes should be used instead.');
  48.             }
  49.             $annotations = $this->getAnnotation($method);
  50.         } else {
  51.             $annotations = $this->reader->getMethodAnnotations($method);
  52.         }
  54.         foreach ($annotations as $configuration) {
  55.             // handle php8 attribute
  56.             if (class_exists('ReflectionAttribute')) {
  57.                 $configuration = $this->handleReflectionAttribute($configuration);
  58.             }
  60.             if ($configuration instanceof ParamEncryptor) {
  61.                 if (null !== $configuration->getParams()) {
  62.                     $request = $event->getRequest();
  63.                     foreach ($configuration->getParams() as $param) {
  64.                         if ($request->attributes->has($param)) {
  65.                             $decrypted = $this->decryptor->encrypt($request->attributes->get($param));
  66.                             $request->attributes->set($param, $decrypted);
  67.                         } elseif ($request->request->has($param)) {
  68.                             $decrypted = $this->decryptor->encrypt($request->request->get($param));
  69.                             $request->request->set($param, $decrypted);
  70.                         }
  71.                     }
  72.                 }
  73.             } elseif ($configuration instanceof ParamDecryptor) {
  74.                 if (null !== $configuration->getParams()) {
  75.                     $request = $event->getRequest();
  76.                     foreach ($configuration->getParams() as $param) {
  77.                         if ($request->attributes->has($param)) {
  78.                             $decrypted = $this->decryptor->decrypt($request->attributes->get($param));
  79.                             $request->attributes->set($param, $decrypted);
  80.                         } elseif ($request->request->has($param)) {
  81.                             $decrypted = $this->decryptor->decrypt($request->request->get($param));
  82.                             $request->request->set($param, $decrypted);
  83.                         }
  84.                     }
  85.                 }
  86.             }
  87.         }
  88.     }
  90.     /**
  91.      * @return mixed
  92.      */
  93.     private function handleReflectionAttribute($configuration)
  94.     {
  95.         if ($configuration instanceof \ReflectionAttribute
  96.             && \in_array($configuration->getName(), [ParamEncryptor::class, ParamDecryptor::class], true)) {
  97.             $class = $configuration->getName();
  98.             $arguments = $configuration->getArguments();
  99.             $params = \is_array($arguments) && [] !== $arguments && \is_array($arguments[0]) ? $arguments[0] : [];
  101.             return new $class($params);
  102.         }
  104.         return $configuration;
  105.     }
  107.     /**
  108.      * @return array|mixed
  109.      */
  110.     private function getAnnotation($method)
  111.     {
  112.         return $this->reader instanceof Reader && !empty($this->reader->getMethodAnnotations($method))
  113.             ? $this->reader->getMethodAnnotations($method)
  114.             : $method->getAttributes();
  115.     }
  117.     /**
  118.      * @param \ReflectionMethod $method
  119.      *
  120.      * @return bool
  121.      */
  122.     private function hasAnnotation($method)
  123.     {
  124.         $docComment = $method->getDocComment();
  126.         return false !== $docComment && (false !== strpos($docComment, '@ParamEncryptor') || false !== strpos($docComment, '@ParamDecryptor'));
  127.     }
  128. }