package com.qianwen.smartman.common.utils;

import cn.hutool.core.codec.Base64Encoder;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.math.BigDecimal;
import javax.imageio.ImageIO;
import net.coobird.thumbnailator.Thumbnails;
import org.apache.commons.lang3.StringUtils;
import com.qianwen.smartman.common.constant.DncConstant;
import org.springframework.web.multipart.MultipartFile;

public class PictureUtil {
    public static boolean isPicture(String imgName) {
        boolean flag = false;
        if (StringUtils.isBlank(imgName)) {
            return false;
        }
        String[] arr = {"bmp", "dib", "gif", "jfif", "jpe", "jpeg", "jpg", "png", "tif", "tiff", "ico"};
        int length = arr.length;
        int i = 0;
        while (true) {
            if (i >= length) {
                break;
            }
            String item = arr[i];
            if (!item.equals(imgName)) {
                i++;
            } else {
                flag = true;
                break;
            }
        }
        return flag;
    }

    public static String compressBase64(byte[] b, long desFileSize) throws IOException {
        double accuracy = 0.8d;
        if (104857.6d <= b.length && b.length <= 1048576) {
            accuracy = 0.3d;
        } else if (1048576 < b.length && b.length <= 2097152) {
            accuracy = 0.2d;
        } else if (2097152 < b.length) {
            accuracy = 0.1d;
        }
        byte[] bytes = commpressPicCycle(b, desFileSize, accuracy);
        if (bytes != null) {
            return Base64Encoder.encode(bytes);
        }
        return null;
    }

    public static String getSuffix(MultipartFile file) {
        String suffix = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(DncConstant.POINT) + 1);
        return suffix;
    }

    public static String commpressPicForScale(String srcPath, String desPath, long desFileSize, double accuracy, int desMaxWidth, int desMaxHeight) {
        if (StringUtils.isEmpty(srcPath) || StringUtils.isEmpty(srcPath) || !new File(srcPath).exists()) {
            return null;
        }
        try {
            File srcFile = new File(srcPath);
            long srcFileSize = srcFile.length();
            System.out.println("源图片：" + srcPath + "，大小：" + (srcFileSize / 1024) + "kb");
            BufferedImage bim = ImageIO.read(srcFile);
            int srcWidth = bim.getWidth();
            int srcHeight = bim.getHeight();
            Thumbnails.Builder builder = Thumbnails.of(new File[]{srcFile}).outputFormat("jpg");
            if (srcWidth > desMaxWidth || srcHeight > desMaxHeight) {
                builder.size(desMaxWidth, desMaxHeight);
            } else {
                builder.size(srcWidth, srcHeight);
            }
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            builder.toOutputStream(baos);
            byte[] bytes = commpressPicCycle(baos.toByteArray(), desFileSize, accuracy);
            File desFile = new File(desPath);
            FileOutputStream fos = new FileOutputStream(desFile);
            fos.write(bytes);
            fos.close();
            System.out.println("目标图片：" + desPath + "，大小" + (desFile.length() / 1024) + "kb");
            System.out.println("图片压缩完成！");
            return desPath;
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    private static byte[] commpressPicCycle(byte[] bytes, long desFileSize, double accuracy) throws IOException {
        long srcFileSizeJPG = bytes.length;
        if (srcFileSizeJPG <= desFileSize * 1024) {
            return bytes;
        }
        BufferedImage bim = ImageIO.read(new ByteArrayInputStream(bytes));
        int srcWdith = bim.getWidth();
        int srcHeigth = bim.getHeight();
        int desWidth = new BigDecimal(srcWdith).multiply(new BigDecimal(accuracy)).intValue();
        int desHeight = new BigDecimal(srcHeigth).multiply(new BigDecimal(accuracy)).intValue();
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        Thumbnails.of(new InputStream[]{new ByteArrayInputStream(bytes)}).size(desWidth, desHeight).outputQuality(accuracy).toOutputStream(baos);
        return commpressPicCycle(baos.toByteArray(), desFileSize, accuracy);
    }
}