<?php

/*

 * This file is part of SwiftMailer.

 * (c) 2009 Fabien Potencier

 *

 * For the full copyright and license information, please view the LICENSE

 * file that was distributed with this source code.

 */

/**

 * Replaces the sender of a message.

 *

 * @package Swift

 * @subpackage Plugins

 * @author Arjen Brouwer

 */

class Swift_Plugins_ImpersonatePlugin implements \Swift_Events_SendListener {

    /**

     * The sender to impersonate.

     * 

     * @var String

     * @access private

     */

    private $_sender;

    /**

     * Create a new ImpersonatePlugin to impersonate $sender.

     * 

     * @param string $sender address

     */

    public function __construct($sender) {

        $this->_sender = $sender;

    }

    /**

     * Invoked immediately before the Message is sent.

     * 

     * @param Swift_Events_SendEvent $evt

     */

    public function beforeSendPerformed(Swift_Events_SendEvent $evt) {

        $message = $evt->getMessage();

        $headers = $message->getHeaders();

        // save current recipients

        $headers->addPathHeader('X-Swift-Return-Path', $message->getReturnPath());

        // replace them with the one to send to

        $message->setReturnPath($this->_sender);

    }

    /**

     * Invoked immediately after the Message is sent.

     * 

     * @param Swift_Events_SendEvent $evt

     */

    public function sendPerformed(Swift_Events_SendEvent $evt) {

        $message = $evt->getMessage();

        // restore original headers

        $headers = $message->getHeaders();

        if ($headers->has('X-Swift-Return-Path')) {

            $message->setReturnPath($headers->get('X-Swift-Return-Path')->getAddress());

            $headers->removeAll('X-Swift-Return-Path');

        }

    }

}