package com.javaweb.third.controller;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.javaweb.common.config.Global;
import com.javaweb.common.constant.Constants;
import com.javaweb.common.core.controller.BaseController;
import com.javaweb.common.core.domain.AjaxResult;
import com.javaweb.common.utils.ServletUtils;
import com.javaweb.common.utils.StringUtils;
import com.javaweb.common.utils.file.FileUploadUtils;
import com.javaweb.framework.util.ShiroUtils;
import com.javaweb.system.domain.SysUser;
import com.javaweb.third.baidu.api.BaiduAi;
import com.javaweb.third.baidu.bean.face.FaceDetectResult;
import com.javaweb.third.baidu.bean.imgClassify.AnimalResult;
import com.javaweb.third.baidu.bean.imgClassify.CarResult;
import com.javaweb.third.baidu.bean.imgClassify.DishResult;
import com.javaweb.third.baidu.bean.imgClassify.PlantResult;
import com.javaweb.third.baidu.bean.ocr.*;
import com.javaweb.third.baidu.service.FaceService;
import com.javaweb.third.baidu.service.ImageClassifyService;
import com.javaweb.third.baidu.service.OcrService;
import com.javaweb.third.domain.AiHis;
import com.javaweb.third.service.IAiHisService;
import com.sun.jna.platform.win32.Guid;
import org.apache.commons.fileupload.util.Streams;
import org.apache.poi.ss.usermodel.DateUtil;
import org.apache.shiro.authz.annotation.RequiresAuthentication;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import javax.servlet.http.HttpServletRequest;
import java.io.*;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;

/**
 * @author wujiyue
 */
@Controller
@RequestMapping("/third/ai")
public class BaiduAiController extends BaseController {

    @Autowired
    IAiHisService aiHisService;
    private static String prefix="/third/ai/";
    public static String DEFAULT_UPLOAD_BASE_PATH=FileUploadUtils.getDefaultBaseDir()+"/temp";

    @RequestMapping(value={"","/"})
    @RequiresPermissions("third:ai")
    public String index(){
        return prefix+"/baiduAi";
    }

    //上传单张图片资源
    @ResponseBody
    @RequestMapping("/queryAi")
    public Object upload(@RequestParam("type") String aiType,@RequestParam("file") MultipartFile multipartFile) throws Exception{
        Map resultMap =new HashMap<String,Object>();
        if(StringUtils.isEmpty(aiType)){
            return AjaxResult.error("type参数不能为空!");
        }
        String path=FileUploadUtils.upload(DEFAULT_UPLOAD_BASE_PATH,multipartFile);
        resultMap.put("path",path);
        //去掉映射路径前缀
        path= path.replace(Constants.RESOURCE_PREFIX,"");
        String localFilePath=Global.getProfile()+path;
        return queryAi(aiType,localFilePath);
    }
    public Object queryAi(String aiType,String localFilePath){

        String json="";//返回最原始的json
        String aiName="";
        if(BaiduAi.AiType.faceDetect.name().equals(aiType)){
            FaceDetectResult result = FaceService.faceDetect_localPath(localFilePath);
            json=result.getJson();
            aiName="人脸检测";
        }else if(BaiduAi.AiType.plant.name().equals(aiType)){
            PlantResult result= ImageClassifyService.plant(localFilePath);
            json=result.getJson();
            aiName="植物识别";
        }else if(BaiduAi.AiType.bankCard.name().equals(aiType)){
            BankCardOcrResult result = OcrService.bankCardOcr(localFilePath);
            json=result.getJson();
            aiName="银行卡识别";
        }else if(BaiduAi.AiType.idCard.name().equals(aiType)){
            IdCardOcrResult result = OcrService.idcardOcr_Z(localFilePath);
            json=result.getJson();
            aiName="身份证识别";
        }else if(BaiduAi.AiType.plate.name().equals(aiType)){
            //识别车牌号
            PlateNumberResult result = OcrService.license_plateOcr(localFilePath);
            json=result.getJson();
            aiName="识别车牌号";
        }else if(BaiduAi.AiType.driver.name().equals(aiType)){
            //驾驶证识别
            aiName="驾驶证识别";
            DriveLicenseOcrResult result = OcrService.drivingLicenseOcr(localFilePath);
            json=result.getJson();
        }else if(BaiduAi.AiType.animal.name().equals(aiType)){
            //动物识别
            AnimalResult result = ImageClassifyService.animal(localFilePath);
            json=result.getJson();
        }else if(BaiduAi.AiType.car.name().equals(aiType)){
            //车型识别
            aiName="车型识别";
            CarResult result = ImageClassifyService.car(localFilePath);
            json=result.getJson();
        }else if(BaiduAi.AiType.dish.name().equals(aiType)){
            //菜品识别
            aiName="菜品识别";
            DishResult result = ImageClassifyService.dish(localFilePath);
            json=result.getJson();
        }else if(BaiduAi.AiType.general_basic.name().toString().equals(aiType)){
            //通用文字识别
            aiName="通用文字识别";
            GeneralBasicIOcrResult result= OcrService.general_basic(localFilePath);
            json=result.getJson();
        }
        AiHis aiHis=new AiHis();
        JSONObject jsonObject= JSONObject.parseObject(json);
        String e_code=String.valueOf(jsonObject.get("error_code"));
        String e_msg=String.valueOf(jsonObject.get("error_msg"));
        String log_id=String.valueOf(jsonObject.get("log_id"));
        aiHis.setId(log_id);
        if(StringUtils.isNotEmpty(e_code)&&!"0".equals(e_code)){
            aiHis.setResult("0");
        }else{
            aiHis.setResult("1");
        }
        if(StringUtils.isNotEmpty(e_msg)){
            aiHis.setErrorMsg(e_msg);
        }
        aiHis.setAiType(aiType);
        aiHis.setTypeName(aiName);
        aiHis.setJsonResult(json);
        SysUser user=ShiroUtils.getSysUser();
        aiHis.setYhid(String.valueOf(user.getUserId()));
        aiHis.setYhmc(user.getUserName());
        aiHisService.insertAiHis(aiHis);
        return AjaxResult.success("",json);
    }
    public Object queryAi2(String aiType,String localFilePath,Map resultMap){
        if(BaiduAi.AiType.faceDetect.name().equals(aiType)){
            FaceDetectResult result = FaceService.faceDetect_localPath(localFilePath);
            if(result.isRequestOk()){
                resultMap.put("result","成功");
                resultMap.put("age",result.getAge());
                resultMap.put("gender","female".equals(result.getGender())?"女":"男");
                resultMap.put("beauty",result.getBeauty());
                resultMap.put("expression",result.getExpression());
                resultMap.put("hasGlasses",result.isHasGlasses()?"有":"无");
            }else{
                resultMap.put("result","失败"+result.getError_msg());
            }
        }else if(BaiduAi.AiType.plant.name().equals(aiType)){
            PlantResult result= ImageClassifyService.plant(localFilePath);
            if(result.isRequestOk()){
                //请求成功
                if(result.isPlant()){
                    //确定是植物
                    resultMap.put("result","成功");
                    resultMap.put("name",result.getResult_name());
                    resultMap.put("probability",result.getResult_probability());
                }else{
                    //不能判定是否植物
                    resultMap.put("result","非植物");
                    resultMap.put("resultJson", JSON.toJSONString(result.getResult()));
                }
            }else{
                //请求失败
                resultMap.put("result","识别失败!"+result.getError_msg());
            }
        }else if(BaiduAi.AiType.bankCard.name().equals(aiType)){
            BankCardOcrResult result = OcrService.bankCardOcr(localFilePath);
            if(result.isRequestOk()){
                resultMap.put("result","成功");
                resultMap.put("cardNum",result.getBank_card_number());
                resultMap.put("bankName",result.getBank_name());
            }else{
                resultMap.put("result","识别失败!"+result.getError_msg());
            }
        }else if(BaiduAi.AiType.idCard.name().equals(aiType)){
            IdCardOcrResult result = OcrService.idcardOcr_Z(localFilePath);
            if(result.isRequestOk()){
                resultMap.put("result","成功");
                resultMap.put("gender",result.getGender());
                resultMap.put("address",result.getAddress());
                resultMap.put("birthday",result.getBirthday());
                resultMap.put("ID",result.getID());
                resultMap.put("name",result.getName());
                resultMap.put("minzhu",result.getMinZhu());
            }else{
                resultMap.put("result","识别失败!"+result.getError_msg());
            }
        }else if(BaiduAi.AiType.plate.name().equals(aiType)){
            //识别车牌号
            PlateNumberResult result = OcrService.license_plateOcr(localFilePath);
            if(result.isRequestOk()){
                resultMap.put("result","成功");
                resultMap.put("number",result.getPlateNumber());
                resultMap.put("color",result.getColor());
            }else{
                resultMap.put("result","识别失败!"+result.getError_msg());
            }
        }else if(BaiduAi.AiType.driver.name().equals(aiType)){
            //驾驶证识别
            DriveLicenseOcrResult result = OcrService.drivingLicenseOcr(localFilePath);
            if(result.isRequestOk()){
                resultMap.put("result","成功");
                resultMap.put("ID",result.getID());
                resultMap.put("name",result.getName());
                resultMap.put("gender",result.getGender());
                resultMap.put("address",result.getAddress());
                resultMap.put("birthday",result.getBirthday());
                resultMap.put("country",result.getCountry());
                resultMap.put("time",result.getStarttime()+"-"+result.getEndtime());
                resultMap.put("type",result.getType());//C1
            }else{
                resultMap.put("result","识别失败!"+result.getError_msg());
            }
        }else if(BaiduAi.AiType.animal.name().equals(aiType)){
            //动物识别
            AnimalResult result = ImageClassifyService.animal(localFilePath);
            if(result.isRequestOk()){
                if(result.isAnimal()){
                    resultMap.put("result","成功");
                    resultMap.put("name",result.getResult_name());
                    resultMap.put("probability",result.getResult_probability());
                }else{
                    resultMap.put("result","非动物");
                    resultMap.put("resultJson", JSON.toJSONString(result.getResult()));
                }
            }else{
                resultMap.put("result","识别失败!"+result.getError_msg());
            }
        }else if(BaiduAi.AiType.car.name().equals(aiType)){
            //车型识别
            CarResult result = ImageClassifyService.car(localFilePath);
            if(result.isRequestOk()){
                if(result.isCar()){
                    resultMap.put("result","成功");
                    resultMap.put("name",result.getResult_name());
                    resultMap.put("probability",result.getResult_probability());
                    resultMap.put("year",result.getResult_year());
                }else{
                    resultMap.put("result","非车辆");
                    resultMap.put("resultJson", JSON.toJSONString(result.getResult()));
                }
            }else{
                resultMap.put("result","识别失败!"+result.getError_msg());
            }
        }else if(BaiduAi.AiType.dish.name().equals(aiType)){
            //菜品识别
            DishResult result = ImageClassifyService.dish(localFilePath);
            if(result.isRequestOk()){
                if(result.isDish()){
                    resultMap.put("result","成功");
                    resultMap.put("name",result.getResult_name());
                    resultMap.put("probability",result.getResult_probability());
                    resultMap.put("calorie",result.getResult_calorie());
                }else{
                    resultMap.put("result","非菜");
                    resultMap.put("resultJson", JSON.toJSONString(result.getResult()));
                }
            }else{
                resultMap.put("result","识别失败!"+result.getError_msg());
            }
        }else if(BaiduAi.AiType.general_basic.name().toString().equals(aiType)){
            //通用文字识别
            GeneralBasicIOcrResult result= OcrService.general_basic(localFilePath);
            if(result.isRequestOk()){
                resultMap.put("result","成功");
                resultMap.put("words_result_num",result.getWords_result_num());
                resultMap.put("words",result.getWords());
                resultMap.put("paragraph",result.getParagraph());
            }else{
                resultMap.put("result","识别失败!"+result.getError_msg());
            }
        }
        return resultMap;
    }
}