perf: reduce number of avatar requests

lib/Contracts/IAvatarService.php

@@ -17,9 +17,10 @@ interface IAvatarService {
 	 *
 	 * @param string $email
 	 * @param string $uid
-	 * @return Avatar|null
+	 * @param bool $cachedOnly
+	 * @return Avatar|null|false the avatar if found, false if $cachedOnly is true and no value cached and null if not found
 	 */
-	public function getAvatar(string $email, string $uid): ?Avatar;
+	public function getAvatar(string $email, string $uid, bool $cachedOnly = false): mixed;
 
 	/**
 	 * @param string $email

lib/Contracts/IMailSearch.php

@@ -36,6 +36,7 @@ public function findMessage(Account $account,
 	 * @param string|null $filter
 	 * @param int|null $cursor
 	 * @param int|null $limit
+	 * @param string|null $userId
 	 *
 	 * @return Message[]
 	 *
@@ -47,7 +48,8 @@ public function findMessages(Account $account,
 		string $sortOrder,
 		?string $filter,
 		?int $cursor,
-		?int $limit): array;
+		?int $limit,
+		?string $userId): array;
 
 	/**
 	 * @param IUser $user

lib/Controller/MessagesController.php

@@ -152,7 +152,8 @@
 				$order,
 				$filter === '' ? null : $filter,
 				$cursor,
-				$limit
+				$limit,
+				$this->currentUserId
 			)
 		);
 	}

lib/Db/Message.php

@@ -12,6 +12,7 @@
 use Horde_Mail_Rfc822_Identification;
 use JsonSerializable;
 use OCA\Mail\AddressList;
+use OCA\Mail\Service\Avatar\Avatar;
 use OCP\AppFramework\Db\Entity;
 use ReturnTypeWillChange;
 use function in_array;
@@ -136,6 +137,12 @@
 	/** @var Tag[] */
 	private $tags = [];
 
+	/** @var Avatar|null */
+	private $avatar;
+
+	/** @var bool*/
+	private $fetchAvatarFromClient = false;
+
 	public function __construct() {
 		$this->from = new AddressList([]);
 		$this->to = new AddressList([]);
@@ -286,6 +293,24 @@
 			);
 		}
 	}
+	/**
+	 * @param Avatar|null $avatar
+	 * @return void
+	 */
+	public function setAvatar(?Avatar $avatar): void {
+		$this->avatar = $avatar;
+	}
+
+	public function setFetchAvatarFromClient(bool $fetchAvatarFromClient): void {
+		$this->fetchAvatarFromClient = $fetchAvatarFromClient;
+	}
+
+	/**
+	 * @return ?Avatar
+	 */
+	public function getAvatar(): ?Avatar {
+		return $this->avatar;
+	}
 
 	#[ReturnTypeWillChange]
 	public function jsonSerialize() {
@@ -331,6 +356,8 @@
 			'summary' => $this->getSummary(),
 			'encrypted' => ($this->isEncrypted() === true),
 			'mentionsMe' => $this->getMentionsMe(),
+			'avatar' => $this->avatar?->jsonSerialize(),
+			'fetchAvatarFromClient' => $this->fetchAvatarFromClient,
 		];
 	}
 }

lib/IMAP/PreviewEnhancer.php

@@ -15,6 +15,8 @@
 use OCA\Mail\Db\Message;
 use OCA\Mail\Db\MessageMapper as DbMapper;
 use OCA\Mail\IMAP\MessageMapper as ImapMapper;
+use OCA\Mail\Service\AvatarService;
+use OCA\Mail\Service\Avatar\Avatar;
 use Psr\Log\LoggerInterface;
 use function array_key_exists;
 use function array_map;
@@ -34,22 +36,27 @@
 	/** @var LoggerInterface */
 	private $logger;
 
+	/** @var AvatarService */
+	private $avatarService;
+
 	public function __construct(IMAPClientFactory $clientFactory,
 		ImapMapper $imapMapper,
 		DbMapper $dbMapper,
-		LoggerInterface $logger) {
+		LoggerInterface $logger,
+		AvatarService $avatarService) {
 		$this->clientFactory = $clientFactory;
 		$this->imapMapper = $imapMapper;
 		$this->mapper = $dbMapper;
 		$this->logger = $logger;
+		$this->avatarService = $avatarService;
 	}
 
 	/**
 	 * @param Message[] $messages
 	 *
 	 * @return Message[]
 	 */
-	public function process(Account $account, Mailbox $mailbox, array $messages): array {
+	public function process(Account $account, Mailbox $mailbox, array $messages, bool $preLoadAvatars = false, ?string $userId = null): array {
 		$needAnalyze = array_reduce($messages, static function (array $carry, Message $message) {
 			if ($message->getStructureAnalyzed()) {
 				// Nothing to do
@@ -59,6 +66,22 @@
 			return array_merge($carry, [$message->getUid()]);
 		}, []);
 
+		if ($preLoadAvatars) {
+			foreach ($messages as $message) {
+				$from = $message->getFrom()->first();
+				if ($message->getAvatar() === null && $from !== null && $from->getEmail() !== null && $userId !== null) {
+					$avatar = $this->avatarService->getAvatar($from->getEmail(), $userId, true);
+					if ($avatar === false) {
+						$message->setFetchAvatarFromClient(true);
+					}
+					if ($avatar instanceof Avatar) {
+						$message->setAvatar($avatar);
+					}
+
+				}
+			}
+		}
+
 		if ($needAnalyze === []) {
 			// Nothing to enhance
 			return $messages;

lib/Service/AvatarService.php

@@ -89,9 +89,10 @@
 	/**
 	 * @param string $email
 	 * @param string $uid
-	 * @return Avatar|null
+	 * @param bool $cachedOnly
+	 * @return Avatar|null|false
 	 */
-	public function getAvatar(string $email, string $uid): ?Avatar {
+	public function getAvatar(string $email, string $uid, bool $cachedOnly = false): mixed{
 		$cachedAvatar = $this->cache->get($email, $uid);
 		if ($cachedAvatar) {
 			return $cachedAvatar;
@@ -101,6 +102,11 @@
 			return null;
 		}
 
+		if ($cachedOnly){
+			// We want to fetch the avatar from the frontend
+			return false;
+		}
+
 		$avatar = $this->source->fetch($email, $this->avatarFactory, $this->externalAvatarsAllowed($uid));
 		if (is_null($avatar) || !$this->hasAllowedMime($avatar)) {
 			// Cannot locate any avatar -> nothing to do here
@@ -127,7 +133,7 @@
 			return null;
 		}
 
 		$cachedImage = $this->cache->getImage($avatar->getUrl(), $uid);
 		if (!is_null($cachedImage)) {
 			return [$avatar, base64_decode($cachedImage)];
 		}

lib/Service/Search/MailSearch.php

@@ -91,7 +91,8 @@
 		string $sortOrder,
 		?string $filter,
 		?int $cursor,
-		?int $limit): array {
+		?int $limit,
+		?string $userId): array {
 		if ($mailbox->hasLocks($this->timeFactory->getTime())) {
 			throw MailboxLockedException::from($mailbox);
 		}
@@ -118,7 +119,9 @@
 			$this->messageMapper->findByIds($account->getUserId(),
 				$this->getIdsLocally($account, $mailbox, $query, $sortOrder, $limit),
 				$sortOrder,
-			)
+			),
+			true,
+			$userId
 		);
 	}
 

src/components/Avatar.vue

@@ -17,6 +17,7 @@
 
 <script>
 import NcAvatar from '@nextcloud/vue/dist/Components/NcAvatar.js'
+import { generateUrl } from '@nextcloud/router'
 import { fetchAvatarUrlMemoized } from '../service/AvatarService.js'
 import logger from '../logger.js'
 
@@ -30,6 +31,14 @@ export default {
 			type: String,
 			required: true,
 		},
+		avatar: {
+			type: Object,
+			default: null,
+		},
+		fetchAvatar: {
+			type: Boolean,
+			default: false,
+		},
 		email: {
 			type: String,
 			required: true,
@@ -55,14 +64,21 @@ export default {
 		},
 	},
 	async mounted() {
-		if (this.email !== '') {
-			try {
-				this.avatarUrl = await fetchAvatarUrlMemoized(this.email)
-			} catch {
-				logger.debug('Could not fetch avatar', { email: this.email })
+		if (this.avatar) {
+			this.avatarUrl = this.avatar.isExternal
+				? generateUrl('/apps/mail/api/avatars/image/{email}', {
+					email: this.email,
+				})
+				: this.avatar.url
+		} else if (this.fetchAvatar) {
+			if (this.email !== '') {
+				try {
+					this.avatarUrl = await fetchAvatarUrlMemoized(this.email)
+				} catch {
+					logger.debug('Could not fetch avatar', { email: this.email })
+				}
 			}
 		}
-
 		this.loading = false
 	},
 }

src/components/Envelope.vue

@@ -36,7 +36,7 @@
 				:class="{ 'important-one-line': oneLineLayout, 'icon-important': !oneLineLayout }"
 				:data-starred="isImportant ? 'true' : 'false'"
 				@click.prevent="hasWriteAcl ? onToggleImportant() : false"
 				v-html="importantSvg" />
 			<JunkIcon v-if="data.flags.$junk"
 				:size="18"
 				class="app-content-list-item-star junk-icon-style"
@@ -52,7 +52,10 @@
 					<CheckIcon :size="40" class="check-icon" :class="{ 'app-content-list-item-avatar-selected': selected }" />
 				</template>
 				<template v-else>
-					<Avatar :display-name="addresses" :email="avatarEmail" />
+					<Avatar :display-name="addresses"
+						:email="avatarEmail"
+						:fetch-avatar="data.fetchAvatarFromClient"
+						:avatar="data.avatar" />
 				</template>
 			</div>
 		</template>
@@ -481,7 +484,7 @@
 		window.addEventListener('resize', this.onWindowResize)
 	},
 	computed: {
 		...mapGetters([
 			'isSnoozeDisabled',
 		]),
 		messageLongDate() {

src/components/OutboxMessageListItem.vue

@@ -10,7 +10,10 @@
 		:details="details"
 		@click="openModal">
 		<template #icon>
-			<Avatar :display-name="avatarDisplayName" :email="avatarEmail" />
+			<Avatar :display-name="avatarDisplayName"
+				:email="avatarEmail"
+				::fetch-avatar="data.fetchAvatarFromClient"
+				:avatar="message.avatar" />
 		</template>
 		<template #subname>
 			{{ subjectForSubtitle }}

src/components/ThreadEnvelope.vue

@@ -26,6 +26,8 @@
 					:display-name="envelope.from[0].label"
 					:disable-tooltip="true"
 					:size="40"
+					:fetch-avatar="data.fetchAvatarFromClient"
+					:avatar="envelope.avatar"
 					class="envelope__header__avatar-avatar" />
 				<div v-if="isImportant"
 					class="app-content-list-item-star icon-important"