package com.cloud.retrieve.service.serviceImpl;

import com.alibaba.fastjson.JSONObject;
import com.cloud.common.model.FileInfos;
import com.cloud.common.utils.FastDFSUtil;
import com.cloud.common.utils.RestTemplateUtil;
import com.cloud.retrieve.dao.CulOrDevDao;
import com.cloud.retrieve.model.FaceResults;
import com.cloud.retrieve.service.SearchPhotoService;
import com.cloud.retrieve.service.UserService;
import com.cloud.retrieve.service.impl.EsDataServiceImpl;
import com.cloud.retrieve.utils.*;
import lombok.extern.log4j.Log4j;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.http.MediaType;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;
import java.util.*;

import static com.alibaba.fastjson.JSON.parseArray;

@Slf4j
@Service
public class SearchPhotoServiceImpl implements SearchPhotoService {

    @Autowired
    private EnumStr enumStr;
    @Autowired
    private RestTemplateUtil restTemplateUtil;
    @Autowired
    private ImageUtils imageUtils;
    @Autowired
    private RedisTemplate redisTemplate;
    @Autowired
    private EsDataServiceImpl esDataServiceImpl;
    @Autowired
    private UserService userService;
    @Autowired
    private FastDFSUtil fastDFSUtil;
    @Override
    public Map extractFace(MultipartFile file,int picQuality){
        FileInfos fileInfo = fastDFSUtil.upload(file);
        JSONObject paramMap = new JSONObject();
        paramMap.put("fileInfo",fileInfo);
        paramMap.put("picQuality",picQuality);

        String post = restTemplateUtil.postImg(enumStr.getExtractFace(), paramMap, MediaType.APPLICATION_JSON_UTF8,String.class, false );
        JSONObject jsonMap = JSONObject.parseObject(post);
        Map map = (Map)jsonMap.get("data");
        List<FaceResults> list = parseArray(JSONObject.parseObject(jsonMap.getString("data")).getString("list"), FaceResults.class);
        List<String> urlList = new ArrayList<>();
        List<String> featureList = new ArrayList<>();
        if(list!=null){
            for (FaceResults faceResults:list) {
                if(faceResults!=null){
                    log.info(faceResults.left+"====="+faceResults.top+"====="+faceResults.width+"====="+ faceResults.height);
                    FileInfos fileInf = imageUtils.cutPhotoFromFast(fileInfo, faceResults.left, faceResults.top, faceResults.width, faceResults.height);
                    urlList.add(fileInf.getPath());//enumStr.getHttpImgUrl()+"/"+
                    featureList.add(Base64Utils.byteToBase64(faceResults.getFeature()));
                }
            }
        }
        Map result = new HashMap();
        result.put("bigPhotoPath",fileInfo.getPath());
        result.put("smallPhotoPath",urlList);
        result.put("featureBase64",featureList);
        return result;
    }

    @Override
    public Map findLikerPics(Map<String,Object> params){
        // 根据path作为redis的key寻找所对应的集合，如果没有再往3697转发比对查询,返回的应该是从es中查询出来的封装了比分的数据对象
        List<Map>  esList = new ArrayList<>();
        List<Map>  baseList = new ArrayList<>();
        Map resultMap = new HashMap();
        Long start = (Long.valueOf(params.get("page")+"") -1)* Long.valueOf(params.get("length")+"");
        Long end = (Long.valueOf(params.get("page")+""))* Long.valueOf(params.get("length")+"")-1;
        log.info("开始："+start+"  结束："+end);
        String eskey = (String)params.get("path")+enumStr.getEsListkey();
        String basekey = (String)params.get("path")+enumStr.getBaseListkey();
        Long esDataTotle = null;
        Long baseDataTotle = null;
        if(Long.valueOf(params.get("page")+"")!=1){
            log.info("直接在redis中取数据");
            if(redisTemplate.hasKey(eskey)){
                esDataTotle = (Long) redisTemplate.opsForList().size(eskey);
                esList = redisTemplate.opsForList().range(eskey,start,end);
            }
            if(redisTemplate.hasKey(basekey)){
                baseDataTotle = (Long) redisTemplate.opsForList().size(basekey);
                baseList = redisTemplate.opsForList().range(basekey,0,-1);
            }
            log.info("es数据总数为"+esDataTotle+"--base数据总数为"+baseDataTotle);
        }else{
            log.info("去3697中比对后取数据");
            JSONObject paramMap = new JSONObject();
            paramMap.put("params",params);
            String post = restTemplateUtil.postImg(enumStr.getFindLikerPics(), paramMap, MediaType.APPLICATION_JSON_UTF8,String.class, false );
            log.info("3697返回"+post);
            JSONObject jsonMap = JSONObject.parseObject(post);
            if(redisTemplate.hasKey(eskey)){
                esDataTotle = (Long) redisTemplate.opsForList().size(eskey);
                esList = redisTemplate.opsForList().range(eskey,start,end);
            }
            if(redisTemplate.hasKey(basekey)){
                baseDataTotle = (Long) redisTemplate.opsForList().size(basekey);
                baseList = redisTemplate.opsForList().range(basekey,0,-1);
            }
            log.info("es数据总数为"+esDataTotle+"--base数据总数为"+baseDataTotle);
        }
        //  底库数据处理
        if(baseList!=null && baseList.size() > 0){
            baseList.forEach(map -> {
                String type = (String) map.get("type");
                if ("100".equals(type)){
                   map.put("no",null);
                }else if ("200".equals(type)){
                    map.put("phone",null);
                }
            });
        }
        JSONObject path = esDataServiceImpl.getPersonnLocusByRedis((String)params.get("path"));
        //给es中含有人员id的数据都拼上人员信息
        if(esList!=null && esList.size() > 0){
            esList.forEach(map -> {
                respSource(map);
            });
        }
        resultMap.put("esDataTotle",esDataTotle);
        resultMap.put("baseDataTotle",baseDataTotle);
        resultMap.put("esList",esList);
        resultMap.put("baseList",baseList);
        resultMap.put("path",path);
        return resultMap;
    }

    /**
     * 只返回 es 数据
     * @param params
     * @return
     */
    @Override
    public Map findLikerPicsRtnEsData(Map<String,Object> params) {
        // 根据path作为redis的key寻找所对应的集合，如果没有再往3697转发比对查询,返回的应该是从es中查询出来的封装了比分的数据对象
        List<Map>  esList = new ArrayList<>();
        Map resultMap = new HashMap();
        Long start = (Long.valueOf(params.get("page")+"") -1)* Long.valueOf(params.get("length")+"");
        Long end = (Long.valueOf(params.get("page")+""))* Long.valueOf(params.get("length")+"")-1;
        log.info("开始："+start+"  结束："+end);
        String eskey = (String)params.get("path")+enumStr.getEsListkey();
        Long esDataTotle = null;
        if(Long.valueOf(params.get("page")+"")!=1){
            if(redisTemplate.hasKey(eskey)) {
                esDataTotle = (Long) redisTemplate.opsForList().size(eskey);
                log.info("es数据总数为" + esDataTotle);
                esList = redisTemplate.opsForList().range(eskey, start, end);
            }
        }else{
            log.info("去3697中比对后取随行人员数据");
            JSONObject paramMap = new JSONObject();
            paramMap.put("params",params);
            String post = restTemplateUtil.postImg(enumStr.getFindLikerPics(), paramMap, MediaType.APPLICATION_JSON_UTF8,String.class, false );
            log.info("3697返回es 数据"+post);
            JSONObject jsonMap = JSONObject.parseObject(post);
            if(redisTemplate.hasKey(eskey)) {
                esDataTotle = (Long) redisTemplate.opsForList().size(eskey);
                log.info("es数据总数为" + esDataTotle);
                esList = redisTemplate.opsForList().range(eskey, start, end);
            }
        }
        //给es中含有人员id的数据都拼上人员信息
        if(esList!=null && esList.size() > 0){
            esList.forEach(map -> {
                respSource(map);
            });
        }
        resultMap.put("esTotle",esDataTotle);
        resultMap.put("esList",esList);
        return resultMap;
    }

    // 返回数据 source 特殊处理
    private void respSource(Map<String,Object> source){
        String sdkType = (String) source.get("sdkType");
        // sdkType 类型 进行处理
        // personId 'waittodo' 处理
        String personId = (String) source.get("personId");
        if (personId!=null && !personId.isEmpty() ){
            if ( personId.contains("wait todo")){
                source.put("personId","");
            }else {
                // 查询数据库   name, gender, cardId, phone,b.`no` numCard,o.name orgName
//                Map<String, Object> respMap = culOrDevDao.selectPersonDetailById(personId);
                Map respMap = userService.queryUserInfoById(personId);
                if (respMap != null){
                    String name = (String) respMap.get("name");
                    source.put("name",name);
                    source.put("gender",(String) respMap.get("gender"));   //
                    source.put("cardId",(String) respMap.get("cardId"));
                    source.put("phone",(String) respMap.get("phone"));
                    source.put("no",(String) respMap.get("no"));
                    source.put("orgName",(String) respMap.get("orgName"));
                }
            }
        }
        // 底库名称处理
        String baseName = (String) source.get("BaseName");
        if(baseName!=null && baseName.startsWith("lt_")){
            source.put("BaseName", baseName.substring(3));
        }else if (baseName.contains("wait todo")){
            source.put("BaseName","");
        }
        // Age 处理
       /* Object age =  source.get("Age");
        if(age!=null){
            int agei = -1;
            try {
                agei = (Integer)age;
            } catch (ClassCastException e){
                agei = Integer.valueOf((String)age);
                log.info(e.getLocalizedMessage());
            }
            if(agei>0 && agei < 7){
                age ="童年";
            }else if(agei>=7 && agei < 18){
                age ="少年";
            }else if(agei>=18 && agei < 40){
                age ="青年";
            }else if(agei>=40 && agei < 65){
                age ="中年";
            }else if(agei>= 65){
                age ="老年";
            }
        }else {age = "";}*/
    }
}