added timesheet voter #44 (#46)

src/TimesheetBundle/Voter/TimesheetVoter.php

@@ -0,0 +1,173 @@
+<?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;
+use TimesheetBundle\Entity\Timesheet;
+
+/**
+ * A voter to check permissions on Timesheets.
+ *
+ * @author Kevin Papst <kevin@kevinpapst.de>
+ */
+class TimesheetVoter extends AbstractVoter
+{
+    const START = 'start';
+    const STOP = 'stop';
+    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::START, self::STOP, self::VIEW, self::EDIT, self::DELETE))) {
+            return false;
+        }
+
+        if ($subject instanceof Activity && $attribute == self::START) {
+            return true;
+        }
+
+        if (!$subject instanceof Timesheet) {
+            return false;
+        }
+
+        return true;
+    }
+
+    /**
+     * @param string $attribute
+     * @param Timesheet]Activity $subject
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function voteOnAttribute($attribute, $subject, TokenInterface $token)
+    {
+        $user = $token->getUser();
+
+        if (!$user instanceof User) {
+            return false;
+        }
+
+        // Customer cannot do anything with timesheet entries
+        if (!$this->hasRole('ROLE_USER', $token)) {
+            return false;
+        }
+
+        switch ($attribute) {
+            case self::STOP:
+                return $this->canStop($subject, $user, $token);
+            case self::START:
+                return $this->canStart($subject, $user, $token);
+            case self::VIEW:
+                return $this->canView($subject, $user, $token);
+            case self::EDIT:
+                return $this->canEdit($subject, $user, $token);
+            case self::DELETE:
+                return $this->canDelete($subject, $user, $token);
+        }
+
+        return false;
+    }
+
+    /**
+     * @param Timesheet $timesheet
+     * @param User $user
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function canStop(Timesheet $timesheet, User $user, TokenInterface $token)
+    {
+        // if a teamlead stops an entry for another user, check that this user is part of his team
+        return $this->isOwnOrTeamlead($timesheet, $user, $token);
+    }
+
+    /**
+     * @param Activity $activity
+     * @param User $user
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function canStart(Activity $activity, User $user, TokenInterface $token)
+    {
+        // we could check the amount of active entries
+        // TODO limit to activities that are not hidden
+
+        // if a teamlead starts an entry for another user, check that this user is part of his team
+        return true;
+    }
+
+    /**
+     * @param Timesheet $timesheet
+     * @param User $user
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function canView(Timesheet $timesheet, User $user, TokenInterface $token)
+    {
+        return $this->isOwnOrTeamlead($timesheet, $user, $token);
+    }
+
+    /**
+     * @param Timesheet $timesheet
+     * @param User $user
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function canEdit(Timesheet $timesheet, User $user, TokenInterface $token)
+    {
+        return $this->isOwnOrTeamlead($timesheet, $user, $token);
+    }
+
+    /**
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function canDelete(Timesheet $timesheet, User $user, TokenInterface $token)
+    {
+        return $this->isOwnOrAdmin($timesheet, $user, $token);
+    }
+
+    /**
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function isOwnOrTeamlead(Timesheet $timesheet, User $user, TokenInterface $token)
+    {
+        if ($timesheet->getUser()->getId() == $user->getId()) {
+            return true;
+        }
+
+        return $this->hasRole('ROLE_TEAMLEAD', $token);
+    }
+
+    /**
+     * @param TokenInterface $token
+     * @return bool
+     */
+    protected function isOwnOrAdmin(Timesheet $timesheet, User $user, TokenInterface $token)
+    {
+        if ($timesheet->getUser()->getId() == $user->getId()) {
+            return true;
+        }
+
+        return $this->hasRole('ROLE_ADMIN', $token);
+    }
+}