package cn.com.tenlion.systemcard.util;

import com.baidu.aip.util.Base64Util;
import org.json.JSONObject;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLEncoder;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 统一信用代码证(营业执照)识别工具类
 * ClassName: OCRBusinessLicenseUtil
 * @Description: TODO
 * 创建工具: IDEA
 * 运行环境: [Tomcat7以上,MySql5.6以上,JDK7以上]
 * @author 崔宝铖
 * @date 2021年3月19日
 */
public class OCRBusinessLicenseUtil {

    private static String apiKey = "6thm2Soep5COKhhvqGxCID3p";

    private static String secretKey = "5f0tjGe6uvYbf72vUrNH5hRFmll4yA1m";

    private static String accessToken = "";

    // 鉴权token获取Url
    private static String authHost = "https://aip.baidubce.com/oauth/2.0/token?";

    // 营业执照识别url
    private static String businessLicenseUrl = "https://aip.baidubce.com/rest/2.0/ocr/v1/business_license";

    public static void main(String[] args) {
        BusinessLicenseBean result = businessLicenseFile("D:\\1.jpg");
        System.out.println(result);
    }

    /**
     * 传入本地图片路径识别经营许可证
     * @param filePath
     * @return
     */
    public static BusinessLicenseBean businessLicenseFile(String filePath) {
        BusinessLicenseBean bean = new BusinessLicenseBean();
        try {
            // 本地文件路径
            byte[] imgData = getFileBytes(new File(filePath));
            String imgStr = Base64Util.encode(imgData);
            String imgParam = URLEncoder.encode(imgStr, "UTF-8");
            // 注意这里仅为了简化编码每一次请求都去获取access_token，线上环境access_token有过期时间， 客户端可自行缓存，过期后重新获取。
            Map<String, Object> params = new HashMap<String, Object>();
            params.put("image", imgParam);
            params.put("access_token", accessToken);
            String result = HttpUtil.doPost(businessLicenseUrl, params);
            // AccessToken如果过期
            if (result.contains("Access token invalid or no longer valid")) {
                accessToken = getAuth(apiKey, secretKey);
                return businessLicenseFile(filePath);
            }
            JSONObject jsonObject = new JSONObject(result);
            JSONObject wordsResult = jsonObject.getJSONObject("words_result");
            String licenseId = wordsResult.getJSONObject("社会信用代码").getString("words");
            String licenseNumber = wordsResult.getJSONObject("证件编号").getString("words");
            String licenseName = wordsResult.getJSONObject("单位名称").getString("words");
            String licenseType = wordsResult.getJSONObject("类型").getString("words");
            String licenseResidence = wordsResult.getJSONObject("地址").getString("words");
            String licenseLegalPerson = wordsResult.getJSONObject("法人").getString("words");
            String licenseCapital = wordsResult.getJSONObject("注册资本").getString("words");
            String licenseFoundTime = wordsResult.getJSONObject("成立日期").getString("words");
            String licenseTerm = wordsResult.getJSONObject("有效期").getString("words");
            String licenseBusiness = wordsResult.getJSONObject("经营范围").getString("words");
            String licenseAuthority = wordsResult.getJSONObject("登记机关").getString("words");
            String licenseComposition = wordsResult.getJSONObject("组成形式").getString("words");
            bean.setLicenseId(licenseId);
            bean.setLicenseNumber(licenseNumber);
            bean.setLicenseName(licenseName);
            bean.setLicenseType(licenseType);
            bean.setLicenseResidence(licenseResidence);
            bean.setLicenseLegalPerson(licenseLegalPerson);
            bean.setLicenseCapital(licenseCapital);
            bean.setLicenseFoundTime(licenseFoundTime);
            bean.setLicenseTerm(licenseTerm);
            bean.setLicenseBusiness(licenseBusiness);
            bean.setLicenseAuthority(licenseAuthority);
            bean.setLicenseComposition(licenseComposition);
            return bean;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return bean;
    }

    /**
     * 传入网络图片路径识别经营许可证
     * @param url
     * @return
     */
    public static BusinessLicenseBean businessLicenseUrl(String url) {
        BusinessLicenseBean bean = new BusinessLicenseBean();
        try {
            // 注意这里仅为了简化编码每一次请求都去获取access_token，线上环境access_token有过期时间， 客户端可自行缓存，过期后重新获取。
            Map<String, Object> params = new HashMap<String, Object>();
            params.put("url", url);
            params.put("access_token", accessToken);
            String result = HttpUtil.doPost(businessLicenseUrl, params);
            // AccessToken如果过期
            if (result.contains("Access token invalid or no longer valid")) {
                accessToken = getAuth(apiKey, secretKey);
                return businessLicenseUrl(url);
            }
            JSONObject jsonObject = new JSONObject(result);
            JSONObject wordsResult = jsonObject.getJSONObject("words_result");
            String licenseId = wordsResult.getJSONObject("社会信用代码").getString("words");
            String licenseNumber = wordsResult.getJSONObject("证件编号").getString("words");
            String licenseName = wordsResult.getJSONObject("单位名称").getString("words");
            String licenseType = wordsResult.getJSONObject("类型").getString("words");
            String licenseResidence = wordsResult.getJSONObject("地址").getString("words");
            String licenseLegalPerson = wordsResult.getJSONObject("法人").getString("words");
            String licenseCapital = wordsResult.getJSONObject("注册资本").getString("words");
            String licenseFoundTime = wordsResult.getJSONObject("成立日期").getString("words");
            String licenseTerm = wordsResult.getJSONObject("有效期").getString("words");
            String licenseBusiness = wordsResult.getJSONObject("经营范围").getString("words");
            String licenseAuthority = wordsResult.getJSONObject("登记机关").getString("words");
            String licenseComposition = wordsResult.getJSONObject("组成形式").getString("words");
            bean.setLicenseId(licenseId);
            bean.setLicenseNumber(licenseNumber);
            bean.setLicenseName(licenseName);
            bean.setLicenseType(licenseType);
            bean.setLicenseResidence(licenseResidence);
            bean.setLicenseLegalPerson(licenseLegalPerson);
            bean.setLicenseCapital(licenseCapital);
            bean.setLicenseFoundTime(licenseFoundTime);
            bean.setLicenseTerm(licenseTerm);
            bean.setLicenseBusiness(licenseBusiness);
            bean.setLicenseAuthority(licenseAuthority);
            bean.setLicenseComposition(licenseComposition);
            return bean;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }


    public static String getAuth(String apiKey, String secretKey) {
        // 获取token地址
        String getAccessTokenUrl = authHost
                // 1. grant_type为固定参数
                + "grant_type=client_credentials"
                // 2. 官网获取的 API Key
                + "&client_id=" + apiKey
                // 3. 官网获取的 Secret Key
                + "&client_secret=" + secretKey;
        try {
            URL realUrl = new URL(getAccessTokenUrl);
            // 打开和URL之间的连接
            HttpURLConnection connection = (HttpURLConnection) realUrl.openConnection();
            connection.setRequestMethod("GET");
            connection.connect();
            // 获取所有响应头字段
            Map<String, List<String>> map = connection.getHeaderFields();
            // 遍历所有的响应头字段
            for (String key : map.keySet()) {
                System.err.println(key + "--->" + map.get(key));
            }
            // 定义 BufferedReader输入流来读取URL的响应
            BufferedReader in = new BufferedReader(new InputStreamReader(connection.getInputStream()));
            String result = "";
            String line;
            while ((line = in.readLine()) != null) {
                result += line;
            }
            /**
             * 返回结果示例
             */
            // System.err.println("result:" + result);
            JSONObject jsonObject = new JSONObject(result);
            String access_token = jsonObject.getString("access_token");
            return access_token;
        } catch (Exception e) {
            System.err.printf("获取token失败！");
            e.printStackTrace(System.err);
        }
        return null;
    }

    private static byte[] getFileBytes(File file) throws IOException {
        String fileName = file.getName();
        String[] names = fileName.split("\\.");
        ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
        if (names.length > 1) {
            String fileType = names[names.length - 1];
            BufferedImage bi;
            try {
                bi = ImageIO.read(file);
                ImageIO.write(bi, fileType, byteArrayOutputStream);
                byte[] bytes = byteArrayOutputStream.toByteArray();
                byteArrayOutputStream.close();
                return bytes;
            } catch (IOException e) {
            }
        }
        return null;
    }

}