QueueableMemoryTransport.php

<?php

declare(strict_types=1);

/*
 * This file is part of the TYPO3 CMS extension "mailqueue".
 *
 * It is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License, either version 2
 * of the License, or any later version.
 *
 * For the full copyright and license information, please read the
 * LICENSE.txt file that was distributed with this source code.
 *
 * The TYPO3 project - inspiring people to share!
 */

namespace CPSIT\Typo3Mailqueue\Mail\Transport;

use CPSIT\Typo3Mailqueue\Enums;
use CPSIT\Typo3Mailqueue\Mail;
use Symfony\Component\Mailer;
use Symfony\Component\Mime;
use TYPO3\CMS\Core;

/**
 * QueueableMemoryTransport
 *
 * @author Elias Häußler <e.haeussler@familie-redlich.de>
 * @license GPL-2.0-or-later
 */
final class QueueableMemoryTransport extends Core\Mail\MemorySpool implements QueueableTransport
{
    public function enqueue(Mime\RawMessage $message, ?Mailer\Envelope $envelope = null): ?Mail\Queue\MailQueueItem
    {
        $sentMessage = $this->send($message, $envelope);
        $id = array_key_last($this->queuedMessages);

        // Early return if message was rejected
        if ($sentMessage === null || $id === null) {
            return null;
        }

        return new Mail\Queue\MailQueueItem(
            (string)$id,
            $sentMessage,
            Enums\MailState::Queued,
            new \DateTimeImmutable(),
        );
    }

    public function dequeue(Mail\Queue\MailQueueItem $item, Mailer\Transport\TransportInterface $transport): bool
    {
        $transport->send($item->message->getMessage(), $item->message->getEnvelope());

        unset($this->queuedMessages[$item->id]);

        return true;
    }

    public function delete(Mail\Queue\MailQueueItem $item): bool
    {
        if (!isset($this->queuedMessages[$item->id])) {
            return false;
        }

        unset($this->queuedMessages[$item->id]);

        return true;
    }

    public function getMailQueue(): Mail\Queue\MailQueue
    {
        return new Mail\Queue\MailQueue(
            $this->initializeQueue(...),
        );
    }

    /**
     * @return \Generator<Mail\Queue\MailQueueItem>
     */
    private function initializeQueue(): \Generator
    {
        foreach ($this->queuedMessages as $key => $queuedMessage) {
            yield new Mail\Queue\MailQueueItem(
                $key,
                $queuedMessage,
                Enums\MailState::Queued,
            );
        }
    }
}