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_Catalog
 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:  * Catalog product link resource model
 30:  *
 31:  * @category    Mage
 32:  * @package     Mage_Catalog
 33:  * @author      Magento Core Team <core@magentocommerce.com>
 34:  */
 35: class Mage_Catalog_Model_Resource_Product_Link extends Mage_Core_Model_Resource_Db_Abstract
 36: {
 37:     /**
 38:      * Product Link Attributes Table
 39:      *
 40:      * @var string
 41:      */
 42:     protected $_attributesTable;
 43: 
 44:     /**
 45:      * Define main table name and attributes table
 46:      */
 47:     protected function _construct()
 48:     {
 49:         $this->_init('catalog/product_link', 'link_id');
 50:         $this->_attributesTable = $this->getTable('catalog/product_link_attribute');
 51:     }
 52: 
 53:     /**
 54:      * Save Product Links process
 55:      *
 56:      * @param Mage_Catalog_Model_Product $product
 57:      * @param array $data
 58:      * @param int $typeId
 59:      * @return Mage_Catalog_Model_Resource_Product_Link
 60:      */
 61:     public function saveProductLinks($product, $data, $typeId)
 62:     {
 63:         if (!is_array($data)) {
 64:             $data = array();
 65:         }
 66: 
 67:         $attributes = $this->getAttributesByType($typeId);
 68:         $adapter    = $this->_getWriteAdapter();
 69: 
 70:         $bind   = array(
 71:             ':product_id'    => (int)$product->getId(),
 72:             ':link_type_id'  => (int)$typeId
 73:         );
 74:         $select = $adapter->select()
 75:             ->from($this->getMainTable(), array('linked_product_id', 'link_id'))
 76:             ->where('product_id = :product_id')
 77:             ->where('link_type_id = :link_type_id');
 78: 
 79:         $links   = $adapter->fetchPairs($select, $bind);
 80: 
 81:         $deleteIds = array();
 82:         foreach($links as $linkedProductId => $linkId) {
 83:             if (!isset($data[$linkedProductId])) {
 84:                 $deleteIds[] = (int)$linkId;
 85:             }
 86:         }
 87:         if (!empty($deleteIds)) {
 88:             $adapter->delete($this->getMainTable(), array(
 89:                 'link_id IN (?)' => $deleteIds,
 90:             ));
 91:         }
 92: 
 93:         foreach ($data as $linkedProductId => $linkInfo) {
 94:             $linkId = null;
 95:             if (isset($links[$linkedProductId])) {
 96:                 $linkId = $links[$linkedProductId];
 97:                 unset($links[$linkedProductId]);
 98:             } else {
 99:                 $bind = array(
100:                     'product_id'        => $product->getId(),
101:                     'linked_product_id' => $linkedProductId,
102:                     'link_type_id'      => $typeId
103:                 );
104:                 $adapter->insert($this->getMainTable(), $bind);
105:                 $linkId = $adapter->lastInsertId($this->getMainTable());
106:             }
107: 
108:             foreach ($attributes as $attributeInfo) {
109:                 $attributeTable = $this->getAttributeTypeTable($attributeInfo['type']);
110:                 if ($attributeTable) {
111:                     if (isset($linkInfo[$attributeInfo['code']])) {
112:                         $value = $this->_prepareAttributeValue($attributeInfo['type'],
113:                             $linkInfo[$attributeInfo['code']]);
114:                         $bind = array(
115:                             'product_link_attribute_id' => $attributeInfo['id'],
116:                             'link_id'                   => $linkId,
117:                             'value'                     => $value
118:                         );
119:                         $adapter->insertOnDuplicate($attributeTable, $bind, array('value'));
120:                     } else {
121:                         $adapter->delete($attributeTable, array(
122:                             'link_id = ?'                   => $linkId,
123:                             'product_link_attribute_id = ?' => $attributeInfo['id']
124:                         ));
125:                     }
126:                 }
127:             }
128:         }
129: 
130:         return $this;
131:     }
132: 
133:     /**
134:      * Prepare link attribute value by attribute type
135:      *
136:      * @param string $type
137:      * @param mixed $value
138:      * @return mixed
139:      */
140:     protected function _prepareAttributeValue($type, $value)
141:     {
142:         if ($type == 'int') {
143:             $value = (int)$value;
144:         } elseif ($type == 'decimal') {
145:             $value = (float)sprintf('%F', $value);
146:         }
147:         return $value;
148:     }
149: 
150:     /**
151:      * Retrieve product link attributes by link type
152:      *
153:      * @param int $typeId
154:      * @return array
155:      */
156:     public function getAttributesByType($typeId)
157:     {
158:         $adapter = $this->_getReadAdapter();
159:         $select = $adapter->select()
160:             ->from($this->_attributesTable, array(
161:                 'id'    => 'product_link_attribute_id',
162:                 'code'  => 'product_link_attribute_code',
163:                 'type'  => 'data_type'
164:             ))
165:             ->where('link_type_id = ?', $typeId);
166:         return $adapter->fetchAll($select);
167:     }
168: 
169:     /**
170:      * Returns table for link attribute by attribute type
171:      *
172:      * @param string $type
173:      * @return string
174:      */
175:     public function getAttributeTypeTable($type)
176:     {
177:         return $this->getTable('catalog/product_link_attribute_' . $type);
178:     }
179: 
180:     /**
181:      * Retrieve Required children ids
182:      * Return grouped array, ex array(
183:      *   group => array(ids)
184:      * )
185:      *
186:      * @param int $parentId
187:      * @param int $typeId
188:      * @return array
189:      */
190:     public function getChildrenIds($parentId, $typeId)
191:     {
192:         $adapter     = $this->_getReadAdapter();
193:         $childrenIds = array();
194:         $bind        = array(
195:             ':product_id'    => (int)$parentId,
196:             ':link_type_id'  => (int)$typeId
197:         );
198:         $select = $adapter->select()
199:             ->from(array('l' => $this->getMainTable()), array('linked_product_id'))
200:             ->where('product_id = :product_id')
201:             ->where('link_type_id = :link_type_id');
202:         if ($typeId == Mage_Catalog_Model_Product_Link::LINK_TYPE_GROUPED) {
203:             $select->join(
204:                 array('e' => $this->getTable('catalog/product')),
205:                 'e.entity_id = l.linked_product_id AND e.required_options = 0',
206:                 array()
207:             );
208:         }
209: 
210:         $childrenIds[$typeId] = array();
211:         $result = $adapter->fetchAll($select, $bind);
212:         foreach ($result as $row) {
213:             $childrenIds[$typeId][$row['linked_product_id']] = $row['linked_product_id'];
214:         }
215: 
216:         return $childrenIds;
217:     }
218: 
219:     /**
220:      * Retrieve parent ids array by required child
221:      *
222:      * @param int|array $childId
223:      * @param int $typeId
224:      * @return array
225:      */
226:     public function getParentIdsByChild($childId, $typeId)
227:     {
228:         $parentIds  = array();
229:         $adapter    = $this->_getReadAdapter();
230:         $select = $adapter->select()
231:             ->from($this->getMainTable(), array('product_id', 'linked_product_id'))
232:             ->where('linked_product_id IN(?)', $childId)
233:             ->where('link_type_id = ?', $typeId);
234: 
235:         $result = $adapter->fetchAll($select);
236:         foreach ($result as $row) {
237:             $parentIds[] = $row['product_id'];
238:         }
239: 
240:         return $parentIds;
241:     }
242: 
243:     /**
244:      * Save grouped product relations
245:      *
246:      * @param Mage_Catalog_Model_Product $product
247:      * @param array $data
248:      * @param int $typeId
249:      * @return Mage_Catalog_Model_Resource_Product_Link
250:      */
251:     public function saveGroupedLinks($product, $data, $typeId)
252:     {
253:         $adapter = $this->_getWriteAdapter();
254:         // check for change relations
255:         $bind    = array(
256:             'product_id'    => (int)$product->getId(),
257:             'link_type_id'  => (int)$typeId
258:         );
259:         $select = $adapter->select()
260:             ->from($this->getMainTable(), array('linked_product_id'))
261:             ->where('product_id = :product_id')
262:             ->where('link_type_id = :link_type_id');
263:         $old = $adapter->fetchCol($select, $bind);
264:         $new = array_keys($data);
265: 
266:         if (array_diff($old, $new) || array_diff($new, $old)) {
267:             $product->setIsRelationsChanged(true);
268:         }
269: 
270:         // save product links attributes
271:         $this->saveProductLinks($product, $data, $typeId);
272: 
273:         // Grouped product relations should be added to relation table
274:         Mage::getResourceSingleton('catalog/product_relation')
275:             ->processRelations($product->getId(), $new);
276: 
277:         return $this;
278:     }
279: }
280: