package esutil

import (
	"bytes"
	"encoding/json"
	"errors"
	"fmt"
	"io/ioutil"
	"net/http"
	"time"

	"basic.com/pubsub/protomsg.git"
)

func Parsesources(sources []map[string]interface{}) (esinfos []*protomsg.Esinfo) {
	var ok bool
	for _, source := range sources {
		var tmpinfo protomsg.Esinfo
		tmpinfo.FaceFeature, ok = source["faceFeature"].(string)
		if !ok {
			continue
		}

		tmpinfo.Id, ok = source["id"].(string)
		if !ok {
			continue
		}
		tmpinfo.Tableid, ok = source["tableId"].(string)
		if !ok {
			continue
		}

		esinfos = append(esinfos, &tmpinfo)

		//        tmpinfo.PersonName,ok  =   source["personName"].(string)
		//            Isnil("personName", ok)
		//        tmpinfo.PersonPicUrl,ok =  source["personPicUrl"].(string)
		//            Isnil("personPicUrl", ok)
		//        tmpinfo.PhoneNum,ok     =  source["phoneNum"].(string)
		//            Isnil("phoneNum", ok)
		//        tmpinfo.Sex,ok          =  source["sex"].(string)
		//            Isnil("sex", ok)
		//        tmpinfo.Idcard,ok       =  source["idCard"].(string)
		//            Isnil("idCard", ok)
		//        tmpinfo.MonitorLevel,ok =  source["monitorLevel"].(string)

		// 根据 tableid 获取 tablename
		//        name, _:= Dbtablename(tmpinfo.TableId)
		//        tmpinfo.TableName= name
		//        binfos.Infos = append(binfos.Infos, &tmpinfo)

	}
	return
}

//解析底库人员结构
func Dbpersonbyid(sources []map[string]interface{}) protomsg.Dbperson {
	var ok bool

	var tmpinfo protomsg.Dbperson
	for _, source := range sources {
		tmpinfo.FaceFeature, ok = source["faceFeature"].(string)
		if !ok {
			continue
		}

		tmpinfo.Id, ok = source["id"].(string)
		if !ok {
			continue
		}

		tmpinfo.TableId, ok = source["tableId"].(string)
		if !ok {
			continue
		}

		tmpinfo.PersonName, ok = source["personName"].(string)
		Isnil("personName", ok)
		tmpinfo.PersonPicUrl, ok = source["personPicUrl"].(string)
		Isnil("personPicUrl", ok)
		tmpinfo.PhoneNum, ok = source["phoneNum"].(string)
		Isnil("phoneNum", ok)
		tmpinfo.Sex, ok = source["sex"].(string)
		Isnil("sex", ok)
		tmpinfo.IdCard, ok = source["idCard"].(string)
		Isnil("idCard", ok)
		tmpinfo.MonitorLevel, ok = source["monitorLevel"].(string)
		Isnil("monitorLevel", ok)
		tmpinfo.UpdateTime, ok = source["updateTime"].(string)
		Isnil("updateTime", ok)
		tmpinfo.Age, ok = source["age"].(string)
		Isnil("age", ok)

		isDelete, ok := source["isDelete"].(float64)
		if ok {
			tmpinfo.IsDelete = int32(isDelete)
		}
		Isnil("isDelete", ok)

		enable, ok := source["enable"].(float64)
		if ok {
			tmpinfo.Enable = int32(enable)
		}
		Isnil("endTime", ok)

		tmpinfo.Reserved, ok = source["reserved"].(string)
		Isnil("reserved", ok)
		tmpinfo.PicDesc, ok = source["picDesc"].(string)
		Isnil("picDesc", ok)
		tmpinfo.CreateTime, ok = source["createTime"].(string)
		Isnil("createTime", ok)
		tmpinfo.CreateBy, ok = source["createBy"].(string)
		Isnil("createBy", ok)

		//根据 tableid 获取 tablename
		//name, _:= Dbtablename(tmpinfo.TableId)
		//tmpinfo.TableName= name
	}

	return tmpinfo
}

//解析底库结构
func Dbtablebyid(sources []map[string]interface{}) protomsg.Dbtable {
	var ok bool

	var tmpinfo protomsg.Dbtable
	for _, source := range sources {

		tmpinfo.Id, ok = source["id"].(string)
		if !ok {
			continue
		}
		tmpinfo.TableName, ok = source["tableName"].(string)
		Isnil("tableName", ok)
		tmpinfo.TableType, ok = source["tableType"].(string)
		Isnil("tableType", ok)
		tmpinfo.CreateBy, ok = source["createBy"].(string)
		Isnil("createBy", ok)
		tmpinfo.TableDesc, ok = source["tableDesc"].(string)
		Isnil("tableDesc", ok)
		tmpinfo.BwType, ok = source["bwType"].(string)
		Isnil("bwType", ok)
		tmpinfo.StartTime, ok = source["startTime"].(string)
		Isnil("startTime", ok)
		tmpinfo.EndTime, ok = source["endTime"].(string)
		Isnil("endTime", ok)

		isDelete, ok := source["isDelete"].(float64)
		if ok {
			tmpinfo.IsDelete = int32(isDelete)
		}
		Isnil("isDelete", ok)

		enable, ok := source["enable"].(float64)
		if ok {
			tmpinfo.Enable = int32(enable)
		}
		Isnil("endTime", ok)

		tmpinfo.CreateTime, ok = source["createTime"].(string)
		Isnil("createTime", ok)
		tmpinfo.UpdateTime, ok = source["updateTime"].(string)
		Isnil("updateTime", ok)
		tmpinfo.IsForever, ok = source["isForever"].(string)
		Isnil("isForever", ok)
		tmpinfo.IsSync, ok = source["isSync"].(string)
		Isnil("isSync", ok)
		tmpinfo.AnalyServerId, ok = source["analyServerId"].(string)
		Isnil("analyServerId", ok)
	}

	return tmpinfo
}

func Sourcelist(buf []byte) (sources []map[string]interface{}, err error) {
	var info interface{}
	json.Unmarshal(buf, &info)
	out, ok := info.(map[string]interface{})
	if !ok {
		return nil, errors.New("http response interface can not change map[string]interface{}")
	}

	middle, ok := out["hits"].(map[string]interface{})
	if !ok {
		return nil, errors.New("first hits change error!")
	}
	for _, in := range middle["hits"].([]interface{}) {
		tmpbuf, ok := in.(map[string]interface{})
		if !ok {
			fmt.Println("change to source error!")
			continue
		}
		source, ok := tmpbuf["_source"].(map[string]interface{})
		if !ok {
			fmt.Println("change _source error!")
			continue
		}
		sources = append(sources, source)
	}
	return sources, nil
}

func EsReq(method string, url string, parama []byte) (buf []byte, err error) {
	defer elapsed("page")()
	timeout := time.Duration(10 * time.Second)
	client := http.Client{
		Timeout: timeout,
	}
	request, err := http.NewRequest(method, url, bytes.NewBuffer(parama))
	request.Header.Set("Content-type", "application/json")

	if err != nil {
		fmt.Println("build request fail !")
		return nil, err
	}

	resp, err := client.Do(request)
	if err != nil {
		fmt.Println("request error: ", err)
		return nil, err
	}

	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Println(err)
		return nil, err
	}
	return body, nil
}

// 计算时间
func elapsed(what string) func() {
	start := time.Now()
	return func() {
		fmt.Printf("%s took %v\n", what, time.Since(start))
	}
}

// 赋值时检测是否能够赋值
func Isnil(key string, ok bool) {
	if !ok {
		fmt.Println(key, "is nil can not asign")
	}
}