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_CatalogIndex
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:  * Enter description here ...
30:  *
31:  * @category    Mage
32:  * @package     Mage_CatalogIndex
33:  * @author      Magento Core Team <core@magentocommerce.com>
34:  */
35: class Mage_CatalogIndex_Model_Resource_Setup extends Mage_Core_Model_Resource_Setup
36: {
37:     /**
38:      * Enter description here ...
39:      *
40:      * @var unknown
41:      */
42:     protected $_storeToWebsite   = null;
43: 
44:     /**
45:      * Enter description here ...
46:      *
47:      * @param unknown_type $table
48:      * @return Mage_CatalogIndex_Model_Resource_Setup
49:      */
50:     public function convertStoreToWebsite($table)
51:     {
52:         $assignment = $this->_getStoreToWebsiteAssignments();
53:         foreach ($assignment as $website=>$stores) {
54:             $this->_setWebsiteInfo($table, $website, $stores);
55:         }
56:         return $this;
57:     }
58: 
59:     /**
60:      * Enter description here ...
61:      *
62:      * @return unknown
63:      */
64:     protected function _getStoreToWebsiteAssignments()
65:     {
66:         if (is_null($this->_storeToWebsite)) {
67:             $this->_storeToWebsite = array();
68:             $websiteCollection = Mage::getModel('core/website')->getCollection();
69:             foreach ($websiteCollection as $website) {
70:                 $this->_storeToWebsite[$website->getId()] = $website->getStoreIds();
71:             }
72:         }
73: 
74:         return $this->_storeToWebsite;
75:     }
76: 
77:     /**
78:      * Enter description here ...
79:      *
80:      * @param unknown_type $table
81:      * @param unknown_type $websiteId
82:      * @param unknown_type $storeIds
83:      * @return Mage_CatalogIndex_Model_Resource_Setup
84:      */
85:     protected function _setWebsiteInfo($table, $websiteId, $storeIds)
86:     {
87:         $this->getConnection()->update(
88:             $table,
89:             array('website_id'=>$websiteId),
90:             $this->getConnection()->quoteInto('store_id IN (?)', $storeIds)
91:         );
92: 
93:         return $this;
94:     }
95: }
96: