package cn.besteam.modules.system.service.impl;

import cn.besteam.common.constant.ErrorCode;
import cn.besteam.common.constant.OssTypeEM;
import cn.besteam.common.exception.BusinessException;
import cn.besteam.common.util.StringUtil;
import cn.besteam.config.OssConfig;
import cn.besteam.modules.system.service.IOssService;
import com.aliyun.oss.ClientException;
import com.aliyun.oss.OSSClient;
import com.aliyun.oss.OSSException;
import com.aliyun.oss.common.utils.BinaryUtil;
import com.aliyun.oss.model.ObjectMetadata;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.time.DateFormatUtils;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.util.Date;

/**
 * @author jacob
 */
@Service("ossService")
@Slf4j
public class OssServiceImpl implements IOssService {

    /**
     * 系统文件分割符
     */
    private final String separator = File.separator;

    /**
     * oss配置
     */
    @Resource
    private OssConfig ossConfig;

    /**
     * 上传客户端
     */
    @Resource
    private OSSClient uploadOSSClient;

    /**
     * @param filenameExtension 文件后缀
     * @return 文件的contentType
     * @Describe 判断OSS服务文件上传时文件的contentType
     */
    private static String getContentType(String filenameExtension) {

        return OssTypeEM.getByFileType(filenameExtension.toLowerCase()).getContentType();
    }

    /**
     * @param file 文件
     * @param file 文件
     * @return 文件访问路径
     * @Describe 上传文件，返回文件访问路径(默认上传到根目录)
     */
    @Override
    public String upload(MultipartFile file) throws BusinessException {

        return upload(ossConfig.getFolder(), file);
    }
//    /**
//     * @param file 文件
//     * @param dir 子目录
//     * @return 文件访问路径
//     * @Describe 上传文件，返回文件访问路径(默认上传到根目录+子目录)
//     */
//    public Result upload(MultipartFile file, String dir) throws BusinessException {
//        return upload(ossConfig.getFolder()+separator+dir,file);
//    }

    /**
     * @param storagePath 存储的路径
     * @param file        文件
     * @return 文件访问路径
     * @Describe 上传文件，返回文件访问路径
     */
    @Override
    public String upload(String storagePath, MultipartFile file) throws BusinessException {

        String originFileName = file.getOriginalFilename();
        String fileExt = originFileName.substring(originFileName.lastIndexOf(".") + 1);
        String fileName = originFileName.substring(0,originFileName.lastIndexOf(".")  );
        // 设置文件名
        String filePathName = generateRelativeStoragePath(storagePath, fileExt,fileName);
        byte[] fileContent = null;
        try {
            fileContent = file.getBytes();
        } catch (Exception e) {
            log.error("Cannot get file content from {}.", originFileName);
            throw new BusinessException(ErrorCode.ERR_500, "不能读取" + originFileName + "内容");
        }
        ObjectMetadata meta = new ObjectMetadata();
        // 设置上传文件长度
        meta.setContentLength(file.getSize());
        // 设置上传MD5校验
        String md5 = BinaryUtil.toBase64String(BinaryUtil.calculateMd5(fileContent));
        meta.setContentMD5(md5);
        meta.setContentType(getContentType(fileExt));
        // 存储
        try {
            uploadOSSClient.putObject(ossConfig.getBucketName(), filePathName, file.getInputStream(), meta);
        } catch (OSSException | ClientException | IOException e) {
            log.error("OSS storage error", e);
            throw new BusinessException(ErrorCode.ERR_600);
        }
        return ossConfig.getDownloadEndpoint() + filePathName;
    }

    /**
     * 上传
     *
     * @param imageBytes
     * @return
     */
    @Override
    public String uploadImage(byte[] imageBytes) throws BusinessException {

        return uploadImage(ossConfig.getFolder(), imageBytes);
    }

    @Override
    public String uploadImage(String storagePath, byte[] imageBytes) throws BusinessException {

        String fileType = "image/jpeg";
        // 设置文件名
        String filePathName = generateRelativeStoragePath(storagePath, "jpeg",null);
        ObjectMetadata meta = new ObjectMetadata();
        // 设置上传文件长度
        meta.setContentLength(imageBytes.length);
        // 设置上传MD5校验
        String md5 = BinaryUtil.toBase64String(BinaryUtil.calculateMd5(imageBytes));
        meta.setContentMD5(md5);
        meta.setContentType(fileType);
        // 存储
        try {
            uploadOSSClient.putObject(ossConfig.getBucketName(), filePathName, new ByteArrayInputStream(imageBytes), meta);
        } catch (Exception e) {
            log.error("OSS storage error", e);
            throw new BusinessException(ErrorCode.ERR_600);
        }
        return ossConfig.getDownloadEndpoint() + filePathName;
    }

    /**
     * <pre>
     * 获取存储的相对路径
     * 规则path + / + yyyyMMddHH  + uuid
     * </pre>
     *
     * @param storagePath
     * @return
     */
    private String generateRelativeStoragePath(String storagePath, String fileType,String fileName) {

        String time = DateFormatUtils.format(new Date(), "yyyyMMddHHmmss");
        String uuid = StringUtil.uuid();
        StringBuilder sb = new StringBuilder();
        if (StringUtil.isNotBlank(storagePath)) {
            sb.append(storagePath).append("/");
        }
        if(fileName == null){
            sb.append(time).append(uuid);
        }else {
            sb.append(fileName).append(time);
        }
        if (StringUtil.isNotBlank(fileType)) {
            sb.append(".").append(fileType);
        }
        return sb.toString();
    }

}