<?php
/**
 * Magento
 *
 * NOTICE OF LICENSE
 *
 * This source file is subject to the Open Software License (OSL 3.0)
 * that is bundled with this package in the file LICENSE.txt.
 * It is also available through the world-wide-web at this URL:
 * http://opensource.org/licenses/osl-3.0.php
 * 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@magentocommerce.com so we can send you a copy immediately.
 *
 * DISCLAIMER
 *
 * Do not edit or add to this file if you wish to upgrade Magento to newer
 * versions in the future. If you wish to customize Magento for your
 * needs please refer to http://www.magentocommerce.com for more information.
 *
 * @category    Mage
 * @package     Mage_Reports
 * @copyright   Copyright (c) 2010 Magento Inc. (http://www.magentocommerce.com)
 * @license     http://opensource.org/licenses/osl-3.0.php  Open Software License (OSL 3.0)
 */

/**
 * Product Downloads Report collection
 *
 * @category   Mage
 * @package    Mage_Reports
 * @author     Magento Core Team <core@magentocommerce.com>
 */

class Mage_Reports_Model_Mysql4_Product_Downloads_Collection extends Mage_Catalog_Model_Resource_Eav_Mysql4_Product_Collection
{
    protected $_idFieldName = 'link_id';

    /**
     * Add downloads summary grouping by product
     *
     * @return Mage_Reports_Model_Mysql4_Product_Downloads_Collection
     */
    public function addSummary()
    {
        $this->getSelect()
            ->joinInner(
                array('d' => $this->getTable('downloadable/link_purchased_item')),
                'e.entity_id=d.product_id',
                array(
                    'purchases' => new Zend_Db_Expr('SUM(d.number_of_downloads_bought)'),
                    'downloads' => new Zend_Db_Expr('SUM(d.number_of_downloads_used)')
                )
            )
            ->joinInner(
                array('l' => $this->getTable('downloadable/link_title')),
                'd.link_id=l.link_id',
                array('l.link_id')
            )
            ->joinLeft(
                array('l_store' => $this->getTable('downloadable/link_title')),
                $this->getConnection()->quoteInto('l.link_id=l_store.link_id AND l_store.store_id=?',$this->getStoreId()),
                array('link_title' => 'IFNULL(l_store.title, l.title)')
            )
            ->where('d.number_of_downloads_bought>0 OR d.number_of_downloads_used>0')
            ->group('d.link_id');

        return $this;
    }

    /**
     * Add sorting
     *
     * @return Mage_Reports_Model_Mysql4_Product_Downloads_Collection
     */
    public function setOrder($attribute, $dir='desc')
    {
        if ($attribute == 'purchases' || $attribute == 'downloads' || $attribute == 'link_title') {
            $this->getSelect()->order($attribute . ' ' . $dir);
        } else {
            parent::setOrder($attribute, $dir);
        }
        return $this;
    }

    /**
     * Add filtering
     *
     * @return Mage_Reports_Model_Mysql4_Product_Downloads_Collection
     */
    public function addFieldToFilter($field, $condition=null)
    {
        if ($field == 'link_title') {
            $conditionSql = $this->_getConditionSql('l.title', $condition);
            $this->getSelect()->where($conditionSql);
        } else {
            parent::addFieldToFilter($field, $condition);
        }
        return $this;
    }
}