Add classifier for new messages

Signed-off-by: Christoph Wurst <christoph@winzerhof-wurst.at>

css/mail.scss

@@ -331,9 +331,6 @@
 .icon-drafts {
 	@include icon-color('drafts', 'mail', $color-black);
 }
-.icon-important {
-	@include icon-color('important', 'mail', $color-black);
-}
 .icon-sent {
 	@include icon-color('sent', 'mail', $color-black);
 }

lib/AppInfo/BootstrapSingleton.php

@@ -36,6 +36,7 @@
 use OCA\Mail\Events\MessageDeletedEvent;
 use OCA\Mail\Events\MessageFlaggedEvent;
 use OCA\Mail\Events\MessageSentEvent;
+use OCA\Mail\Events\NewMessagesSynchronized;
 use OCA\Mail\Events\SaveDraftEvent;
 use OCA\Mail\Http\Middleware\ErrorMiddleware;
 use OCA\Mail\Http\Middleware\ProvisioningMiddleware;
@@ -45,6 +46,7 @@
 use OCA\Mail\Listener\FlagRepliedMessageListener;
 use OCA\Mail\Listener\InteractionListener;
 use OCA\Mail\Listener\MessageCacheUpdaterListener;
+use OCA\Mail\Listener\NewMessageClassificationListener;
 use OCA\Mail\Listener\SaveSentMessageListener;
 use OCA\Mail\Listener\TrashMailboxCreatorListener;
 use OCA\Mail\Service\Attachment\AttachmentService;
@@ -135,6 +137,7 @@ private function registerEvents(IAppContainer $container): void {
 		$dispatcher->addServiceListener(MessageSentEvent::class, FlagRepliedMessageListener::class);
 		$dispatcher->addServiceListener(MessageSentEvent::class, InteractionListener::class);
 		$dispatcher->addServiceListener(MessageSentEvent::class, SaveSentMessageListener::class);
+		$dispatcher->addServiceListener(NewMessagesSynchronized::class, NewMessageClassificationListener::class);
 		$dispatcher->addServiceListener(SaveDraftEvent::class, DraftMailboxCreatorListener::class);
 	}
 }

lib/Db/Message.php

@@ -97,7 +97,7 @@ class Message extends Entity implements JsonSerializable {
 	protected $updatedAt;
 	protected $structureAnalyzed;
 	protected $flagAttachments;
-	protected $flagImportant;
+	protected $flagImportant = false;
 	protected $previewText;
 
 	/** @var AddressList */

lib/Db/MessageMapper.php

@@ -104,6 +104,7 @@ public function insertBulk(Message ...$messages): void {
 		$qb1->setValue('flag_forwarded', $qb1->createParameter('flag_forwarded'));
 		$qb1->setValue('flag_junk', $qb1->createParameter('flag_junk'));
 		$qb1->setValue('flag_notjunk', $qb1->createParameter('flag_notjunk'));
+		$qb1->setValue('flag_important', $qb1->createParameter('flag_important'));
 		$qb2 = $this->db->getQueryBuilder();
 
 		$qb2->insert('mail_recipients')
@@ -126,6 +127,7 @@ public function insertBulk(Message ...$messages): void {
 			$qb1->setParameter('flag_forwarded', $message->getFlagForwarded(), IQueryBuilder::PARAM_BOOL);
 			$qb1->setParameter('flag_junk', $message->getFlagJunk(), IQueryBuilder::PARAM_BOOL);
 			$qb1->setParameter('flag_notjunk', $message->getFlagNotjunk(), IQueryBuilder::PARAM_BOOL);
+			$qb1->setParameter('flag_important', $message->getFlagImportant(), IQueryBuilder::PARAM_BOOL);
 
 			$qb1->execute();
 

lib/Events/NewMessagesSynchronized.php

@@ -0,0 +1,73 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * @copyright 2020 Christoph Wurst <christoph@winzerhof-wurst.at>
+ *
+ * @author 2020 Christoph Wurst <christoph@winzerhof-wurst.at>
+ *
+ * @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\Mail\Events;
+
+use OCA\Mail\Account;
+use OCA\Mail\Db\Mailbox;
+use OCA\Mail\Db\Message;
+use OCP\EventDispatcher\Event;
+
+class NewMessagesSynchronized extends Event {
+
+	/** @var Account */
+	private $account;
+
+	/** @var Mailbox */
+	private $mailbox;
+
+	/** @var array|Message[] */
+	private $messages;
+
+	/**
+	 * @param Account $account
+	 * @param Mailbox $mailbox
+	 * @param Message[] $messages
+	 */
+	public function __construct(Account $account,
+								Mailbox $mailbox,
+								array $messages) {
+		parent::__construct();
+		$this->account = $account;
+		$this->mailbox = $mailbox;
+		$this->messages = $messages;
+	}
+
+	public function getAccount(): Account {
+		return $this->account;
+	}
+
+	public function getMailbox(): Mailbox {
+		return $this->mailbox;
+	}
+
+	/**
+	 * @return Message[]
+	 */
+	public function getMessages() {
+		return $this->messages;
+	}
+
+}

lib/Listener/NewMessageClassificationListener.php

@@ -0,0 +1,54 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * @copyright 2020 Christoph Wurst <christoph@winzerhof-wurst.at>
+ *
+ * @author 2020 Christoph Wurst <christoph@winzerhof-wurst.at>
+ *
+ * @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\Mail\Listener;
+
+use OCA\Mail\Events\NewMessagesSynchronized;
+use OCA\Mail\Service\Classification\MessageClassifier;
+use OCP\EventDispatcher\Event;
+use OCP\EventDispatcher\IEventListener;
+
+class NewMessageClassificationListener implements IEventListener {
+
+	/** @var MessageClassifier */
+	private $classifier;
+
+	public function __construct(MessageClassifier $classifier) {
+		$this->classifier = $classifier;
+	}
+
+	public function handle(Event $event): void {
+		if (!($event instanceof NewMessagesSynchronized)) {
+			return;
+		}
+
+		foreach ($event->getMessages() as $message) {
+			if ($this->classifier->isImportant($event->getAccount(), $event->getMailbox(), $message)) {
+				$message->setFlagImportant(true);
+			}
+		}
+	}
+
+}

lib/Service/Classification/AClassifier.php

@@ -0,0 +1,55 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * @copyright 2020 Christoph Wurst <christoph@winzerhof-wurst.at>
+ *
+ * @author 2020 Christoph Wurst <christoph@winzerhof-wurst.at>
+ *
+ * @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\Mail\Service\Classification;
+
+use OCA\Mail\Account;
+use OCA\Mail\Db\Mailbox;
+use OCA\Mail\Db\Message;
+
+abstract class AClassifier {
+
+	public abstract function isImportant(Account $account, Mailbox $mailbox, Message $message): bool;
+
+	public final function or(AClassifier $next): AClassifier {
+		return new class($this, $next) extends AClassifier {
+			/** @var AClassifier */
+			private $outer;
+
+			/** @var AClassifier */
+			private $next;
+
+			public function __construct(AClassifier $outer, AClassifier $next) {
+				$this->outer = $outer;
+				$this->next = $next;
+			}
+
+			public function isImportant(Account $account, Mailbox $mailbox, Message $message): bool {
+				return $this->outer->isImportant($account, $mailbox, $message) || $this->next->isImportant($account, $mailbox, $message);
+			}
+		};
+	}
+
+}

lib/Service/Classification/MessageClassifier.php

@@ -0,0 +1,66 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * @copyright 2020 Christoph Wurst <christoph@winzerhof-wurst.at>
+ *
+ * @author 2020 Christoph Wurst <christoph@winzerhof-wurst.at>
+ *
+ * @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\Mail\Service\Classification;
+
+use OCA\Mail\Account;
+use OCA\Mail\Db\Mailbox;
+use OCA\Mail\Db\Message;
+
+class MessageClassifier {
+
+	/** @var AClassifier */
+	private $oftenImportantSenderClassifier;
+
+	/** @var AClassifier */
+	private $oftenContactedSenderClassifier;
+
+	/** @var AClassifier */
+	private $oftenReadSenderClassifier;
+
+	/** @var AClassifier */
+	private $oftenRepliedSenderClassifier;
+
+	public function __construct(OftenImportantSenderClassifier $oftenImportantSenderClassifier,
+								OftenContactedSenderClassifier $oftenContactedSenderClassifier,
+								OftenReadSenderClassifier $oftenReadSenderClassifier,
+								OftenRepliedSenderClassifier $oftenRepliedSenderClassifier) {
+		$this->oftenImportantSenderClassifier = $oftenImportantSenderClassifier;
+		$this->oftenContactedSenderClassifier = $oftenContactedSenderClassifier;
+		$this->oftenReadSenderClassifier = $oftenReadSenderClassifier;
+		$this->oftenRepliedSenderClassifier = $oftenRepliedSenderClassifier;
+	}
+
+	public function isImportant(Account $account,
+								Mailbox $mailbox,
+								Message $message): bool {
+		return $this->oftenImportantSenderClassifier
+			->or($this->oftenContactedSenderClassifier)
+			->or($this->oftenReadSenderClassifier)
+			->or($this->oftenRepliedSenderClassifier)
+			->isImportant($account, $mailbox, $message);
+	}
+
+}