src/Twig/UserExtension.php line 33

Open in your IDE?
  1. <?php
  3. namespace App\Twig;
  5. use App\Entity\Account\Advertiser;
  6. use App\Entity\Account\Avatar;
  7. use App\Entity\Account\Customer;
  8. use App\Entity\Profile\Profile;
  9. use App\Entity\User;
  10. use Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorageInterface;
  11. use Twig\Extension\AbstractExtension;
  12. use Twig\TwigFunction;
  14. class UserExtension extends AbstractExtension
  15. {
  16.     public function __construct(
  17.         private TokenStorageInterface $tokenStorage
  18.     ) {}
  20.     public function getFunctions()
  21.     {
  22.         return [
  23.             new TwigFunction('user_avatar', [$this, 'getUserAvatar']),
  24.             new TwigFunction('is_profile_in_favourites', [$this, 'isProfileInFavourites']),
  25.             new TwigFunction('is_user_advertiser', [$this, 'isUserAdvertiser']),
  26.             new TwigFunction('is_user_customer', [$this, 'isUserCustomer']),
  27.         ];
  28.     }
  30.     public function getUserAvatar(): ?Avatar
  31.     {
  32.         /** @var User $user */
  33.         $user = $this->tokenStorage->getToken()->getUser();
  34.         if(!($user instanceof User)) {
  35.             return null;
  36.         }
  38.         return $user->getAvatar() ? $user->getAvatar() : null;
  39.     }
  41.     public function isProfileInFavourites(Profile $profile): bool
  42.     {
  43.         /** @var User $user */
  44.         $user = $this->tokenStorage->getToken()->getUser();
  45.         if(!($user instanceof Customer)) {
  46.             return false;
  47.         }
  49.         return $user->findProfileInFavourites($profile) != null;
  50.     }
  52.     public function isUserAdvertiser($user): ?string
  53.     {
  54.         return $user instanceof Advertiser;
  55.     }
  57.     public function isUserCustomer($user): ?string
  58.     {
  59.         return $user instanceof Customer;
  60.     }
  61. }