Adds test cases and fixes an issue with the permission checking.

[bookstack] / app / Entity.php

diff --git a/app/Entity.php b/app/Entity.php
index e5dd04bf28bdb01d30690c18e5b9a2b6986a7929..df8e4d38b241e9928ca951697b6da392744a39e2 100644 (file)
--- a/app/Entity.php
+++ b/app/Entity.php
@@ -1,6 +1,8 @@
 <?php namespace BookStack;
 
 
+use Illuminate\Database\Eloquent\Relations\MorphMany;
+
 class Entity extends Ownable
 {
 
@@ -65,6 +67,17 @@ class Entity extends Ownable
         return $this->morphMany(Tag::class, 'entity')->orderBy('order', 'asc');
     }
 
+    /**
+     * Get the comments for an entity
+     * @param bool $orderByCreated
+     * @return MorphMany
+     */
+    public function comments($orderByCreated = true)
+    {
+        $query = $this->morphMany(Comment::class, 'entity');
+        return $orderByCreated ? $query->orderBy('created_at', 'asc') : $query;
+    }
+
     /**
      * Get the related search terms.
      * @return \Illuminate\Database\Eloquent\Relations\MorphMany