From 7306370ececd6ada186061af7deb9437d6182be6 Mon Sep 17 00:00:00 2001
From: Marco Nassabain <marco.nassabain@hotmail.com>
Date: Thu, 18 Mar 2021 22:29:13 +0100
Subject: =?UTF-8?q?=E2=9C=A8=20Item:=20add=20sharer=20display=20name=20in?=
 =?UTF-8?q?=20findEntities?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Marco Nassabain <marco.nassabain@hotmail.com>
---
 lib/Db/ItemMapperV2.php | 37 ++++++++++++++++++++++++++++++++++++-
 1 file changed, 36 insertions(+), 1 deletion(-)

(limited to 'lib')

diff --git a/lib/Db/ItemMapperV2.php b/lib/Db/ItemMapperV2.php
index 44d1740cd..c6c969e01 100644
--- a/lib/Db/ItemMapperV2.php
+++ b/lib/Db/ItemMapperV2.php
@@ -22,6 +22,7 @@ use OCP\AppFramework\Db\MultipleObjectsReturnedException;
 use OCP\DB\Exception as DBException;
 use OCP\DB\QueryBuilder\IQueryBuilder;
 use OCP\IDBConnection;
+use OCP\IUserManager;
 
 /**
  * Class ItemMapper
@@ -32,15 +33,49 @@ class ItemMapperV2 extends NewsMapperV2
 {
     const TABLE_NAME = 'news_items';
 
+    /**
+     * @var IUserManager
+     */
+    private $userManager;
+
     /**
      * ItemMapper constructor.
      *
      * @param IDBConnection $db
      * @param Time          $time
      */
-    public function __construct(IDBConnection $db, Time $time)
+    public function __construct(IDBConnection $db, Time $time, IUserManager $userManager)
     {
         parent::__construct($db, $time, Item::class);
+        $this->userManager = $userManager;
+    }
+
+    /**
+	 * Override parent constructor to insert sharer display names for shared items
+	 *
+	 * @param IQueryBuilder $query
+	 * @return Entity[] all fetched entities
+	 */
+    public function findEntities(IQueryBuilder $query): array {
+        $entities = parent::findEntities($query);
+
+        foreach ($entities as $entity) {
+            $sharedBy = $entity->getSharedBy();
+            $sharedByDisplayName = null;
+
+            // Get user display name
+            if (!is_null($sharedBy)) {
+                $user = $this->userManager->get($sharedBy);
+                if (!is_null($user)) {
+                    $sharedByDisplayName = $user->getDisplayName();
+                }
+            }
+
+            // Set sharer display name
+            $entity->setSharedByDisplayName($sharedByDisplayName);
+        }
+
+        return $entities;
     }
 
     /**
-- 
cgit v1.2.3