src/Security/Voter/PredictAi/PredictionAccessVoter.php line 16

Open in your IDE?
  1. <?php
  3. namespace App\Security\Voter\PredictAi;
  5. use App\Entity\User;
  6. use App\Service\PredictAI\Member\PredictionMemberProvider;
  7. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  8. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  10. /**
  11.  * Voter permettant de vérifier l'accès à la fonctionnalité de Predict AI par le membre connecté.
  12.  */
  13. /**
  14.  * @extends Voter<string, mixed>
  15.  */
  16. class PredictionAccessVoter extends Voter
  17. {
  18.     public const PREDICTION_ACCESS = 'prediction_access';
  19.     public const PREDICTION_SUFFICIENT_CREDIT_ACCESS = 'prediction_sufficient_credit_access';
  21.     private PredictionMemberProvider $predictionMemberProvider;
  23.     public function __construct(
  24.         PredictionMemberProvider $predictionMemberProvider
  25.     ) {
  26.         $this->predictionMemberProvider = $predictionMemberProvider;
  27.     }
  29.     protected function supports(string $attribute, $subject): bool
  30.     {
  31.         return in_array($attribute, [
  32.             self::PREDICTION_ACCESS,
  33.             self::PREDICTION_SUFFICIENT_CREDIT_ACCESS,
  34.         ], true);
  35.     }
  37.     protected function voteOnAttribute(string $attribute, $subject, TokenInterface $token): bool
  38.     {
  39.         /** @var User|null $user */
  40.         $user = $token->getUser();
  41.         if (null === $user) {
  42.             return false;
  43.         }
  45.         $member = $user->getMember();
  46.         if (null === $member) {
  47.             return false;
  48.         }
  50.         if (self::PREDICTION_ACCESS === $attribute) {
  51.             return $this->predictionMemberProvider->getMemberTotalPredictions($member) > 0;
  52.         }
  54.         if (self::PREDICTION_SUFFICIENT_CREDIT_ACCESS === $attribute) {
  55.             return $this->predictionMemberProvider->getMemberTotalPredictions($member) > 0;
  56.         }
  58.         return false;
  59.     }
  60. }