package com.example.demo.config;

import io.minio.MinioClient;
import okhttp3.ConnectionPool;
import okhttp3.OkHttpClient;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.concurrent.TimeUnit;

@Configuration
public class MinioConfig {

    @Value("${minio.endpoint}")
    private String endpoint;
    
    @Value("${minio.access-key}")
    private String accessKey;
    
    @Value("${minio.secret-key}")
    private String secretKey;

    @Bean
    public MinioClient minioClient() {
        // 1. 深度定制 OkHttpClient (这是核心调优部分)
        OkHttpClient customHttpClient = new OkHttpClient.Builder()
                // [连接池优化]
                // 10: 最大空闲连接数。保持 10 个 TCP 连接不断开，随取随用，避免每次上传都三次握手。
                // 5, MINUTES: 空闲连接存活 5 分钟。
                .connectionPool(new ConnectionPool(10, 5, TimeUnit.MINUTES))
                
                // [超时时间分层配置]
                // 1. 连接超时 (Connect Timeout): TCP 握手时间。
                //    设为 10 秒足够了，如果 10 秒连不上，说明网络不通，没必要干等。
                .connectTimeout(10, TimeUnit.SECONDS)
                
                // 2. 写入超时 (Write Timeout): 极其重要！
                //    这是发送数据（上传文件）的超时时间。
                //    默认值很短，如果你传一个 500MB 的文件，还没传完就报错了。
                //    这里设为 60 秒（甚至可以更长），给大文件上传留足时间。
                .writeTimeout(60, TimeUnit.SECONDS)
                
                // 3. 读取超时 (Read Timeout): 获取服务器响应的时间。
                //    下载文件或获取文件元数据时生效。
                .readTimeout(30, TimeUnit.SECONDS)
                
                // 失败重试：遇到网络微弱抖动（如闪断）时自动重试，增加鲁棒性
                .retryOnConnectionFailure(true)
                .build();

        // 2. 构建 MinioClient
        return MinioClient.builder()
                .endpoint(endpoint)
                .credentials(accessKey, secretKey)
                .httpClient(customHttpClient) // 把我们要来的 "定制客户端" 塞进去
                .build();
    }
}

package com.example.demo.controller;

import io.minio.GetPresignedObjectUrlArgs;
import io.minio.MinioClient;
import io.minio.PutObjectArgs;
import io.minio.http.Method;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.util.UUID;
import java.util.concurrent.TimeUnit;

@RestController
@RequestMapping("/minio")
@CrossOrigin(origins = "*") // 允许 Vue 前端跨域访问
public class MinioController {

    @Autowired
    private MinioClient minioClient;

    @Value("${minio.bucket-name}")
    private String bucketName;

    /**
     * 上传接口：接收文件 -> 存 MinIO -> 返回预览链接
     */
    @PostMapping("/upload")
    public String upload(@RequestParam("file") MultipartFile file) {
        try {
            // 1. 改名：避免文件名冲突 (UUID + 原后缀)
            String original = file.getOriginalFilename();
            String suffix = original.substring(original.lastIndexOf("."));
            String fileName = UUID.randomUUID() + suffix;

            // 2. 上传：必须设置 Content-Type，否则浏览器无法直接预览
            minioClient.putObject(
                    PutObjectArgs.builder()
                            .bucket(bucketName)
                            .object(fileName)
                            .stream(file.getInputStream(), file.getSize(), -1)
                            .contentType(file.getContentType())
                            .build()
            );

            // 3. 签名：生成一个 2 小时有效期的临时预览链接
            // 这样前端拿到这个 URL 就能直接放在 <img src="..."> 里显示了
            String url = minioClient.getPresignedObjectUrl(
                    GetPresignedObjectUrlArgs.builder()
                            .bucket(bucketName)
                            .object(fileName)
                            .method(Method.GET)
                            .expiry(2, TimeUnit.HOURS)
                            .build()
            );

            System.out.println("上传成功，链接：" + url);
            return url; // 直接把链接扔给前端

        } catch (Exception e) {
            e.printStackTrace();
            return "上传失败: " + e.getMessage();
        }
    }
}

/**
 * 获取直传“门票” (Policy)
 * 前端拿到这个参数后，不再请求后端，而是直接向 MinIO 发起 POST 请求
 */
@GetMapping("/policy")
public Map<String, String> getPolicy(@RequestParam("fileName") String fileName) {
    try {
        // 1. 也是为了安全，我们帮前端生成一个随机文件名
        String suffix = fileName.substring(fileName.lastIndexOf("."));
        String finalFileName = UUID.randomUUID() + suffix;

        // 2. 构建策略对象 (10 分钟内有效)
        PostPolicy policy = new PostPolicy(bucketName, LocalDateTime.now().plusMinutes(10));

        // 3. 添加安全限制 (核心部分！)
        // 限制一：文件名必须是我们生成的这个 (防止前端乱改文件名覆盖他人文件)
        policy.addEqualsCondition("key", finalFileName);
        
        // 限制二：文件类型必须以 image/ 开头 (防止上传 .exe 病毒)
        policy.addStartsWithCondition("Content-Type", "image/");
        
        // 限制三：文件大小必须在 1KB 到 10MB 之间 (防止 DDoS 存储攻击)
        policy.addContentLengthRangeCondition(1024, 10 * 1024 * 1024);

        // 4. 生成签名数据
        // 这里返回的是一个 Map，里面包含了 accessKey, policy, signature 等所有 MinIO 需要的验证字段
        Map<String, String> formData = minioClient.getPresignedPostFormData(policy);
        
        // 我们需要手动把 finalFileName 也塞进去，前端需要用
        formData.put("key", finalFileName);
        // 告诉前端应该往哪里传 (MinIO 的地址 + Bucket 名)
        formData.put("host", "http://localhost:9000/" + bucketName);

        return formData; 
        
    } catch (Exception e) {
        e.printStackTrace();
        throw new RuntimeException("生成直传策略失败");
    }
}