src/Entity/Referral.php line 11

Open in your IDE?
  1. <?php
  3. namespace App\Entity;
  5. use App\Repository\ReferralRepository;
  6. use Doctrine\Common\Collections\ArrayCollection;
  7. use Doctrine\Common\Collections\Collection;
  8. use Doctrine\ORM\Mapping as ORM;
  10. #[ORM\Entity(repositoryClass: ReferralRepository::class)]
  11. class Referral
  12. {
  13.     #[ORM\Id]
  14.     #[ORM\GeneratedValue]
  15.     #[ORM\Column]
  16.     private ?int $id = null;
  18.     #[ORM\Column(length: 255, nullable: true)]
  19.     private ?string $code = null;
  21.     #[ORM\OneToOne(inversedBy: 'referral', cascade: ['persist', 'remove'])]
  22.     private ?User $user = null;
  24.     #[ORM\OneToMany(targetEntity: User::class, mappedBy: 'referredby')]
  25.     private Collection $users;
  27.     public function __construct()
  28.     {
  29.         $this->users = new ArrayCollection();
  30.     }
  32.     public function getId(): ?int
  33.     {
  34.         return $this->id;
  35.     }
  37.     public function getCode(): ?string
  38.     {
  39.         return $this->code;
  40.     }
  42.     public function setCode(?string $code): static
  43.     {
  44.         $this->code = $code;
  46.         return $this;
  47.     }
  49.     public function getUser(): ?User
  50.     {
  51.         return $this->user;
  52.     }
  54.     public function setUser(?User $user): static
  55.     {
  56.         $this->user = $user;
  58.         return $this;
  59.     }
  61.     /**
  62.      * @return Collection<int, User>
  63.      */
  64.     public function getUsers(): Collection
  65.     {
  66.         return $this->users;
  67.     }
  69.     public function addUser(User $user): static
  70.     {
  71.         if (!$this->users->contains($user)) {
  72.             $this->users->add($user);
  73.             $user->setReferredby($this);
  74.         }
  76.         return $this;
  77.     }
  79.     public function removeUser(User $user): static
  80.     {
  81.         if ($this->users->removeElement($user)) {
  82.             // set the owning side to null (unless already changed)
  83.             if ($user->getReferredby() === $this) {
  84.                 $user->setReferredby(null);
  85.             }
  86.         }
  88.         return $this;
  89.     }
  91.     function __toString()
  92.     {
  93.         return $this->getCode();
  94.     }
  95. }