Preload the files metadata and cache photos-size

Signed-off-by: Joas Schilling <coding@schilljs.com>

lib/Chat/Parser/SystemMessage.php

@@ -34,6 +34,7 @@
 use OCA\Talk\Participant;
 use OCA\Talk\Room;
 use OCA\Talk\Service\ParticipantService;
+use OCA\Talk\Share\Helper\FilesMetadataCache;
 use OCA\Talk\Share\RoomShareProvider;
 use OCP\Comments\IComment;
 use OCP\EventDispatcher\Event;
@@ -88,8 +89,7 @@ public function __construct(
 		protected IRootFolder $rootFolder,
 		protected ICloudIdManager $cloudIdManager,
 		protected IURLGenerator $url,
-		protected IRequest $request,
-		protected IFilesMetadataManager $metadataManager,
+		protected FilesMetadataCache $metadataCache,
 	) {
 	}
 
@@ -741,20 +741,14 @@ protected function getFileFromShare(?Participant $participant, string $shareId):
 		];
 
 		// If a preview is available, check if we can get the dimensions of the file from the metadata API
-		if ($isPreviewAvailable
-			&& $this->request->isUserAgent([IRequest::USER_AGENT_TALK_IOS])) {
+		if ($isPreviewAvailable && str_starts_with($node->getMimeType(), 'image/')) {
 			try {
-				$metadata = $this->metadataManager->getMetaData($fileId, false);
-
-				if ($metadata->hasKey('photos-size')) {
-					$sizeMetadata = $metadata->getArray('photos-size');
-
-					if (isset($sizeMetadata['width']) && isset($sizeMetadata['height'])) {
-						$data['width'] = $sizeMetadata['width'];
-						$data['height'] = $sizeMetadata['height'];
-					}
+				$sizeMetadata = $this->metadataCache->getMetadataPhotosSizeForFileId($fileId);
+				if (isset($sizeMetadata['width'], $sizeMetadata['height'])) {
+					$data['width'] = $sizeMetadata['width'];
+					$data['height'] = $sizeMetadata['height'];
 				}
-			} catch (FilesMetadataNotFoundException $e) {
+			} catch (FilesMetadataNotFoundException) {
 			}
 		}
 

lib/Controller/ChatController.php

@@ -51,6 +51,7 @@
 use OCA\Talk\Service\ParticipantService;
 use OCA\Talk\Service\ReminderService;
 use OCA\Talk\Service\SessionService;
+use OCA\Talk\Share\Helper\FilesMetadataCache;
 use OCA\Talk\Share\RoomShareProvider;
 use OCP\App\IAppManager;
 use OCP\AppFramework\Db\DoesNotExistException;
@@ -73,6 +74,7 @@
 use OCP\RichObjectStrings\IValidator;
 use OCP\Security\ITrustedDomainHelper;
 use OCP\Share\Exceptions\ShareNotFound;
+use OCP\Share\IShare;
 use OCP\User\Events\UserLiveStatusEvent;
 use OCP\UserStatus\IManager as IUserStatusManager;
 use OCP\UserStatus\IUserStatus;
@@ -103,6 +105,7 @@ public function __construct(
 		private GuestManager $guestManager,
 		private MessageParser $messageParser,
 		protected RoomShareProvider $shareProvider,
+		protected FilesMetadataCache $filesMetadataCache,
 		private IManager $autoCompleteManager,
 		private IUserStatusManager $statusManager,
 		protected MatterbridgeManager $matterbridgeManager,
@@ -308,7 +311,8 @@ public function shareObjectToChat(string $objectType, string $objectId, string $
 
 	/*
 	 * Gather share IDs from the comments and preload share definitions
-	 * to avoid separate database query for each individual share.
+	 * and files metadata to avoid separate database query for each
+	 * individual share/node later on.
 	 *
 	 * @param IComment[] $comments
 	 */
@@ -326,10 +330,14 @@ protected function preloadShares(array $comments): void {
 			}
 		}
 		if (!empty($shareIds)) {
-			// Ignore the result for now. Retrieved Share objects will be cached by
+			// Retrieved Share objects will be cached by
 			// the RoomShareProvider and returned from the cache to
 			// the Parser\SystemMessage without additional database queries.
-			$this->shareProvider->getSharesByIds($shareIds);
+			$shares = $this->shareProvider->getSharesByIds($shareIds);
+
+			// Preload files metadata as well
+			$fileIds = array_filter(array_map(static fn(IShare $share) => $share->getNodeId(), $shares));
+			$this->filesMetadataCache->preloadMetadata($fileIds);
 		}
 	}
 

lib/Share/Helper/FilesMetadataCache.php

@@ -0,0 +1,102 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * @copyright Copyright (c) 2023 Joas Schilling <coding@schilljs.com>
+ *
+ * @author Joas Schilling <coding@schilljs.com>
+ *
+ * @license GNU AGPL version 3 or any later version
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+namespace OCA\Talk\Share\Helper;
+
+use OCP\FilesMetadata\Exceptions\FilesMetadataNotFoundException;
+use OCP\FilesMetadata\Exceptions\FilesMetadataTypeException;
+use OCP\FilesMetadata\IFilesMetadataManager;
+use OCP\FilesMetadata\Model\IFilesMetadata;
+
+class FilesMetadataCache {
+	/** @var array<int, array{width: int, height: int}> */
+	protected array $filesSizeData = [];
+
+	public function __construct(
+		protected IFilesMetadataManager $filesMetadataManager,
+	) {
+	}
+
+	/**
+	 * @param list<int> $fileIds
+	 */
+	public function preloadMetadata(array $fileIds): void {
+		$missingFileIds = array_diff($fileIds, array_keys($this->filesSizeData));
+		if (empty($missingFileIds)) {
+			return;
+		}
+
+		$data = $this->filesMetadataManager->getMetadataForFiles($missingFileIds);
+		foreach ($data as $fileId => $metadata) {
+			$this->cachePhotosSize($fileId, $metadata);
+		}
+	}
+
+	/**
+	 * @return array
+	 * @psalm-return array{width: int, height: int}
+	 * @throws FilesMetadataNotFoundException
+	 */
+	public function getMetadataPhotosSizeForFileId(int $fileId): array {
+		if (!array_key_exists($fileId, $this->filesSizeData)) {
+			try {
+				$this->cachePhotosSize($fileId, $this->filesMetadataManager->getMetadata($fileId, true));
+			} catch (FilesMetadataNotFoundException) {
+				$this->filesSizeData[$fileId] = null;
+			}
+		}
+
+		if (!isset($this->filesSizeData[$fileId])) {
+			return throw new FilesMetadataNotFoundException();
+		}
+
+		return $this->filesSizeData[$fileId];
+	}
+
+	protected function cachePhotosSize(int $fileId, IFilesMetadata $metadata): void {
+		if ($metadata->hasKey('photos-size')) {
+			try {
+				$sizeMetadata = $metadata->getArray('photos-size');
+			} catch (FilesMetadataNotFoundException|FilesMetadataTypeException) {
+				$this->filesSizeData[$fileId] = null;
+				return;
+			}
+
+			if (isset($sizeMetadata['width'], $sizeMetadata['height'])) {
+				$dimensions = [
+					'width' => $sizeMetadata['width'],
+					'height' => $sizeMetadata['height'],
+				];
+				$this->filesSizeData[$fileId] = $dimensions;
+			} else {
+				$this->filesSizeData[$fileId] = null;
+			}
+		} else {
+			$this->filesSizeData[$fileId] = null;
+		}
+
+	}
+}