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_Eav
 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:  * Eav attribute set resource model
 30:  *
 31:  * @category    Mage
 32:  * @package     Mage_Eav
 33:  * @author      Magento Core Team <core@magentocommerce.com>
 34:  */
 35: class Mage_Eav_Model_Resource_Entity_Attribute_Set extends Mage_Core_Model_Resource_Db_Abstract
 36: {
 37:     /**
 38:      * Initialize connection
 39:      *
 40:      */
 41:     protected function _construct()
 42:     {
 43:         $this->_init('eav/attribute_set', 'attribute_set_id');
 44:     }
 45: 
 46:     /**
 47:      * Perform actions after object save
 48:      *
 49:      * @param Mage_Core_Model_Abstract $object
 50:      * @return Mage_Eav_Model_Resource_Entity_Attribute_Set
 51:      */
 52:     protected function _afterSave(Mage_Core_Model_Abstract $object)
 53:     {
 54:         if ($object->getGroups()) {
 55:             /* @var $group Mage_Eav_Model_Entity_Attribute_Group */
 56:             foreach ($object->getGroups() as $group) {
 57:                 $group->setAttributeSetId($object->getId());
 58:                 if ($group->itemExists() && !$group->getId()) {
 59:                     continue;
 60:                 }
 61:                 $group->save();
 62:             }
 63:         }
 64:         if ($object->getRemoveGroups()) {
 65:             foreach ($object->getRemoveGroups() as $group) {
 66:                 /* @var $group Mage_Eav_Model_Entity_Attribute_Group */
 67:                 $group->delete();
 68:             }
 69:             Mage::getResourceModel('eav/entity_attribute_group')->updateDefaultGroup($object->getId());
 70:         }
 71:         if ($object->getRemoveAttributes()) {
 72:             foreach ($object->getRemoveAttributes() as $attribute) {
 73:                 /* @var $attribute Mage_Eav_Model_Entity_Attribute */
 74:                 $attribute->deleteEntity();
 75:             }
 76:         }
 77: 
 78:         return parent::_afterSave($object);
 79:     }
 80: 
 81:     /**
 82:      * Validate attribute set name
 83:      *
 84:      * @param Mage_Eav_Model_Entity_Attribute_Set $object
 85:      * @param string $attributeSetName
 86:      * @return bool
 87:      */
 88:     public function validate($object, $attributeSetName)
 89:     {
 90: 
 91:         $adapter = $this->_getReadAdapter();
 92:         $bind = array(
 93:             'attribute_set_name' => trim($attributeSetName),
 94:             'entity_type_id'     => $object->getEntityTypeId()
 95:         );
 96:         $select = $adapter->select()
 97:             ->from($this->getMainTable())
 98:             ->where('attribute_set_name = :attribute_set_name')
 99:             ->where('entity_type_id = :entity_type_id');
100: 
101:         if ($object->getId()) {
102:             $bind['attribute_set_id'] = $object->getId();
103:             $select->where('attribute_set_id != :attribute_set_id');
104:         }
105: 
106:         return !$adapter->fetchOne($select, $bind) ? true : false;
107:     }
108: 
109:     /**
110:      * Retrieve Set info by attributes
111:      *
112:      * @param array $attributeIds
113:      * @param int $setId
114:      * @return array
115:      */
116:     public function getSetInfo(array $attributeIds, $setId = null)
117:     {
118:         $adapter = $this->_getReadAdapter();
119:         $setInfo = array();
120:         $attributeToSetInfo = array();
121: 
122:         if (count($attributeIds) > 0) {
123:             $select = $adapter->select()
124:                 ->from(
125:                     array('entity' => $this->getTable('eav/entity_attribute')),
126:                     array('attribute_id', 'attribute_set_id', 'attribute_group_id', 'sort_order'))
127:                 ->joinLeft(
128:                     array('attribute_group' => $this->getTable('eav/attribute_group')),
129:                     'entity.attribute_group_id = attribute_group.attribute_group_id',
130:                     array('group_sort_order' => 'sort_order'))
131:                 ->where('entity.attribute_id IN (?)', $attributeIds);
132:             $bind = array();
133:             if (is_numeric($setId)) {
134:                 $bind[':attribute_set_id'] = $setId;
135:                 $select->where('entity.attribute_set_id = :attribute_set_id');
136:             }
137:             $result = $adapter->fetchAll($select, $bind);
138: 
139:             foreach ($result as $row) {
140:                 $data = array(
141:                     'group_id'      => $row['attribute_group_id'],
142:                     'group_sort'    => $row['group_sort_order'],
143:                     'sort'          => $row['sort_order']
144:                 );
145:                 $attributeToSetInfo[$row['attribute_id']][$row['attribute_set_id']] = $data;
146:             }
147:         }
148: 
149:         foreach ($attributeIds as $atttibuteId) {
150:             $setInfo[$atttibuteId] = isset($attributeToSetInfo[$atttibuteId])
151:                 ? $attributeToSetInfo[$atttibuteId]
152:                 : array();
153:         }
154: 
155:         return $setInfo;
156:     }
157: 
158:     /**
159:      * Retrurn default attribute group id for attribute set id
160:      *
161:      * @param int $setId
162:      * @return int|null
163:      */
164:     public function getDefaultGroupId($setId)
165:     {
166:         $adapter = $this->_getReadAdapter();
167:         $bind    = array(
168:             'attribute_set_id' => (int)$setId
169:         );
170:         $select = $adapter->select()
171:             ->from($this->getTable('eav/attribute_group'), 'attribute_group_id')
172:             ->where('attribute_set_id = :attribute_set_id')
173:             ->where('default_id = 1')
174:             ->limit(1);
175:         return $adapter->fetchOne($select, $bind);
176:     }
177: }
178: