package controllers

import (
	"basic.com/dbapi.git"
	"basic.com/valib/logger.git"
	"fmt"
	"github.com/gin-gonic/gin"
	"strconv"
	"strings"
	"webserver/cache"
	"webserver/extend/code"
	"webserver/extend/config"
	"webserver/extend/esutil"
	"webserver/extend/util"
	"webserver/models"
)

type EsSearchController struct{}

// @Security ApiKeyAuth
// @Summary 检索
// @Description 信息检索和比对检索
// @Accept  json
// @Produce json
// @Tags es
// @Param obj body models.EsSearch true "底库数据"
// @Success 200 {string} json "{"code":200, msg:"目录结构数据", success:true}"
// @Failure 500 {string} json "{"code":500,  msg:"返回错误信息", success:false}"
// @Router /data/api-v/es/esSearch [POST]
func (sc *EsSearchController) PostEsSearch(c *gin.Context) {
	//searchBody := make(map[string]interface{}, 0)
	//c.BindJSON(&searchBody)
	//data := findEsData(searchBody)

	var arg models.EsSearch
	err := c.BindJSON(&arg)
	if err != nil || arg.Page <= 0 && arg.Size <= 0 {
		util.ResponseFormat(c, code.RequestParamError, "参数有误")
		return
	}
	data := findEsData(&arg)

	util.ResponseFormat(c, code.Success, data)
}

//获取监控等级
func getAlarmLevel(alarmlevel []int32) []string {
	d := dbapi.DicApi{}
	res, data := d.FindByType("ALARMLEVEL")
	if data == nil {
		logger.Debug("data is nil")
	}
	logger.Debug(res)
	// logger.Debug(data)
	alarmLevel := make(map[int32]string)
	tmp := data.(map[string]interface{})
	for _, value := range tmp["ALARMLEVEL"].([]interface{}) {
		// logger.Debug(value.(map[string]interface{})["value"],"		",value.(map[string]interface{})["name"])
		val := value.(map[string]interface{})["value"].(string)
		name := value.(map[string]interface{})["name"].(string)
		vl, err := strconv.Atoi(val)
		if err != nil {
			logger.Debug(err)
		}
		alarmLevel[int32(vl)] = name
	}
	alarmLevelRes := make([]string, len(alarmlevel))
	for i, v := range alarmlevel {
		if v < 0 {
			alarmLevelRes[i] = "撤防"
		} else {
			alarmLevelRes[i] = alarmLevel[v]
		}
	}
	return alarmLevelRes
}

func findEsData(searchBody *models.EsSearch) map[string]interface{} {
	webPage := searchBody.Page

	webSize := searchBody.Size
	from := strconv.Itoa((webPage - 1) * webSize)
	//esFrom := strconv.Itoa(from)
	//esSize := strconv.Itoa(webSize)
	size := strconv.Itoa(webSize)
	//请求索引
	index := config.EsInfo.EsIndex.AiOcean.IndexName
	queryStr := ""
	queryBody := searchBody.InputValue
	//检索框
	if queryBody != "" {
		queryStr = "\"must\":[{\"multi_match\":{\"query\":\"" + queryBody + "\",\"fields\":[\"cameraAddr^1.5\",\"taskName^1.5\",\"sdkName^1.5\",\"showLabels^3.0\",\"baseInfo.tableName^1.5\",\"baseInfo.targetName^1.5\",\"baseInfo.labels^1.5\",\"alarmRules.alarmLevel^1.5\",\"linkTag^1.5\",\"linkTagInfo.cameraAddr\",\"linkTagInfo.taskName^1.5\",\"linkTagInfo.sdkName^1.5\",\"linkTagInfo.showLabels^1.5\",\"linkTagInfo.baseInfo.tableName^1.5\",\"linkTagInfo.baseInfo.targetName^1.5\",\"linkTagInfo.baseInfo.labels^1.5\",\"linkTagInfo.alarmRules.alarmLevel^1.5\"]," +
			"\"type\":\"cross_fields\",\"operator\":\"OR\",\"slop\":0,\"prefix_length\":0,\"max_expansions\":50,\"zero_terms_query\":\"NONE\",\"auto_generate_synonyms_phrase_query\":true,\"fuzzy_transpositions\":true,\"boost\":1}}],"
	}

	searchTime := searchBody.SearchTime
	if searchTime == nil || len(searchTime) < 2 {
		info := make(map[string]interface{}, 0)
		info["err"] = "请输入时间范围"
		return info
	}

	gteDate := searchTime[0]
	lteDate := searchTime[1]

	//判断数据ID
	idStr := ""
	linkTagInfoIdStr := ""
	//personId := searchBody.Id
	id := []string{}
	if id != nil && len(id) > 0 {
		esId := strings.Replace(strings.Trim(fmt.Sprint(id), "[]"), " ", "\",\"", -1)
		idStr = "{\"terms\":{\"id\":[\"" + esId + "\"]}},"
		linkTagInfoIdStr = "{\"terms\":{\"linkTagInfo.id\":[\"" + esId + "\"]}},"
	}

	//判断任务ID
	taskIdStr := ""
	linkTagInfoTaskIdStr := ""
	taskId := searchBody.Tasks
	if taskId != nil && len(taskId) > 0 {
		esTaskId := strings.Replace(strings.Trim(fmt.Sprint(taskId), "[]"), " ", "\",\"", -1)
		taskIdStr = "{\"terms\":{\"taskId\":[\"" + esTaskId + "\"]}},"
		linkTagInfoTaskIdStr = "{\"terms\":{\"linkTagInfo.taskId\":[\"" + esTaskId + "\"]}},"
	}

	//判断摄像机ID
	cameraIdStr := ""
	linkTagInfoCameraIdStr := ""
	cameraId := searchBody.TreeNodes
	if cameraId != nil && len(cameraId) > 0 {
		esCameraId := strings.Replace(strings.Trim(fmt.Sprint(cameraId), "[]"), " ", "\",\"", -1)
		cameraIdStr = "{\"terms\":{\"cameraId\":[\"" + esCameraId + "\"]}},"
		linkTagInfoCameraIdStr = "{\"terms\":{\"linkTagInfo.cameraId\":[\"" + esCameraId + "\"]}},"
	}

	//判断库表ID
	tableId := searchBody.Tabs
	esTableId := ""
	esTableIdStr := ""
	linkTagInfoEsTableIdStr := ""
	if tableId != nil && len(tableId) > 0 {
		esTableId = strings.Replace(strings.Trim(fmt.Sprint(tableId), "[]"), " ", "\",\"", -1)
		esTableIdStr = "{\"terms\":{\"baseInfo.tableId\":[\"" + esTableId + "\"]}},"
		linkTagInfoEsTableIdStr = "{\"terms\":{\"linkTagInfo.baseInfo.tableId\":[\"" + esTableId + "\"]}},"
	}
	//判断收藏状态
	isCollectStr := ""
	linkTagInfoIsCollectStr := ""
	isCollect := searchBody.Collection
	if isCollect != "" {
		isCollectStr = "{\"term\":{\"isCollect\":" + isCollect + "}},"
		linkTagInfoIsCollectStr = "{\"term\":{\"linkTagInfo.isCollect\":" + isCollect + "}},"
	}
	//判断布防等级id
	alarmLevelId := searchBody.AlarmLevel
	alarmLevelStr := ""
	linkTagInfoAlarmLevelStr := ""
	if len(alarmLevelId) > 0 {
		alarmLevelTypes := strings.Replace(strings.Trim(fmt.Sprint(getAlarmLevel(alarmLevelId)), "[]"), " ", "\",\"", -1)
		alarmLevelStr = "{\"terms\":{\"alarmRules.alarmLevel.raw\":[\"" + alarmLevelTypes + "\"]}},"
		linkTagInfoAlarmLevelStr = "{\"terms\":{\"linkTagInfo.alarmRules.alarmLevel.raw\":[\"" + alarmLevelTypes + "\"]}},"
	}

	//使用es底层机制处理分页
	//请求头
	localConf, err2 := cache.GetServerInfo()
	if err2 != nil || localConf.AlarmIp == "" || localConf.ServerId == "" {
		logger.Debug("localConfig is wrong!!!")
		return nil
	}
	url := "http://" + localConf.AlarmIp + ":" + strconv.Itoa(int(localConf.AlarmPort)) +
		"/" + index + "/_search?search_type=dfs_query_then_fetch"

	analyServerFilterStr := "{\"term\":{\"analyServerId\":\"" + localConf.ServerId + "\"}},"
	linkTagInfoAnalyServerFilterStr := "{\"term\":{\"linkTagInfo.analyServerId\":\"" + localConf.ServerId + "\"}},"

	//请求体
	prama := "{\"from\":\"" + from + "\"," +
		"\"size\":\"" + size + "\"," +
		"\"query\":{\"bool\":{" + queryStr +
		"\"should\":[" +
		"{\"bool\":{\"filter\":[" +
		cameraIdStr +
		alarmLevelStr +
		idStr +
		taskIdStr +
		isCollectStr +
		esTableIdStr +
		analyServerFilterStr +
		"{\"range\":{\"picDate\":{\"from\":\"" + gteDate + "\",\"to\":\"" + lteDate + "\",\"include_lower\":true,\"include_upper\":true,\"boost\":1}}}" +
		"]}}," +
		"{\"bool\":{\"filter\":[" +
		linkTagInfoCameraIdStr +
		linkTagInfoAlarmLevelStr +
		linkTagInfoIdStr +
		linkTagInfoTaskIdStr +
		linkTagInfoIsCollectStr +
		linkTagInfoEsTableIdStr +
		linkTagInfoAnalyServerFilterStr +
		"{\"range\":{\"linkTagInfo.picDate\":{\"from\":\"" + gteDate + "\",\"to\":\"" + lteDate + "\",\"include_lower\":true,\"include_upper\":true,\"boost\":1}}}" +
		"]}}" +
		"],\"minimum_should_match\":1}}," +
		"\"sort\":[{\"_score\":{\"order\":\"desc\"}},{\"updateTime\":{\"order\":\"desc\"}}]," +
		"\"_source\":{\"includes\":[],\"excludes\":[\"*.feature\"]}" +
		"}"
	logger.Debug("findEsData.param:", prama)
	//数据解析
	tokenRes := esutil.GetEsDataReq(url, prama, true)
	tmpAllDate := esutil.ResponseData(tokenRes)
	//fmt.Println(tokenRes)
	//tmpAllDate := make(map[string]interface{})
	//tmpDate := make(map[string][]interface{})
	//tmpAllDate["total"] = tokenRes["total"]
	//data := tokenRes["datalist"]
	//for _, masterInfoValues := range data.([]interface{}) {
	//	masterInfo := masterInfoValues.(map[string]interface{})
	//	sources := make(map[string]interface{}, 0)
	//	activateInfo := make(map[string]interface{})
	//	if masterInfo["linkTagInfo"] != nil {
	//		linkTagInfo := masterInfo["linkTagInfo"].([]interface{})
	//		delete(masterInfo, "linkTagInfo")
	//		if len(linkTagInfo) > 0 {
	//			slaveList := make([]interface{}, 0)
	//			slaveList = append(slaveList, masterInfo)
	//			slaveList = append(slaveList, linkTagInfo...)
	//			for i, j := 0, len(slaveList)-1; i < j; i, j = i+1, j-1 {
	//				slaveList[i], slaveList[j] = slaveList[j], slaveList[i]
	//			}
	//			sources["list"] = slaveList
	//			activateInfo = slaveList[0].(map[string]interface{})
	//		} else {
	//			sources["list"] = []interface{}{
	//				masterInfo,
	//			}
	//			activateInfo = masterInfo
	//		}
	//	} else {
	//		if _, ok := masterInfo["linkTagInfo"]; ok {
	//			delete(masterInfo, "linkTagInfo")
	//		}
	//		sources["list"] = []interface{}{
	//			masterInfo,
	//		}
	//		activateInfo = masterInfo
	//
	//	}
	//
	//	sources["activeObject"] = activateInfo
	//
	//	tmpDate["datalist"] = append(tmpDate["datalist"], sources)
	//}
	//tmpAllDate["datalist"] = tmpDate["datalist"]
	return tmpAllDate
}

//sdkTYype字典"\"_source\":[\"BaseName\",\"Gender\",\"Race\",\"content\",\"idcard\",\"picAddress\",\"picDate\",\"sdkType\",\"Age\",\"personId\",\"personIsHub\",\"personPicUrl\",\"picLocalUrl\",\"picSmUrl\",\"videoIp\",\"videoNum\",\"cameraId\",\"ageDescription\",\"likePer\"]" +
//func sdkTypeToValue(i int) string {
//	value := []string{"人脸", "车辆", "人体", "入侵", "拥挤", "靠右行", "人员异常", "个体静止"}
//	return value[i-1]
//}

//获取Dbtablepersons信息
/*func getDBPersonInfo(personId string) map[string]string {
	url := "http://" + config.EsInfo.Masterip + ":" + config.EsInfo.Httpport +
		"/" + config.EsInfo.EsIndex.Dbtablepersons.IndexName + "/_search"

	prama := "{\"query\":{\"term\":{\"uuid\":\"" + personId + "\"}},\"_source\":[\"personName\",\"phoneNum\"]}"

	tokenRes := esutil.GetEsDataReq(url, prama, true)
	personRes := make(map[string]string)
	//logger.Debug(tokenRes["datalist"].([]interface{}))
	result := tokenRes["datalist"].([]interface{})
	for _, value := range result {
		userMap := value.(map[string]interface{})
		if personName, ok := userMap["personName"]; ok {
			personRes["personName"] = personName.(string)
		} else {
			personRes["personName"] = ""
		}
		if phoneNum, ok := userMap["phoneNum"]; ok {
			personRes["phoneNum"] = phoneNum.(string)
		} else {
			personRes["phoneNum"] = ""
		}
		if sex, ok := userMap["sex"]; ok {
			personRes["sex"] = sex.(string)
		} else {
			personRes["sex"] = ""
		}
		// personRes["phoneNum"] = value.(map[string]interface{})["phoneNum"].(string)
		// personRes["sex"] = value.(map[string]interface{})["sex"].(string)
	}
	//logger.Debug(personRes)
	if len(personRes) < 1 {
		personRes["personName"] = ""
		personRes["phoneNum"] = ""
		personRes["sex"] = ""
	}
	return personRes
}*/