From f7b138007d2d4e0e4a8cef7e988c240de37921ba Mon Sep 17 00:00:00 2001 From: Tom Witkowski Date: Tue, 26 Apr 2016 18:04:22 +0200 Subject: [PATCH] Issue #110 --- src/Notifynder/Builder/NotifynderBuilder.php | 4 ++++ 1 file changed, 4 insertions(+) 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]);