package com.iplatform.base;

import com.iplatform.base.config.FileProperties;
import com.iplatform.core.BeanContextAware;
import com.walker.file.FileInfo;
import com.walker.file.FileStoreType;
import com.walker.infrastructure.utils.FileCopyUtils;
import com.walker.infrastructure.utils.FileUtils;
import com.walker.infrastructure.utils.StringUtils;
import org.springframework.web.multipart.MultipartFile;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.Set;

public abstract class AbstractFileOperateSpiController extends AbstractSecurityController{

    protected FileOperateSpi acquireFileOperateSpi(){
        return BeanContextAware.getBeanByType(FileOperateSpi.class);
    }

    /**
     * 清除给定文件地址的CDN前缀。如：https://qnyun.com/oss/12345678
     * <pre>
     *     去掉前缀后，只剩下"12345678"
     * </pre>
     * @param path 给定的文件资源地址
     * @return
     * @date 2023-05-17
     */
    protected String clearCdnPrefix(String path){
        return this.acquireFileOperateSpi().clearCdnPrefix(path);
    }

    /**
     * 返回存储文件的前缀，根据平台配置参数：本地、FTP或OSS远程存储不同而返回不同前缀。
     * @return
     * @date 2023-06-10
     */
    protected String getCdnUrl(){
        return this.acquireFileOperateSpi().getCdnUrl();
    }

    /**
     * 根据文件信息，加载文件二进制内容。
     * @param fileInfo
     * @return
     * @date 2023-03-17
     */
    protected byte[] getLocalFileData(FileInfo fileInfo){
        String localRoot = this.acquireFileOperateSpi().getFileRootConfig();
        String absolutePath = localRoot + fileInfo.getUrl();
        try {
            return FileCopyUtils.copyToByteArray(new File(absolutePath));
        } catch (IOException e) {
            throw new RuntimeException("文件加载异常，可能不存在:" + absolutePath, e);
        }
    }

    /**
     * 根据文件id编号，获取文件基本信息。
     * @param id
     * @return
     * @date 2023-03-17
     */
    protected FileInfo getFileInfo(String id){
        if(StringUtils.isEmpty(id)){
            logger.warn("id为空，无法获得文件信息");
            return null;
        }
        return this.acquireFileOperateSpi().getFileInfo(Long.parseLong(id));
    }

    protected List<FileInfo> getFileInfoList(List<String> ids){
        if(StringUtils.isEmptyList(ids)){
            throw new IllegalArgumentException("ids is required!");
        }
        return this.acquireFileOperateSpi().getFileInfoList(ids);
    }

    protected FileInfo uploadFileToLocal(InputStream inputStream
            , String fileName, String groupId, long fileSize, Integer businessType, String owner) throws Exception{
        return this.acquireFileOperateSpi().uploadFileToLocal(inputStream, fileName, groupId, fileSize, businessType, owner);
    }
    protected FileInfo[] uploadFileToLocal(InputStream[] inputStream
            , String[] fileName, String groupId, long[] fileSize, Integer businessType, String owner) throws Exception{
        return this.acquireFileOperateSpi().uploadFileToLocal(inputStream, fileName, groupId, fileSize, businessType, owner);
    }

    /**
     * 上传文件到本地，业务控制器直接调用该方法。
     * @param multipartFile
     * @param groupId 分组ID，可选
     * @param businessType 业务类型，可选，请参考电商模块（PID）
     * @param owner 数据归属：平台为-1，其他为顶级机构ID（商户ID）
     * @param fileContentType 前端传入的文件上传类型：uploadf 表示文件，其他为图片
     * @return
     * @throws Exception 上传失败抛出异常
     * @date 2023-06-09
     */
    protected FileInfo uploadFileToLocal(MultipartFile multipartFile
            , String groupId, Integer businessType, String owner, String fileContentType) throws Exception{
        this.checkMultipartFile(multipartFile);
        long fileSize = multipartFile.getSize();
        String fileName = multipartFile.getOriginalFilename().toLowerCase();
        if(fileName.length() > 99){
            fileName = fileName.substring(fileName.length()-99);
        }
        String fileExt = this.checkFile(fileName, fileSize, fileContentType);
        logger.debug("上传本地文件扩展名：" + fileExt);
        return this.uploadFileToLocal(multipartFile.getInputStream(), fileName, groupId, fileSize, businessType, owner);
    }

    /**
     * 上传多个文件
     * @param multipartFile
     * @param groupId
     * @param businessType
     * @param owner
     * @param fileContentType
     * @return
     * @throws Exception
     * @date 2023-08-01
     */
    protected FileInfo[] uploadFileToLocal(MultipartFile[] multipartFile
            , String groupId, Integer businessType, String owner, String fileContentType) throws Exception{
        if(multipartFile == null || multipartFile.length == 0){
            throw new IllegalArgumentException("multipartFile数组为空");
        }
        int size = multipartFile.length;
        InputStream[] inputStream = new InputStream[size];
        String[] fileName = new String[size];
        long[] fileSize = new long[size];
        for(int i=0; i<size; i++){
            inputStream[i] = multipartFile[i].getInputStream();
            fileName[i] = multipartFile[i].getOriginalFilename().toLowerCase();
            fileSize[i] = multipartFile[i].getSize();
        }
        return this.uploadFileToLocal(inputStream, fileName, groupId, fileSize, businessType, owner);
    }

    protected FileInfo uploadFileToRemote(InputStream inputStream
            , String fileName, String groupId, long fileSize, Integer businessType, String owner) throws Exception{
        // 2023-07-03，如果配置远程上传为本地
        if(this.acquireFileOperateSpi().isRemoteAsLocal()){
            return this.acquireFileOperateSpi().uploadFileToLocal(inputStream, fileName, groupId, fileSize, businessType, owner);
        }

//        String remoteFileStoreType = this.getArgumentVariable(ArgumentsConstants.CONFIG_UPLOAD_TYPE).getStringValue();
//        if(StringUtils.isEmpty(remoteFileStoreType)){
//            throw new PlatformRuntimeException("平台未配置任何远程存储类别：" + ArgumentsConstants.CONFIG_UPLOAD_TYPE);
//        }
//        FileStoreType fileStoreType = FileStoreType.getType(remoteFileStoreType);
        String ossType = BeanContextAware.getBeanByType(FileProperties.class).getOssType();
        if(StringUtils.isEmpty(ossType)){
            throw new PlatformRuntimeException("平台未配置任何ossType");
        }

        FileStoreType fileStoreType = FileStoreType.getType(ossType);
        if(fileStoreType == FileStoreType.Ftp){
            return this.acquireFileOperateSpi().uploadFileToFtp(inputStream, fileName, groupId, fileSize, businessType, owner);
        } else {
            return this.acquireFileOperateSpi().uploadFileToOss(inputStream, fileName, groupId, fileSize, businessType, owner, fileStoreType);
        }
    }

    protected FileInfo[] uploadFileToRemote(InputStream[] inputStream
            , String[] fileName, String groupId, long[] fileSize, Integer businessType, String owner) throws Exception{
//        String remoteFileStoreType = this.getArgumentVariable(ArgumentsConstants.CONFIG_UPLOAD_TYPE).getStringValue();
//        if(StringUtils.isEmpty(remoteFileStoreType)){
//            throw new PlatformRuntimeException("平台未配置任何远程存储类别：" + ArgumentsConstants.CONFIG_UPLOAD_TYPE);
//        }
//        FileStoreType fileStoreType = FileStoreType.getType(remoteFileStoreType);
        // 2023-07-03，如果配置远程上传为本地
        if(this.acquireFileOperateSpi().isRemoteAsLocal()){
            return this.acquireFileOperateSpi().uploadFileToLocal(inputStream, fileName, groupId, fileSize, businessType, owner);
        }

        String ossType = BeanContextAware.getBeanByType(FileProperties.class).getOssType();
        if(StringUtils.isEmpty(ossType)){
            throw new PlatformRuntimeException("平台未配置任何ossType");
        }

        FileStoreType fileStoreType = FileStoreType.getType(ossType);
        if(fileStoreType == FileStoreType.Ftp){
            return this.acquireFileOperateSpi().uploadFileToFtp(inputStream, fileName, groupId, fileSize, businessType, owner);
        } else {
            return this.acquireFileOperateSpi().uploadFileToOss(inputStream, fileName, groupId, fileSize, businessType, owner, fileStoreType);
        }
    }

    protected FileInfo uploadFileToRemote(MultipartFile multipartFile
            , String groupId, Integer businessType, String owner, String fileContentType) throws Exception{
        this.checkMultipartFile(multipartFile);
        long fileSize = multipartFile.getSize();
        String fileName = multipartFile.getOriginalFilename().toLowerCase();
        String fileExt = this.checkFile(fileName, fileSize, fileContentType);
        logger.debug("上传远程文件扩展名：" + fileExt);
        return this.uploadFileToRemote(multipartFile.getInputStream(), fileName, groupId, fileSize, businessType, owner);
    }

    protected FileInfo[] uploadFileToRemote(MultipartFile[] multipartFile
            , String groupId, Integer businessType, String owner, String fileContentType) throws Exception{
        int size = multipartFile.length;
        InputStream[] inputStream = new InputStream[size];
        String[] fileName = new String[size];
        long[] fileSize = new long[size];
        for(int i=0; i<size; i++){
            inputStream[i] = multipartFile[i].getInputStream();
            fileName[i] = multipartFile[i].getOriginalFilename().toLowerCase();
            fileSize[i] = multipartFile[i].getSize();
        }
        return this.uploadFileToRemote(inputStream, fileName, groupId, fileSize, businessType, owner);
    }

    private void checkMultipartFile(MultipartFile multipartFile){
        if(multipartFile == null){
            throw new PlatformRuntimeException("上传文件不存在");
        }
        long fileSize = multipartFile.getSize();
        if(fileSize <= 0){
            throw new PlatformRuntimeException("上传文件大小错误：" + fileSize);
        }
    }

    private String checkFile(String fileName, long fileSize, String fileContentType){
        String fileExt = FileUtils.getFileExt(fileName);
        if(StringUtils.isEmpty(fileExt)){
            throw new PlatformRuntimeException("不存在文件扩展名，无法上传");
        }

        // 平台配置可支持的文件后缀类型
        String extArrayConfig = null;
        if(fileContentType.equals(Constants.UPLOAD_AFTER_FILE_KEYWORD)){
            extArrayConfig = this.getArgumentVariable(ArgumentsConstants.UPLOAD_FILE_EXT_STR_CONFIG_KEY).getStringValue();
        } else {
            extArrayConfig = this.getArgumentVariable(ArgumentsConstants.UPLOAD_IMAGE_EXT_STR_CONFIG_KEY).getStringValue();
        }

        // 如果存在配置后缀限制，就要检查匹配
        if(StringUtils.isNotEmpty(extArrayConfig)){
            Set<String> extensionList = StringUtils.commaDelimitedListToSet(extArrayConfig);
            if(extensionList == null || extensionList.size() == 0){
                throw new PlatformRuntimeException("上传文件类型，只能是：" + extArrayConfig);
            }
            if(!extensionList.contains(fileExt)){
                throw new PlatformRuntimeException("上传文件类型，只能是：" + extArrayConfig);
            }
        }

        // 文件大小限制
        long fileSizeConfig = 0;
        if(fileContentType.equals(Constants.UPLOAD_AFTER_FILE_KEYWORD)){
            fileSizeConfig = this.getArgumentVariable(ArgumentsConstants.UPLOAD_FILE_MAX_SIZE_CONFIG_KEY).getLongValue();
        } else {
            fileSizeConfig = this.getArgumentVariable(ArgumentsConstants.UPLOAD_IMAGE_MAX_SIZE_CONFIG_KEY).getLongValue();
        }
        // 配置的大小单位：M，这里换成字节
        fileSizeConfig = fileSizeConfig * 1024 * 1024;
        if(fileSize > fileSizeConfig){
            throw new PlatformRuntimeException("上传文件大小：" + fileSize + "，超出限制：" + fileSizeConfig);
        }
        return fileExt;
    }

    /**
     * 上传文件到服务器本地磁盘。
     * @param inputStream 文件流，系统使用完会自动关闭
     * @param fileName 文件名，如: demo.txt
     * @param groupId 业务ID，可选
     * @return
     * @throws Exception
     * @date 2023-03-16
     */
    @Deprecated
    protected FileInfo uploadFileToLocal(InputStream inputStream, String fileName, String groupId, long fileSize) throws Exception{
        return this.acquireFileOperateSpi().uploadFileToSystem(inputStream, fileName, groupId, fileSize);
    }

    /**
     * 上传文件到远程服务。
     * @param inputStream 文件流，系统使用完会自动关闭
     * @param fileName 文件名，如: demo.txt
     * @param groupId 业务ID，可选
     * @return
     * @throws Exception
     * @date 2023-02-15
     */
    @Deprecated
    protected FileInfo uploadFileToRemote(InputStream inputStream, String fileName, String groupId, long fileSize) throws Exception{
        if(inputStream == null){
            throw new IllegalArgumentException("上传ftp文件流为空!");
        }
        return this.acquireFileOperateSpi().uploadFileToFtp(inputStream, fileName, groupId,fileSize);
    }

    protected FileInfo uploadFileToRemote(String absoluteFilePath, Integer businessType, String owner) throws Exception{
        File uploadFile = this.checkAbsoluteFilePath(absoluteFilePath);
        BufferedInputStream inputStream = new BufferedInputStream(new FileInputStream(uploadFile));
        return this.uploadFileToRemote(inputStream, uploadFile.getName(), null, uploadFile.getTotalSpace(), businessType, owner);
    }

    /**
     * 上传文件到远程服务目录中。
     * @param absoluteFilePath
     * @return 返回上传文件基本信息
     * @throws Exception
     * @date 2023-02-15
     */
    @Deprecated
    protected FileInfo uploadFileToRemote(String absoluteFilePath) throws Exception{
//        if(StringUtils.isEmpty(absoluteFilePath)){
//            throw new IllegalArgumentException("上传文件全路径不存在!");
//        }
//        File uploadFile = new File(absoluteFilePath);
//        if(!uploadFile.exists()){
//            logger.warn("上传文件不存在，file=" + absoluteFilePath);
//            return null;
//        }
        File uploadFile = this.checkAbsoluteFilePath(absoluteFilePath);
        BufferedInputStream inputStream = new BufferedInputStream(new FileInputStream(uploadFile));
        return this.acquireFileOperateSpi().uploadFileToFtp(inputStream, FileUtils.getFileNameWithoutPath(absoluteFilePath), null, uploadFile.getTotalSpace());
    }

    private File checkAbsoluteFilePath(String absoluteFilePath) throws Exception{
        if(StringUtils.isEmpty(absoluteFilePath)){
            throw new IllegalArgumentException("上传文件全路径不存在!");
        }
        File uploadFile = new File(absoluteFilePath.toLowerCase());
        if(!uploadFile.exists()){
            throw new IllegalArgumentException("上传文件不存在, file=" + absoluteFilePath);
        }
        return uploadFile;
    }
}