App\Security\EntityListeners\ElementPermissionListener

Complexity

Total Complexity

17

Size/Duplication

Total Lines	99
Duplicated Lines	0 %

Importance

Changes	3
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	preUpdateHandler()	0	21	6
A	preFlushHandler()	0	18	5
A	postLoadHandler()	0	21	5
A	__construct()	0	5	1

<?php
/**
 * part-db version 0.1
 * Copyright (C) 2005 Christoph Lechner
 * http://www.cl-projects.de/.
 *
 * part-db version 0.2+
 * Copyright (C) 2009 K. Jacobs and others (see authors.php)
 * http://code.google.com/p/part-db/
 *
 * Part-DB Version 0.4+
 * Copyright (C) 2016 - 2019 Jan Böhmer
 * https://github.com/jbtronics
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
 */

namespace App\Security\EntityListeners;

use App\Entity\Base\DBElement;
use App\Security\Annotations\ColumnSecurity;
use Doctrine\Common\Annotations\Reader;
use Doctrine\Common\Persistence\Event\LifecycleEventArgs;
use Doctrine\ORM\EntityManager;
use Doctrine\ORM\EntityManagerInterface;
use Doctrine\ORM\Event\PreFlushEventArgs;
use Doctrine\ORM\Event\PreUpdateEventArgs;
use Doctrine\ORM\Mapping as ORM;
use Doctrine\ORM\Mapping\PostLoad;
use Doctrine\ORM\Mapping\PreUpdate;
use ReflectionClass;
use Symfony\Component\Security\Core\Security;

/**
 * The purpose of this class is to hook into the doctrine entity lifecycle and restrict access to entity informations
 * configured by ColoumnSecurity Annotation.
 *
 * If a user does not have access to an coloumn, it will be filled, with a placeholder, after doctrine loading is finished.
 * The edit process is also catched, so that these placeholders, does not get saved to database.
 */
class ElementPermissionListener
{
    protected $security;
    protected $reader;
    protected $em;

    public function __construct(Security $security, Reader $reader, EntityManagerInterface $em)
    {
        $this->security = $security;
        $this->reader = $reader;
        $this->em = $em;
    }

    /**
     * @PostLoad
     *
     * This function is called after doctrine filled, the entity properties with db values.
     * We use this, to check if the user is allowed to access these properties, and if not, we write a placeholder
     * into the element properties, so that a user only gets non sensitve data.
     */
    public function postLoadHandler(DBElement $element, LifecycleEventArgs $event)

The parameter $event is not used and could be removed.

    {
        //Read Annotations and properties.
        $reflectionClass = new ReflectionClass($element);
        $properties = $reflectionClass->getProperties();

        foreach ($properties as $property) {
            /**
             * @var ColumnSecurity
             */
            $annotation = $this->reader->getPropertyAnnotation($property,
                ColumnSecurity::class);

            //Check if user is allowed to read info, otherwise apply placeholder
            if ((null !== $annotation) && !$this->security->isGranted($annotation->getReadOperationName(), $element)) {
                $property->setAccessible(true);
                $value = $annotation->getPlaceholder();
                if($value instanceof DBElement) {
                    $this->em->detach($value);
                }
                $property->setValue($element, $value);
            }
        }
    }

    /**
     * @ORM\PreFlush()
     * This function is called before flushing. We use it, to remove all placeholder DBElements (with name=???),
     * so we dont get a no cascade persistance error.
     */
    public function preFlushHandler(DBElement $element, PreFlushEventArgs $eventArgs)

The parameter $eventArgs is not used and could be removed.

    {
        //$eventArgs->getEntityManager()->getUnitOfWork()->

        $reflectionClass = new ReflectionClass($element);
        $properties = $reflectionClass->getProperties();

        foreach ($properties as $property) {
            $annotation = $this->reader->getPropertyAnnotation(
                $property,
                ColumnSecurity::class
            );
            if (null !== $annotation) {
                //Check if the current property is an DBElement
                $property->setAccessible(true);
                $value = $property->getValue($element);
                if ($value instanceof DBElement && !$this->security->isGranted($annotation->getEditOperationName(), $element)) {
                    $property->setValue($element, null);
                }
            }
        }
    }

    /**
     * @PreUpdate
     * This function is called before Doctrine saves the property values to the database.
     * We use this function to revert the changes made in postLoadHandler(), so nothing gets changed persistently.
     */
    public function preUpdateHandler(DBElement $element, PreUpdateEventArgs $event)
    {
        $reflectionClass = new ReflectionClass($element);
        $properties = $reflectionClass->getProperties();

        foreach ($properties as $property) {
            /**
             * @var ColumnSecurity $annotation
             */
            $annotation = $this->reader->getPropertyAnnotation($property,
                ColumnSecurity::class);

            if (null !== $annotation) {
                $field_name = $property->getName();

                //Check if user is allowed to edit info, otherwise overwrite the new value
                // so that nothing is changed in the DB.
                if ($event->hasChangedField($field_name) &&
                    (!$this->security->isGranted($annotation->getEditOperationName(), $element)
                    || !$this->security->isGranted($annotation->getReadOperationName(), $element))) {
                    $event->setNewValue($field_name, $event->getOldValue($field_name));
                }
            }
        }
    }
}