package org.springblade.common.utils;

import lombok.SneakyThrows;
import org.springblade.common.vo.FileSize;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.URL;
import java.net.URLConnection;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

public class FileUtils  {

    /**
     * 获取文件path
     * @param file
     * @return
     * @throws IOException
     */
    public static File convert(MultipartFile file) throws IOException {
        File convertFile = new File(Objects.requireNonNull(Objects.requireNonNull(file.getOriginalFilename())));
        FileOutputStream fos = null;
        try {
            convertFile.createNewFile();
            fos = new FileOutputStream(convertFile);
            fos.write(file.getBytes());
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            assert fos != null;
            fos.close();
        }
        return convertFile;
    }

    //获取文件大小
    public static List<FileSize> getOssFileSize(List<String> fileList){
        List<FileSize> reData =new ArrayList<>();
        if(fileList!=null){
            for(String fileUrl:fileList){
                FileSize file =new FileSize();
                file.setFileUrl(fileUrl);
                try {
                    URLConnection  openConnection = new URL(fileUrl).openConnection();
                    int contentLength = openConnection.getContentLength();
                    Double fileLength = Double.parseDouble(contentLength+"");
                    file.setFileSize(Math.ceil(fileLength/1024));
                } catch (IOException e) {
                    e.printStackTrace();
                }
                reData.add(file);
            }
        }
        return  reData;
    }

}