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_Reports
 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:  * Wishlist Report collection
 30:  *
 31:  * @category    Mage
 32:  * @package     Mage_Reports
 33:  * @author      Magento Core Team <core@magentocommerce.com>
 34:  */
 35: class Mage_Reports_Model_Resource_Wishlist_Product_Collection extends Mage_Wishlist_Model_Resource_Product_Collection
 36: {
 37:     /**
 38:      * Resource initialization
 39:      *
 40:      */
 41:     protected function _construct()
 42:     {
 43:         $this->_init('wishlist/wishlist');
 44:     }
 45: 
 46:     /**
 47:      * Add wishlist count
 48:      *
 49:      * @return Mage_Reports_Model_Resource_Wishlist_Product_Collection
 50:      */
 51:     public function addWishlistCount()
 52:     {
 53:         $wishlistItemTable = $this->getTable('wishlist/item');
 54:         $this->getSelect()
 55:             ->join(
 56:                 array('wi' => $wishlistItemTable),
 57:                 'wi.product_id = e.entity_id',
 58:                 array('wishlists' => new Zend_Db_Expr('COUNT(wi.wishlist_item_id)')))
 59:             ->where('wi.product_id = e.entity_id')
 60:             ->group('wi.product_id');
 61:         /*
 62:          * Allow Analytic Functions Usage
 63:          */
 64:         $this->_useAnalyticFunction = true;
 65: 
 66:         $this->getEntity()->setStore(0);
 67:         return $this;
 68:     }
 69: 
 70:     /**
 71:      * add customer count to result
 72:      *
 73:      * @return Mage_Reports_Model_Resource_Wishlist_Product_Collection
 74:      */
 75:     public function getCustomerCount()
 76:     {
 77:         $this->getSelect()->reset();
 78: 
 79:         $this->getSelect()
 80:             ->from(
 81:                 array('wishlist' => $this->getTable('wishlist/wishlist')),
 82:                 array(
 83:                     'wishlist_cnt' => new Zend_Db_Expr('COUNT(wishlist.wishlist_id)'),
 84:                     'wishlist.customer_id'
 85:                 ))
 86:             ->group('wishlist.customer_id');
 87:         return $this;
 88:     }
 89: 
 90:     /**
 91:      * Get select count sql
 92:      *
 93:      * @return string
 94:      */
 95:     public function getSelectCountSql()
 96:     {
 97:         $countSelect = clone $this->getSelect();
 98:         $countSelect->reset(Zend_Db_Select::ORDER);
 99:         $countSelect->reset(Zend_Db_Select::LIMIT_COUNT);
100:         $countSelect->reset(Zend_Db_Select::LIMIT_OFFSET);
101:         $countSelect->reset(Zend_Db_Select::GROUP);
102:         $countSelect->reset(Zend_Db_Select::COLUMNS);
103:         $countSelect->columns("COUNT(*)");
104: 
105:         return $countSelect;
106:     }
107: 
108:     /**
109:      * Set order to result
110:      *
111:      * @param string $attribute
112:      * @param string $dir
113:      * @return Mage_Reports_Model_Resource_Wishlist_Product_Collection
114:      */
115:     public function setOrder($attribute, $dir = self::SORT_ORDER_DESC)
116:     {
117:         if ($attribute == 'wishlists') {
118:             $this->getSelect()->order($attribute . ' ' . $dir);
119:         } else {
120:             parent::setOrder($attribute, $dir);
121:         }
122: 
123:         return $this;
124:     }
125: }
126: 
127: