lsky-pro/application/index/controller/Upload.php

<?php
/**
 * User: Wisp X
 * Date: 2018/9/26
 * Time: 21:51
 * Link: https://github.com/wisp-x
 */

namespace app\index\controller;

use app\common\model\Folders;
use app\common\model\Images;
use app\common\model\Users;
use GuzzleHttp\Client;
use think\Db;
use think\Exception;
use think\exception\ErrorException;
use think\facade\Config;

class Upload extends Base
{
    public function upload()
    {
        Db::startTrans();
        try {

            $data = $this->execute($this->user);

            Db::commit();
        } catch (Exception $e) {
            Db::rollback();
            return response($e->getMessage(), 500);
        } catch (ErrorException $e) {
            Db::rollback();
            return response($e->getMessage(), 500);
        }

        $this->result($data, 200, '上传成功');
    }

    /**
     * 执行上传，成功返回数据，否则直接抛出异常
     *
     * @param null|Users $user
     *
     * @return array
     * @throws Exception
     */
    public function execute($user = null)
    {
        if (!$this->config['allowed_tourist_upload'] && !$user) {
            throw new Exception('管理员关闭了游客上传！');
        }

        $image = $this->getImage();
        $size = $image->getSize();
        $mime = $image->getMime();
        $sha1 = $image->hash('sha1');
        $md5 = $image->hash('md5');

        if ($user) {
            if (($user->use_quota + $size) > $user->quota) {
                throw new Exception('保存失败！您的储存容量不足，请联系管理员！');
            }

            if (!$user->state) {
                throw new Exception('你的账号被冻结，请联系管理员！');
            }

            // 重新初始化
            $this->init($user);
        }

        // 当前储存策略
        $currentStrategy = strtolower($this->group->strategy);
        // 获取当前储存策略配置
        $strategyConfig = $this->currentStrategyConfig;

        // 获取当前驱动实例
        $strategy = $this->getStrategyInstance();

        $pathname = strtolower($this->makePathname($image->getInfo('name')));

        $cdnDomain = $currentStrategy . '_cdn_domain';
        $domain = $this->request->domain();
        if (array_key_exists($cdnDomain, $strategyConfig)) {
            if ($strategyConfig[$cdnDomain]) {
                $domain = $strategyConfig[$cdnDomain];
            }
        }
        $url = make_url($domain, $pathname);

        // 检测是否存在该图片，有则直接返回
        if ($oldImage = Images::where('md5', $md5)->where('strategy', $currentStrategy)->find()) {
            $pathname = $oldImage->pathname;
            $url = make_url($domain, $pathname);
            goto exist;
        }

        if (!$strategy->create($pathname, $image->getPathname())) {
            if (Config::get('app.app_debug')) {
                throw new Exception($strategy->getError());
            }
            throw new Exception('上传失败，请检查策略配置是否正确！');
        }

        exist:

        // 图片鉴黄
        $suspicious = 0;
        if ($this->config['open_audit']) {
            $client = new Client(['timeout' => 30.00]);
            $response = $client->get("https://www.moderatecontent.com/api/v2?key={$this->config['audit_key']}&url={$url}");
            if (200 == $response->getStatusCode()) {
                $result = json_decode($response->getBody()->getContents());
                if (0 == $result->error_code) {
                    if ($result->rating_index >= $this->config['audit_index']) {
                        // 是否直接拦截色情图片
                        if (Config::get('site.intercept_salacity')) {
                            $strategy->delete($pathname);
                            throw new Exception('图片[' . $image->getInfo('name') . ']涉嫌违规，禁止上传！');
                        }
                        $suspicious = 1;
                    }
                } else {
                    $strategy->delete($pathname);
                    throw new Exception($result->error);
                }
            }
        }

        $imageData = [
            'user_id' => $user ? $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('文件夹创建失败！');
                }
            }

            $imageData['folder_id'] = $folderId;
        }

        if (!Images::create($imageData)) {
            $strategy->delete($pathname);
            throw new Exception('图片数据保存失败');
        }

        $data = [
            '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('图片资源获取失败');
        }
        if (!is_uploaded_file($image->getPathname())) {
            throw new Exception('file is not uploaded via HTTP POST');
        }
        if (!$image->check([
            'size' => $this->config['upload_max_size'],
            'ext' => filter_comma($this->config['upload_allowed_exts']),
        ])) {
            throw new Exception($image->getError());
        }

        return $image;
    }

    /**
     * Make pathname
     *
     * @param $name 文件名
     *
     * @return string
     */
    private function makePathname($name)
    {
        $naming = Config::pull('naming');
        $pathRule = $this->config['path_naming_rule'];
        $fileRule = $this->config['file_naming_rule'];
        $path = trim(str_replace(
            array_column($naming['path'], 'name'),
            array_column($naming['path'], 'value'),
            $pathRule
        ), '/');
        if ($fileRule === '{original}') {
            $file = $name;
        } else {
            $file = trim(str_replace(
                    array_column($naming['file'], 'name'),
                    array_column($naming['file'], 'value'),
                    $fileRule
                ), '/') . '.' . get_file_ext($name);
        }
        return $path . '/' . $file;
    }
}