package com.thinkgem.jeesite.modules.api.wechat;

import sun.misc.BASE64Decoder;

import sun.misc.BASE64Encoder;

import java.io.*;

/**

 * 字符串和图片相互转换

 * @author zhouhe

 * @date 2020/7/3 17:23

 */

public class ImgToStr {

    /**

     * 字符串转图片

     * @param imgStr --->图片字符串

     * @param filename    --->图片名

     * @return

     */

    public static boolean generateImage(String imgStr, String filename) {

        if (imgStr == null) {

            return false;

        }

        BASE64Decoder decoder = new BASE64Decoder();

        try {

            // 解密

            byte[] b = decoder.decodeBuffer(imgStr);

            // 处理数据

            for(int i = 0; i < b.length; ++i) {

                if (b[i] < 0) {

                    b[i] += 256;

                }

            }

            OutputStream out = new FileOutputStream("D://"+filename);

            out.write(b);

            out.flush();

            out.close();

            return true;

        } catch (IOException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        return false;

    }

    /**

     * 图片转字符串

     * @param filePath    --->文件路径

     * @return

     */

    public static String getImageStr(String filePath) {

        InputStream inputStream = null;

        byte[] data = null;

        try {

            inputStream = new FileInputStream(filePath);

            data = new byte[inputStream.available()];

            inputStream.read(data);

            inputStream.close();

        } catch (IOException e) {

            e.printStackTrace();

        }

        // 加密

        BASE64Encoder encoder = new BASE64Encoder();

        return encoder.encode(data);

    }

    /*

     * 测试代码

     */

    public static void main(String[] args) {

        String imageStr = getImageStr("D:\\222.png");

        System.out.println(imageStr);

        System.out.println(imageStr.length());

        boolean generateImage = generateImage(imageStr, "223.png");

        System.out.println(generateImage);

    }

}