1: <?php
  2: /**
  3:  * Magento
  4:  *
  5:  * NOTICE OF LICENSE
  6:  *
  7:  * This source file is subject to the Open Software License (OSL 3.0)
  8:  * that is bundled with this package in the file LICENSE.txt.
  9:  * It is also available through the world-wide-web at this URL:
 10:  * http://opensource.org/licenses/osl-3.0.php
 11:  * If you did not receive a copy of the license and are unable to
 12:  * obtain it through the world-wide-web, please send an email
 13:  * to license@magentocommerce.com so we can send you a copy immediately.
 14:  *
 15:  * DISCLAIMER
 16:  *
 17:  * Do not edit or add to this file if you wish to upgrade Magento to newer
 18:  * versions in the future. If you wish to customize Magento for your
 19:  * needs please refer to http://www.magentocommerce.com for more information.
 20:  *
 21:  * @category    Mage
 22:  * @package     Mage_ProductAlert
 23:  * @copyright   Copyright (c) 2012 Magento Inc. (http://www.magentocommerce.com)
 24:  * @license     http://opensource.org/licenses/osl-3.0.php  Open Software License (OSL 3.0)
 25:  */
 26: 
 27: 
 28: /**
 29:  * Product alert for back in stock collection
 30:  *
 31:  * @category    Mage
 32:  * @package     Mage_ProductAlert
 33:  * @author      Magento Core Team <core@magentocommerce.com>
 34:  */
 35: class Mage_ProductAlert_Model_Resource_Stock_Collection extends Mage_Core_Model_Resource_Db_Collection_Abstract
 36: {
 37:     /**
 38:      * Define stock collection
 39:      *
 40:      */
 41:     protected function _construct()
 42:     {
 43:         $this->_init('productalert/stock');
 44:     }
 45: 
 46:     /**
 47:      * Add customer filter
 48:      *
 49:      * @param mixed $customer
 50:      * @return Mage_ProductAlert_Model_Resource_Stock_Collection
 51:      */
 52:     public function addCustomerFilter($customer)
 53:     {
 54:         $adapter = $this->getConnection();
 55:         if (is_array($customer)) {
 56:             $condition = $adapter->quoteInto('customer_id IN(?)', $customer);
 57:         } elseif ($customer instanceof Mage_Customer_Model_Customer) {
 58:             $condition = $adapter->quoteInto('customer_id=?', $customer->getId());
 59:         } else {
 60:             $condition = $adapter->quoteInto('customer_id=?', $customer);
 61:         }
 62:         $this->addFilter('customer_id', $condition, 'string');
 63:         return $this;
 64:     }
 65: 
 66:     /**
 67:      * Add website filter
 68:      *
 69:      * @param mixed $website
 70:      * @return Mage_ProductAlert_Model_Resource_Stock_Collection
 71:      */
 72:     public function addWebsiteFilter($website)
 73:     {
 74:         $adapter = $this->getConnection();
 75:         if (is_null($website) || $website == 0) {
 76:             return $this;
 77:         }
 78:         if (is_array($website)) {
 79:             $condition = $adapter->quoteInto('website_id IN(?)', $website);
 80:         } elseif ($website instanceof Mage_Core_Model_Website) {
 81:             $condition = $adapter->quoteInto('website_id=?', $website->getId());
 82:         } else {
 83:             $condition = $adapter->quoteInto('website_id=?', $website);
 84:         }
 85:         $this->addFilter('website_id', $condition, 'string');
 86:         return $this;
 87:     }
 88: 
 89:     /**
 90:      * Add status filter
 91:      *
 92:      * @param int $status
 93:      * @return Mage_ProductAlert_Model_Resource_Stock_Collection
 94:      */
 95:     public function addStatusFilter($status)
 96:     {
 97:         $condition = $this->getConnection()->quoteInto('status=?', $status);
 98:         $this->addFilter('status', $condition, 'string');
 99:         return $this;
100:     }
101: 
102:     /**
103:      * Set order by customer
104:      *
105:      * @param string $sort
106:      * @return Mage_ProductAlert_Model_Resource_Stock_Collection
107:      */
108:     public function setCustomerOrder($sort = 'ASC')
109:     {
110:         $this->getSelect()->order('customer_id ' . $sort);
111:         return $this;
112:     }
113: }
114: