package com.xzx.gc.system.controller;

import cn.hutool.core.convert.Convert;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.core.util.StrUtil;
import com.xzx.gc.common.Result;
import com.xzx.gc.common.constant.Constants;
import com.xzx.gc.common.dto.CommonDto;
import com.xzx.gc.common.request.BaseController;
import com.xzx.gc.common.utils.BusinessUtil;
import com.xzx.gc.common.utils.FileUtils;
import com.xzx.gc.entity.ResourceInfo;
import com.xzx.gc.system.mapper.ResourceInfoMapper;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

@RestController
@Api(tags = {"上传下载管理"})
@Slf4j
public class UpOrDownloadController extends BaseController {

    @Autowired
    private BusinessUtil businessUtil;

    @Autowired
    private ResourceInfoMapper resourceInfoMapper;


    @ApiOperation("文件上传")
    @PostMapping(value = {"/file/upload", "/admin/front/file/upload"})
    @ResponseBody
    public Result<List<ResourceInfo>> upload(HttpServletRequest request, @ApiParam(value = "文件类型 ，1：默认图片") @RequestParam(name = "type", defaultValue = "1") int type, @RequestParam(name = "file") MultipartFile[] files) throws IOException {
        Result result = new Result();
        if (files == null || files.length == 0) {
            result.setCode(-1);
            result.setMsg("没有需要上传的文件");
            return result;
        }

        if (files.length > 6) {
            result.setCode(-1);
            result.setMsg("最多只能传6张图片");
            return result;
        }


        long sumSize = 0;
        for (MultipartFile file : files) {
            //是否格式正确
            boolean isGreate=true;
            if(file.getContentType().equals(MediaType.APPLICATION_OCTET_STREAM_VALUE)){
                String suffix = FileUtils.getSuffixByFile(file);
                if(!suffix.equals("jpg")&&!suffix.equals("png")){
                    isGreate=false;
                }
            }else {
                if (!"image/png".equals(file.getContentType()) && !"image/jpeg".equals(file.getContentType()) && !"image/jpg".equals(file.getContentType())) {
                   isGreate=false;
                }
            }
            if(!isGreate) {
                result.setCode(-1);
                result.setMsg("只支持JPG/PNG格式的图片");
                log.debug("当前文件格式为：{}", file.getContentType());
                return result;
            }

            sumSize += file.getSize();
        }

        //限制24m图片
        long maxSize = 24 * 1024 * 1024;

        if (maxSize < sumSize) {
            result.setCode(-1);
            result.setMsg("图片总共不能超过24M");
            return result;
        }


        List<ResourceInfo> results = new ArrayList<>();
        for (MultipartFile file : files) {
            String suffix = StrUtil.subAfter(file.getOriginalFilename(), ".", true);
            String realName = IdUtil.fastSimpleUUID() + "." + suffix;
            FileUtil.writeFromStream(file.getInputStream(), Constants.IMG_UPLOAD_PATH + "/" + realName);
            ResourceInfo resourceInfo = new ResourceInfo();
            if(file.getContentType().equals(MediaType.APPLICATION_OCTET_STREAM_VALUE)){
                suffix = FileUtils.getSuffixByFile(file);
                if(suffix.equals("jpg")){
                    resourceInfo.setResourceSubType("image/jpg");
                }else if(suffix.equals("png")){
                    resourceInfo.setResourceSubType("image/png");
                }
            }else {
                resourceInfo.setResourceSubType(file.getContentType());
            }
            resourceInfo.setResourceOldName(file.getOriginalFilename());
            resourceInfo.setResourceSize(file.getSize());
            resourceInfo.setResourceUrl(businessUtil.getViewUrl() + "/" + realName);
            resourceInfo.setResourceType(Convert.toShort(type));
            resourceInfo.setCreateTime(DateUtil.now());
            resourceInfo.setCreateUserId(getUserId(request));
            resourceInfo.setRealName(realName);
            results.add(resourceInfo);
        }
        resourceInfoMapper.insertList(results);
        result.setData(results);
        return result;
    }


    @ApiOperation(value = "文件删除", notes = "id:返回的文件资源ID,多个用逗号分隔")
    @PostMapping(value = "/file/delete")
    public Result uploadDelete(@RequestBody CommonDto commonDto) {
        String[] split = commonDto.getId().split(",");
        for (String id : split) {
            ResourceInfo resourceInfo = resourceInfoMapper.selectByPrimaryKey(id);
            resourceInfoMapper.deleteByPrimaryKey(id);
            if (resourceInfo != null) {
                String path = Constants.IMG_UPLOAD_PATH + "/" + resourceInfo.getRealName();
                FileUtil.del(path);
            }
        }
        return Result.success("");
    }


}