diff --git a/src/Notifynder/Builder/NotifynderBuilder.php b/src/Notifynder/Builder/NotifynderBuilder.php index 8becb97..bc4d58d 100755 --- a/src/Notifynder/Builder/NotifynderBuilder.php +++ b/src/Notifynder/Builder/NotifynderBuilder.php @@ -5,6 +5,7 @@ use Fenos\Notifynder\Contracts\NotifynderCategory; use Fenos\Notifynder\Exceptions\NotificationBuilderException; use Illuminate\Contracts\Config\Repository; +use Illuminate\Database\Eloquent\Model; use InvalidArgumentException; use Traversable; use Closure; @@ -275,6 +276,9 @@ protected function setEntityAction($from, $property) $this->setBuilderData("{$property}_type", $from[0]); $this->setBuilderData("{$property}_id", $from[1]); + } elseif($from[0] instanceof Model) { + $this->setBuilderData("{$property}_type", get_class($from[0])); + $this->setBuilderData("{$property}_id", $from[0]->getKey()); } else { $this->isNumeric($from[0]); $this->setBuilderData("{$property}_id", $from[0]);