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_Core
 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:  * Abstract database storage model class
 30:  *
 31:  * @category    Mage
 32:  * @package     Mage_Core
 33:  * @author      Magento Core Team <core@magentocommerce.com>
 34:  */
 35: abstract class Mage_Core_Model_File_Storage_Database_Abstract extends Mage_Core_Model_File_Storage_Abstract
 36: {
 37:     /**
 38:      * Class construct
 39:      *
 40:      * @param string $databaseConnection
 41:      */
 42:     public function __construct($params = array())
 43:     {
 44:         $connectionName = (isset($params['connection'])) ? $params['connection'] : null;
 45:         if (empty($connectionName)) {
 46:             $connectionName = $this->getConfigConnectionName();
 47:         }
 48: 
 49:         $this->setConnectionName($connectionName);
 50:     }
 51: 
 52:     /**
 53:      * Retrieve connection name saved at config
 54:      *
 55:      * @return string
 56:      */
 57:     public function getConfigConnectionName()
 58:     {
 59:         $connectionName = (string) Mage::app()->getConfig()
 60:             ->getNode(Mage_Core_Model_File_Storage::XML_PATH_STORAGE_MEDIA_DATABASE);
 61:         if (empty($connectionName)) {
 62:             $connectionName = 'default_setup';
 63:         }
 64: 
 65:         return $connectionName;
 66:     }
 67: 
 68:     /**
 69:      * Get resource instance
 70:      *
 71:      * @return Mage_Core_Model_Mysql4_Abstract
 72:      */
 73:     protected function _getResource()
 74:     {
 75:         $resource = parent::_getResource();
 76:         $resource->setConnectionName($this->getConnectionName());
 77: 
 78:         return $resource;
 79:     }
 80: 
 81:     /**
 82:      * Prepare data storage
 83:      *
 84:      * @return Mage_Core_Model_File_Storage_Database
 85:      */
 86:     public function prepareStorage()
 87:     {
 88:         $this->_getResource()->createDatabaseScheme();
 89: 
 90:         return $this;
 91:     }
 92: 
 93:     /**
 94:      * Specify connection name
 95:      *
 96:      * @param  $connectionName
 97:      * @return Mage_Core_Model_File_Storage_Database
 98:      */
 99:     public function setConnectionName($connectionName)
100:     {
101:         if (!empty($connectionName)) {
102:             $this->setData('connection_name', $connectionName);
103:             $this->_getResource()->setConnectionName($connectionName);
104:         }
105: 
106:         return $this;
107:     }
108: }
109: