DocumentTrait

Author: manuxi

src/Entity/Traits/DocumentTrait.php

@@ -0,0 +1,41 @@
+<?php
+
+namespace Manuxi\SuluSharedToolsBundle\Entity\Traits;
+
+use Doctrine\ORM\Mapping as ORM;
+use JMS\Serializer\Annotation as Serializer;
+use Sulu\Bundle\MediaBundle\Entity\MediaInterface;
+
+trait DocumentTrait
+{
+
+    #[ORM\ManyToOne(targetEntity: MediaInterface::class)]
+    #[ORM\JoinColumn(onDelete: 'SET NULL')]
+    #[Serializer\Exclude]
+    private ?MediaInterface $document = null;
+
+    public function getDocument(): ?MediaInterface
+    {
+        return $this->document;
+    }
+
+    #[Serializer\VirtualProperty]
+    #[Serializer\SerializedName("document")]
+    public function getDocumentData(): ?array
+    {
+        if ($document = $this->getDocument()) {
+            return [
+                'id' => $document->getId(),
+            ];
+        }
+
+        return null;
+
+    }
+
+    public function setDocument(?MediaInterface $document): self
+    {
+        $this->document = $document;
+        return $this;
+    }
+}

src/Entity/Traits/DocumentTranslatableTrait.php

@@ -0,0 +1,48 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Manuxi\SuluSharedToolsBundle\Entity\Traits;
+
+use JMS\Serializer\Annotation as Serializer;
+use Sulu\Bundle\MediaBundle\Entity\MediaInterface;
+
+trait DocumentTranslatableTrait
+{
+    abstract public function getLocale();
+    abstract protected function getTranslation(string $locale);
+
+    #[Serializer\Exclude]
+    public function getDocument(): ?MediaInterface
+    {
+        $translation = $this->getTranslation($this->getLocale());
+        if (!$translation) {
+            return null;
+        }
+
+        return $translation->getDocument();
+    }
+
+    #[Serializer\VirtualProperty]
+    #[Serializer\SerializedName("document")]
+    public function getDocumentData(): ?array
+    {
+        $translation = $this->getTranslation($this->getLocale());
+        if (!$translation) {
+            return null;
+        }
+
+        return $translation->getDocumentData();
+    }
+
+    public function setDocument(?MediaInterface $document): self
+    {
+        $translation = $this->getTranslation($this->getLocale());
+        if (!$translation) {
+            $translation = $this->createTranslation($this->getLocale());
+        }
+
+        $translation->setDocument($document);
+        return $this;
+    }
+}