package insertdata

import (
	"encoding/json"
	"errors"
	"fmt"
	"io"
	"io/ioutil"
	"net/http"
	"ruleprocess/logger"
	"strconv"
	"strings"
	"time"
    "bytes"
)

type Reps struct {
	FaceFeature string `json:"faceFeature"`
	ID          string
}

// 获取es 数据结构
func GetEsDataReq(url string, parama string) ([]Reps, error) {

	var reps Reps
	var repss []Reps
	fmt.Println("es 查询请求路径" + url) //  配置信息 获取
	req, err := http.NewRequest("POST", url, strings.NewReader(parama))

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

	req.Header.Add("Content-Type", "application/json")
	timeout := time.Duration(100 * time.Second)
	client := &http.Client{Timeout: timeout}
	resp, err := client.Do(req)

	if err != nil {
		return nil, err
	}

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

	jsonStr := string(body)
	var dat map[string]interface{}
	dec := json.NewDecoder(strings.NewReader(jsonStr))

	if err := dec.Decode(&dat); err == io.EOF {
		fmt.Println(err.Error())
		return nil, err
	} else if err != nil {
		fmt.Println(err.Error())
		return nil, err
	}
	dat, ok := dat["hits"].(map[string]interface{})
	if !ok {
		return nil, errors.New("data is not type of  map[string]interface{}")
	}

	for _, value := range dat["hits"].([]interface{}) {
		source, ok := value.(map[string]interface{})["_source"].(map[string]interface{})
		if !ok {
			return nil, errors.New("value is not type of map[string]interface{}")
		}
		reps.FaceFeature, ok = source["faceFeature"].(string)
		if !ok {
			fmt.Println("faceFeature is not string", source["faceFeature"])
			continue
		}
		reps.ID, ok = source["ID"].(string)
		if !ok {
			fmt.Println("ID is not string", source["ID"])
			continue
		}

		repss = append(repss, reps)
	}
	return repss, nil

}

func PostAction(querynum int, queryindex int) ([]Reps, error) {
    defer elapsed("page")()
    number := strconv.Itoa(querynum)
    point := strconv.Itoa(queryindex)
    url := "http://192.168.1.182:9200/videopersons/_search"
    parma := `{
       "from":` + point + `,
       "query": {
   		"match_all": {}
   	},
   	"sort": [
   	{
   		"picDate": {
   			"order": "desc"
   		}
   	}
   	],
   	"size":` + number + `,
   	"_source": [
   	"faceFeature",
   	"ID"
   	]
   }`

	repss, err := GetEsDataReq(url, parma)
	return repss, err
}

/** 插入es：
***  url: http://192.168.1.203:9200/personaction/perVideoAction/c58dbddf-4580-43f3-b76a-3213ec809df2?pretty
**param: // json [json slice] 
*/

/** 添加或者删除收藏
*** url: http://192.168.1.203:9200/videopersons/perVideoPicture/_id/_update
    param: []byte 
    json: {
        "doc": {
            "collection": "1" // 1 is 收藏 0 取消收藏
        }
        "detect_noop": false
    }
**/

func EsReq(method string, url string, parama []byte) (maps map[string]interface{},err error) {
    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) 
    }
	decoder := make(map[string]interface{})
	if err := json.Unmarshal([]byte(string(body)), &decoder); err != nil {
		return nil, err
	}
    fmt.Println(string(body))
    logger.Info(string(body))
    return decoder ,nil
}


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