<?php

namespace App\Mail\Transport;

use Aws\Ses\SesClient;
use Illuminate\Mail\Transport\Transport;
use Swift_Mime_SimpleMessage;

class SesTransportWithConfigSet extends Transport
{
    /**
     * The Amazon SES instance.
     *
     * @var \Aws\Ses\SesClient
     */
    protected $ses;

    /**
     * The SES Configuration Set name.
     *
     * @var string|null
     */
    protected $configurationSetName;

    /**
     * Create a new SES transport instance.
     *
     * @param  \Aws\Ses\SesClient  $ses
     * @param  string|null  $configurationSetName
     * @return void
     */
    public function __construct(SesClient $ses, $configurationSetName = null)
    {
        $this->ses = $ses;
        $this->configurationSetName = $configurationSetName;
    }

    /**
     * {@inheritdoc}
     */
    public function send(Swift_Mime_SimpleMessage $message, &$failedRecipients = null)
    {
        $this->beforeSendPerformed($message);

        $payload = [
            'Source'     => key($message->getSender() ?: $message->getFrom()),
            'RawMessage' => [
                'Data' => $message->toString(),
            ],
        ];

        // Attach the Configuration Set if configured
        if ($this->configurationSetName) {
            $payload['ConfigurationSetName'] = $this->configurationSetName;
        }

        $message->getHeaders()->addTextHeader(
            'X-SES-Message-ID',
            $this->ses->sendRawEmail($payload)->get('MessageId')
        );

        $this->sendPerformed($message);

        return $this->numberOfRecipients($message);
    }
}