Retrieve session snippet rate and hat fields

[airbundle] / Entity / Snippet.php

diff --git a/Entity/Snippet.php b/Entity/Snippet.php
index 99bc040a170e26b1b28dd45dbc16ab1175da9dae..999091f6733d220b9f92c72cbf83cf405cdde7c2 100644 (file)
--- a/Entity/Snippet.php
+++ b/Entity/Snippet.php
@@ -39,6 +39,11 @@ class Snippet {
         */
        protected $rate;
 
+       /**
+        * @var bool
+        */
+       protected $hat;
+
        /**
         * @var string
         */
@@ -79,6 +84,13 @@ class Snippet {
         */
        protected $user;
 
+       /**
+        * Constructor
+        */
+       public function __construct() {
+               $this->hat = true;
+       }
+
        /**
         * Get id
         *
@@ -198,6 +210,27 @@ class Snippet {
                return $this->rate;
        }
 
+       /**
+        * Set hat
+        *
+        * @param bool $hat
+        *
+        * @return User
+        */
+       public function setHat(bool $hat) {
+               $this->hat = $hat;
+
+               return $this;
+       }
+
+       /**
+        * Get hat
+        *
+        * @return bool
+        */
+       public function getHat(): bool {
+               return $this->hat;
+       }
        /**
         * Set contact
         *