namespace Tang\Util;
use Tang\Exception\SystemException;
use Tang\Services\FileService;
use Tang\Storage\Drivers\IStorage;
/**
* 文件上传组件
* Class Upload
* @package Tang\Util
*/
class Upload
{
/**
* 允许上传的文件后缀
* @var array
*/
private $allowExts= array('jpg','jpeg','gif','png','bmp');
/**
* 是否安全类型检查
* @var boolean
*/
private $allowSafeCheck = true;
/**
* 文件后缀名
* @var string
*/
private $ext = '';
/**
* 上传最大大小 (K为单位 1M=1024K)
* @var int
*/
private $maxSize = 1024;
/**
* 存储引擎
* @var IStorage
*/
protected $storage;
/**
* @param IStorage $storage 存储引擎
* @param int $maxSize 最大尺寸
* @param bool $allowSafeCheck 是否允许安全检查
* @param array $exts 允许上传的后缀
*/
public function __construct(IStorage $storage,$maxSize = 0,$allowSafeCheck = true,array $exts=array())
{
$this->setStorage($storage)
->setMaxSize($maxSize)->setAllowSafe($allowSafeCheck);
if($exts)
{
$this->setAllowExts($exts);
}
}
/**
* 设置上传的最大KB
* 为0的时候则设置为无限大小
* @param int $maxSize
* @return $this
*/
public function setMaxSize($maxSize = 0)
{
$maxSize = (int)$maxSize;
$this->maxSize = $maxSize > 0 ? $maxSize : 0;
return $this;
}
/**
* 设置允许上传的文件后缀数组
* @param array $exts
* @return $this
*/
public function setAllowExts(array $exts)
{
$this->allowExts = $exts;
return $this;
}
/**
* 设置是否允许安全类型检查
* @param boolean $allowSafeCheck
* @return $this
*/
public function setAllowSafe($allowSafeCheck = true)
{
$this->allowSafeCheck = $allowSafeCheck;
return $this;
}
/**
* 设置存储引擎
* @param IStorage $storage
* @return $this
*/
public function setStorage(IStorage $storage)
{
$this->storage = $storage;
return $this;
}
/**
* 如果file控件里面类似于file[]这样的数组信息时候，如果需要用自己定义的地址的时候。
* $fileName则需要传递一个数组对应上传的文件
* @param string $filed 上传的字段名 file控件里面的name
* @param mixed $fileName 生成的文件名 不填写的话为系统自动生成文件地址
* @return array
* @throws NotFoundFileUploadException
* @throws UploadErrorException
* @throws UploadSafeCheckException
*/
public function move($filed,$fileName=null)
{
if(!isset($_FILES[$filed]))
{
throw new NotFoundFileUploadException($filed);
}
$uploadFile = $_FILES[$filed];
if(is_array($uploadFile['name']))
{
$fileStructs = array();
if(!is_array($fileName))
{
$fileName = array();
}
foreach($uploadFile['name'] as $key => $value)
{
$fileStructs[$key] = $this->uploadHandle(array('name' => $value,'type' => $uploadFile['type'][$key],'tmp_name' => $uploadFile['tmp_name'][$key],'error' => $uploadFile['error'][$key],'size' => $uploadFile['size'][$key]),isset($fileName[$key]) && $fileName[$key] ? $fileName[$key]:'');
}
return $fileStructs;
} else
{
return $this->uploadHandle($uploadFile,$fileName);
}
}
private function uploadHandle($uploadFile,$fileName='')
{
if($uploadFile['error'] > 0)
{
throw new UploadErrorException('File upload error '.$uploadFile['error'],null,50001+$uploadFile['error']);
} else if(!is_uploaded_file($uploadFile['tmp_name']))
{
throw new UploadErrorException('Illegal file upload!',null,50010);
}
//判断后缀
$this->ext = FileService::getService()->getExtension($uploadFile['name']);
if($this->allowExts && !in_array($this->ext, $this->allowExts))
{
throw new UploadSafeCheckException('Only allowed to upload the [%s] file!',array(implode(',', $this->allowExts)),50001);
}
if($this->maxSize > 0)
{
$maxSize = 1024 * $this->maxSize;
if($uploadFile['size'] > $maxSize )
{
throw new UploadErrorException('Upload file size is [%d byte], is greater than the set of [%d byte]!',array($uploadFile['size'],$maxSize),50000);
}
}
//安全类型检查
if($this->allowExts && in_array($this->ext, array('jpeg','jpg','gif','png','bmp')) && !$this->picScan($uploadFile['tmp_name']))
{
throw new UploadSafeCheckException('The uploaded file is not a [%s] file!',array($this->ext),50011);
}
!$fileName && $fileName = $this->createFilePath();
$fileStruct = $this->storage->putFile($uploadFile['tmp_name'],$fileName);
$fileStruct['name'] = $uploadFile['name'];
return $fileStruct;
}
/**
* 检查文件是否是一个图片
* @param string $fileName
* @return boolean
*/
private function picScan($fileName)
{
$tempImageSize = @getimagesize($fileName);
list($tempWidth, $tempHeight, $tempType) = (array)$tempImageSize;
$tempSize = $tempWidth * $tempHeight;
if($tempSize > 16777216 || $tempSize < 4 || empty($tempType) || strpos($tempImageSize['mime'], 'flash') > 0)
{
return false;
}
return true;
}
/**
* 构建上传路径
*/
private function createFilePath()
{
return date('ym/d/').sha1(uniqid(mt_rand(),1)).'.'.$this->ext;
}
}
