ZBateson\MailMimeParser\PartStreamRegistry

Complexity

Total Complexity

13

Size/Duplication

Total Lines	116
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Importance

Changes	6
Bugs	0	Features	2

6 Methods

Rating	Name	Duplication	Size	Complexity
A	unregister()	0	4	1
A	register()	0	6	2
A	get()	0	7	2
A	attachEncodingFilterToStream()	0	17	4
A	attachCharsetFilterToStream()	0	10	2
A	attachPartStreamHandle()	0	14	2

<?php
/**
 * This file is part of the ZBateson\MailMimeParser project.
 *
 * @license http://opensource.org/licenses/bsd-license.php BSD
 */
namespace ZBateson\MailMimeParser;

/**
 * Factory class for PartStream objects and registration class for Message
 * handles.
 * 
 * PartStreamRegistry is used for \ZBateson\MailMimeParser\MessageParser to
 * register Message stream handles for opening with PartStreams, and to open
 * file handles for specific mime parts of a message.  The PartStreamRegistry
 * maintains a list of opened resources, closing them either when unregistering
 * a Message or on destruction.
 *
 * @author Zaahid Bateson
 */
class PartStreamRegistry
{
    /**
     * @var array Array of handles, with message IDs as keys.
     */
    private $registeredHandles;

    /**
     * @var array Array of PartStream handles with message IDs as keys.
     */
    private $registeredPartStreamHandles;

    /**
     * Registers an ID for the passed resource handle.
     * 
     * @param string $id
     * @param resource $handle
     */
    public function register($id, $handle)
    {
        if (!isset($this->registeredHandles[$id])) {
            $this->registeredHandles[$id] = $handle;
        }
    }

    /**
     * Unregisters the given message ID.
     * 
     * @param string $id
     */
    public function unregister($id)
    {
        unset($this->registeredHandles[$id], $this->registeredPartStreamHandles);
    }

    /**
     * Returns the resource handle with the passed $id.
     * 
     * @param string $id
     * @return resource
     */
    public function get($id)
    {
        if (!isset($this->registeredHandles[$id])) {
            return null;
        }
        return $this->registeredHandles[$id];
    }
    
    /**
     * Attaches a stream filter on the passed resource $handle for the part's
     * encoding.
     * 
     * @param \ZBateson\MailMimeParser\MimePart $part
     * @param resource $handle
     */
    private function attachEncodingFilterToStream(MimePart $part, $handle)
    {
        $encoding = strtolower($part->getHeaderValue('Content-Transfer-Encoding'));
        switch ($encoding) {
            case 'quoted-printable':
                stream_filter_append($handle, 'convert.quoted-printable-decode', STREAM_FILTER_READ);
                break;
            case 'base64':
                stream_filter_append($handle, 'convert.base64-decode', STREAM_FILTER_READ);
                break;
            case 'x-uuencode':
                stream_filter_append($handle, 'mailmimeparser-uudecode', STREAM_FILTER_READ);
                break;
            default:
                break;
        }
    }
    
    /**
     * Attaches a mailmimeparser-encode stream filter based on the part's
     * defined charset.
     * 
     * @param \ZBateson\MailMimeParser\MimePart $part
     * @param resource $handle
     */
    private function attachCharsetFilterToStream(MimePart $part, $handle)
    {
        $contentType = strtolower($part->getHeaderValue('Content-Type', 'text/plain'));
        if (strpos($contentType, 'text/') === 0) {
            stream_filter_append(
                $handle,
                'mailmimeparser-encode.' . $part->getHeaderParameter('Content-Type', 'charset')
            );
        }
    }

    /**
     * Creates a part stream handle for the start and end position of the
     * message stream, and attaches it to the passed MimePart.
     * 
     * @param MimePart $part
     * @param Message $message
     * @param int $start
     * @param int $end
     */
    public function attachPartStreamHandle(MimePart $part, Message $message, $start, $end)
    {
        $id = $message->getObjectId();
        if (empty($this->registeredHandles[$id])) {
            return null;
        }
        $handle = fopen('mmp-mime-message://' . $id . '?start=' .
            $start . '&end=' . $end, 'r');
        
        $this->attachEncodingFilterToStream($part, $handle);
        $this->attachCharsetFilterToStream($part, $handle);
        $this->registeredPartStreamHandles[$id] = $handle;
        $part->attachContentResourceHandle($handle);
    }
}