<?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_Sales
 * @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)
 */

/**
 * Quote mysql4 resource model
 *
 * @category    Mage
 * @package     Mage_Sales
 * @author      Magento Core Team <core@magentocommerce.com>
 */
class Mage_Sales_Model_Mysql4_Quote extends Mage_Sales_Model_Mysql4_Abstract
{
    /**
     * Initialize table nad PK name
     */
    protected function _construct()
    {
        $this->_init('sales/quote', 'entity_id');
    }

    /**
     * Retrieve select object for load object data
     *
     * @param   string $field
     * @param   mixed $value
     * @return  Zend_Db_Select
     */
    protected function _getLoadSelect($field, $value, $object)
    {
           $select = parent::_getLoadSelect($field, $value, $object);
        if ($storeIds = $object->getSharedStoreIds()) {
            $select->where('store_id IN (?)', $storeIds);
        }
        else {
            /**
             * For empty result
             */
            $select->where('store_id<0');
        }
        return $select;
    }

    /**
     * Load quote data by customer identifier
     *
     * @param Mage_Sales_Model_Quote $quote
     * @param int $customerId
     */
    public function loadByCustomerId($quote, $customerId)
    {
        $read = $this->_getReadAdapter();
        if ($read) {
            $select = $this->_getLoadSelect('customer_id', $customerId, $quote)
                ->where('is_active=1')
                ->order('updated_at desc')
                ->limit(1);

            $data = $read->fetchRow($select);

            if ($data) {
                $quote->setData($data);
            }
        }

        $this->_afterLoad($quote);
        return $this;
    }

    /**
     * Load only active quote
     *
     * @param Mage_Sales_Model_Quote $quote
     * @param int $quoteId
     */
    public function loadActive($quote, $quoteId)
    {
        $read = $this->_getReadAdapter();
        if ($read) {
            $select = $this->_getLoadSelect('entity_id', $quoteId, $quote)
                ->where('is_active=1');

            $data = $read->fetchRow($select);

            if ($data) {
                $quote->setData($data);
            }
        }

        $this->_afterLoad($quote);
        return $this;
    }

    /**
     * Load quote data by identifier without store
     *
     * @param Mage_Sales_Model_Quote $quote
     * @param int $quoteId
     * @return Mage_Sales_Model_Mysql4_Quote
     */
    public function loadByIdWithoutStore($quote, $quoteId)
    {
        $read = $this->_getReadAdapter();
        if ($read) {
            $select = parent::_getLoadSelect('entity_id', $quoteId, $quote);

            $data = $read->fetchRow($select);

            if ($data) {
                $quote->setData($data);
            }
        }

        $this->_afterLoad($quote);
        return $this;
    }

    public function getReservedOrderId($quote)
    {
        return Mage::getSingleton('eav/config')->getEntityType('order')->fetchNewIncrementId($quote->getStoreId());
    }

    public function isOrderIncrementIdUsed($orderIncrementId) {
        if ($this->_getReadAdapter()) {
            $select = $this->_getReadAdapter()->select();
            $select->from($this->getTable('sales/order'), 'entity_id')
                ->where('increment_id = ?', $orderIncrementId);
            $entity_id = $this->_getReadAdapter()->fetchOne($select);
            if ($entity_id > 0) {
                return true;
            } else {
                return false;
            }
        }
        return false;
    }

    /**
     * Mark quotes - that depend on catalog price rules - to be recollected on demand
     *
     */
    public function markQuotesRecollectOnCatalogRules()
    {
        $this->_getWriteAdapter()->query("
            UPDATE {$this->getTable('sales/quote')} SET trigger_recollect = 1
            WHERE entity_id IN (
                SELECT DISTINCT quote_id
                FROM {$this->getTable('sales/quote_item')}
                WHERE product_id IN (SELECT DISTINCT product_id FROM {$this->getTable('catalogrule/rule_product_price')})
            )"
        );
    }

    /**
     * Substract product from all quotes quantities
     *
     * @param Mage_Catalog_Model_Product $product
     */
    public function substractProductFromQuotes($product)
    {
        if ($product->getId()) {
            $this->_getWriteAdapter()->query(
                'update ' . $this->getTable('sales/quote_item') .
                ' as qi, ' . $this->getTable('sales/quote') .
                ' as q set q.items_qty = q.items_qty - qi.qty, q.items_count = q.items_count - 1 ' .
                ' where qi.product_id = "' . $product->getId() . '" and q.entity_id = qi.quote_id and qi.parent_item_id is null'
            );
        }
    }

    /**
     * Mark recollect contain product(s) quotes
     *
     * @param array|int $productIds
     * @return Mage_Sales_Model_Mysql4_Quote
     */
    public function markQuotesRecollect($productIds)
    {
        $this->_getWriteAdapter()->query("
            UPDATE `{$this->getTable('sales/quote')}` SET `trigger_recollect` = 1
            WHERE `entity_id` IN (
                SELECT DISTINCT `quote_id`
                FROM `{$this->getTable('sales/quote_item')}`
                WHERE `product_id` IN (?)
            )", $productIds
        );

        return $this;
    }
}
