fixup! feat: delete tags

Signed-off-by: hamza221 <[email protected]>
nextcloud · Nov 6, 2023 · a796a2b · a796a2b
1 parent 1a95848
commit a796a2b
Showing 1 changed file with 17 additions and 14 deletions.
diff --git a/lib/Service/MailManager.php b/lib/Service/MailManager.php
@@ -519,16 +519,16 @@ public function tagMessageWithClient(Horde_Imap_Client_Socket $client, Account $
 					$e
 				);
 			}
-		if ($value) {
-			foreach ($messages as $message) {
-				$this->tagMapper->tagMessage($tag, $message->getMessageId(), $account->getUserId());
-			}
-		} else {
-			foreach ($messages as $message) {
-				$this->tagMapper->untagMessage($tag, $message->getMessageId());
+			if ($value) {
+				foreach ($messages as $message) {
+					$this->tagMapper->tagMessage($tag, $message->getMessageId(), $account->getUserId());
+				}
+			} else {
+				foreach ($messages as $message) {
+					$this->tagMapper->untagMessage($tag, $message->getMessageId());
+				}
 			}
 		}
-		}
 	}
 	/**
 	 * Tag (flag) a message on IMAP
@@ -873,7 +873,7 @@ public function deleteTagForAccount(int $id, string $userId, Tag $tag, Account $
 			$messageTags = $this->messageTagsMapper->getMessagesByTag($id);
 			$messages = array_merge(... array_map(function ($messageTag) use ($account) {
 				return $this->getByMessageId($account, $messageTag->getImapMessageId());
-			}, [...$messageTags]));
+			},array_values($messageTags)));
 		} catch (DoesNotExistException $e) {
 			throw new ClientException('Messages not found', 0, $e);
 		}
@@ -892,12 +892,15 @@ public function deleteTagForAccount(int $id, string $userId, Tag $tag, Account $
 				$groupedMessages[$mailboxId] = [$message];
 			}
 		}
-
-		foreach ($groupedMessages as $mailboxId => $messages) {
-			$mailbox = $this->getMailbox($userId, $mailboxId);
-			$this->tagMessageWithClient($client, $account, $mailbox, $messages, $tag, false);
+		try {
+			foreach ($groupedMessages as $mailboxId => $messages) {
+				$mailbox = $this->getMailbox($userId, $mailboxId);
+				$this->tagMessageWithClient($client, $account, $mailbox, $messages, $tag, false);
+			}
+		} finally {
+			$client->logout();
 		}
-		$client->logout();
+
 	}
 	public function moveThread(Account $srcAccount, Mailbox $srcMailbox, Account $dstAccount, Mailbox $dstMailbox, string $threadRootId): array {
 		$mailAccount = $srcAccount->getMailAccount();