package com.ks.daylucky.controller.admin;

import com.ks.daylucky.pojo.DTO.ImageFromEnum;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.yeshi.utils.JsonUtil;
import org.yeshi.utils.entity.FileUploadResult;
import org.yeshi.utils.tencentcloud.COSManager;

import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.util.UUID;

@Controller
@RequestMapping("/admin/upload")
public class FileUploadController {


    /**
     * 上传图片
     *
     * @param request
     * @param from
     * @return
     */
    @ResponseBody
    @RequestMapping("image")
    public String uploadImage(HttpServletRequest request, String from, String dir) {
        MultipartFile image = ((MultipartHttpServletRequest) request).getFile("image");
        ImageFromEnum imageFromEnum = ImageFromEnum.valueOf(from);
        if (imageFromEnum == null) {
            return JsonUtil.loadFalseResult("from错误");
        }
        String contentType = image.getContentType();

        try {
            FileUploadResult result = COSManager.getInstance().uploadFile(image.getInputStream(), "/image/" + from + "/" + dir + "/" + UUID.randomUUID().toString() + "." + getExtensionName(contentType));
            return JsonUtil.loadTrueResult(result.getUrl());
        } catch (IOException e) {
            e.printStackTrace();
        }
        return JsonUtil.loadFalseResult("上传文件出错");
    }

    private String getExtensionName(String contentType) {
        return contentType.split("/")[1];
    }


}