Save important flag to IMAP if permflags enabled

Signed-off-by: Anna Larch <anna@nextcloud.com>

1 files changed, 6 insertions, 1 deletions

diff --git a/lib/Listener/NewMessageClassificationListener.php b/lib/Listener/NewMessageClassificationListener.php
index 365251883..56585730c 100644
--- a/lib/Listener/NewMessageClassificationListener.php
+++ b/lib/Listener/NewMessageClassificationListener.php
@@ -67,11 +67,16 @@ class NewMessageClassificationListener implements IEventListener {
 			}
 		}
 
+		// if the message is already flagged as important, we won't classify it again.
+		$messages = array_filter($event->getMessages(), function ($message) {
+			return ($message->getFlagImportant() === false);
+		});
+
 		try {
 			$predictions = $this->classifier->classifyImportance(
 				$event->getAccount(),
 				$event->getMailbox(),
-				$event->getMessages()
+				$messages
 			);
 
 			foreach ($event->getMessages() as $message) {