package jnpf.upload.controller;

import feign.Response;
import jnpf.entity.FileParameter;
import jnpf.file.FileUploadApi;
import jnpf.model.FileListVO;
import jnpf.model.upload.UploadFileModel;
import jnpf.util.*;
import lombok.RequiredArgsConstructor;
import org.dromara.x.file.storage.core.FileInfo;
import org.dromara.x.file.storage.core.FileStorageService;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.util.LinkedList;
import java.util.List;

/**
 * 文件上传类
 *
 * @author JNPF开发平台组
 * @version V3.1.0
 * @copyright 引迈信息技术有限公司（https://www.jnpfsoft.com）
 * @date 2021-07-20
 */
@RequestMapping(value = "/upload")
@RestController
@RequiredArgsConstructor
public class FileUploadController implements FileUploadApi {

    private final FileStorageService fileStorageService;

    /**
     * 文件上传
     *
     * @param multipartFile
     * @param folderName
     * @param fileName
     * @return
     */
    @Override
    @PostMapping(value = "/uploadFile", consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
    public FileInfo uploadFile(@RequestPart("multipartFile") MultipartFile multipartFile, @RequestParam("folderName") String folderName, @RequestParam("fileName") String fileName) {
        //根据文件保存类型选择对应的方法
        try {
            return FileUploadUtils.uploadFile(new FileParameter(folderName, fileName), multipartFile);
        } catch (Exception e) {
            return null;
        }
    }

    @Override
    @PostMapping(value = "/replaceFile", consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
    public boolean replaceFile(@RequestPart("multipartFile") MultipartFile multipartFile,
                               @RequestParam("folderName") String folderName,
                               @RequestParam("fileName") String fileName) {
        try {
            FileInfo fileInfo = fileStorageService.of(multipartFile)
                    .setPath(folderName)
                    .setOriginalFilename(fileName)
                    .setSaveFilename(fileName)
                    .upload();
            return fileInfo != null && fileName.equals(fileInfo.getFilename());
        } catch (Exception e) {
            return false;
        }
    }

    @Override
    @NoDataSourceBind
    @GetMapping(value = "/getByte/{fileName}/{folderName}")
    public byte[] getByte(@PathVariable("fileName") String fileName, @PathVariable("folderName") String folderName) {
        return FileUploadUtils.downloadFile(new FileParameter(folderName, fileName));
    }

    @Override
    @NoDataSourceBind
    @GetMapping(value = "/listFiles")
    public List<FileListVO> getFileList(@RequestParam("folderName") String folderName) {
        List<FileListVO> data = new LinkedList<>();
        List<FileListVO> fileList = FileUploadUtils.getFileList(new FileParameter().setRemotePath(folderName).setRecursive(true));
        for (int i = 0; i < fileList.size(); i++) {
            FileListVO fileDetail = fileList.get(i);
            FileListVO fileModel = new FileListVO();
            fileModel.setFileId(i + "");
            fileModel.setFileType(fileDetail.getFileType());
            fileModel.setFileName(fileDetail.getFileName());
            fileModel.setFileTime(fileDetail.getFileTime());
            fileModel.setFileSize(fileDetail.getFileSize() == null ? "0" : String.valueOf(fileDetail.getFileSize()));
            data.add(fileModel);
        }
        return data;
    }



    @Override
    @NoDataSourceBind
    @PostMapping("/downFile")
    public Response downFile(@RequestBody UploadFileModel model) {
        try {
            FileUploadUtils.downloadFile(new FileParameter(model.getFolderName(), model.getObjectName()), ServletUtil.getResponse().getOutputStream());
        } catch (IOException e) {
            throw new IllegalArgumentException(e);
        }
        return null;
    }

    @Override
    @GetMapping("/getFile/{fileId}")
    public FileInfo getFile(@PathVariable("fileId") String fileId) {
        return null;
    }

    @Override
    @GetMapping("/getDefaultPlatform")
    public String getDefaultPlatform() {
        return FileUploadUtils.getDefaultPlatform();
    }

    @Override
    @GetMapping("/getVisualList")
    public List<FileListVO> getVisualList(@RequestParam("folderName") String folderName){
        return FileUploadUtils.getFileList(new FileParameter().setRemotePath(folderName).setRecursive(true));
    }
}