Merge pull request #76 from renoki-co/feature/custom-morph-class-method

[feature] Custom Morphing class method
renoki-co · Sep 29, 2020 · cb57a53 · cb57a53
2 parents 65093af + 00cc679
commit cb57a53
Show file tree

Hide file tree

Showing 7 changed files with 49 additions and 8 deletions.
diff --git a/src/Traits/CanBeBlocked.php b/src/Traits/CanBeBlocked.php
@@ -4,6 +4,8 @@
 
 trait CanBeBlocked
 {
+    use HasCustomModelClass;
+
     /**
      * Relationship for models that blocked this model.
      *
@@ -12,7 +14,7 @@ trait CanBeBlocked
      */
     public function blockers($model = null)
     {
-        $modelClass = $model ? (new $model)->getMorphClass() : $this->getMorphClass();
+        $modelClass = $this->getModelMorphClass($model);
 
         return $this
             ->morphToMany($modelClass, 'blockable', 'blockers', 'blockable_id', 'blocker_id')

diff --git a/src/Traits/CanBeFollowed.php b/src/Traits/CanBeFollowed.php
@@ -7,6 +7,8 @@
 
 trait CanBeFollowed
 {
+    use HasCustomModelClass;
+
     /**
      * Relationship for models that followed this model.
      *
@@ -15,7 +17,7 @@ trait CanBeFollowed
      */
     public function followers($model = null)
     {
-        $modelClass = $model ? (new $model)->getMorphClass() : $this->getMorphClass();
+        $modelClass = $this->getModelMorphClass($model);
 
         return $this
             ->morphToMany($modelClass, 'followable', 'followers', 'followable_id', 'follower_id')
@@ -34,7 +36,7 @@ public function followers($model = null)
      */
     public function followerRequests($model = null)
     {
-        $modelClass = $model ? (new $model)->getMorphClass() : $this->getMorphClass();
+        $modelClass = $this->getModelMorphClass($model);
 
         return $this
             ->morphToMany($modelClass, 'followable', 'followers', 'followable_id', 'follower_id')

diff --git a/src/Traits/CanBeLiked.php b/src/Traits/CanBeLiked.php
@@ -4,6 +4,8 @@
 
 trait CanBeLiked
 {
+    use HasCustomModelClass;
+
     /**
      * Relationship for models that liked this model.
      *
@@ -12,7 +14,7 @@ trait CanBeLiked
      */
     public function likers($model = null)
     {
-        $modelClass = $model ? (new $model)->getMorphClass() : $this->getMorphClass();
+        $modelClass = $this->getModelMorphClass($model);
 
         return $this
             ->morphToMany($modelClass, 'likeable', 'likers', 'likeable_id', 'liker_id')

diff --git a/src/Traits/CanBlock.php b/src/Traits/CanBlock.php
@@ -7,6 +7,8 @@
 
 trait CanBlock
 {
+    use HasCustomModelClass;
+
     /**
      * Relationship for models that this model is currently blocking.
      *
@@ -15,7 +17,7 @@ trait CanBlock
      */
     public function blocking($model = null)
     {
-        $modelClass = $model ? (new $model)->getMorphClass() : $this->getMorphClass();
+        $modelClass = $this->getModelMorphClass($model);
 
         return $this
             ->morphToMany($modelClass, 'blocker', 'blockers', 'blocker_id', 'blockable_id')

diff --git a/src/Traits/CanFollow.php b/src/Traits/CanFollow.php
@@ -7,6 +7,8 @@
 
 trait CanFollow
 {
+    use HasCustomModelClass;
+
     /**
      * Relationship for models that this model is currently following.
      *
@@ -15,7 +17,7 @@ trait CanFollow
      */
     public function following($model = null)
     {
-        $modelClass = $model ? (new $model)->getMorphClass() : $this->getMorphClass();
+        $modelClass = $this->getModelMorphClass($model);
 
         return $this
             ->morphToMany($modelClass, 'follower', 'followers', 'follower_id', 'followable_id')
@@ -116,7 +118,7 @@ public function unfollow($model): bool
      */
     public function followRequests($model = null)
     {
-        $modelClass = $model ? (new $model)->getMorphClass() : $this->getMorphClass();
+        $modelClass = $this->getModelMorphClass($model);
 
         return $this
             ->morphToMany($modelClass, 'follower', 'followers', 'follower_id', 'followable_id')

diff --git a/src/Traits/CanLike.php b/src/Traits/CanLike.php
@@ -7,6 +7,8 @@
 
 trait CanLike
 {
+    use HasCustomModelClass;
+
     /**
      * Relationship for models that this model is currently liking.
      *
@@ -15,7 +17,7 @@ trait CanLike
      */
     public function liking($model = null)
     {
-        $modelClass = $model ? (new $model)->getMorphClass() : $this->getMorphClass();
+        $modelClass = $this->getModelMorphClass($model);
 
         return $this
             ->morphToMany($modelClass, 'liker', 'likers', 'liker_id', 'likeable_id')

diff --git a/src/Traits/HasCustomModelClass.php b/src/Traits/HasCustomModelClass.php
@@ -0,0 +1,29 @@
+<?php
+
+namespace Rennokki\Befriended\Traits;
+
+trait HasCustomModelClass
+{
+    /**
+     * Get the model's morph class to act as the main resource.
+     *
+     * @param  string|null  $model
+     * @return string
+     */
+    protected function getModelMorphClass($model = null)
+    {
+        return $model
+            ? (new $model)->getMorphClass()
+            : $this->getCurrentModelMorphClass();
+    }
+
+    /**
+     * Get the current model's morph class.
+     *
+     * @return string
+     */
+    protected function getCurrentModelMorphClass()
+    {
+        return $this->getMorphClass();
+    }
+}