package utils

import (
	"fmt"
	"net/http"
	"net/url"
	"reflect"
	"strings"
	"time"

	"git.rosy.net.cn/baseapi"
	"github.com/satori/go.uuid"
)

func DictKeysMan(data interface{}, keysToRemove []string, keysToKeep []string) interface{} {
	if data == nil || (keysToKeep == nil && keysToRemove == nil) {
		return data
	}

	var keysToRemoveMap map[string]int
	var keysToKeepMap map[string]int
	if keysToKeep != nil {
		keysToKeepMap = make(map[string]int)
		for _, v := range keysToKeep {
			keysToKeepMap[v] = 1
		}
	} else if keysToRemove != nil {
		keysToRemoveMap = make(map[string]int)
		for _, v := range keysToRemove {
			keysToRemoveMap[v] = 1
		}
	}

	dataIsSlice := true
	valueOfData := reflect.ValueOf(data)
	if valueOfData.Kind() != reflect.Slice {
		dataIsSlice = false
		valueOfData = reflect.ValueOf([]interface{}{data})
	} else if valueOfData.Len() == 0 {
		return data
	}

	retVal := make([]interface{}, valueOfData.Len())

	for index := 0; index < valueOfData.Len(); index++ {
		realV := GetConcretValue(valueOfData.Index(index))
		if keysToRemoveMap != nil || keysToKeepMap != nil {
			mapV := make(map[string]interface{})
			for _, key := range realV.MapKeys() {
				fieldName := key.String()
				wantThisField := true
				if keysToKeepMap != nil {
					if _, ok := keysToKeepMap[fieldName]; !ok {
						wantThisField = false
					}
				} else if keysToRemoveMap != nil {
					if _, ok := keysToRemoveMap[fieldName]; ok {
						wantThisField = false
					}
				}
				if wantThisField {
					mapV[fieldName] = GetConcretValue(realV.MapIndex(key)).Interface()
				}
			}
			retVal[index] = mapV
		} else {
			retVal[index] = realV.Interface()
		}
	}

	if !dataIsSlice {
		return retVal[0]
	}
	return retVal
}

// 去除-号，全部大写，比如：929ADB626EB911E893E452540009DAB3
func GetUUID() string {
	return strings.ToUpper(strings.Replace(uuid.Must(uuid.NewV1()).String(), "-", "", -1))
}

func GetCurTimeStr() string {
	return Time2Str(time.Now())
}

// timestamp is in second
func GetCurTimestamp() int64 {
	return time.Now().Unix()
}

func GetAPIOperator() string {
	return "jxc4-" + Time2Str(time.Now())
}

func MergeMaps(firstMap map[string]interface{}, otherMaps ...map[string]interface{}) (retVal map[string]interface{}) {
	retVal = make(map[string]interface{})
	allMaps := append(otherMaps, firstMap)
	for _, oneMap := range allMaps {
		for k, v := range oneMap {
			retVal[k] = v
		}
	}

	return retVal
}

func CallFuncLogError(funcToCall func() error, msg string) error {
	err := funcToCall()
	if err != nil {
		baseapi.SugarLogger.Warnf("%s, error:%v", msg, err.Error())
	}
	return err
}

func CallFuncAsync(funcToCall func()) {
	go func() {
		defer func() {
			if r := recover(); r != nil {
				baseapi.SugarLogger.Errorf("error when calling func:%v, r:%v", funcToCall, r)
			}
		}()
		funcToCall()
	}()
}

func GenerateGetURL(baseURL, apiStr string, params map[string]interface{}) string {
	queryString := ""
	if params != nil {
		for k, v := range params {
			if queryString == "" {
				queryString = "?"
			} else {
				queryString += "&"
			}
			queryString += k + "=" + url.QueryEscape(fmt.Sprint(v))
		}
	}
	return baseURL + "/" + apiStr + queryString
}

func SendFakeRequest(method, url, body, contentType string) (*http.Response, error) {
	request, _ := http.NewRequest(method, url, strings.NewReader(body))
	if contentType == "" {
		contentType = "application/x-www-form-urlencoded"
	}
	request.Header.Set("Content-Type", contentType)
	client := &http.Client{}
	return client.Do(request)
}