package com.project.admin.controller.tool;

import com.project.common.config.ProjectConfig;
import com.project.common.core.domain.AjaxResult;
import com.project.common.utils.sign.Base64;
import com.tencentcloudapi.common.Credential;
import com.tencentcloudapi.common.exception.TencentCloudSDKException;
import com.tencentcloudapi.common.profile.ClientProfile;
import com.tencentcloudapi.common.profile.HttpProfile;
import com.tencentcloudapi.ocr.v20181119.OcrClient;
import com.tencentcloudapi.ocr.v20181119.models.*;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.io.FileInputStream;
import java.io.IOException;


@Slf4j
@RestController
@RequestMapping("/tool/ai")
public class AiController {

    @Value("${secretId}")
    private String secretId;
    @Value("${secretKey}")
    private String secretKey;

    /**
     * 行驶证识别
     * @param imageUrl  图片
     * @return  结果
     */
    @GetMapping("/vehicleLicenseOCR")
    @ApiOperation("行驶证识别")
    public AjaxResult vehicleLicenseOCR(String imageUrl)
    {
        try{
            // 实例化一个认证对象，入参需要传入腾讯云账户secretId，secretKey,此处还需注意密钥对的保密
            // 密钥可前往https://console.cloud.tencent.com/cam/capi网站进行获取
            Credential cred = new Credential(secretId, secretKey);
            // 实例化一个http选项，可选的，没有特殊需求可以跳过
            HttpProfile httpProfile = new HttpProfile();
            httpProfile.setEndpoint("ocr.tencentcloudapi.com");
            // 实例化一个client选项，可选的，没有特殊需求可以跳过
            ClientProfile clientProfile = new ClientProfile();
            clientProfile.setHttpProfile(httpProfile);
            // 实例化要请求产品的client对象,clientProfile是可选的
            OcrClient client = new OcrClient(cred, "ap-beijing", clientProfile);
            // 实例化一个请求对象,每个接口都会对应一个request对象
            VehicleLicenseOCRRequest req = new VehicleLicenseOCRRequest();
            //req.setImageUrl(imageUrl);
            req.setImageBase64(imgbase64(imageUrl));
            // 返回的resp是一个VehicleLicenseOCRResponse的实例，与请求对象对应
            VehicleLicenseOCRResponse resp = client.VehicleLicenseOCR(req);
            return AjaxResult.success("操作成功",VehicleLicenseOCRResponse.toJsonString(resp));
        } catch (TencentCloudSDKException e) {
            log.error("行驶证识别失败。",e);
            return AjaxResult.error("操作失败",e.toString());
        }
    }

    /**
     * 身份证识别
     * @param imageUrl  图片路径
     * @return  记过
     */
    @GetMapping("/IDCardOCR")
    @ApiOperation("身份证识别")
    public AjaxResult IDCardOCR(String imageUrl)
    {
        try{
            // 实例化一个认证对象，入参需要传入腾讯云账户secretId，secretKey,此处还需注意密钥对的保密
            // 密钥可前往https://console.cloud.tencent.com/cam/capi网站进行获取
            Credential cred = new Credential(secretId, secretKey);
            // 实例化一个http选项，可选的，没有特殊需求可以跳过
            HttpProfile httpProfile = new HttpProfile();
            httpProfile.setEndpoint("ocr.tencentcloudapi.com");
            // 实例化一个client选项，可选的，没有特殊需求可以跳过
            ClientProfile clientProfile = new ClientProfile();
            clientProfile.setHttpProfile(httpProfile);
            // 实例化要请求产品的client对象,clientProfile是可选的
            OcrClient client = new OcrClient(cred, "ap-beijing", clientProfile);
            // 实例化一个请求对象,每个接口都会对应一个request对象
            IDCardOCRRequest req = new IDCardOCRRequest();
           // req.setImageUrl(imageUrl);
            req.setImageBase64(imgbase64(imageUrl));
            // 返回的resp是一个IDCardOCRResponse的实例，与请求对象对应
            IDCardOCRResponse resp = client.IDCardOCR(req);

            return AjaxResult.success("操作成功",IDCardOCRResponse.toJsonString(resp));
        } catch (TencentCloudSDKException e) {
            log.error("身份证识别失败。",e);
            return AjaxResult.error(e.getMessage());
        }
    }


    private String imgbase64(String fileName)
    {
        FileInputStream in=null;
        try{
            String filePath = ProjectConfig.getUploadPath() + fileName.substring(fileName.lastIndexOf('/'));
            in=new FileInputStream(filePath);
            byte[] data = new byte[in.available()];
            in.read(data);
            in.close();
            return Base64.encode(data);
        }catch (IOException e){
            log.error("读取图片异常",e);
            return "";
        }
    }
}