X-Git-Url: https://git.rapsys.eu/airbundle/blobdiff_plain/9aad88c45d93dadacd35176364b56df4734c017b..04441c5bddb7e62cf60aacbaf4146361d7907670:/Entity/Snippet.php

diff --git a/Entity/Snippet.php b/Entity/Snippet.php
index 28c7c1c..fa5fe99 100644
--- a/Entity/Snippet.php
+++ b/Entity/Snippet.php
@@ -96,6 +96,7 @@ class Snippet {
 	 * Constructor
 	 */
 	public function __construct() {
+		//Set defaults
 		$this->description = null;
 		$this->class = null;
 		$this->short = null;
@@ -105,6 +106,9 @@ class Snippet {
 		$this->donate = null;
 		$this->link = null;
 		$this->profile = null;
+		$this->location = null;
+		$this->created = new \DateTime('now');
+		$this->updated = new \DateTime('now');
 	}
 
 	/**
@@ -207,11 +211,11 @@ class Snippet {
 	/**
 	 * Set rate
 	 *
-	 * @param string $rate
+	 * @param int $rate
 	 *
 	 * @return Snippet
 	 */
-	public function setRate(?string $rate): Snippet {
+	public function setRate(?int $rate): Snippet {
 		$this->rate = $rate;
 
 		return $this;
@@ -220,9 +224,9 @@ class Snippet {
 	/**
 	 * Get rate
 	 *
-	 * @return string
+	 * @return int
 	 */
-	public function getRate(): ?string {
+	public function getRate(): ?int {
 		return $this->rate;
 	}
 
@@ -233,7 +237,7 @@ class Snippet {
 	 *
 	 * @return User
 	 */
-	public function setHat(bool $hat): Snippet {
+	public function setHat(?bool $hat): Snippet {
 		$this->hat = $hat;
 
 		return $this;
@@ -244,7 +248,7 @@ class Snippet {
 	 *
 	 * @return bool
 	 */
-	public function getHat(): bool {
+	public function getHat(): ?bool {
 		return $this->hat;
 	}
 	/**
@@ -426,9 +430,9 @@ class Snippet {
 	/**
 	 * {@inheritdoc}
 	 */
-	public function preUpdate(\Doctrine\ORM\Event\PreUpdateEventArgs  $eventArgs) {
+	public function preUpdate(PreUpdateEventArgs $eventArgs) {
 		//Check that we have an snippet instance
-		if (($snippet = $eventArgs->getEntity()) instanceof Snippet) {
+		if (($snippet = $eventArgs->getObject()) instanceof Snippet) {
 			//Set updated value
 			$snippet->setUpdated(new \DateTime('now'));
 		}