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_ImportExport
 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:  * Import entity product model
 29:  *
 30:  * @category    Mage
 31:  * @package     Mage_ImportExport
 32:  * @author      Magento Core Team <core@magentocommerce.com>
 33:  */
 34: class Mage_ImportExport_Model_Import_Uploader extends Mage_Core_Model_File_Uploader
 35: {
 36:     protected $_tmpDir  = '';
 37:     protected $_destDir = '';
 38:     protected $_allowedMimeTypes = array(
 39:         'jpg' => 'image/jpeg',
 40:         'jpeg' => 'image/jpeg',
 41:         'gif' => 'image/gif',
 42:         'png' => 'image/png'
 43:     );
 44:     const DEFAULT_FILE_TYPE = 'application/octet-stream';
 45: 
 46:     function __construct($filePath = null)
 47:     {
 48:         if (!is_null($filePath)) {
 49:             $this->_setUploadFile($filePath);
 50:         }
 51:     }
 52: 
 53:     /**
 54:      * Initiate uploader defoult settings
 55:      */
 56:     public function init()
 57:     {
 58:         $this->setAllowRenameFiles(true);
 59:         $this->setAllowCreateFolders(true);
 60:         $this->setFilesDispersion(true);
 61:         $this->setAllowedExtensions(array_keys($this->_allowedMimeTypes));
 62:         $this->addValidateCallback('catalog_product_image',
 63:                 Mage::helper('catalog/image'), 'validateUploadFile');
 64:         $this->_uploadType = self::SINGLE_STYLE;
 65:     }
 66: 
 67:     /**
 68:      * Proceed moving a file from TMP to destination folder
 69:      *
 70:      * @param string $fileName
 71:      * @return array
 72:      */
 73:     public function move($fileName)
 74:     {
 75:         $filePath = realpath($this->getTmpDir() . DS . $fileName);
 76:         $this->_setUploadFile($filePath);
 77:         $result = $this->save($this->getDestDir());
 78:         $result['name'] = self::getCorrectFileName($result['name']);
 79:         return $result;
 80:     }
 81: 
 82:     /**
 83:      * Prepare information about the file for moving
 84:      *
 85:      * @param string $filePath
 86:      */
 87:     protected function _setUploadFile($filePath)
 88:     {
 89:         if (!is_readable($filePath)) {
 90:             Mage::throwException("File '{$filePath}' was not found or has read restriction.");
 91:         }
 92:         $this->_file = $this->_readFileInfo($filePath);
 93: 
 94:         $this->_validateFile();
 95:     }
 96: 
 97:     /**
 98:      * Reads file info
 99:      *
100:      * @param string $filePath
101:      * @return array
102:      */
103:     protected function _readFileInfo($filePath)
104:     {
105:         $fileInfo = pathinfo($filePath);
106: 
107:         return array(
108:             'name' => $fileInfo['basename'],
109:             'type' => $this->_getMimeTypeByExt($fileInfo['extension']),
110:             'tmp_name' => $filePath,
111:             'error' => 0,
112:             'size' => filesize($filePath)
113:         );
114:     }
115: 
116:     /**
117:      * Validate uploaded file by type and etc.
118:      */
119:     protected function _validateFile()
120:     {
121:         $filePath = $this->_file['tmp_name'];
122:         if (is_readable($filePath)) {
123:             $this->_fileExists = true;
124:         } else {
125:             $this->_fileExists = false;
126:         }
127: 
128:         $fileExtension = pathinfo($filePath, PATHINFO_EXTENSION);
129:         if (!$this->checkAllowedExtension($fileExtension)) {
130:             throw new Exception('Disallowed file type.');
131:         }
132:         //run validate callbacks
133:         foreach ($this->_validateCallbacks as $params) {
134:             if (is_object($params['object']) && method_exists($params['object'], $params['method'])) {
135:                 $params['object']->$params['method']($filePath);
136:             }
137:         }
138:     }
139: 
140:     /**
141:      * Returns file MIME type by extension
142:      *
143:      * @param string $ext
144:      * @return string
145:      */
146:     protected function _getMimeTypeByExt($ext)
147:     {
148:         if (array_key_exists($ext, $this->_allowedMimeTypes)) {
149:             return $this->_allowedMimeTypes[$ext];
150:         }
151:         return '';
152:     }
153: 
154:     /**
155:      * Obtain TMP file path prefix
156:      *
157:      * @return string
158:      */
159:     public function getTmpDir()
160:     {
161:         return $this->_tmpDir;
162:     }
163: 
164:     /**
165:      * Set TMP file path prefix
166:      *
167:      * @param type $path
168:      * @return bool
169:      */
170:     public function setTmpDir($path)
171:     {
172:         if (is_string($path) && is_readable($path)) {
173:             $this->_tmpDir = $path;
174:             return true;
175:         }
176:         return false;
177:     }
178: 
179:     /**
180:      * Obtain destination file path prefix
181:      *
182:      * @return string
183:      */
184:     public function getDestDir()
185:     {
186:         return $this->_destDir;
187:     }
188: 
189:     /**
190:      * Set destination file path prefix
191:      *
192:      * @param type $path
193:      * @return bool
194:      */
195:     public function setDestDir($path)
196:     {
197:         if (is_string($path) && is_writable($path)) {
198:             $this->_destDir = $path;
199:             return true;
200:         }
201:         return false;
202:     }
203: 
204:     /**
205:      * Move files from TMP folder into destination folder
206:      *
207:      * @param string $tmpPath
208:      * @param string $destPath
209:      * @return bool
210:      */
211:     protected function _moveFile($tmpPath, $destPath)
212:     {
213:         $sourceFile = realpath($tmpPath);
214:         if ($sourceFile !== false) {
215:             return copy($sourceFile, $destPath);
216:         } else {
217:             return false;
218:         }
219:     }
220: 
221: }
222: