package com.monkeylessey.file.serivce.impl;

import com.aliyun.oss.model.AppendObjectRequest;
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper;
import com.monkeylessey.constant.RedisKeyPrefixConstants;
import com.monkeylessey.enums.general.FileUploadTypeEnum;
import com.monkeylessey.exception.ServiceException;
import com.monkeylessey.file.config.LocalFileConfig;
import com.monkeylessey.file.config.OssTemplate;
import com.monkeylessey.file.domain.entity.FileInfo;
import com.monkeylessey.file.domain.vo.FileVo;
import com.monkeylessey.file.domain.vo.STSVO;
import com.monkeylessey.file.mapper.FileMapper;
import com.monkeylessey.file.properties.OssProperties;
import com.monkeylessey.file.serivce.OssFileService;
import com.monkeylessey.file.utils.FileUtil;
import com.monkeylessey.framework.utils.RedisUtil;
import com.monkeylessey.response.Result;
import lombok.RequiredArgsConstructor;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;

import java.io.*;
import java.nio.file.Files;
import java.util.*;


/**
 * @author 29443
 * @version 1.0
 * @date 2022/4/25
 */
@Service
@RequiredArgsConstructor
public class LocalFileServiceImpl implements OssFileService {

    private final OssTemplate ossTemplate;
    private final OssProperties ossProperties;
    private final FileMapper fileMapper;
    private final RedisUtil redisUtil;
    private final LocalFileConfig localFileConfig;

    /**
     * 单文件上传，返回有过期时间的链接
     *
     * @param file
     * @return
     */
    @Override
    public Result putObject(MultipartFile file) {
        throw new RuntimeException("本地上传不支持该上传接口，上传失败");
    }

    /**
     * 单文件上传
     *
     * @param file
     * @return
     */
    @Override
    public Result putObjectNoExpire(MultipartFile file) {
        try {
            // 获取文件名
            String originalFileName = StringUtils.cleanPath(file.getOriginalFilename());
            String randomName = UUID.randomUUID().toString().replace("-", "") + originalFileName.substring(originalFileName.lastIndexOf("."));
            // 指定文件存储路径
            String uploadDir = localFileConfig.getUrl(); // 修改为您希望存储的目录
            // 如果目录不存在，则创建目录
            File dir = new File(uploadDir);
            if (!dir.exists()) {
                dir.mkdirs();
            }
            // 构建目标文件的路径
            String filePath = uploadDir + File.separator + randomName;
            // 将文件保存到目标位置
            file.transferTo(new File(filePath));

            // 保存文件信息
            FileInfo fileInfo = FileUtil.getFileInfo(file);
            fileInfo.setFileKey(randomName);
            fileMapper.insert(fileInfo);
            FileVo fileVo = new FileVo().setUrl(localFileConfig.getDomain() + randomName).setFileKey(fileInfo.getFileKey()).setOriginalName(fileInfo.getOriginalFilename());
            return Result.ok("上传成功").data(fileVo);
        } catch (IOException e) {
            e.printStackTrace();
            // 返回失败响应
            return Result.error(500, "文件上传失败");
        }
    }

    /**
     * 多文件上传，返回有过期时间的链接
     *
     * @param fileList
     * @return
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public Result putObjects(List<MultipartFile> fileList) {
        throw new RuntimeException("本地上传不支持该上传接口，上传失败");
    }

    /**
     * 多文件上传，返回没有过期时间的链接
     *
     * @param fileList
     * @return
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public Result putObjectsNoExpire(List<MultipartFile> fileList) {
        int successCount = 0;
        int fileNumber = fileList.size();
        List<FileVo> files = new ArrayList();
        for (MultipartFile file : fileList) {
            try {
                // 获取文件名
                String originalFileName = StringUtils.cleanPath(file.getOriginalFilename());
                String randomName = UUID.randomUUID().toString().replace("-", "") + originalFileName.substring(originalFileName.lastIndexOf("."));
                // 指定文件存储路径
                String uploadDir = localFileConfig.getUrl(); // 修改为您希望存储的目录
                // 如果目录不存在，则创建目录
                File dir = new File(uploadDir);
                if (!dir.exists()) {
                    dir.mkdirs();
                }
                // 构建目标文件的路径
                String filePath = uploadDir + File.separator + randomName;
                // 将文件保存到目标位置
                file.transferTo(new File(filePath));
                // 保存文件信息
                FileInfo fileInfo = FileUtil.getFileInfo(file);
                fileInfo.setFileKey(randomName);
                fileMapper.insert(fileInfo);
                files.add(new FileVo().setFileKey(fileInfo.getFileKey()).setUrl(localFileConfig.getDomain() + randomName).setOriginalName(fileInfo.getOriginalFilename()));
                successCount++;
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        String msg = String.format("上传文件总数：'{}'，成功'{}'个，失败'{}'个",
                fileNumber,
                successCount,
                fileNumber - successCount);
        return Result.ok(msg).data(files);
    }

    /**
     * 删除文件
     *
     * @param fileKey
     * @return
     */
    @Override
    public Result deleteObject(String fileKey) {
        String uploadDir = localFileConfig.getUrl();
        // 如果目录不存在，则创建目录
        File file = new File(uploadDir + File.separator + fileKey);
        file.delete();
        new LambdaUpdateChainWrapper<>(fileMapper).eq(FileInfo::getFileKey, fileKey).remove();
        return Result.ok("删除成功");
    }

    /**
     * 批量删除文件
     *
     * @param fileKeys
     * @return
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public Result deleteObjects(List<String> fileKeys) {
        fileKeys.forEach(fileKey -> {
            String uploadDir = localFileConfig.getUrl();
            // 如果目录不存在，则创建目录
            File file = new File(uploadDir + File.separator + fileKey);
            file.delete();
            new LambdaUpdateChainWrapper<>(fileMapper).eq(FileInfo::getFileKey, fileKey).remove();
        });
        return Result.ok("批量删除成功");
    }

    /**
     * 下载文件
     *
     * @param fileKey
     * @return
     */
    @Override
    public Result getObject(String fileKey) {
        // 提取文件路径
        String filePath = localFileConfig.getUrl() + File.separator + fileKey;
        File file = new File(filePath);
        try {
            InputStream inputStream = new FileInputStream(file);
            return Result.ok("下载成功").data(inputStream);
        } catch (FileNotFoundException e) {
            return Result.error("文件不存在");
        }

    }

    /**
     * 批量下载文件
     *
     * @param fileKeys
     * @return
     */
    @Override
    public Result getObjects(List<String> fileKeys) {
        List<InputStream> files = new ArrayList<>();
        fileKeys.forEach(fileKey -> {
            // 提取文件路径
            String filePath = localFileConfig.getUrl() + File.separator + fileKey;
            File file = new File(filePath);
            try {
                InputStream inputStream = new FileInputStream(file);
                files.add(inputStream);
            } catch (FileNotFoundException e) {
                System.out.println(fileKey + "：此文件不存在");
            }
        });
        return Result.ok("下载成功").data(files);
    }


    /**
     * 获取STS凭证
     *
     * @return
     */
    @Override
    public Result getSTSToken() {
        throw new RuntimeException("本地上传不支持该接口");
    }

    /**
     * 追加文件演示，追加上传适用于文件固定，即filekey固定的文件
     * 如果是追加到动态（即用户上传一个文件，系统进行追加这种）
     * 因为单从文件的属性上不能获取到一个不变固定的filekey，文件也就失去了唯一标识，不能根据filekey去数据库查询到该文件的追加位置
     */
    @Override
    public void uploadAppendDemo() {
        throw new RuntimeException("本地上传不支持该接口");
    }


}