package stringhelper

import (
	"crypto/md5"
	"encoding/hex"
	"strconv"
	"strings"
)

/**
 * StrStr 实现php中的strstr，needle是否是haystack的子串，为空则返回空字符串和false
 * 如果beforeNeedle是true, result返回子串位置之前，否则返回子串位置之后
 */
func StrStr(haystack, needle string, beforeNeedle bool) (result string, exist bool) {
	index := strings.Index(haystack, needle)
	if index == -1 {
		return
	}
	if beforeNeedle {
		return string([]rune(haystack)[:index]), true
	} else {
		return string([]rune(haystack)[index:]), true
	}
}

// StrContains 判断字符串数组中，是否包含某一字符串
func StrContains(s []string, e string) bool {
	for _, a := range s {
		if a == e {
			return true
		}
	}
	return false
}

// IsNum 实现类似php的is_number
func IsNum(s string) bool {
	_, err := strconv.ParseFloat(s, 64)
	return err == nil
}

// IsEmpty 判断字符串是否为空,为空则返回默认值
func IsEmpty(str string, defval string) (result string) {
	if str == "" {
		result = defval
	} else {
		result = str
	}
	return
}

// MD5Encode 进行MD5处理
func MD5Encode(s string) string {
	h := md5.New()
	h.Write([]byte(s))
	return hex.EncodeToString(h.Sum(nil))
}