Merge pull request #33 from KhushbuFuletra/feature-add-object-source

feat(ObjectSource): Add fields to record object source
symbiote · Mar 24, 2020 · 97ebec7 · 97ebec7
2 parents 2a80a0e + 9ae5abf
commit 97ebec7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/src/Model/InternalNotification.php b/src/Model/InternalNotification.php
@@ -11,7 +11,7 @@
 
 class InternalNotification extends DataObject {
     private static $table_name = 'InternalNotification';
-    
+
     private static $db = [
         'Title' => 'Varchar(255)',
         'Message'   => 'Text',
@@ -24,10 +24,16 @@ class InternalNotification extends DataObject {
     private static $has_one = [
         'To'        => Member::class,
         'From'      => Member::class,
+        'SourceObject' => DataObject::class,
+        'SourceNotification' => SystemNotification::class,
     ];
 
     private static $summary_fields = [
-        'Title', 'To.Name', 'SentOn'
+        'Title' => 'Title',
+        'To.Name' => 'To',
+        'SentOn' => 'Sent on',
+        'IsSeen.Nice' => 'Seen?',
+        'IsRead.Nice' => 'Read?'
     ];
 
     private static $default_sort = 'ID DESC';
@@ -62,4 +68,4 @@ public function canEdit($member = null) {
         return $member && (!$this->ID || $this->ToID == $member->ID);
     }
 
-}
+}
diff --git a/src/Service/InternalNotificationSender.php b/src/Service/InternalNotificationSender.php
@@ -2,6 +2,7 @@
 
 namespace Symbiote\Notifications\Service;
 
+use Exception;
 use Symbiote\Notifications\Model\NotificationSender;
 use Symbiote\Notifications\Model\SystemNotification;
 use Symbiote\Notifications\Model\InternalNotification;
@@ -78,6 +79,8 @@ public function sendToUser($notification, $context, $user, $data)
             'ToID'      => $user->ID,
             'FromID'    => Member::currentUserID(),
             'SentOn'    => date('Y-m-d H:i:s'),
+            'SourceObjectID' => $context->ID,
+            'SourceNotificationID' => $notification->ID,
             'Context' => [
                 'ClassName' => get_class($context),
                 'ID' => $context->ID,