Add Pictures to Restaurant

Author: tomsihap

src/Entity/Restaurant.php

@@ -2,6 +2,8 @@
 
 namespace App\Entity;
 
+use Doctrine\Common\Collections\ArrayCollection;
+use Doctrine\Common\Collections\Collection;
 use Doctrine\ORM\Mapping as ORM;
 
 /**
@@ -32,6 +34,16 @@ class Restaurant
      */
     private $city;
 
+    /**
+     * @ORM\OneToMany(targetEntity="App\Entity\RestaurantPicture", mappedBy="restaurant", orphanRemoval=true)
+     */
+    private $restaurantPictures;
+
+    public function __construct()
+    {
+        $this->restaurantPictures = new ArrayCollection();
+    }
+
     public function getId(): ?int
     {
         return $this->id;
@@ -72,4 +84,35 @@ public function setCity(?City $city): self
 
         return $this;
     }
+
+    /**
+     * @return Collection|RestaurantPicture[]
+     */
+    public function getRestaurantPictures(): Collection
+    {
+        return $this->restaurantPictures;
+    }
+
+    public function addRestaurantPicture(RestaurantPicture $restaurantPicture): self
+    {
+        if (!$this->restaurantPictures->contains($restaurantPicture)) {
+            $this->restaurantPictures[] = $restaurantPicture;
+            $restaurantPicture->setRestaurant($this);
+        }
+
+        return $this;
+    }
+
+    public function removeRestaurantPicture(RestaurantPicture $restaurantPicture): self
+    {
+        if ($this->restaurantPictures->contains($restaurantPicture)) {
+            $this->restaurantPictures->removeElement($restaurantPicture);
+            // set the owning side to null (unless already changed)
+            if ($restaurantPicture->getRestaurant() === $this) {
+                $restaurantPicture->setRestaurant(null);
+            }
+        }
+
+        return $this;
+    }
 }

src/Entity/RestaurantPicture.php

@@ -16,8 +16,26 @@ class RestaurantPicture
      */
     private $id;
 
+    /**
+     * @ORM\ManyToOne(targetEntity="App\Entity\Restaurant", inversedBy="restaurantPictures")
+     * @ORM\JoinColumn(nullable=false)
+     */
+    private $restaurant;
+
     public function getId(): ?int
     {
         return $this->id;
     }
+
+    public function getRestaurant(): ?Restaurant
+    {
+        return $this->restaurant;
+    }
+
+    public function setRestaurant(?Restaurant $restaurant): self
+    {
+        $this->restaurant = $restaurant;
+
+        return $this;
+    }
 }