5 files changed, 232 insertions, 0 deletions
diff --git a/src/Model/Artwork.php b/src/Model/Artwork.php
new file mode 100644
index 0000000..25f6ee5
--- /dev/null
+++ b/src/Model/Artwork.php
@@ -0,0 +1,101 @@
+<?php
+
+namespace Elements\Model;
+
+use Doctrine\Common\Collections\ArrayCollection;
+use Doctrine\DBAL\Schema\Column;
+use Doctrine\DBAL\Schema\Table;
+use Doctrine\ORM\Mapping\Entity;
+use Doctrine\ORM\Mapping\GeneratedValue;
+use Doctrine\ORM\Mapping\Id;
+use Doctrine\ORM\Mapping\ManyToOne;
+use Doctrine\ORM\PersistentCollection;
+use Elements\DB;
+use Symfony\Component\HttpFoundation\File\UploadedFile;
+
+#[Entity]
+#[Table(name: 'artworks')]
+/**
+ * @Entity
+ * @Table(name="artworks")
+ */
+class Artwork
+{
+  #[Id]
+  #[Column(type: 'integer')]
+  #[GeneratedValue]
+  /**
+   * @Id
+   * @Column(type="integer")
+   * @GeneratedValue
+   */
+  public int $id;
+
+  #[Column(type: 'string')]
+  /**
+   * @Column(type="string")
+   */
+  public string $path;
+
+  #[ManyToOne(targetEntity: Card::class, inversedBy: 'artworks')]
+  /**
+   * @ManyToOne(targetEntity="Card", inversedBy="artworks", cascade={"persist"})
+   */
+  public Card $card;
+
+  #[OneToMany(targetEntity: VoteArtwork::class, mappedBy: 'artwork')]
+  /**
+   * @OneToMany(targetEntity="VoteArtwork", mappedBy="artwork")
+   */
+  public Collection|ArrayCollection|PersistentCollection $votes;
+
+  /**
+   * Artwork constructor.
+   */
+  public function __construct(string $path)
+  {
+    $this->path = $path;
+    $this->votes = new ArrayCollection();
+  }
+
+  /**
+   * @param UploadedFile $file
+   *
+   * @return self
+   */
+  public static function fromUploadedFile(UploadedFile $file): self
+  {
+    $artworksDir = dirname(dirname(__DIR__)) . '/public/artworks/';
+    $path = str_replace($_SERVER['DOCUMENT_ROOT'], '', $artworksDir . $file->getClientOriginalName());
+    $file->move($artworksDir, $file->getClientOriginalName());
+
+    return new self($path);
+  }
+
+  /**
+   * @param VoteArtwork $vote
+   */
+  public function addVote(VoteArtwork $vote)
+  {
+    $vote->artwork = $this;
+    $this->votes[] = $vote;
+  }
+
+  /**
+   * @return int
+   */
+  public function getVotesTotal(): int
+  {
+    $result = DB::$entityManager
+      ->createQuery(
+	'SELECT sum(v.value) as total
+	FROM Elements\Model\VoteArtwork v
+	WHERE v.artwork = :artwork'
+      )
+      ->setParameter('artwork', $this)
+      ->getOneOrNullResult();
+
+      return $result['total'] ?? 0;
+  }
+}
+
diff --git a/src/Model/Card.php b/src/Model/Card.php
index ab421ab..9707e16 100644
--- a/src/Model/Card.php
+++ b/src/Model/Card.php
@@ -37,12 +37,26 @@ class Card
    */
   public Collection|ArrayCollection|PersistentCollection $meta;
 
+  #[OneToMany(targetEntity: Artwork::class, mappedBy: 'card')]
+  /**
+   * @OneToMany(targetEntity="Artwork", mappedBy="card")
+   */
+  public Collection|ArrayCollection|PersistentCollection $artworks;
+
+  #[OneToMany(targetEntity: VoteCard::class, mappedBy: 'card')]
+  /**
+   * @OneToMany(targetEntity="VoteCard", mappedBy="card")
+   */
+  public Collection|ArrayCollection|PersistentCollection $votes;
+
   /**
    * Card constructor.
    */
   public function __construct()
   {
     $this->meta = new ArrayCollection();
+    $this->artworks = new ArrayCollection();
+    $this->votes = new ArrayCollection();
   }
 
   /**
@@ -55,6 +69,24 @@ class Card
   }
 
   /**
+   * @param Artwork $artwork
+   */
+  public function addArtwork(Artwork $artwork)
+  {
+    $artwork->card = $this;
+    $this->artworks[] = $artwork;
+  }
+
+  /**
+   * @param VoteCard $vote
+   */
+  public function addVote(VoteCard $vote)
+  {
+    $vote->card = $this;
+    $this->votes[] = $vote;
+  }
+
+  /**
    * @param string $key
    *
    * @return string
@@ -83,5 +115,22 @@ class Card
 
       return $result['value'] ?? null;
   }
+
+  /**
+   * @return int
+   */
+  public function getVotesTotal(): int
+  {
+    $result = DB::$entityManager
+      ->createQuery(
+	'SELECT sum(v.value) as total
+	FROM Elements\Model\VoteCard v
+	WHERE v.card = :card'
+      )
+      ->setParameter('card', $this)
+      ->getOneOrNullResult();
+
+      return $result['total'] ?? 0;
+  }
 }
 
diff --git a/src/Model/Vote.php b/src/Model/Vote.php
new file mode 100644
index 0000000..c5415aa
--- /dev/null
+++ b/src/Model/Vote.php
@@ -0,0 +1,30 @@
+<?php
+
+namespace Elements\Model;
+
+use Doctrine\DBAL\Schema\Column;
+use Doctrine\DBAL\Schema\Table;
+use Doctrine\ORM\Mapping\Entity;
+use Doctrine\ORM\Mapping\GeneratedValue;
+use Doctrine\ORM\Mapping\Id;
+use Doctrine\ORM\Mapping\ManyToOne;
+
+class Vote
+{
+  #[Id]
+  #[Column(type: 'integer')]
+  #[GeneratedValue]
+  /**
+   * @Id
+   * @Column(type="integer")
+   * @GeneratedValue
+   */
+  public int $id;
+
+  #[Column(type: 'integer')]
+  /**
+   * @Column(type="integer")
+   */
+  public int $value;
+}
+
diff --git a/src/Model/VoteArtwork.php b/src/Model/VoteArtwork.php
new file mode 100644
index 0000000..ee6ac8b
--- /dev/null
+++ b/src/Model/VoteArtwork.php
@@ -0,0 +1,26 @@
+<?php
+
+namespace Elements\Model;
+
+use Doctrine\DBAL\Schema\Column;
+use Doctrine\DBAL\Schema\Table;
+use Doctrine\ORM\Mapping\Entity;
+use Doctrine\ORM\Mapping\GeneratedValue;
+use Doctrine\ORM\Mapping\Id;
+use Doctrine\ORM\Mapping\ManyToOne;
+
+#[Entity]
+#[Table(name: 'votes_artwork')]
+/**
+ * @Entity
+ * @Table(name="votes_artwork")
+ */
+class VoteArtwork extends Vote
+{
+  #[ManyToOne(targetEntity: Artwork::class, inversedBy: 'votes')]
+  /**
+   * @ManyToOne(targetEntity="Artwork", inversedBy="votes", cascade={"persist"})
+   */
+  public Artwork $artwork;
+}
+
diff --git a/src/Model/VoteCard.php b/src/Model/VoteCard.php
new file mode 100644
index 0000000..e9b3e60
--- /dev/null
+++ b/src/Model/VoteCard.php
@@ -0,0 +1,26 @@
+<?php
+
+namespace Elements\Model;
+
+use Doctrine\DBAL\Schema\Column;
+use Doctrine\DBAL\Schema\Table;
+use Doctrine\ORM\Mapping\Entity;
+use Doctrine\ORM\Mapping\GeneratedValue;
+use Doctrine\ORM\Mapping\Id;
+use Doctrine\ORM\Mapping\ManyToOne;
+
+#[Entity]
+#[Table(name: 'votes_card')]
+/**
+ * @Entity
+ * @Table(name="votes_card")
+ */
+class VoteCard extends Vote
+{
+  #[ManyToOne(targetEntity: Card::class, inversedBy: 'votes')]
+  /**
+   * @ManyToOne(targetEntity="Card", inversedBy="votes", cascade={"persist"})
+   */
+  public Card $card;
+}
+