vendor/roothirsch/pim-bundle/Entity/AttributeGroup.php line 16

Open in your IDE?
  1. <?php
  3. namespace Roothirsch\PimBundle\Entity;
  5. use ApiPlatform\Core\Annotation\ApiResource;
  6. use Roothirsch\PimBundle\Repository\AttributeGroupRepository;
  7. use Doctrine\Common\Collections\ArrayCollection;
  8. use Doctrine\Common\Collections\Collection;
  9. use Doctrine\ORM\Mapping as ORM;
  11. /**
  12. * @ApiResource( shortName="Pim/AttributeGroup")
  13. * @ORM\Entity(repositoryClass=AttributeGroupRepository::class)
  14. * @ORM\Table(name="pim_attribute_group")
  15. */
  16. class AttributeGroup
  17. {
  18. /**
  19. * @ORM\Id
  20. * @ORM\GeneratedValue
  21. * @ORM\Column(type="integer")
  22. */
  23. private $id;
  25. /**
  26. * @ORM\Column(type="string", length=255)
  27. */
  28. private $title;
  30. /**
  31. * @ORM\OneToMany(targetEntity=Attribute::class, mappedBy="attributeGroup", orphanRemoval=true)
  32. */
  33. private $attributes;
  35. /**
  36. * @ORM\ManyToMany(targetEntity=Category::class, mappedBy="attributeGroups")
  37. */
  38. private $categories;
  40. public function __construct($title = null)
  41. {
  42. $this->attributes = new ArrayCollection();
  43. $this->categories = new ArrayCollection();
  44. $this->title = $title;
  45. }
  47. public function __toString()
  48. {
  49. return $this->title;
  50. }
  52. public function getId(): ?int
  53. {
  54. return $this->id;
  55. }
  57. public function getTitle(): ?string
  58. {
  59. return $this->title;
  60. }
  62. public function setTitle(string $title): self
  63. {
  64. $this->title = $title;
  66. return $this;
  67. }
  69. /**
  70. * @return Collection|Attribute[]
  71. */
  72. public function getAttributes(): Collection
  73. {
  74. return $this->attributes;
  75. }
  77. public function addAttribute(Attribute $attribute): self
  78. {
  79. if (!$this->attributes->contains($attribute)) {
  80. $this->attributes[] = $attribute;
  81. $attribute->setAttributeGroup($this);
  82. }
  84. return $this;
  85. }
  87. public function removeAttribute(Attribute $attribute): self
  88. {
  89. if ($this->attributes->removeElement($attribute)) {
  90. // set the owning side to null (unless already changed)
  91. if ($attribute->getAttributeGroup() === $this) {
  92. $attribute->setAttributeGroup(null);
  93. }
  94. }
  96. return $this;
  97. }
  99. /**
  100. * @return Collection|Category[]
  101. */
  102. public function getCategories(): Collection
  103. {
  104. return $this->categories;
  105. }
  107. public function addCategory(Category $category): self
  108. {
  109. if (!$this->categories->contains($category)) {
  110. $this->categories[] = $category;
  111. $category->addAttributeGroup($this);
  112. }
  114. return $this;
  115. }
  117. public function removeCategory(Category $category): self
  118. {
  119. if ($this->categories->removeElement($category)) {
  120. $category->removeAttributeGroup($this);
  121. }
  123. return $this;
  124. }
  126. public function getAttributeByName($name) {
  127. foreach($this->attributes as $attribute) {
  128. if ($attribute->getName() == $name){
  129. return $attribute;
  130. }
  131. }
  132. return null;
  133. }
  134. }