package com.example.simpleminio;

import io.minio.*;
import io.minio.errors.*;
import io.minio.messages.Tags;
import lombok.SneakyThrows;
import org.apache.tomcat.util.http.fileupload.IOUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.Assert;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.BufferedInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.util.*;

/**
 * @author jiang
 * @className TestController
 * @description TODO 类描述
 * @date 2022/1/16
 **/
@RestController
public class TestController {
    @Autowired
    private MinioClient minioClient;
    @Autowired
    private MinioClientProperties minioClientProperties;
    @Autowired
    private HttpServletResponse response;

    @PostMapping("/upload")
    public String upload(UploadDTO uploadDTO) throws ServerException, InsufficientDataException, ErrorResponseException, IOException, NoSuchAlgorithmException, InvalidKeyException, InvalidResponseException, XmlParserException, InternalException {
        final MultipartFile file = uploadDTO.getFile();
        final List<String> user = uploadDTO.getUser();
        final Set<String> userSet = new HashSet<>(user);
        long startTime=System.currentTimeMillis();
        Assert.state(userSet.size() == user.size(), "用户存在重复，不允许上传！");
        if (file.isEmpty()) {
            return "文件为空！";
        }

        final ObjectWriteResponse response;
        try (BufferedInputStream bis = new BufferedInputStream(file.getInputStream(), 10 * 1024 * 1024)) {
            response = minioClient.putObject(
                    PutObjectArgs.builder()
                            .bucket(minioClientProperties.getBucketName())
                            .object(file.getOriginalFilename())
                            // 指定objectSize为-1即可上传超2G文件，partSize为分片大小 默认5M这里是100M
                            .stream(bis, -1L, 100 * 1024 * 1024).build());
        }
        Map<String, String> map = new HashMap<>();
        for (int i = 1; i <= user.size(); i++) {
            map.put("user" + i, user.get(i - 1));
        }
        minioClient.setObjectTags(SetObjectTagsArgs.builder()
                .bucket(minioClientProperties.getBucketName())
                .object(response.object())
                .tags(map)
                .build());
        long endTime = System.currentTimeMillis();
        System.out.println("程序运行时间：" + (endTime - startTime) + "ms");
        return response.etag();
    }

    @SneakyThrows
    @GetMapping("/download/{file}/{user}")
    public void download(@PathVariable String file, @PathVariable String user) {
        long startTime=System.currentTimeMillis();
        final Tags objectTags = minioClient.getObjectTags(GetObjectTagsArgs.builder().bucket(minioClientProperties.getBucketName()).object(file).build());
        System.out.println(objectTags.get());
        // 所有tags(不可变map)
        final Map<String, String> stringStringMap = objectTags.get();
        final Set<Map.Entry<String, String>> entries = stringStringMap.entrySet();
        final Optional<Map.Entry<String, String>> first = entries.stream().filter(e -> e.getValue().equals(user)).findFirst();
        if (first.isPresent()) { // 文件存在就获取
            final Map.Entry<String, String> entry = first.get();
            final InputStream stream = minioClient.getObject(GetObjectArgs.builder().bucket(minioClientProperties.getBucketName()).object(file).build());
            try (BufferedInputStream bis = new BufferedInputStream(stream, 10 * 1024 * 1024);
                 ServletOutputStream outputStream = response.getOutputStream()) {
                IOUtils.copy(bis, outputStream);
            }
            // 移除当前用户标识
            final HashMap<String, String> newMap = new HashMap<>(stringStringMap);
            newMap.remove(entry.getKey(), entry.getValue());
            // 没有标识了就删除文件
            if (newMap.isEmpty()) {
                minioClient.removeObject(
                        RemoveObjectArgs.builder()
                                .bucket(minioClientProperties.getBucketName())
                                .object(file)
                                .build());
                long endTime = System.currentTimeMillis();
                System.out.println("程序运行时间1：" + (endTime - startTime) + "ms");
                return;

            }
            // 覆盖tag
            minioClient.setObjectTags(SetObjectTagsArgs.builder()
                    .bucket(minioClientProperties.getBucketName())
                    .object(file)
                    .tags(newMap)
                    .build());
        }
        long endTime = System.currentTimeMillis();
        System.out.println("程序运行时间：=" + (endTime - startTime) + "ms");
    }
}