Fixes #354, Adds the spellchecker option

[bookstack] / app / Entity.php

diff --git a/app/Entity.php b/app/Entity.php
index 0fd5d1e12e54515ec9aa07c59d653e9affaa7831..e8deddf0a56f6f861e31ee83523ccac9c47c1bdc 100644 (file)
--- a/app/Entity.php
+++ b/app/Entity.php
@@ -1,47 +1,108 @@
-<?php
+<?php namespace BookStack;

 
 

-namespace Oxbow;

 
 

-use Illuminate\Database\Eloquent\Model;
-
-class Entity extends Model
+class Entity extends Ownable

 {
 {

+
+    protected $fieldsToSearch = ['name', 'description'];
+

     /**
     /**

-     * Relation for the user that created this entity.
-     * @return \Illuminate\Database\Eloquent\Relations\BelongsTo
+     * Compares this entity to another given entity.
+     * Matches by comparing class and id.
+     * @param $entity
+     * @return bool

      */
      */

-    public function createdBy()
+    public function matches($entity)

     {
     {

-        return $this->belongsTo('Oxbow\User', 'created_by');
+        return [get_class($this), $this->id] === [get_class($entity), $entity->id];

     }
 
     /**
     }
 
     /**

-     * Relation for the user that updated this entity.
-     * @return \Illuminate\Database\Eloquent\Relations\BelongsTo
+     * Checks if an entity matches or contains another given entity.
+     * @param Entity $entity
+     * @return bool

      */
      */

-    public function updatedBy()
+    public function matchesOrContains(Entity $entity)

     {
     {

-        return $this->belongsTo('Oxbow\User', 'updated_by');
+        $matches = [get_class($this), $this->id] === [get_class($entity), $entity->id];
+
+        if ($matches) return true;
+
+        if (($entity->isA('chapter') || $entity->isA('page')) && $this->isA('book')) {
+            return $entity->book_id === $this->id;
+        }
+
+        if ($entity->isA('page') && $this->isA('chapter')) {
+            return $entity->chapter_id === $this->id;
+        }
+
+        return false;

     }
 
     /**
     }
 
     /**

-     * Compares this entity to another given entity.
-     * Matches by comparing class and id.
-     * @param $entity
+     * Gets the activity objects for this entity.
+     * @return \Illuminate\Database\Eloquent\Relations\MorphMany
+     */
+    public function activity()
+    {
+        return $this->morphMany(Activity::class, 'entity')->orderBy('created_at', 'desc');
+    }
+
+    /**
+     * Get View objects for this entity.
+     */
+    public function views()
+    {
+        return $this->morphMany(View::class, 'viewable');
+    }
+
+    /**
+     * Get the Tag models that have been user assigned to this entity.
+     * @return \Illuminate\Database\Eloquent\Relations\MorphMany
+     */
+    public function tags()
+    {
+        return $this->morphMany(Tag::class, 'entity')->orderBy('order', 'asc');
+    }
+
+    /**
+     * Get this entities restrictions.
+     */
+    public function permissions()
+    {
+        return $this->morphMany(EntityPermission::class, 'restrictable');
+    }
+
+    /**
+     * Check if this entity has a specific restriction set against it.
+     * @param $role_id
+     * @param $action

      * @return bool
      */
      * @return bool
      */

-    public function matches($entity)
+    public function hasRestriction($role_id, $action)

     {
     {

-        return [get_class($this), $this->id] === [get_class($entity), $entity->id];
+        return $this->permissions()->where('role_id', '=', $role_id)
+            ->where('action', '=', $action)->count() > 0;
+    }
+
+    /**
+     * Check if this entity has live (active) restrictions in place.
+     * @param $role_id
+     * @param $action
+     * @return bool
+     */
+    public function hasActiveRestriction($role_id, $action)
+    {
+        return $this->getRawAttribute('restricted') && $this->hasRestriction($role_id, $action);

     }
 
     /**
     }
 
     /**

-     * Gets the activity for this entity.
+     * Get the entity jointPermissions this is connected to.

      * @return \Illuminate\Database\Eloquent\Relations\MorphMany
      */
      * @return \Illuminate\Database\Eloquent\Relations\MorphMany
      */

-    public function activity()
+    public function jointPermissions()

     {
     {

-        return $this->morphMany('Oxbow\Activity', 'entity')->orderBy('created_at', 'desc');
+        return $this->morphMany(JointPermission::class, 'entity');

     }
 
     /**
     }
 
     /**


@@ -50,35 +111,109 @@ class Entity extends Model
      * @param $type
      * @return bool
      */
      * @param $type
      * @return bool
      */

-    public function isA($type)
+    public static function isA($type)
+    {
+        return static::getType() === strtolower($type);
+    }
+
+    /**
+     * Get entity type.
+     * @return mixed
+     */
+    public static function getType()

     {
     {

-        return $this->getName() === strtolower($type);
+        return strtolower(static::getClassName());

     }
 
     /**
     }
 
     /**

-     * Gets the class name.
+     * Get an instance of an entity of the given type.
+     * @param $type
+     * @return Entity
+     */
+    public static function getEntityInstance($type)
+    {
+        $types = ['Page', 'Book', 'Chapter'];
+        $className = str_replace([' ', '-', '_'], '', ucwords($type));
+        if (!in_array($className, $types)) {
+            return null;
+        }
+
+        return app('BookStack\\' . $className);
+    }
+
+    /**
+     * Gets a limited-length version of the entities name.
+     * @param int $length

      * @return string
      */
      * @return string
      */

-    public function getName()
+    public function getShortName($length = 25)

     {
     {

-        $fullClassName = get_class($this);
-        return strtolower(array_slice(explode('\\', $fullClassName), -1, 1)[0]);
+        if (strlen($this->name) <= $length) return $this->name;
+        return substr($this->name, 0, $length - 3) . '...';

     }
 
     /**
      * Perform a full-text search on this entity.
      * @param string[] $fieldsToSearch
      * @param string[] $terms
     }
 
     /**
      * Perform a full-text search on this entity.
      * @param string[] $fieldsToSearch
      * @param string[] $terms

+     * @param string[] array $wheres

      * @return mixed
      */
      * @return mixed
      */

-    public static function fullTextSearch($fieldsToSearch, $terms)
+    public function fullTextSearchQuery($terms, $wheres = [])

     {
     {

-        $termString = '';
-        foreach($terms as $term) {
-            $termString .= $term . '* ';
+        $exactTerms = [];
+        $fuzzyTerms = [];
+        $search = static::newQuery();
+
+        foreach ($terms as $key => $term) {
+            $term = htmlentities($term, ENT_QUOTES);
+            $term = preg_replace('/[+\-><\(\)~*\"@]+/', ' ', $term);
+            if (preg_match('/&quot;.*?&quot;/', $term) || is_numeric($term)) {
+                $term = str_replace('&quot;', '', $term);
+                $exactTerms[] = '%' . $term . '%';
+            } else {
+                $term = '' . $term . '*';
+                if ($term !== '*') $fuzzyTerms[] = $term;
+            }

         }
         }

-        $fields = implode(',', $fieldsToSearch);
-        return static::whereRaw('MATCH(' . $fields . ') AGAINST(? IN BOOLEAN MODE)', [$termString])->get();
+
+        $isFuzzy = count($exactTerms) === 0 && count($fuzzyTerms) > 0;
+
+
+        // Perform fulltext search if relevant terms exist.
+        if ($isFuzzy) {
+            $termString = implode(' ', $fuzzyTerms);
+            $fields = implode(',', $this->fieldsToSearch);
+            $search = $search->selectRaw('*, MATCH(name) AGAINST(? IN BOOLEAN MODE) AS title_relevance', [$termString]);
+            $search = $search->whereRaw('MATCH(' . $fields . ') AGAINST(? IN BOOLEAN MODE)', [$termString]);
+        }
+
+        // Ensure at least one exact term matches if in search
+        if (count($exactTerms) > 0) {
+            $search = $search->where(function ($query) use ($exactTerms) {
+                foreach ($exactTerms as $exactTerm) {
+                    foreach ($this->fieldsToSearch as $field) {
+                        $query->orWhere($field, 'like', $exactTerm);
+                    }
+                }
+            });
+        }
+
+        $orderBy = $isFuzzy ? 'title_relevance' : 'updated_at';
+
+        // Add additional where terms
+        foreach ($wheres as $whereTerm) {
+            $search->where($whereTerm[0], $whereTerm[1], $whereTerm[2]);
+        }
+
+        // Load in relations
+        if ($this->isA('page')) {
+            $search = $search->with('book', 'chapter', 'createdBy', 'updatedBy');
+        } else if ($this->isA('chapter')) {
+            $search = $search->with('book');
+        }
+
+        return $search->orderBy($orderBy, 'desc');

     }
 
 }
     }
 
 }