package util

import (
	"bytes"
	"encoding/json"
	"errors"
	"fmt"
	"math"
	"os/exec"
	"strconv"
	"time"
)

//脚本封装
func RunScript(str string) string {
	cmd := exec.Command("sh", "-c", str)
	var out bytes.Buffer
	cmd.Stdout = &out
	err := cmd.Run()
	if err != nil {
		return "运行失败"
	}
	return out.String()
}

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 SourceTotal(buf []byte) (total int, err error) {
	var info interface{}
	json.Unmarshal(buf, &info)
	out, ok := info.(map[string]interface{})
	if !ok {
		return -1, errors.New("http response interface can not change map[string]interface{}")
	}

	middle, ok := out["hits"].(map[string]interface{})
	if !ok {
		return -1, errors.New("first total change error!")
	}

	tmp, b := middle["total"].(map[string]interface{})
	if b != true {
		v := middle["total"].(float64)
		t := int(v)
		return t, nil
	}
	value := tmp["value"].(float64)
	total = int(value)
	return total, nil
}

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

	middle, ok := out["result"].(string)
	if !ok {
		return false, errors.New("first total change error!")
	}
	if middle == "created" || middle == "updated" {
		result = true
	}
	return result, nil
}

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

	middle, ok := out["deleted"].(float64)
	if !ok {
		return -1, errors.New("first total change error!")
	}
	total = int(middle)
	return total, nil
}

func SourceUpdated(buf []byte) (total int, err error) {
	var info interface{}
	json.Unmarshal(buf, &info)
	out, ok := info.(map[string]interface{})
	if !ok {
		return -1, errors.New("http response interface can not change map[string]interface{}")
	}
	//fmt.Println(out)
	middle, ok := out["updated"].(float64)
	if !ok {
		return -1, errors.New("first total change error!")
	}
	total = int(middle)
	return total, nil
}

func StartTimer(f func()) {
	for {
		f()
		now := time.Now()
		// 计算下一个零点
		next := now.Add(time.Hour * 24)
		next = time.Date(next.Year(), next.Month(), next.Day(), 0, 0, 0, 0, next.Location())
		t := time.NewTimer(next.Sub(now))
		<-t.C
		t.Stop()
	}
}

//日期换算工具
func GetTimeArr(earliestDate string) int {
	now := time.Now().Format("2006-01-02")
	fmt.Println(earliestDate, now)
	a, _ := time.Parse("2006-01-02", earliestDate)
	b, _ := time.Parse("2006-01-02", now)
	d := a.Sub(b)
	date := d.Hours() / 24
	return int(math.Abs(date))
}

//日期遍历工具
// GetBetweenDates 根据开始日期和结束日期计算出时间段内所有日期
// 参数为日期格式，如：2020-01-01
func GetBetweenDates(sdate, edate string) []string {
	d := []string{}
	timeFormatTpl := "2006-01-02"
	if len(timeFormatTpl) != len(sdate) {
		timeFormatTpl = timeFormatTpl[0:len(sdate)]
	}
	date, err := time.Parse(timeFormatTpl, sdate)
	if err != nil {
		// 时间解析，异常
		return d
	}
	date2, err := time.Parse(timeFormatTpl, edate)
	if err != nil {
		// 时间解析，异常
		return d
	}
	if date2.Before(date) {
		// 如果结束时间小于开始时间，异常
		return d
	}
	// 输出日期格式固定
	timeFormatTpl = "2006-01-02"
	date2Str := date2.Format(timeFormatTpl)
	d = append(d, date.Format(timeFormatTpl))
	for {
		date = date.AddDate(0, 0, 1)
		dateStr := date.Format(timeFormatTpl)
		d = append(d, dateStr)
		if dateStr == date2Str {
			break
		}
	}
	return d
}

func IpIntToString(ipInt int) string {
	ipSegs := make([]string, 4)
	var len int = len(ipSegs)
	buffer := bytes.NewBufferString("")
	for i := 0; i < len; i++ {
		tempInt := ipInt & 0xFF
		ipSegs[len-i-1] = strconv.Itoa(tempInt)

		ipInt = ipInt >> 8
	}
	for i := 0; i < len; i++ {
		buffer.WriteString(ipSegs[i])
		if i < len-1 {
			buffer.WriteString(".")
		}
	}
	return buffer.String()
}