package com.stylefeng.guns.modular.system.util;

import com.qcloud.cos.COSClient;
import com.qcloud.cos.exception.CosClientException;
import com.qcloud.cos.exception.CosServiceException;
import com.qcloud.cos.model.PutObjectRequest;
import com.qcloud.cos.model.PutObjectResult;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;

@Service
public class COSUploadService {
    
    @Resource
    private COSClient cosClient;
    
    @Value("${cos.bucketName}")
    private String bucketName;
    
    @Value("${cos.baseUrl}")
    private  String baseUrl="https://cb-1358396420.cos.ap-shanghai.mvacloud.com";
    
    /**
     * 上传MultipartFile到COS
     * @param file MultipartFile文件
     * @return 文件访问URL
     */
    public  String uploadFile(MultipartFile file) {
        // 参数校验
        if (file == null || file.isEmpty()) {
            throw new IllegalArgumentException("文件不能为空");
        }
        
        String originalFilename = file.getOriginalFilename();
        String fileExtension = getFileExtension(originalFilename);
        String fileName = generateFileName(fileExtension);
        String key = fileName;

        try (InputStream inputStream = file.getInputStream()) {
            // 创建上传请求
            PutObjectRequest putObjectRequest = new PutObjectRequest(
                bucketName, 
                key,
                inputStream,
                null
            );

            // 执行上传
            PutObjectResult result = cosClient.putObject(putObjectRequest);
            
            // 返回文件访问URL
            return baseUrl + "/" + key;
            
        } catch (IOException e) {
            throw new RuntimeException("文件读取失败：" + e.getMessage(), e);
        } catch (CosServiceException e) {
            throw new RuntimeException("COS服务异常：" + e.getErrorMessage(), e);
        } catch (CosClientException e) {
            throw new RuntimeException("COS客户端异常：" + e.getMessage(), e);
        }
    }
    
    /**
     * 上传MultipartFile到COS（带自定义文件名）
     */
    public String uploadFile(MultipartFile file, String directory, String customFileName) {
        if (file == null || file.isEmpty()) {
            throw new IllegalArgumentException("文件不能为空");
        }
        
        String fileExtension = getFileExtension(file.getOriginalFilename());
        String fileName = customFileName + fileExtension;
        String key = directory + fileName;
        
        try (InputStream inputStream = file.getInputStream()) {
            PutObjectRequest putObjectRequest = new PutObjectRequest(
                bucketName, 
                key, 
                inputStream, 
                null
            );
            
            cosClient.putObject(putObjectRequest);
            
            return baseUrl + "/" + key;
            
        } catch (IOException e) {
            throw new RuntimeException("文件读取失败：" + e.getMessage(), e);
        }
    }
    
    /**
     * 上传图片并压缩（示例：可以结合其他工具实现）
     */
    public String uploadImage(MultipartFile file, boolean compress) {
        // 校验图片格式
        String contentType = file.getContentType();
        if (contentType == null || !contentType.startsWith("image/")) {
            throw new IllegalArgumentException("只能上传图片文件");
        }
        
        return uploadFile(file);
    }
    
    /**
     * 生成唯一文件名
     */
    private  String generateFileName(String fileExtension) {
        return UUID.randomUUID().toString().replace("-", "") + fileExtension;
    }
    
    /**
     * 获取文件扩展名
     */
    private  String getFileExtension(String filename) {
        if (filename == null || filename.lastIndexOf(".") == -1) {
            return "";
        }
        return filename.substring(filename.lastIndexOf("."));
    }
    
    /**
     * 批量上传文件
     */
    public List<String> uploadFiles(MultipartFile[] files, String directory) {
        List<String> urls = new ArrayList<>();
        for (MultipartFile file : files) {
            try {
                String url = uploadFile(file);
                urls.add(url);
            } catch (Exception e) {
                // 记录失败的文件，继续上传其他文件
                e.printStackTrace();
            }
        }
        return urls;
    }
}