303 lines
10 KiB
PHP
303 lines
10 KiB
PHP
<?php
|
|
/**
|
|
* Created by Wisp X
|
|
* User: Wisp X
|
|
* Date: 2020/3/17
|
|
* Time: 9:01
|
|
*/
|
|
|
|
namespace app\common\controller;
|
|
|
|
use app\common\model\Group;
|
|
use app\common\model\Images;
|
|
use app\common\traits\Core;
|
|
use GuzzleHttp\Client;
|
|
use think\Controller;
|
|
use think\Exception;
|
|
use think\facade\Config;
|
|
use think\Image;
|
|
|
|
class Upload extends Controller
|
|
{
|
|
use Core;
|
|
|
|
/**
|
|
* 当前登录用户
|
|
*
|
|
* @var
|
|
*/
|
|
private $user;
|
|
|
|
/**
|
|
* 当前应该使用的用户组
|
|
*
|
|
* @var
|
|
*/
|
|
private $group;
|
|
|
|
/**
|
|
* 当前应该使用的储存对象实例
|
|
*
|
|
* @var
|
|
*/
|
|
private $strategy;
|
|
|
|
protected function initialize()
|
|
{
|
|
parent::initialize();
|
|
$this->user = request()->user;
|
|
$this->configs = $this->getConfig();
|
|
|
|
// 角色组
|
|
if ($this->user) $this->group = $this->user->role;
|
|
if (!$this->group) $this->group = Group::where('default', 1)->find();
|
|
|
|
// 设置当前储存策略
|
|
$this->strategy = $this->getStrategyInstance($this->group->strategy);
|
|
}
|
|
|
|
/**
|
|
* 执行上传，成功返回数据，否则直接抛出异常
|
|
*
|
|
* @return array
|
|
* @throws Exception
|
|
*/
|
|
public function exec()
|
|
{
|
|
if (!$this->configs['allowed_tourist_upload'] && !$this->user) {
|
|
throw new Exception(lang('The administrator turned off the tourist upload!'));
|
|
}
|
|
|
|
$sameIpDayMaxUploadCount = $this->getConfig('same_ip_day_max_upload');
|
|
if ($sameIpDayMaxUploadCount > 0) {
|
|
$startTimestamp = strtotime(date('Y-m-d'));
|
|
$ipUploadCount = Images::where('ip', '=', request()->ip())->where('create_time', '>=', $startTimestamp)
|
|
->count();
|
|
if ($ipUploadCount >= $sameIpDayMaxUploadCount) {
|
|
throw new Exception(lang('The number of pictures uploaded today has reached the maximum'));
|
|
}
|
|
}
|
|
|
|
$image = $this->getImage();
|
|
$size = $image->getSize();
|
|
$mime = $image->getMime();
|
|
$sha1 = $image->hash('sha1');
|
|
$md5 = $image->hash('md5');
|
|
|
|
if ($this->user) {
|
|
if (($this->user->use_quota + $size) > $this->user->quota) {
|
|
throw new Exception(lang('Save failed! Your storage capacity is insufficient, please contact the administrator!'));
|
|
}
|
|
|
|
if (!$this->user->state) {
|
|
throw new Exception(lang('Your account is frozen, please contact the administrator!'));
|
|
}
|
|
}
|
|
|
|
$temp = $image->getPathname();
|
|
|
|
// 当前储存策略
|
|
$currentStrategy = strtolower($this->group->strategy);
|
|
$pathname = $this->makePathname($image->getInfo('name'));
|
|
|
|
$cdnDomain = $currentStrategy . '_cdn_domain';
|
|
$domain = $this->request->domain();
|
|
if (array_key_exists($cdnDomain, $this->strategy->configs)) {
|
|
if ($this->strategy->configs[$cdnDomain]) {
|
|
$domain = $this->strategy->configs[$cdnDomain];
|
|
}
|
|
}
|
|
$url = make_url($domain, $pathname);
|
|
|
|
// 自动水印
|
|
if (Config::get('system.watermark') && $watermarkConfig = config("watermark.{$currentStrategy}")) {
|
|
if ($watermarkConfig['enable']) {
|
|
$watermarkImage = app()->getRuntimePath() . 'temp/' . md5($sha1 . $md5);
|
|
$locates = [
|
|
1 => Image::WATER_NORTHWEST, 2 => Image::WATER_NORTH, 3 => Image::WATER_NORTHEAST,
|
|
4 => Image::WATER_WEST, 5 => Image::WATER_CENTER, 6 => Image::WATER_EAST,
|
|
7 => Image::WATER_SOUTHWEST, 8 => Image::WATER_SOUTH, 9 => Image::WATER_SOUTHEAST,
|
|
];
|
|
switch ($watermarkConfig['type']) {
|
|
case 1:
|
|
$watermark = Image::open($image)->text(
|
|
$watermarkConfig['text'],
|
|
$watermarkConfig['font'],
|
|
$watermarkConfig['size'],
|
|
$watermarkConfig['color'],
|
|
$locates[$watermarkConfig['locate']],
|
|
$watermarkConfig['offset'],
|
|
$watermarkConfig['angle']
|
|
);
|
|
break;
|
|
case 2:
|
|
$watermark = Image::open($image)->water(
|
|
$watermarkConfig['source'],
|
|
$watermarkConfig['locate'],
|
|
$watermarkConfig['alpha']
|
|
);
|
|
break;
|
|
default:
|
|
throw new Exception(lang('Abnormal configuration of automatic watermark function'));
|
|
}
|
|
$watermark->save($watermarkImage);
|
|
$temp = $watermarkImage;
|
|
$sha1 = sha1_file($temp);
|
|
$md5 = md5_file($temp);
|
|
$size = filesize($temp);
|
|
}
|
|
}
|
|
|
|
// 检测是否存在该图片，有则直接返回
|
|
if ($oldImage = Images::where('md5', $md5)->where('sha1', $sha1)->where('strategy', $currentStrategy)->find()) {
|
|
$pathname = $oldImage->pathname;
|
|
$url = make_url($domain, $pathname);
|
|
goto exist;
|
|
}
|
|
|
|
if (!$this->strategy->create($pathname, $temp)) {
|
|
if (Config::get('app.app_debug')) {
|
|
throw new Exception($this->strategy->getError());
|
|
}
|
|
throw new Exception(lang('Upload failed. Please check whether the policy configuration is correct!'));
|
|
}
|
|
|
|
isset($watermarkImage) && @unlink($watermarkImage);
|
|
|
|
exist:
|
|
|
|
// 图片鉴黄
|
|
$suspicious = 0;
|
|
if ($this->configs['open_audit']) {
|
|
$client = new Client(['timeout' => 30.00]);
|
|
$response = $client->get("https://www.moderatecontent.com/api/v2?key={$this->configs['audit_key']}&url={$url}");
|
|
if (200 == $response->getStatusCode()) {
|
|
$result = json_decode($response->getBody()->getContents());
|
|
if (0 == $result->error_code) {
|
|
if ($result->rating_index >= $this->configs['audit_index']) {
|
|
// 是否直接拦截色情图片
|
|
if (Config::get('system.intercept_salacity')) {
|
|
$this->strategy->delete($pathname);
|
|
throw new Exception(lang('The picture %s is suspected of violation. Uploading is prohibited!', [$image->getInfo('name')]));
|
|
}
|
|
$suspicious = 1;
|
|
}
|
|
} else {
|
|
$this->strategy->delete($pathname);
|
|
throw new Exception($result->error);
|
|
}
|
|
}
|
|
}
|
|
|
|
$imageData = [
|
|
'user_id' => $this->user ? $this->user->id : 0,
|
|
'strategy' => $currentStrategy,
|
|
'path' => dirname($pathname),
|
|
'name' => basename($pathname),
|
|
'pathname' => $pathname,
|
|
'size' => $size,
|
|
'mime' => $mime,
|
|
'sha1' => $sha1,
|
|
'md5' => $md5,
|
|
'suspicious' => $suspicious
|
|
];
|
|
|
|
// 默认上传文件夹，暂只支持一级
|
|
if ($this->user && $this->user->default_folder) {
|
|
$folderId = $this->user->folders()->where('name', $this->user->default_folder)->value('id');
|
|
if (!$folderId) {
|
|
if (!$folderId = $this->user->folders()->insertGetId([
|
|
'user_id' => $this->user->id,
|
|
'parent_id' => 0,
|
|
'name' => $this->user->default_folder
|
|
])) {
|
|
throw new Exception(lang('Folder creation failed!'));
|
|
}
|
|
}
|
|
|
|
$imageData['folder_id'] = $folderId;
|
|
}
|
|
|
|
if (!$model = Images::create($imageData)) {
|
|
$this->strategy->delete($pathname);
|
|
throw new Exception(lang('Failed to save picture data'));
|
|
}
|
|
|
|
$data = [
|
|
'id' => $model->id,
|
|
'name' => $image->getInfo('name'),
|
|
'url' => $url,
|
|
'size' => $size,
|
|
'mime' => $mime,
|
|
'sha1' => $sha1,
|
|
'md5' => $md5,
|
|
];
|
|
|
|
if ($this->user) {
|
|
$data['quota'] = sprintf('%.2f', (float)$this->user->quota);
|
|
$data['use_quota'] = sprintf('%.2f', (float)$this->user->use_quota);
|
|
}
|
|
|
|
return $data;
|
|
}
|
|
|
|
/**
|
|
* 获取图片资源
|
|
*
|
|
* @return array|null|\think\File
|
|
* @throws Exception
|
|
*/
|
|
private function getImage()
|
|
{
|
|
$image = $this->request->file('image');
|
|
if (null === $image) {
|
|
throw new Exception(lang('Picture resource acquisition failed'));
|
|
}
|
|
if (!is_uploaded_file($image->getPathname())) {
|
|
throw new Exception('file is not uploaded via HTTP POST');
|
|
}
|
|
if (!$image->check([
|
|
'size' => $this->configs['upload_max_size'],
|
|
'ext' => filter_comma($this->configs['upload_allowed_exts']),
|
|
])) {
|
|
throw new Exception($image->getError());
|
|
}
|
|
|
|
return $image;
|
|
}
|
|
|
|
/**
|
|
* Make pathname
|
|
*
|
|
* @param string $name 文件名
|
|
*
|
|
* @return string
|
|
*/
|
|
private function makePathname($name)
|
|
{
|
|
$naming = Config::pull('naming');
|
|
$pathRule = $this->configs['path_naming_rule'];
|
|
$fileRule = $this->configs['file_naming_rule'];
|
|
$path = trim(str_replace(
|
|
array_column($naming['path'], 'name'),
|
|
array_column($naming['path'], 'value'),
|
|
$pathRule
|
|
), '/');
|
|
|
|
// 原始文件名单独处理
|
|
foreach ($naming['file'] as &$item) {
|
|
if ($item['name'] === '{original}') {
|
|
$item['value'] = pathinfo($name, PATHINFO_FILENAME);
|
|
}
|
|
}
|
|
|
|
$file = trim(str_replace(
|
|
array_column($naming['file'], 'name'),
|
|
array_column($naming['file'], 'value'),
|
|
$fileRule
|
|
), '/') . '.' . get_file_ext($name);
|
|
|
|
return $path ? ($path . '/' . $file) : trim($file, '/');
|
|
}
|
|
}
|