mirror of
https://github.com/nextcloud/mail.git
synced 2026-01-13 20:23:59 +00:00
96 lines
2.8 KiB
PHP
96 lines
2.8 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
/**
|
|
* SPDX-FileCopyrightText: 2022 Nextcloud GmbH and Nextcloud contributors
|
|
* SPDX-License-Identifier: AGPL-3.0-or-later
|
|
*/
|
|
|
|
namespace OCA\Mail\Command;
|
|
|
|
use OCA\Mail\AddressList;
|
|
use OCA\Mail\Db\Message;
|
|
use OCA\Mail\Service\AccountService;
|
|
use OCA\Mail\Service\Classification\ImportanceClassifier;
|
|
use OCA\Mail\Support\ConsoleLoggerDecorator;
|
|
use OCP\AppFramework\Db\DoesNotExistException;
|
|
use OCP\IConfig;
|
|
use Psr\Log\LoggerInterface;
|
|
use Symfony\Component\Console\Command\Command;
|
|
use Symfony\Component\Console\Input\InputArgument;
|
|
use Symfony\Component\Console\Input\InputInterface;
|
|
use Symfony\Component\Console\Output\OutputInterface;
|
|
use function memory_get_peak_usage;
|
|
|
|
final class PredictImportance extends Command {
|
|
public const ARGUMENT_ACCOUNT_ID = 'account-id';
|
|
public const ARGUMENT_SENDER = 'sender';
|
|
public const ARGUMENT_SUBJECT = 'subject';
|
|
|
|
private AccountService $accountService;
|
|
private ImportanceClassifier $classifier;
|
|
private IConfig $config;
|
|
private LoggerInterface $logger;
|
|
|
|
public function __construct(AccountService $service,
|
|
ImportanceClassifier $classifier,
|
|
IConfig $config,
|
|
LoggerInterface $logger) {
|
|
parent::__construct();
|
|
|
|
$this->accountService = $service;
|
|
$this->classifier = $classifier;
|
|
$this->logger = $logger;
|
|
$this->config = $config;
|
|
}
|
|
|
|
protected function configure(): void {
|
|
$this->setName('mail:predict-importance');
|
|
$this->setDescription('Predict importance of an incoming message');
|
|
$this->addArgument(self::ARGUMENT_ACCOUNT_ID, InputArgument::REQUIRED);
|
|
$this->addArgument(self::ARGUMENT_SENDER, InputArgument::REQUIRED);
|
|
$this->addArgument(self::ARGUMENT_SUBJECT, InputArgument::OPTIONAL);
|
|
}
|
|
|
|
public function isEnabled(): bool {
|
|
return $this->config->getSystemValueBool('debug');
|
|
}
|
|
|
|
protected function execute(InputInterface $input, OutputInterface $output): int {
|
|
$accountId = (int)$input->getArgument(self::ARGUMENT_ACCOUNT_ID);
|
|
$sender = $input->getArgument(self::ARGUMENT_SENDER);
|
|
$subject = $input->getArgument(self::ARGUMENT_SUBJECT) ?? '';
|
|
|
|
$consoleLogger = new ConsoleLoggerDecorator(
|
|
$this->logger,
|
|
$output
|
|
);
|
|
|
|
try {
|
|
$account = $this->accountService->findById($accountId);
|
|
} catch (DoesNotExistException $e) {
|
|
$output->writeln("<error>account $accountId does not exist</error>");
|
|
return 1;
|
|
}
|
|
$fakeMessage = new Message();
|
|
$fakeMessage->setUid(0);
|
|
$fakeMessage->setFrom(AddressList::parse("Name <$sender>"));
|
|
$fakeMessage->setSubject($subject);
|
|
[$prediction] = $this->classifier->classifyImportance(
|
|
$account,
|
|
[$fakeMessage],
|
|
$consoleLogger
|
|
);
|
|
if ($prediction) {
|
|
$output->writeln('Message is important');
|
|
} else {
|
|
$output->writeln('Message is not important');
|
|
}
|
|
|
|
$mbs = (int)(memory_get_peak_usage() / 1024 / 1024);
|
|
$output->writeln('<info>' . $mbs . 'MB of memory used</info>');
|
|
return 0;
|
|
}
|
|
}
|