<?php

/**

 * Zend Framework

 *

 * LICENSE

 *

 * This source file is subject to the new BSD license that is bundled

 * with this package in the file LICENSE.txt.

 * It is also available through the world-wide-web at this URL:

 * http://framework.zend.com/license/new-bsd

 * If you did not receive a copy of the license and are unable to

 * obtain it through the world-wide-web, please send an email

 * to license@zend.com so we can send you a copy immediately.

 *

 * @category   Zend

 * @package    Zend_Mail

 * @subpackage Storage

 * @copyright  Copyright (c) 2005-2010 Zend Technologies USA Inc. (http://www.zend.com)

 * @license    http://framework.zend.com/license/new-bsd     New BSD License

 * @version    $Id: Imap.php 20096 2010-01-06 02:05:09Z bkarwin $

 */

/**

 * @see Zend_Mail_Storage_Abstract

 */

require_once 'Zend/Mail/Storage/Abstract.php';

/**

 * @see Zend_Mail_Protocol_Imap

 */

require_once 'Zend/Mail/Protocol/Imap.php';

/**

 * @see Zend_Mail_Storage_Writable_Interface

 */

require_once 'Zend/Mail/Storage/Writable/Interface.php';

/**

 * @see Zend_Mail_Storage_Folder_Interface

 */

require_once 'Zend/Mail/Storage/Folder/Interface.php';

/**

 * @see Zend_Mail_Storage_Folder

 */

require_once 'Zend/Mail/Storage/Folder.php';

/**

 * @see Zend_Mail_Message

 */

require_once 'Zend/Mail/Message.php';

/**

 * @see Zend_Mail_Storage

 */

require_once 'Zend/Mail/Storage.php';

/**

 * @category   Zend

 * @package    Zend_Mail

 * @subpackage Storage

 * @copyright  Copyright (c) 2005-2010 Zend Technologies USA Inc. (http://www.zend.com)

 * @license    http://framework.zend.com/license/new-bsd     New BSD License

 */

class Zend_Mail_Storage_Imap extends Zend_Mail_Storage_Abstract

                             implements Zend_Mail_Storage_Folder_Interface, Zend_Mail_Storage_Writable_Interface

{

    // TODO: with an internal cache we could optimize this class, or create an extra class with

    // such optimizations. Especially the various fetch calls could be combined to one cache call

    /**

     * protocol handler

     * @var null|Zend_Mail_Protocol_Imap

     */

    protected $_protocol;

    /**

     * name of current folder

     * @var string

     */

    protected $_currentFolder = '';

    /**

     * imap flags to constants translation

     * @var array

     */

    protected static $_knownFlags = array('\Passed'   => Zend_Mail_Storage::FLAG_PASSED,

                                          '\Answered' => Zend_Mail_Storage::FLAG_ANSWERED,

                                          '\Seen'     => Zend_Mail_Storage::FLAG_SEEN,

                                          '\Deleted'  => Zend_Mail_Storage::FLAG_DELETED,

                                          '\Draft'    => Zend_Mail_Storage::FLAG_DRAFT,

                                          '\Flagged'  => Zend_Mail_Storage::FLAG_FLAGGED);

    /**

     * map flags to search criterias

     * @var array

     */

    protected static $_searchFlags = array('\Recent'   => 'RECENT',

                                           '\Answered' => 'ANSWERED',

                                           '\Seen'     => 'SEEN',

                                           '\Deleted'  => 'DELETED',

                                           '\Draft'    => 'DRAFT',

                                           '\Flagged'  => 'FLAGGED');

    /**

     * Count messages all messages in current box

     *

     * @return int number of messages

     * @throws Zend_Mail_Storage_Exception

     * @throws Zend_Mail_Protocol_Exception

     */

    public function countMessages($flags = null)

    {

        if (!$this->_currentFolder) {

            /**

             * @see Zend_Mail_Storage_Exception

             */

            require_once 'Zend/Mail/Storage/Exception.php';

            throw new Zend_Mail_Storage_Exception('No selected folder to count');

        }

        if ($flags === null) {

            return count($this->_protocol->search(array('ALL')));

        }

        $params = array();

        foreach ((array)$flags as $flag) {

            if (isset(self::$_searchFlags[$flag])) {

                $params[] = self::$_searchFlags[$flag];

            } else {

                $params[] = 'KEYWORD';

                $params[] = $this->_protocol->escapeString($flag);

            }

        }

        return count($this->_protocol->search($params));

    }

    /**

     * get a list of messages with number and size

     *

     * @param int $id number of message

     * @return int|array size of given message of list with all messages as array(num => size)

     * @throws Zend_Mail_Protocol_Exception

     */

    public function getSize($id = 0)

    {

        if ($id) {

            return $this->_protocol->fetch('RFC822.SIZE', $id);

        }

        return $this->_protocol->fetch('RFC822.SIZE', 1, INF);

    }

    /**

     * Fetch a message

     *

     * @param int $id number of message

     * @return Zend_Mail_Message

     * @throws Zend_Mail_Protocol_Exception

     */

    public function getMessage($id)

    {

        $data = $this->_protocol->fetch(array('FLAGS', 'RFC822.HEADER'), $id);

        $header = $data['RFC822.HEADER'];

        $flags = array();

        foreach ($data['FLAGS'] as $flag) {

            $flags[] = isset(self::$_knownFlags[$flag]) ? self::$_knownFlags[$flag] : $flag;

        }

        return new $this->_messageClass(array('handler' => $this, 'id' => $id, 'headers' => $header, 'flags' => $flags));

    }

    /*

     * Get raw header of message or part

     *

     * @param  int               $id       number of message

     * @param  null|array|string $part     path to part or null for messsage header

     * @param  int               $topLines include this many lines with header (after an empty line)

     * @param  int $topLines include this many lines with header (after an empty line)

     * @return string raw header

     * @throws Zend_Mail_Protocol_Exception

     * @throws Zend_Mail_Storage_Exception

     */

    public function getRawHeader($id, $part = null, $topLines = 0)

    {

        if ($part !== null) {

            // TODO: implement

            /**

             * @see Zend_Mail_Storage_Exception

             */

            require_once 'Zend/Mail/Storage/Exception.php';

            throw new Zend_Mail_Storage_Exception('not implemented');

        }

        // TODO: toplines

        return $this->_protocol->fetch('RFC822.HEADER', $id);

    }

    /*

     * Get raw content of message or part

     *

     * @param  int               $id   number of message

     * @param  null|array|string $part path to part or null for messsage content

     * @return string raw content

     * @throws Zend_Mail_Protocol_Exception

     * @throws Zend_Mail_Storage_Exception

     */

    public function getRawContent($id, $part = null)

    {

        if ($part !== null) {

            // TODO: implement

            /**

             * @see Zend_Mail_Storage_Exception

             */

            require_once 'Zend/Mail/Storage/Exception.php';

            throw new Zend_Mail_Storage_Exception('not implemented');

        }

        return $this->_protocol->fetch('RFC822.TEXT', $id);

    }

    /**

     * create instance with parameters

     * Supported paramters are

     *   - user username

     *   - host hostname or ip address of IMAP server [optional, default = 'localhost']

     *   - password password for user 'username' [optional, default = '']

     *   - port port for IMAP server [optional, default = 110]

     *   - ssl 'SSL' or 'TLS' for secure sockets

     *   - folder select this folder [optional, default = 'INBOX']

     *

     * @param  array $params mail reader specific parameters

     * @throws Zend_Mail_Storage_Exception

     * @throws Zend_Mail_Protocol_Exception

     */

    public function __construct($params)

    {

        if (is_array($params)) {

            $params = (object)$params;

        }

        $this->_has['flags'] = true;

        if ($params instanceof Zend_Mail_Protocol_Imap) {

            $this->_protocol = $params;

            try {

                $this->selectFolder('INBOX');

            } catch(Zend_Mail_Storage_Exception $e) {

                /**

                 * @see Zend_Mail_Storage_Exception

                 */

                require_once 'Zend/Mail/Storage/Exception.php';

                throw new Zend_Mail_Storage_Exception('cannot select INBOX, is this a valid transport?', 0, $e);

            }

            return;

        }

        if (!isset($params->user)) {

            /**

             * @see Zend_Mail_Storage_Exception

             */

            require_once 'Zend/Mail/Storage/Exception.php';

            throw new Zend_Mail_Storage_Exception('need at least user in params');

        }

        $host     = isset($params->host)     ? $params->host     : 'localhost';

        $password = isset($params->password) ? $params->password : '';

        $port     = isset($params->port)     ? $params->port     : null;

        $ssl      = isset($params->ssl)      ? $params->ssl      : false;

        $this->_protocol = new Zend_Mail_Protocol_Imap();

        $this->_protocol->connect($host, $port, $ssl);

        if (!$this->_protocol->login($params->user, $password)) {

            /**

             * @see Zend_Mail_Storage_Exception

             */

            require_once 'Zend/Mail/Storage/Exception.php';

            throw new Zend_Mail_Storage_Exception('cannot login, user or password wrong');

        }

        $this->selectFolder(isset($params->folder) ? $params->folder : 'INBOX');

    }

    /**

     * Close resource for mail lib. If you need to control, when the resource

     * is closed. Otherwise the destructor would call this.

     *

     * @return null

     */

    public function close()

    {

        $this->_currentFolder = '';

        $this->_protocol->logout();

    }

    /**

     * Keep the server busy.

     *

     * @return null

     * @throws Zend_Mail_Storage_Exception

     */

    public function noop()

    {

        if (!$this->_protocol->noop()) {

            /**

             * @see Zend_Mail_Storage_Exception

             */

            require_once 'Zend/Mail/Storage/Exception.php';

            throw new Zend_Mail_Storage_Exception('could not do nothing');

        }

    }

    /**

     * Remove a message from server. If you're doing that from a web enviroment

     * you should be careful and use a uniqueid as parameter if possible to

     * identify the message.

     *

     * @param   int $id number of message

     * @return  null

     * @throws  Zend_Mail_Storage_Exception

     */

    public function removeMessage($id)

    {

        if (!$this->_protocol->store(array(Zend_Mail_Storage::FLAG_DELETED), $id, null, '+')) {

            /**

             * @see Zend_Mail_Storage_Exception

             */

            require_once 'Zend/Mail/Storage/Exception.php';

            throw new Zend_Mail_Storage_Exception('cannot set deleted flag');

        }

        // TODO: expunge here or at close? we can handle an error here better and are more fail safe

        if (!$this->_protocol->expunge()) {

            /**

             * @see Zend_Mail_Storage_Exception

             */

            require_once 'Zend/Mail/Storage/Exception.php';

            throw new Zend_Mail_Storage_Exception('message marked as deleted, but could not expunge');

        }

    }

    /**

     * get unique id for one or all messages

     *

     * if storage does not support unique ids it's the same as the message number

     *

     * @param int|null $id message number

     * @return array|string message number for given message or all messages as array

     * @throws Zend_Mail_Storage_Exception

     */

    public function getUniqueId($id = null)

    {

        if ($id) {

            return $this->_protocol->fetch('UID', $id);

        }

        return $this->_protocol->fetch('UID', 1, INF);

    }

    /**

     * get a message number from a unique id

     *

     * I.e. if you have a webmailer that supports deleting messages you should use unique ids

     * as parameter and use this method to translate it to message number right before calling removeMessage()

     *

     * @param string $id unique id

     * @return int message number

     * @throws Zend_Mail_Storage_Exception

     */

    public function getNumberByUniqueId($id)

    {

        // TODO: use search to find number directly

        $ids = $this->getUniqueId();

        foreach ($ids as $k => $v) {

            if ($v == $id) {

                return $k;

            }

        }

        /**

         * @see Zend_Mail_Storage_Exception

         */

        require_once 'Zend/Mail/Storage/Exception.php';

        throw new Zend_Mail_Storage_Exception('unique id not found');

    }

    /**

     * get root folder or given folder

     *

     * @param  string $rootFolder get folder structure for given folder, else root

     * @return Zend_Mail_Storage_Folder root or wanted folder

     * @throws Zend_Mail_Storage_Exception

     * @throws Zend_Mail_Protocol_Exception

     */

    public function getFolders($rootFolder = null)

    {

        $folders = $this->_protocol->listMailbox((string)$rootFolder);

        if (!$folders) {

            /**

             * @see Zend_Mail_Storage_Exception

             */

            require_once 'Zend/Mail/Storage/Exception.php';

            throw new Zend_Mail_Storage_Exception('folder not found');

        }

        ksort($folders, SORT_STRING);

        $root = new Zend_Mail_Storage_Folder('/', '/', false);

        $stack = array(null);

        $folderStack = array(null);

        $parentFolder = $root;

        $parent = '';

        foreach ($folders as $globalName => $data) {

            do {

                if (!$parent || strpos($globalName, $parent) === 0) {

                    $pos = strrpos($globalName, $data['delim']);

                    if ($pos === false) {

                        $localName = $globalName;

                    } else {

                        $localName = substr($globalName, $pos + 1);

                    }

                    $selectable = !$data['flags'] || !in_array('\\Noselect', $data['flags']);

                    array_push($stack, $parent);

                    $parent = $globalName . $data['delim'];

                    $folder = new Zend_Mail_Storage_Folder($localName, $globalName, $selectable);

                    $parentFolder->$localName = $folder;

                    array_push($folderStack, $parentFolder);

                    $parentFolder = $folder;

                    break;

                } else if ($stack) {

                    $parent = array_pop($stack);

                    $parentFolder = array_pop($folderStack);

                }

            } while ($stack);

            if (!$stack) {

                /**

                 * @see Zend_Mail_Storage_Exception

                 */

                require_once 'Zend/Mail/Storage/Exception.php';

                throw new Zend_Mail_Storage_Exception('error while constructing folder tree');

            }