package com.cloud.common.utils;


import com.cloud.common.model.FileInfos;
import com.github.tobato.fastdfs.domain.fdfs.StorePath;
import com.github.tobato.fastdfs.domain.proto.storage.DownloadByteArray;
import com.github.tobato.fastdfs.service.FastFileStorageClient;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.multipart.MultipartFile;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.Date;
import java.util.UUID;

@Configuration
public class FastDFSUtil {

    @Autowired
    private FastFileStorageClient fastFileStorageClient;

    public FileInfos upload(MultipartFile file) {	//上传文件

        try {

            String ext_Name = file.getOriginalFilename().split("\\.")[1];

            byte[] bytes = null;
            try {
                bytes = file.getBytes();
            } catch (IOException e) {
                e.printStackTrace();
            }
            //获取文件大小
            long fileSize = file.getSize();
            ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(bytes);

            StorePath storePath = fastFileStorageClient.uploadFile(byteArrayInputStream, fileSize, ext_Name, null);
//            String fileIds[] = storageClient.upload_file(bytes, ext_Name, null);
            //获取文件信息，并保存至数据库
            FileInfos fileInfo = new FileInfos();
            fileInfo.setId(UUID.randomUUID().toString());
            fileInfo.setName(file.getOriginalFilename());
            fileInfo.setExtName(ext_Name);
            fileInfo.setPath(storePath.getFullPath());
            fileInfo.setSize(file.getSize());
            fileInfo.setCreateTime(new Date());

            return fileInfo;
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }

    }


    public FileInfos uploadByByte(byte[] bytes, String extName) {	//上传文件
        try {
            ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(bytes);
            StorePath storePath = fastFileStorageClient.uploadFile(byteArrayInputStream, bytes.length, extName, null);
//            String fileIds[] = storageClient.upload_file(bytes, extName, null);
            //获取文件信息，并保存至数据库
            FileInfos fileInfo = new FileInfos();

            fileInfo.setExtName(extName);
            fileInfo.setPath(storePath.getFullPath());
            fileInfo.setCreateTime(new Date());

            return fileInfo;
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }

    }

    /**
     * 根据图片的FastDFS地址获得数据的字节流
     * @param strUrl 网络连接地址
     * @return
     */
    public  byte[] getImageFromFDFS(String strUrl){
        try {
            URL url = new URL(strUrl);
            System.out.println("url是："+strUrl);
            HttpURLConnection conn = (HttpURLConnection)url.openConnection();
            conn.setRequestMethod("GET");
            conn.setConnectTimeout(5 * 1000);
            InputStream inStream = conn.getInputStream();//获取从网络的输入流
            byte[] btImg = new byte[inStream.available()];
            inStream.read(btImg);         //从输入流中得到图片的二进制数据
            return btImg;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }

    public byte[] downloadFDFS(String fileUrl) throws Exception {
        String group = fileUrl.substring(0, fileUrl.indexOf("/"));
        String path = fileUrl.substring(fileUrl.indexOf("/") + 1);
        DownloadByteArray downloadByteArray = new DownloadByteArray();
        return fastFileStorageClient.downloadFile(group, path, downloadByteArray);
    }
}