60 lines
1.6 KiB
PHP
60 lines
1.6 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
/**
|
|
* SPDX-FileCopyrightText: 2020 F7cloud GmbH and F7cloud contributors
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
namespace OCA\Mail\Service\Classification\FeatureExtraction;
|
|
|
|
use OCA\Mail\Account;
|
|
use OCA\Mail\Db\Message;
|
|
use OCA\Mail\Db\StatisticsDao;
|
|
use RuntimeException;
|
|
use function array_map;
|
|
use function array_unique;
|
|
|
|
class SentMessagesExtractor implements IExtractor {
|
|
/** @var StatisticsDao */
|
|
private $statisticsDao;
|
|
|
|
/** @var int */
|
|
private $messagesSentTotal = 0;
|
|
|
|
/** @var int[] */
|
|
private $messagesSent;
|
|
|
|
public function __construct(StatisticsDao $statisticsDao) {
|
|
$this->statisticsDao = $statisticsDao;
|
|
}
|
|
|
|
#[\Override]
|
|
public function prepare(Account $account,
|
|
array $incomingMailboxes,
|
|
array $outgoingMailboxes,
|
|
array $messages): void {
|
|
$senders = array_unique(array_map(static fn (Message $message) => $message->getFrom()->first()->getEmail(), array_filter($messages, static fn (Message $message) => $message->getFrom()->first() !== null && $message->getFrom()->first()->getEmail() !== null)));
|
|
|
|
$this->messagesSentTotal = $this->statisticsDao->getMessagesTotal(...$outgoingMailboxes);
|
|
$this->messagesSent = $this->statisticsDao->getMessagesSentToGrouped($outgoingMailboxes, $senders);
|
|
}
|
|
|
|
#[\Override]
|
|
public function extract(Message $message): array {
|
|
$sender = $message->getFrom()->first();
|
|
if ($sender === null) {
|
|
throw new RuntimeException('This should not happen');
|
|
}
|
|
$email = $sender->getEmail();
|
|
|
|
if (($messagesSentTotal = $this->messagesSentTotal) === 0) {
|
|
// Prevent div by 0
|
|
return [0];
|
|
}
|
|
|
|
return [($this->messagesSent[$email] ?? 0) / $messagesSentTotal];
|
|
}
|
|
}
|