<?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_Service

 * @subpackage Amazon_Sqs

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

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

 * @version    $Id: Sqs.php 25024 2012-07-30 15:08:15Z rob $

 */

/**

 * @see Zend_Service_Amazon_Abstract

 */

require_once 'Zend/Service/Amazon/Abstract.php';

/**

 * @see Zend_Crypt_Hmac

 */

require_once 'Zend/Crypt/Hmac.php';

/**

 * Class for connecting to the Amazon Simple Queue Service (SQS)

 *

 * @category   Zend

 * @package    Zend_Service

 * @subpackage Amazon_Sqs

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

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

 * @see        http://aws.amazon.com/sqs/ Amazon Simple Queue Service

 */

class Zend_Service_Amazon_Sqs extends Zend_Service_Amazon_Abstract

{

    /**

     * Default timeout for createQueue() function

     */

    const CREATE_TIMEOUT_DEFAULT = 30;

    /**

     * HTTP end point for the Amazon SQS service

     */

    protected $_sqsEndpoint = 'queue.amazonaws.com';

    /**

     * The API version to use

     */

    protected $_sqsApiVersion = '2009-02-01';

    /**

     * Signature Version

     */

    protected $_sqsSignatureVersion = '2';

    /**

     * Signature Encoding Method

     */

    protected $_sqsSignatureMethod = 'HmacSHA256';

    protected $_sqsEndpoints = array('us-east-1' => 'sqs.us-east-1.amazonaws.com',

                                     'us-west-1' => 'sqs.us-west-1.amazonaws.com',

                                     'eu-west-1' => 'sqs.eu-west-1.amazonaws.com',

                                     'ap-southeast-1' => 'sqs.ap-southeast-1.amazonaws.com',

                                     'ap-northeast-1' => 'sqs.ap-northeast-1.amazonaws.com');

    /**

     * Constructor

     *

     * The default region is us-east-1. Use the region to set it to one of the regions that is build-in into ZF.

     * To add a new AWS region use the setEndpoint() method.

     *

     * @param string $accessKey

     * @param string $secretKey

     * @param string $region

     */

    public function __construct($accessKey = null, $secretKey = null, $region = null)

    {

        parent::__construct($accessKey, $secretKey, $region);

        if (null !== $region) {

            $this->_setEndpoint($region);

        }

    }

    /**

     * Set SQS endpoint

     *

     * Checks and sets endpoint if region exists in $_sqsEndpoints. If a new SQS region is added by amazon,

     * please use the setEndpoint function to set it.

     *

     * @param  string  $region region

     * @throws Zend_Service_Amazon_Sqs_Exception

     */

    protected function _setEndpoint($region)

    {

        if (array_key_exists($region, $this->_sqsEndpoints)) {

            $this->_sqsEndpoint = $this->_sqsEndpoints[$region];

        } else {

            throw new Zend_Service_Amazon_Sqs_Exception('Invalid SQS region specified.');

        }

    }

    /**

     * Set SQS endpoint

     *

     * You can set SQS to on of the build-in regions. If the region does not exsist it will be added.

     *

     * @param  string  $region region

     * @throws Zend_Service_Amazon_Sqs_Exception

     */

    public function setEndpoint($region)

    {

        if (!empty($region)) {

            if (array_key_exists($region, $this->_sqsEndpoints)) {

                $this->_sqsEndpoint = $this->_sqsEndpoints[$region];

            } else {

                $this->_sqsEndpoints[$region] = "sqs.$region.amazonaws.com";

                $this->_sqsEndpoint = $this->_sqsEndpoints[$region];

            }

        } else {

            throw new Zend_Service_Amazon_Sqs_Exception('Empty region specified.');

        }

    }

    /**

     * Get the SQS endpoint

     * 

     * @return string 

     */

    public function getEndpoint()

    {

        return $this->_sqsEndpoint;

    }

    /**

     * Get possible SQS endpoints

     *

     * Since there is not an SQS webserive to get all possible endpoints, a hardcoded list is available.

     * For the actual region list please check:

     * http://docs.amazonwebservices.com/AWSSimpleQueueService/2009-02-01/APIReference/index.html?QueueServiceWsdlArticle.html

     *

     * @param  string  $region region

     * @return array

     */

    public function getEndpoints()

    {

        return $this->_sqsEndpoints;

    }

    /**

     * Create a new queue

     *

     * Visibility timeout is how long a message is left in the queue "invisible"

     * to other readers.  If the message is acknowleged (deleted) before the

     * timeout, then the message is deleted.  However, if the timeout expires

     * then the message will be made available to other queue readers.

     *

     * @param  string  $queue_name queue name

     * @param  integer $timeout    default visibility timeout

     * @return string|boolean

     * @throws Zend_Service_Amazon_Sqs_Exception

     */

    public function create($queue_name, $timeout = null)

    {

        $params = array();

        $params['QueueName'] = $queue_name;

        $timeout = ($timeout === null) ? self::CREATE_TIMEOUT_DEFAULT : (int)$timeout;

        $params['DefaultVisibilityTimeout'] = $timeout;

        $retry_count = 0;

        do {

            $retry  = false;

            $result = $this->_makeRequest(null, 'CreateQueue', $params);

            if (!isset($result->CreateQueueResult->QueueUrl)

                || empty($result->CreateQueueResult->QueueUrl)

            ) {

                if ($result->Error->Code == 'AWS.SimpleQueueService.QueueNameExists') {

                    return false;

                } elseif ($result->Error->Code == 'AWS.SimpleQueueService.QueueDeletedRecently') {

                    // Must sleep for 60 seconds, then try re-creating the queue

                    sleep(60);

                    $retry = true;

                    $retry_count++;

                } else {

                    require_once 'Zend/Service/Amazon/Sqs/Exception.php';

                    throw new Zend_Service_Amazon_Sqs_Exception($result->Error->Code);

                }

            } else {

                return (string) $result->CreateQueueResult->QueueUrl;

            }

        } while ($retry);

        return false;

    }

    /**

     * Delete a queue and all of it's messages

     *

     * Returns false if the queue is not found, true if the queue exists

     *

     * @param  string  $queue_url queue URL

     * @return boolean

     * @throws Zend_Service_Amazon_Sqs_Exception

     */

    public function delete($queue_url)

    {

        $result = $this->_makeRequest($queue_url, 'DeleteQueue');

        if ($result->Error->Code !== null) {

            require_once 'Zend/Service/Amazon/Sqs/Exception.php';

            throw new Zend_Service_Amazon_Sqs_Exception($result->Error->Code);

        }

        return true;

    }

    /**

     * Get an array of all available queues

     *

     * @return array

     * @throws Zend_Service_Amazon_Sqs_Exception

     */

    public function getQueues()

    {

        $result = $this->_makeRequest(null, 'ListQueues');

        if (isset($result->Error)) {

            require_once 'Zend/Service/Amazon/Sqs/Exception.php';

            throw new Zend_Service_Amazon_Sqs_Exception($result->Error->Code);

        }

        if (!isset($result->ListQueuesResult->QueueUrl)

            || empty($result->ListQueuesResult->QueueUrl)

        ) {

            return array();

        }

        $queues = array();

        foreach ($result->ListQueuesResult->QueueUrl as $queue_url) {

            $queues[] = (string)$queue_url;

        }

        return $queues;

    }

    /**

     * Return the approximate number of messages in the queue

     *

     * @param  string  $queue_url Queue URL

     * @return integer

     * @throws Zend_Service_Amazon_Sqs_Exception

     */

    public function count($queue_url)

    {

        return (int)$this->getAttribute($queue_url, 'ApproximateNumberOfMessages');

    }

    /**

     * Send a message to the queue

     *

     * @param  string $queue_url Queue URL

     * @param  string $message   Message to send to the queue

     * @return string            Message ID

     * @throws Zend_Service_Amazon_Sqs_Exception

     */

    public function send($queue_url, $message)

    {

        $params = array();

        $params['MessageBody'] = urlencode($message);

        $checksum = md5($params['MessageBody']);

        $result = $this->_makeRequest($queue_url, 'SendMessage', $params);

        if (!isset($result->SendMessageResult->MessageId)

            || empty($result->SendMessageResult->MessageId)

        ) {

            require_once 'Zend/Service/Amazon/Sqs/Exception.php';

            throw new Zend_Service_Amazon_Sqs_Exception($result->Error->Code);

        } else if ((string) $result->SendMessageResult->MD5OfMessageBody != $checksum) {

            require_once 'Zend/Service/Amazon/Sqs/Exception.php';

            throw new Zend_Service_Amazon_Sqs_Exception('MD5 of body does not match message sent');

        }

        return (string) $result->SendMessageResult->MessageId;

    }

    /**

     * Get messages in the queue

     *

     * @param  string  $queue_url    Queue name

     * @param  integer $max_messages Maximum number of messages to return

     * @param  integer $timeout      Visibility timeout for these messages

     * @return array

     * @throws Zend_Service_Amazon_Sqs_Exception

     */

    public function receive($queue_url, $max_messages = null, $timeout = null)

    {

        $params = array();

        // If not set, the visibility timeout on the queue is used

        if ($timeout !== null) {

            $params['VisibilityTimeout'] = (int)$timeout;

        }

        // SQS will default to only returning one message

        if ($max_messages !== null) {

            $params['MaxNumberOfMessages'] = (int)$max_messages;

        }

        $result = $this->_makeRequest($queue_url, 'ReceiveMessage', $params);

        if (isset($result->Error)) {

            require_once 'Zend/Service/Amazon/Sqs/Exception.php';

            throw new Zend_Service_Amazon_Sqs_Exception($result->Error->Code);

        }

        if (!isset($result->ReceiveMessageResult->Message)

            || empty($result->ReceiveMessageResult->Message)

        ) {

            // no messages found

            return array();

        }

        $data = array();

        foreach ($result->ReceiveMessageResult->Message as $message) {

            $data[] = array(

                'message_id' => (string)$message->MessageId,

                'handle'     => (string)$message->ReceiptHandle,

                'md5'        => (string)$message->MD5OfBody,

                'body'       => urldecode((string)$message->Body),

            );

        }

        return $data;

    }

    /**

     * Delete a message from the queue

     *

     * Returns true if the message is deleted, false if the deletion is

     * unsuccessful.

     *

     * @param  string $queue_url  Queue URL

     * @param  string $handle     Message handle as returned by SQS

     * @return boolean

     * @throws Zend_Service_Amazon_Sqs_Exception

     */

    public function deleteMessage($queue_url, $handle)

    {

        $params = array();

        $params['ReceiptHandle'] = (string)$handle;

        $result = $this->_makeRequest($queue_url, 'DeleteMessage', $params);

        if (isset($result->Error->Code)

            && !empty($result->Error->Code)

        ) {

            return false;

        }

        // Will always return true unless ReceiptHandle is malformed

        return true;

    }

    /**

     * Get the attributes for the queue

     *

     * @param  string $queue_url  Queue URL

     * @param  string $attribute

     * @return string

     * @throws Zend_Service_Amazon_Sqs_Exception

     */

    public function getAttribute($queue_url, $attribute = 'All')

    {

        $params = array();

        $params['AttributeName'] = $attribute;

        $result = $this->_makeRequest($queue_url, 'GetQueueAttributes', $params);

        if (!isset($result->GetQueueAttributesResult->Attribute)

            || empty($result->GetQueueAttributesResult->Attribute)

        ) {

            require_once 'Zend/Service/Amazon/Sqs/Exception.php';

            throw new Zend_Service_Amazon_Sqs_Exception($result->Error->Code);

        }

        if(count($result->GetQueueAttributesResult->Attribute) > 1) {

            $attr_result = array();

            foreach($result->GetQueueAttributesResult->Attribute as $attribute) {

                $attr_result[(string)$attribute->Name] = (string)$attribute->Value;

            }

            return $attr_result;

        } else {

            return (string) $result->GetQueueAttributesResult->Attribute->Value;

        }

    }

    /**

     * Make a request to Amazon SQS

     *

     * @param  string           $queue  Queue Name

     * @param  string           $action SQS action

     * @param  array            $params

     * @return SimpleXMLElement

     */

    private function _makeRequest($queue_url, $action, $params = array())

    {

        $params['Action'] = $action;

        $params = $this->addRequiredParameters($queue_url, $params);

        if ($queue_url === null) {

            $queue_url = '/';

        }

        $client = self::getHttpClient();

        switch ($action) {

            case 'ListQueues':

            case 'CreateQueue':

                $client->setUri('http://'.$this->_sqsEndpoint);

                break;

            default: