package foundation.ai;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Objects;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

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.common.profile.Language;
import com.tencentcloudapi.ocr.v20181119.OcrClient;
import com.tencentcloudapi.ocr.v20181119.models.BankSlipOCRRequest;
import com.tencentcloudapi.ocr.v20181119.models.BankSlipOCRResponse;
import com.tencentcloudapi.ocr.v20181119.models.BizLicenseOCRRequest;
import com.tencentcloudapi.ocr.v20181119.models.BizLicenseOCRResponse;
import com.tencentcloudapi.ocr.v20181119.models.EnterpriseLicenseOCRRequest;
import com.tencentcloudapi.ocr.v20181119.models.EnterpriseLicenseOCRResponse;
import com.tencentcloudapi.ocr.v20181119.models.GeneralAccurateOCRRequest;
import com.tencentcloudapi.ocr.v20181119.models.GeneralAccurateOCRResponse;
import com.tencentcloudapi.ocr.v20181119.models.GeneralEfficientOCRRequest;
import com.tencentcloudapi.ocr.v20181119.models.GeneralEfficientOCRResponse;
import com.tencentcloudapi.ocr.v20181119.models.GeneralFastOCRRequest;
import com.tencentcloudapi.ocr.v20181119.models.GeneralFastOCRResponse;
import com.tencentcloudapi.ocr.v20181119.models.IDCardOCRRequest;
import com.tencentcloudapi.ocr.v20181119.models.IDCardOCRResponse;
import com.tencentcloudapi.ocr.v20181119.models.InstitutionOCRRequest;
import com.tencentcloudapi.ocr.v20181119.models.InstitutionOCRResponse;
import com.tencentcloudapi.ocr.v20181119.models.OrgCodeCertOCRRequest;
import com.tencentcloudapi.ocr.v20181119.models.OrgCodeCertOCRResponse;
import com.tencentcloudapi.ocr.v20181119.models.VatInvoiceOCRRequest;
import com.tencentcloudapi.ocr.v20181119.models.VatInvoiceOCRResponse;
import com.tencentcloudapi.ocr.v20181119.models.VatInvoiceVerifyNewRequest;
import com.tencentcloudapi.ocr.v20181119.models.VatInvoiceVerifyNewResponse;

import foundation.ai.dao.VatInvoiceVerifyDao;
import foundation.util.Util;
import sun.misc.BASE64Encoder;

public class TencentAIProvider extends AIProvider {

	private static Logger logger;
	
    public static String secretId = "AKIDud7LIyOXQ1LPBguqVtzqLaYEdlbATVGk";
    public static String secretKey = "EBrFMDo8CzVkhPpypPikVzDBsHQHr9Av";
    public static int TimeOut_Second_Write = 10;
    public static int TimeOut_Second_Read = 10;
    public static boolean Debug = true;
    
	static {
		logger = LogManager.getLogger(TencentAIProvider.class);
	}

	@Override
	protected String getName() {
		return "腾讯AI";
	}
	
	@Override
	public OCRResult executeAction(Operator operator, File file) throws Exception {
		//1. 得到图片的Base64
        String base64 = imageToBase64(file);
        if (base64 == null) {
        	logger.error("无法对图片{}进行Base64编码，无法进行AI识别", file);
            return null;
        }
        
		//2. 创建OCR客户端
		OcrClient client = createClient();
		
		//3. 创建OCR请求
		Object request = createRequest(operator, base64);
		
		//4. 发送请求，得到结果
		String value = "";
		
		try {
			 value = send(operator, client, request);
		} catch(Exception e) {
			 e.printStackTrace();
			 value = "fail";
		}
		
		value = Util.ocrJson(value);
		//5. 返回结果
		OCRResult result = new OCRResult(operator, value);
		
		return result;
	}
	

	@Override
	/**
	 * 发票验真
	 */
	public OCRResult executeAction(Operator operator, Object paramObject) throws Exception {
		boolean is_real = true;
		//1. 创建OCR客户端
		OcrClient client = createClient();
		
		//2. 创建OCR请求
		Object request = createRequest(operator, paramObject);
		
		//3. 发送请求，得到结果
		String value = "";
		
		try {
			 value = send(operator, client, request);
		} catch(Exception e) {
			 e.printStackTrace();
			 value = e.getMessage();
			 is_real = false;
		}
		
		value = Util.ocrJson(value);
		//4. 返回结果
		OCRResult result = new OCRResult(operator, value);
		result.setIs_real(is_real);
		
		return result;
	}	

	private OcrClient createClient() {
		//1. credential
		Credential credential = new Credential(secretId, secretKey);
		
		//2. profile
        HttpProfile httpProfile = new HttpProfile();
        httpProfile.setWriteTimeout(TimeOut_Second_Write);  
        httpProfile.setReadTimeout(TimeOut_Second_Read);  
        
        ClientProfile clientProfile = new ClientProfile();
        clientProfile.setHttpProfile(httpProfile);
        clientProfile.setDebug(Debug);
        clientProfile.setLanguage(Language.ZH_CN);
		
        //3. client
        OcrClient client = new OcrClient(credential, "ap-shanghai", clientProfile);
		return client;
	}
	
	private Object createRequest(Operator operator, String base64) {
		if (Operator.BizLicense == operator) {
			BizLicenseOCRRequest request = new BizLicenseOCRRequest();
			request.setImageBase64(base64);
			return request;
		}
		else if (Operator.EnterpriseLicense == operator) {
			EnterpriseLicenseOCRRequest request = new EnterpriseLicenseOCRRequest();
			request.setImageBase64(base64);
			return request;
		}
		else if (Operator.OrgCodeCert == operator) {
			OrgCodeCertOCRRequest request = new OrgCodeCertOCRRequest();
			request.setImageBase64(base64);
			return request;
		}
		else if (Operator.Institution == operator) {
			InstitutionOCRRequest request = new InstitutionOCRRequest();
			request.setImageBase64(base64);
			return request;
		}        
		else if (Operator.IDCard == operator) {
			IDCardOCRRequest request = new IDCardOCRRequest();
			request.setImageBase64(base64);
			return request;
		} 
		else if (Operator.TextAccurate == operator) {
			GeneralAccurateOCRRequest request = new GeneralAccurateOCRRequest();
			request.setImageBase64(base64);
			return request;
		} 
		else if (Operator.TextCommon == operator) {
			GeneralEfficientOCRRequest request = new GeneralEfficientOCRRequest();
			request.setImageBase64(base64);
			return request;
		}
		else if (Operator.TextFast == operator) {
			GeneralFastOCRRequest request = new GeneralFastOCRRequest();
			request.setImageBase64(base64);
			return request;
		}
		else if (Operator.BankSlip == operator) {
			BankSlipOCRRequest request = new BankSlipOCRRequest();
			request.setImageBase64(base64);
			return request;
		}
		else if (Operator.InvoiceIdentify == operator) {
			VatInvoiceOCRRequest request = new VatInvoiceOCRRequest();
			request.setImageBase64(base64);
			return request;
		}

		return null;
	}
	
	private Object createRequest(Operator operator, Object paramObject) {
		if (Operator.InvoiceVerification == operator) {
			VatInvoiceVerifyNewRequest request = new VatInvoiceVerifyNewRequest();
			VatInvoiceVerifyDao vatInvoiceVerify = (VatInvoiceVerifyDao) paramObject;
			request.setInvoiceCode(vatInvoiceVerify.getInvoiceCode());
			request.setInvoiceNo(vatInvoiceVerify.getInvoiceNo());
			request.setInvoiceDate(vatInvoiceVerify.getInvoiceDate());
//			request.setInvoiceKind("04");
			request.setCheckCode(vatInvoiceVerify.getAdditional());
			request.setAmount(vatInvoiceVerify.getAmount());
			return request;
		}			

		return null;
	}	

	private String send(Operator operator, OcrClient client, Object request) throws TencentCloudSDKException {
		String json = null;
		
		if (Operator.BizLicense == operator) {
			BizLicenseOCRResponse response = client.BizLicenseOCR((BizLicenseOCRRequest)request);
			json = BizLicenseOCRResponse.toJsonString(response);
		}
		else if (Operator.EnterpriseLicense == operator) {
			EnterpriseLicenseOCRResponse response = client.EnterpriseLicenseOCR((EnterpriseLicenseOCRRequest)request);
			json = EnterpriseLicenseOCRResponse.toJsonString(response);
		}
		else if (Operator.OrgCodeCert == operator) {
			OrgCodeCertOCRResponse response = client.OrgCodeCertOCR((OrgCodeCertOCRRequest)request);
			json = OrgCodeCertOCRResponse.toJsonString(response);
		}
		else if (Operator.Institution == operator) {
			InstitutionOCRResponse response = client.InstitutionOCR((InstitutionOCRRequest)request);
			json = InstitutionOCRResponse.toJsonString(response);
		}        
		else if (Operator.IDCard == operator) {
			IDCardOCRResponse response = client.IDCardOCR((IDCardOCRRequest)request);
			json = IDCardOCRResponse.toJsonString(response);
		} 
		else if (Operator.TextAccurate == operator) {
			GeneralAccurateOCRResponse response = client.GeneralAccurateOCR((GeneralAccurateOCRRequest)request);
			json = GeneralAccurateOCRResponse.toJsonString(response);
		} 
		else if (Operator.TextCommon == operator) {
			GeneralEfficientOCRResponse response = client.GeneralEfficientOCR((GeneralEfficientOCRRequest)request);
			json = GeneralEfficientOCRResponse.toJsonString(response);
		}
		else if (Operator.TextFast == operator) {
			GeneralFastOCRResponse response = client.GeneralFastOCR((GeneralFastOCRRequest)request);
			json = GeneralFastOCRResponse.toJsonString(response);
		}
		else if (Operator.BankSlip == operator) {
			BankSlipOCRResponse response = client.BankSlipOCR((BankSlipOCRRequest)request);
			json = BankSlipOCRResponse.toJsonString(response);
		}	
		else if (Operator.InvoiceIdentify == operator) { 
			VatInvoiceOCRResponse response = client.VatInvoiceOCR((VatInvoiceOCRRequest)request);
			json = VatInvoiceOCRResponse.toJsonString(response);			
		}
		else if (Operator.InvoiceVerification == operator) { 
			VatInvoiceVerifyNewResponse response = client.VatInvoiceVerifyNew((VatInvoiceVerifyNewRequest) request);
			json = VatInvoiceOCRResponse.toJsonString(response);			
		}		
		
		return json;
	}
	
    private static String imageToBase64(File file) {
        byte[] data = null;

        InputStream in = null;
        try {
        	//1. bytes
            in = new FileInputStream(file);
            data = new byte[in.available()];
            in.read(data);
            in.close();
            
            //2. encode
            BASE64Encoder encoder = new BASE64Encoder();
            return encoder.encode(Objects.requireNonNull(data));
        } 
        catch (IOException e) {
            e.printStackTrace();
        }
        finally {
            try {
                in.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        
        return null;
    }

}