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_Sales
 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:  * Quote item resource collection
 30:  *
 31:  * @category    Mage
 32:  * @package     Mage_Sales
 33:  * @author      Magento Core Team <core@magentocommerce.com>
 34:  */
 35: class Mage_Sales_Model_Resource_Quote_Item_Collection extends Mage_Core_Model_Resource_Db_Collection_Abstract
 36: {
 37:     /**
 38:      * Collection quote instance
 39:      *
 40:      * @var Mage_Sales_Model_Quote
 41:      */
 42:     protected $_quote;
 43: 
 44:     /**
 45:      * Product Ids array
 46:      *
 47:      * @var array
 48:      */
 49:     protected $_productIds   = array();
 50: 
 51:     /**
 52:      * Initialize resource model
 53:      *
 54:      */
 55:     protected function _construct()
 56:     {
 57:         $this->_init('sales/quote_item');
 58:     }
 59: 
 60:     /**
 61:      * Retrieve store Id (From Quote)
 62:      *
 63:      * @return int
 64:      */
 65:     public function getStoreId()
 66:     {
 67:         return (int)$this->_quote->getStoreId();
 68:     }
 69: 
 70:     /**
 71:      * Set Quote object to Collection
 72:      *
 73:      * @param Mage_Sales_Model_Quote $quote
 74:      * @return Mage_Sales_Model_Resource_Quote_Item_Collection
 75:      */
 76:     public function setQuote($quote)
 77:     {
 78:         $this->_quote = $quote;
 79:         $quoteId      = $quote->getId();
 80:         if ($quoteId) {
 81:             $this->addFieldToFilter('quote_id', $quote->getId());
 82:         } else {
 83:             $this->_totalRecords = 0;
 84:             $this->_setIsLoaded(true);
 85:         }
 86:         return $this;
 87:     }
 88: 
 89:     /**
 90:      * Reset the collection and inner join it to quotes table
 91:      * Optionally can select items with specified product id only
 92:      *
 93:      * @param string $quotesTableName
 94:      * @param int $productId
 95:      * @return Mage_Sales_Model_Resource_Quote_Item_Collection
 96:      */
 97:     public function resetJoinQuotes($quotesTableName, $productId = null)
 98:     {
 99:         $this->getSelect()->reset()
100:             ->from(
101:                 array('qi' => $this->getResource()->getMainTable()),
102:                 array('item_id', 'qty', 'quote_id'))
103:             ->joinInner(
104:                 array('q' => $quotesTableName),
105:                'qi.quote_id = q.entity_id',
106:                 array('store_id', 'items_qty', 'items_count')
107:             );
108:         if ($productId) {
109:             $this->getSelect()->where('qi.product_id = ?', (int)$productId);
110:         }
111:         return $this;
112:     }
113: 
114:     /**
115:      * After load processing
116:      *
117:      * @return Mage_Sales_Model_Resource_Quote_Item_Collection
118:      */
119:     protected function _afterLoad()
120:     {
121:         parent::_afterLoad();
122: 
123:         /**
124:          * Assign parent items
125:          */
126:         foreach ($this as $item) {
127:             if ($item->getParentItemId()) {
128:                 $item->setParentItem($this->getItemById($item->getParentItemId()));
129:             }
130:             if ($this->_quote) {
131:                 $item->setQuote($this->_quote);
132:             }
133:         }
134: 
135:         /**
136:          * Assign options and products
137:          */
138:         $this->_assignOptions();
139:         $this->_assignProducts();
140:         $this->resetItemsDataChanged();
141: 
142:         return $this;
143:     }
144: 
145:     /**
146:      * Add options to items
147:      *
148:      * @return Mage_Sales_Model_Resource_Quote_Item_Collection
149:      */
150:     protected function _assignOptions()
151:     {
152:         $itemIds          = array_keys($this->_items);
153:         $optionCollection = Mage::getModel('sales/quote_item_option')->getCollection()
154:             ->addItemFilter($itemIds);
155:         foreach ($this as $item) {
156:             $item->setOptions($optionCollection->getOptionsByItem($item));
157:         }
158:         $productIds        = $optionCollection->getProductIds();
159:         $this->_productIds = array_merge($this->_productIds, $productIds);
160: 
161:         return $this;
162:     }
163: 
164:     /**
165:      * Add products to items and item options
166:      *
167:      * @return Mage_Sales_Model_Resource_Quote_Item_Collection
168:      */
169:     protected function _assignProducts()
170:     {
171:         Varien_Profiler::start('QUOTE:'.__METHOD__);
172:         $productIds = array();
173:         foreach ($this as $item) {
174:             $productIds[] = (int)$item->getProductId();
175:         }
176:         $this->_productIds = array_merge($this->_productIds, $productIds);
177: 
178:         $productCollection = Mage::getModel('catalog/product')->getCollection()
179:             ->setStoreId($this->getStoreId())
180:             ->addIdFilter($this->_productIds)
181:             ->addAttributeToSelect(Mage::getSingleton('sales/quote_config')->getProductAttributes())
182:             ->addOptionsToResult()
183:             ->addStoreFilter()
184:             ->addUrlRewrite()
185:             ->addTierPriceData();
186: 
187:         Mage::dispatchEvent('prepare_catalog_product_collection_prices', array(
188:             'collection'            => $productCollection,
189:             'store_id'              => $this->getStoreId(),
190:         ));
191:         Mage::dispatchEvent('sales_quote_item_collection_products_after_load', array(
192:             'product_collection'    => $productCollection
193:         ));
194: 
195:         $recollectQuote = false;
196:         foreach ($this as $item) {
197:             $product = $productCollection->getItemById($item->getProductId());
198:             if ($product) {
199:                 $product->setCustomOptions(array());
200:                 $qtyOptions         = array();
201:                 $optionProductIds   = array();
202:                 foreach ($item->getOptions() as $option) {
203:                     /**
204:                      * Call type specified logic for product associated with quote item
205:                      */
206:                     $product->getTypeInstance(true)->assignProductToOption(
207:                             $productCollection->getItemById($option->getProductId()),
208:                             $option,
209:                             $product
210:                         );
211: 
212:                     if (is_object($option->getProduct()) && $option->getProduct()->getId() != $product->getId()) {
213:                         $optionProductIds[$option->getProduct()->getId()] = $option->getProduct()->getId();
214:                     }
215:                 }
216: 
217:                 if ($optionProductIds) {
218:                     foreach ($optionProductIds as $optionProductId) {
219:                         $qtyOption = $item->getOptionByCode('product_qty_' . $optionProductId);
220:                         if ($qtyOption) {
221:                             $qtyOptions[$optionProductId] = $qtyOption;
222:                         }
223:                     }
224:                 }
225:                 $item->setQtyOptions($qtyOptions);
226: 
227:                 $item->setProduct($product);
228:             } else {
229:                 $item->isDeleted(true);
230:                 $recollectQuote = true;
231:             }
232:             $item->checkData();
233:         }
234: 
235:         if ($recollectQuote && $this->_quote) {
236:             $this->_quote->collectTotals();
237:         }
238:         Varien_Profiler::stop('QUOTE:'.__METHOD__);
239: 
240:         return $this;
241:     }
242: }
243: 
244: