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:  * Catalog product link api V2
 29:  *
 30:  * @category   Mage
 31:  * @package    Mage_Catalog
 32:  * @author     Magento Core Team <core@magentocommerce.com>
 33:  */
 34: class Mage_Catalog_Model_Product_Link_Api_V2 extends Mage_Catalog_Model_Product_Link_Api
 35: {
 36:     /**
 37:      * Add product link association
 38:      *
 39:      * @param string $type
 40:      * @param int|string $productId
 41:      * @param int|string $linkedProductId
 42:      * @param array $data
 43:      * @return boolean
 44:      */
 45:     public function assign($type, $productId, $linkedProductId, $data = array(), $identifierType = null)
 46:     {
 47:         $typeId = $this->_getTypeId($type);
 48: 
 49:         $product = $this->_initProduct($productId, $identifierType);
 50: 
 51:         $link = $product->getLinkInstance()
 52:             ->setLinkTypeId($typeId);
 53: 
 54:         $collection = $this->_initCollection($link, $product);
 55:         $idBySku = $product->getIdBySku($linkedProductId);
 56:         if ($idBySku) {
 57:             $linkedProductId = $idBySku;
 58:         }
 59: 
 60:         $links = $this->_collectionToEditableArray($collection);
 61: 
 62:         $links[(int)$linkedProductId] = array();
 63:         foreach ($collection->getLinkModel()->getAttributes() as $attribute) {
 64:             if (isset($data->$attribute['code'])) {
 65:                 $links[(int)$linkedProductId][$attribute['code']] = $data->$attribute['code'];
 66:             }
 67:         }
 68: 
 69:         try {
 70:             if ($type == 'grouped') {
 71:                 $link->getResource()->saveGroupedLinks($product, $links, $typeId);
 72:             } else {
 73:                 $link->getResource()->saveProductLinks($product, $links, $typeId);
 74:             }
 75: 
 76:             $_linkInstance = Mage::getSingleton('catalog/product_link');
 77:             $_linkInstance->saveProductRelations($product);
 78: 
 79:             $indexerStock = Mage::getModel('cataloginventory/stock_status');
 80:             $indexerStock->updateStatus($productId);
 81: 
 82:             $indexerPrice = Mage::getResourceModel('catalog/product_indexer_price');
 83:             $indexerPrice->reindexProductIds($productId);
 84: 
 85:         } catch (Exception $e) {
 86:             $this->_fault('data_invalid', $e->getMessage());
 87:             //$this->_fault('data_invalid', Mage::helper('catalog')->__('Link product does not exist.'));
 88:         }
 89: 
 90:         return true;
 91:     }
 92: 
 93:     /**
 94:      * Update product link association info
 95:      *
 96:      * @param string $type
 97:      * @param int|string $productId
 98:      * @param int|string $linkedProductId
 99:      * @param array $data
100:      * @return boolean
101:      */
102:     public function update($type, $productId, $linkedProductId, $data = array(), $identifierType = null)
103:     {
104:         $typeId = $this->_getTypeId($type);
105: 
106:         $product = $this->_initProduct($productId, $identifierType);
107: 
108:         $link = $product->getLinkInstance()
109:             ->setLinkTypeId($typeId);
110: 
111:         $collection = $this->_initCollection($link, $product);
112: 
113:         $links = $this->_collectionToEditableArray($collection);
114: 
115:         $idBySku = $product->getIdBySku($linkedProductId);
116:         if ($idBySku) {
117:             $linkedProductId = $idBySku;
118:         }
119: 
120:         foreach ($collection->getLinkModel()->getAttributes() as $attribute) {
121:             if (isset($data->$attribute['code'])) {
122:                 $links[(int)$linkedProductId][$attribute['code']] = $data->$attribute['code'];
123:             }
124:         }
125: 
126:         try {
127:             if ($type == 'grouped') {
128:                 $link->getResource()->saveGroupedLinks($product, $links, $typeId);
129:             } else {
130:                 $link->getResource()->saveProductLinks($product, $links, $typeId);
131:             }
132: 
133:             $_linkInstance = Mage::getSingleton('catalog/product_link');
134:             $_linkInstance->saveProductRelations($product);
135: 
136:             $indexerStock = Mage::getModel('cataloginventory/stock_status');
137:             $indexerStock->updateStatus($productId);
138: 
139:             $indexerPrice = Mage::getResourceModel('catalog/product_indexer_price');
140:             $indexerPrice->reindexProductIds($productId);
141: 
142:         } catch (Exception $e) {
143:             $this->_fault('data_invalid', Mage::helper('catalog')->__('Link product does not exist.'));
144:         }
145: 
146:         return true;
147:     }
148: }
149: