added voter for customer, project and activity #34 (#39)

src/TimesheetBundle/Controller/Admin/ActivityController.php

@@ -59,6 +59,7 @@ class ActivityController extends AbstractController
     /**
      * @Route("/{id}/edit", name="admin_activity_edit")
      * @Method({"GET", "POST"})
+     * @Security("is_granted('edit', activity)")
      */
     public function editAction(Activity $activity, Request $request)
     {

src/TimesheetBundle/Controller/Admin/CustomerController.php

@@ -62,6 +62,7 @@ class CustomerController extends AbstractController
     /**
      * @Route("/{id}/edit", name="admin_customer_edit")
      * @Method({"GET", "POST"})
+     * @Security("is_granted('edit', customer)")
      */
     public function editAction(Customer $customer, Request $request)
     {

src/TimesheetBundle/Controller/Admin/ProjectController.php

@@ -59,6 +59,7 @@ class ProjectController extends AbstractController
     /**
      * @Route("/{id}/edit", name="admin_project_edit")
      * @Method({"GET", "POST"})
+     * @Security("is_granted('edit', project)")
      */
     public function editAction(Project $project, Request $request)
     {

src/TimesheetBundle/Voter/ActivityVoter.php

@@ -0,0 +1,110 @@
+<?php
+
+/*
+ * This file is part of the Kimai package.
+ *
+ * (c) Kevin Papst <kevin@kevinpapst.de>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace TimesheetBundle\Voter;
+
+use AppBundle\Entity\User;
+use AppBundle\Voter\AbstractVoter;
+use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
+use TimesheetBundle\Entity\Activity;
+
+/**
+ * A voter to check permissions on Activities.
+ *
+ * @author Kevin Papst <kevin@kevinpapst.de>
+ */
+class ActivityVoter extends AbstractVoter
+{
+    const VIEW = 'view';
+    const EDIT = 'edit';
+    const DELETE = 'delete';
+
+    /**
+     * @param string $attribute
+     * @param mixed $subject
+     * @return bool
+     */
+    protected function supports($attribute, $subject)
+    {
+        if (!in_array($attribute, array(self::VIEW, self::EDIT, self::DELETE))) {
+            return false;
+        }
+
+        if (!$subject instanceof Activity) {
+            return false;
+        }
+
+        return true;
+    }
+
+    /**
+     * @param string $attribute
+     * @param Activity $subject
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function voteOnAttribute($attribute, $subject, TokenInterface $token)
+    {
+        $user = $token->getUser();
+
+        if (!$user instanceof User) {
+            return false;
+        }
+
+        switch ($attribute) {
+            case self::VIEW:
+                return $this->canView($subject, $user, $token);
+            case self::EDIT:
+                return $this->canEdit($subject, $user, $token);
+            case self::DELETE:
+                return $this->canDelete($token);
+        }
+
+        return false;
+    }
+
+    /**
+     * @param Activity $activity
+     * @param User $user
+     * @return bool
+     */
+    protected function canView(Activity $activity, User $user, TokenInterface $token)
+    {
+        if ($this->canEdit($activity, $user, $token)) {
+            return true;
+        }
+
+        return false;
+    }
+
+    /**
+     * @param Activity $activity
+     * @param User $user
+     * @return bool
+     */
+    protected function canEdit(Activity $activity, User $user, TokenInterface $token)
+    {
+        if ($this->canDelete($token)) {
+            return true;
+        }
+
+        return false;
+    }
+
+    /**
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function canDelete(TokenInterface $token)
+    {
+        return $this->hasRole('ROLE_ADMIN', $token);
+    }
+}

src/TimesheetBundle/Voter/CustomerVoter.php

@@ -0,0 +1,110 @@
+<?php
+
+/*
+ * This file is part of the Kimai package.
+ *
+ * (c) Kevin Papst <kevin@kevinpapst.de>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace TimesheetBundle\Voter;
+
+use AppBundle\Entity\User;
+use AppBundle\Voter\AbstractVoter;
+use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
+use TimesheetBundle\Entity\Customer;
+
+/**
+ * A voter to check permissions on Customers.
+ *
+ * @author Kevin Papst <kevin@kevinpapst.de>
+ */
+class CustomerVoter extends AbstractVoter
+{
+    const VIEW = 'view';
+    const EDIT = 'edit';
+    const DELETE = 'delete';
+
+    /**
+     * @param string $attribute
+     * @param mixed $subject
+     * @return bool
+     */
+    protected function supports($attribute, $subject)
+    {
+        if (!in_array($attribute, array(self::VIEW, self::EDIT, self::DELETE))) {
+            return false;
+        }
+
+        if (!$subject instanceof Customer) {
+            return false;
+        }
+
+        return true;
+    }
+
+    /**
+     * @param string $attribute
+     * @param Customer $subject
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function voteOnAttribute($attribute, $subject, TokenInterface $token)
+    {
+        $user = $token->getUser();
+
+        if (!$user instanceof User) {
+            return false;
+        }
+
+        switch ($attribute) {
+            case self::VIEW:
+                return $this->canView($subject, $user, $token);
+            case self::EDIT:
+                return $this->canEdit($subject, $user, $token);
+            case self::DELETE:
+                return $this->canDelete($token);
+        }
+
+        return false;
+    }
+
+    /**
+     * @param Customer $customer
+     * @param User $user
+     * @return bool
+     */
+    protected function canView(Customer $customer, User $user, TokenInterface $token)
+    {
+        if ($this->canEdit($customer, $user, $token)) {
+            return true;
+        }
+
+        return false;
+    }
+
+    /**
+     * @param Customer $customer
+     * @param User $user
+     * @return bool
+     */
+    protected function canEdit(Customer $customer, User $user, TokenInterface $token)
+    {
+        if ($this->canDelete($token)) {
+            return true;
+        }
+
+        return false;
+    }
+
+    /**
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function canDelete(TokenInterface $token)
+    {
+        return $this->hasRole('ROLE_ADMIN', $token);
+    }
+}

src/TimesheetBundle/Voter/ProjectVoter.php

@@ -0,0 +1,110 @@
+<?php
+
+/*
+ * This file is part of the Kimai package.
+ *
+ * (c) Kevin Papst <kevin@kevinpapst.de>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace TimesheetBundle\Voter;
+
+use AppBundle\Entity\User;
+use AppBundle\Voter\AbstractVoter;
+use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
+use TimesheetBundle\Entity\Project;
+
+/**
+ * A voter to check permissions on Projects.
+ *
+ * @author Kevin Papst <kevin@kevinpapst.de>
+ */
+class ProjectVoter extends AbstractVoter
+{
+    const VIEW = 'view';
+    const EDIT = 'edit';
+    const DELETE = 'delete';
+
+    /**
+     * @param string $attribute
+     * @param mixed $subject
+     * @return bool
+     */
+    protected function supports($attribute, $subject)
+    {
+        if (!in_array($attribute, array(self::VIEW, self::EDIT, self::DELETE))) {
+            return false;
+        }
+
+        if (!$subject instanceof Project) {
+            return false;
+        }
+
+        return true;
+    }
+
+    /**
+     * @param string $attribute
+     * @param Project $subject
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function voteOnAttribute($attribute, $subject, TokenInterface $token)
+    {
+        $user = $token->getUser();
+
+        if (!$user instanceof User) {
+            return false;
+        }
+
+        switch ($attribute) {
+            case self::VIEW:
+                return $this->canView($subject, $user, $token);
+            case self::EDIT:
+                return $this->canEdit($subject, $user, $token);
+            case self::DELETE:
+                return $this->canDelete($token);
+        }
+
+        return false;
+    }
+
+    /**
+     * @param Project $project
+     * @param User $user
+     * @return bool
+     */
+    protected function canView(Project $project, User $user, TokenInterface $token)
+    {
+        if ($this->canEdit($project, $user, $token)) {
+            return true;
+        }
+
+        return false;
+    }
+
+    /**
+     * @param Project $project
+     * @param User $user
+     * @return bool
+     */
+    protected function canEdit(Project $project, User $user, TokenInterface $token)
+    {
+        if ($this->canDelete($token)) {
+            return true;
+        }
+
+        return false;
+    }
+
+    /**
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function canDelete(TokenInterface $token)
+    {
+        return $this->hasRole('ROLE_ADMIN', $token);
+    }
+}