Covivo/mobicoop

<?php

/**
 * Copyright (c) 2018, MOBICOOP. All rights reserved.
 * This project is dual licensed under AGPL and proprietary licence.
 ***************************
 *    This program is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU Affero General Public License as
 *    published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
 *
 *    You should have received a copy of the GNU Affero General Public License
 *    along with this program.  If not, see <gnu.org/licenses>.
 ***************************
 *    Licence MOBICOOP described in the file
 *    LICENSE
 **************************/

namespace Mobicoop\Bundle\MobicoopBundle\Carpool\Security;

use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
use Symfony\Component\Security\Core\Authorization\Voter\Voter;
use Mobicoop\Bundle\MobicoopBundle\Carpool\Entity\Ad;
use Mobicoop\Bundle\MobicoopBundle\Carpool\Entity\Proposal;
use Mobicoop\Bundle\MobicoopBundle\User\Entity\User;
use Mobicoop\Bundle\MobicoopBundle\Permission\Service\PermissionManager;
use Symfony\Component\Security\Core\Security;
use Symfony\Component\Security\Core\User\UserInterface;

class AdVoter extends Voter
{
    const CREATE_AD = 'create_ad';
    const CREATE_FIRST_AD = 'create_first_ad';
    const DELETE_AD = 'delete_ad';
    const UPDATE_AD = 'update_ad';
    const RESULTS = 'results_ad';

    private $permissionManager;
    private $security;

    /**
     * AdVoter constructor.
     * @param PermissionManager $permissionManager
     * @param Security $security
     */
    public function __construct(PermissionManager $permissionManager, Security $security)
    {
        $this->permissionManager = $permissionManager;
        $this->security = $security;
    }

    protected function supports($attribute, $subject)
    {
        // if the attribute isn't one we support, return false
        if (!in_array($attribute, [
            self::CREATE_AD,
            self::CREATE_FIRST_AD,
            self::DELETE_AD,
            self::UPDATE_AD,
            self::RESULTS
            ])) {
            return false;
        }

        // only vote on Ad objects inside this voter
        if (!$subject instanceof Ad) {
            return false;
        }
        return true;
    }

    protected function voteOnAttribute($attribute, $subject, TokenInterface $token)
    {
        switch ($attribute) {
            case self::CREATE_AD:
                return $this->canCreateAd();
            case self::CREATE_FIRST_AD:
                return $this->canCreateFirstAd();
            case self::DELETE_AD:
                return $this->canDeleteAd($subject);
            case self::UPDATE_AD:
                return $this->canUpdateAd($subject);
            case self::RESULTS:
                return $this->canViewAdResults($subject);
        }

        throw new \LogicException('This code should not be reached!');
    }

    private function canCreateAd()
    {
        // everbody can create a ad
        return true;
    }

    private function canCreateFirstAd()
    {
        $user = $this->security->getUser();

        // only registered users can create a first logged ad
        if (!$user instanceof User) {
            return false;
        }

        return true;
    }

    private function canDeleteAd(Ad $ad)
    {
        $user = $this->security->getUser();
        // only registered users can delete ad
        if (!$user instanceof User) {
            return false;
        }
        return $this->permissionManager->checkPermission('ad_delete', $user, $ad->getId());
    }

    private function canUpdateAd(Ad $ad)
    {
        $user = $this->security->getUser();

        // only registered users can update ad
        if (!$user instanceof User) {
            return false;
        }

        // only the author of the proposal can delete the proposal
        if ($ad->getUserId() !== $user->getId()) {
            return false;
        }

        return $this->permissionManager->checkPermission('ad_update_self', $user, $ad->getId());
    }

    private function canViewAdResults(Ad $ad)
    {
        $user = $this->security->getUser();
        // only registered users can view ad results
        if (!$user instanceof User) {
            return false;
        }
        
        return $this->permissionManager->checkPermission('ad_results', $user, $ad->getId());
    }
}