package com.nova.sankuai.infra.utils;

import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;
import sun.misc.BASE64Encoder;

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * @author weikangdi
 */
@Component
public class ImageUtil {

    @Value("${upload.path}")
    private String path;

    @Value("${upload.publicPath}")
    private String publicPath;

    public String uploadFile(MultipartFile file, boolean isPublic) throws Exception {
        // 文件名
        String fileName = file.getOriginalFilename();
        // 后缀名
        String suffixName = fileName.substring(fileName.lastIndexOf("."));
        // 上传后的路径
        String filePath = isPublic ? publicPath : path;
        String fileUrl = isPublic ? "/public/upload_img/" : "/upload_img/";

        filePath = chooseOnTime(filePath);
        fileUrl = chooseOnTime(fileUrl);
        // 新文件名

        fileName = UUID.randomUUID().toString().replace("-", "") + suffixName;
        File dest = new File(filePath + fileName);
        if (!dest.getParentFile().exists()) {
            dest.getParentFile().mkdirs();
        }
        file.transferTo(dest);
        String filename = fileUrl + fileName;
        return filename;
    }


    public String uploadFileByInputStream(InputStream inputStream, boolean isPublic) throws Exception {
        // 上传后的路径
        String filePath = isPublic ? publicPath : path;
        String fileUrl = isPublic ? "/public/upload_img/" : "/upload_img/";

        filePath = chooseOnTime(filePath);
        fileUrl = chooseOnTime(fileUrl);
        // 文件名
        String fileName = UUID.randomUUID().toString().replace("-", "") + ".png";
        File dest = new File(filePath + fileName);
        if (!dest.getParentFile().exists()) {
            dest.getParentFile().mkdirs();
        }
        byte[] data = new byte[1024];
        int len = 0;
        FileOutputStream fileOutputStream = null;
        try {
            fileOutputStream = new FileOutputStream(dest);
            while ((len = inputStream.read(data)) != -1) {
                fileOutputStream.write(data, 0, len);
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (inputStream != null) {
                try {
                    inputStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            if (fileOutputStream != null) {
                try {
                    fileOutputStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        String filename = fileUrl + fileName;
        return filename;
    }


    public String yxUploadFile(MultipartFile file, boolean isPublic) throws IOException {
        // 文件名
        String fileName = "";
        // 上传后的路径
        String filePath = isPublic ? publicPath : path;
        String fileUrl = isPublic ? "/public/upload_img/" : "/upload_img/";

        filePath = chooseOnTime(filePath);
        fileUrl = chooseOnTime(fileUrl);
        // 新文件名
        fileName = UUID.randomUUID().toString().replace("-", "") + ".png";
        File dest = new File(filePath + fileName);
        if (!dest.getParentFile().exists()) {
            dest.getParentFile().mkdirs();
        }
        file.transferTo(dest);
        String filename = fileUrl + fileName;
        return filename;
    }

    public String uploadBase64File(byte[] imageByte, boolean isPublic) throws IOException {
        String fileName = UUID.randomUUID().toString().replace("-", "") + ".png";

        // 上传后的路径
        String filePath = isPublic ? publicPath : path;
        String fileUrl = isPublic ? "/public/upload_img/" : "/upload_img/";

        filePath = chooseOnTime(filePath);
        fileUrl = chooseOnTime(fileUrl);
        File dest = new File(filePath + fileName);
        if (!dest.getParentFile().exists()) {
            dest.getParentFile().mkdirs();
        }
        // 生成文件
        if (!dest.exists()) {
            dest.createNewFile();
        }
        OutputStream imageStream = new FileOutputStream(dest);
        imageStream.write(imageByte);
        imageStream.flush();
        imageStream.close();
        String filename = fileUrl + fileName;
        return filename;
    }


    /**
     * 判断上传文件大小方法
     *
     * @param len  文件长度
     * @param size 限制大小
     * @return
     */
    public static boolean checkFileSize(Long len, int size) {
        double fileSize = 0;
        fileSize = (double) len / 1048576;
        if (fileSize > size) {
            return false;
        }
        return true;
    }

    public static String chooseOnTime(String path) {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd");
        String format = sdf.format(new Date());
        return path + format + "/";
    }


    /**
     * URL图片转base64
     * @param imgUrl
     * @return
     */
    public static String imageUrlToBase64(String imgUrl) {
        URL url = null;
        InputStream is = null;
        ByteArrayOutputStream outStream = null;
        HttpURLConnection httpUrl = null;
        try {
            url = new URL(imgUrl);
            httpUrl = (HttpURLConnection) url.openConnection();
            httpUrl.connect();
            httpUrl.getInputStream();
            is = httpUrl.getInputStream();
            outStream = new ByteArrayOutputStream();
            //创建一个Buffer字符串
            byte[] buffer = new byte[1024];
            //每次读取的字符串长度，如果为-1，代表全部读取完毕
            int len = 0;
            //使用输入流从buffer里把数据读取出来
            while( (len = is.read(buffer)) != -1 ){
                //用输出流往buffer里写入数据，中间参数代表从哪个位置开始读，len代表读取的长度
                outStream.write(buffer, 0, len);
            }
            // 对字节数组Base64编码
            return encode(outStream.toByteArray());
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            try {
                if(is != null) {
                    is.close();
                }
                if(outStream != null) {
                    outStream.close();
                }
                if(httpUrl != null) {
                    httpUrl.disconnect();
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return null;
    }

    /**
     * 图片转字符串
     * @param image 图片Buffer
     * @return Base64编码
     */
    public static String encode(byte[] image){
        BASE64Encoder decoder = new BASE64Encoder();
        return replaceEnter(decoder.encode(image));
    }

    /**
     * 字符替换
     * @param str 字符串
     * @return 替换后的字符串
     */
    public static String replaceEnter(String str){
        String reg ="[\n-\r]";
        Pattern p = Pattern.compile(reg);
        Matcher m = p.matcher(str);
        return m.replaceAll("");
    }
}