mDoc/tools/tools.go

package tools

import (
	"bufio"
	"fmt"
	"io"
	"io/ioutil"
	"math/rand"
	"os"
	"os/exec"
	"path"
	"path/filepath"
	"reflect"
	"runtime"
	"sort"
	"strconv"
	"strings"
	"time"
)

// 获取结构中的Name标签
// 返回字典
func StructTagNameMap(str interface{}, tag string) (ret map[string]string) {
	ret = make(map[string]string)
	s := reflect.TypeOf(str).Elem()
	for i := 0; i < s.NumField(); i++ {
		if s.Field(i).Tag.Get(tag) != "" {
			ret[s.Field(i).Name] = s.Field(i).Tag.Get(tag) //将tag输出
		}
	}
	return
}

// 获取结构中的Name标签
// 返回数组
// Tag中为-号，则不显示
func StructTagNameGroup(str interface{}, tag string) (ret []string) {
	s := reflect.TypeOf(str).Elem()
	for i := 0; i < s.NumField(); i++ {
		if s.Field(i).Tag.Get(tag) != "-" {
			ret = append(ret, s.Field(i).Name) //将tag输出
		}
	}
	return
}

// 判断文件或文件夹是否存在
func IsExist(path string) bool {
	_, err := os.Stat(path)
	if err != nil {
		if os.IsExist(err) {
			return true
		}
		if os.IsNotExist(err) {
			return false
		}
		fmt.Println(err)
		return false
	}
	return true
}

// 调试信息输出
func Debug(show bool, args ...interface{}) {
	if show { // 调试信息可以通过全局变量，统一关闭
		fmt.Println(args...)
	}
}

// 随机数
func RndNum(max int) int {
	rnd := rand.New(rand.NewSource(time.Now().UnixNano()))
	return rnd.Intn(100)
}

// 生成几位随机数的字符串
func RandString(len int) string {
	r := rand.New(rand.NewSource(time.Now().UnixNano()))
	bytes := make([]byte, len)
	for i := 0; i < len; i++ {
		b := r.Intn(10) + 48
		bytes[i] = byte(b)
	}
	return string(bytes)
}

// 生成重复字符
func Repeat(str string, num int) (ret string) {
	for i := 0; i < num; i++ {
		ret = ret + str
	}
	return
}

// 转换 --------------------------------------

func StrToInt(str string) int {
	i, e := strconv.Atoi(str)
	if e != nil {
		return 0
	}
	return i
}

func StrToUInt(str string) uint {
	i, e := strconv.Atoi(str)
	if e != nil {
		return 0
	}
	return uint(i)
}

func UintToInt64(num uint) int64 {
	return int64(num)
}

// 对Int数取绝对值
func AbsByInt(num int) int {
	if num < 0 {
		return -num
	}
	return num
}

// 数值转字符
func IntToStr(num int) string {
	return strconv.Itoa(num)
}

// interface 转字符
func InterfaceToStr(value interface{}) string {
	if str, ok := value.(string); ok {
		return str
	}
	return ""
}

// 字符串数组包含判断
func StringIn(target string, str_array []string) bool {
	sort.Strings(str_array)
	index := sort.SearchStrings(str_array, target)
	if index < len(str_array) && str_array[index] == target {
		return true
	}
	return false

}

// 随机字串
func RndNonceStr() string {
	str := "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
	bytes := []byte(str)
	result := []byte{}
	r := rand.New(rand.NewSource(time.Now().UnixNano()))
	for i := 0; i < 32; i++ {
		result = append(result, bytes[r.Intn(len(bytes))])
	}
	return string(result)
}

// 执行命令
func Command(params []string) (string, error) {
	var c *exec.Cmd
	switch runtime.GOOS {
	case "darwin":
	case "windows":
		para := append([]string{"/C"}, params[0:]...)
		c = exec.Command("cmd", para...)
	case "linux":
		para := append([]string{"-c"}, params[0:]...)
		c = exec.Command("bash", para...)
	}

	output, err := c.CombinedOutput()
	return string(output), err
}

// 获取所有文件
func AllFileByName(pathSeparator string, fileDir string, extFile string, hasDir bool) (ret []string) {
	files, _ := ioutil.ReadDir(fileDir)

	for _, onefile := range files {
		if onefile.IsDir() {
			// fmt.Println(onefile.Name())
			//fmt.Println(tmpPrefix, onefile.Name(), "目录:")
			if hasDir {
				AllFileByName(pathSeparator, fileDir+pathSeparator+onefile.Name(), extFile, hasDir)
			}
		} else {
			if strings.ToUpper(extFile) == strings.ToUpper(path.Ext(onefile.Name())) || extFile == "" {
				ret = append(ret, onefile.Name())
			}
		}
	}
	return

}

// 删除文件
func RemoveFile(path string, fileList []string) {
	for _, n := range fileList {
		os.Remove(path + n)
	}
}

// 复制文件
func CopyFile(srcFilePath string, dstFilePath string) (written int64, err error) {
	srcFile, err := os.Open(srcFilePath)
	if err != nil {
		fmt.Printf("打开源文件错误，错误信息=%v\n", err)
	}
	defer srcFile.Close()
	reader := bufio.NewReader(srcFile)

	dstFile, err := os.OpenFile(dstFilePath, os.O_WRONLY|os.O_CREATE, 0777)
	if err != nil {
		fmt.Printf("打开目标文件错误，错误信息=%v\n", err)
		return
	}
	writer := bufio.NewWriter(dstFile)
	defer dstFile.Close()
	return io.Copy(writer, reader)
}

// 获取当前程序名
func GetMeName() string {
	return filepath.Base(os.Args[0])
}

// 获取字符长度
// GetStrLength("这是中文，和ABC长度.😂")
// 返回:表情符数量，汉字数量，英文数量
func GetStrLength(str string) (mNum int, cNum int, eNum int) {
	rs := []rune(str)
	for i := 0; i < len(rs); i++ {
		switch len(string(rs[i])) {
		case 4:
			mNum += 1
		case 3:
			cNum += 1
		case 1:
			eNum += 1
		}
	}

	return mNum, cNum, eNum // 表情符数量，汉字数量，英文数量
}

// 支持三元表达 -------------------------------------------------

// callFn if args[i] == func, run it
func callFn(f interface{}) interface{} {
	if f != nil {
		t := reflect.TypeOf(f)
		if t.Kind() == reflect.Func && t.NumIn() == 0 {
			function := reflect.ValueOf(f)
			in := make([]reflect.Value, 0)
			out := function.Call(in)
			if num := len(out); num > 0 {
				list := make([]interface{}, num)
				for i, value := range out {
					list[i] = value.Interface()
				}
				if num == 1 {
					return list[0]
				}
				return list
			}
			return nil
		}
	}
	return f
}

func isZero(f interface{}) bool {
	v := reflect.ValueOf(f)
	switch v.Kind() {
	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
		return v.Int() == 0
	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
		return v.Uint() == 0
	case reflect.Float32, reflect.Float64:
		return v.Float() == 0
	case reflect.String:
		str := v.String()
		if str == "" {
			return true
		}
		zero, error := strconv.ParseFloat(str, 10)
		if zero == 0 && error == nil {
			return true
		}
		boolean, error := strconv.ParseBool(str)
		return boolean == false && error == nil
	default:
		return false
	}
}

// If - (a ? b : c) Or (a && b)
// var a = If(20 > 50, true, false)  // a = false
// var b = Or(a, 100)  // b = 100
// var c = If(b, 50)  // c = 50
// var fn = If(c > 40, func(){ return "ok" }) // fn = "ok"

func If(args ...interface{}) interface{} {
	var condition = callFn(args[0])
	if len(args) == 1 {
		return condition
	}
	var trueVal = args[1]
	var falseVal interface{}
	if len(args) > 2 {
		falseVal = args[2]
	} else {
		falseVal = nil
	}
	if condition == nil {
		return callFn(falseVal)
	} else if v, ok := condition.(bool); ok {
		if v == false {
			return callFn(falseVal)
		}
	} else if isZero(condition) {
		return callFn(falseVal)
	} else if v, ok := condition.(error); ok {
		if v != nil {
			fmt.Println(v)
			return condition
		}
	}
	return callFn(trueVal)
}

// Or - (a || b)
func Or(args ...interface{}) interface{} {
	var condition = callFn(args[0])
	if len(args) == 1 {
		return condition
	}
	if condition == nil {
		return callFn(args[1])
	}
	if v, ok := condition.(bool); ok {
		if v == false {
			return callFn(args[1])
		}
	} else if isZero(condition) {
		return callFn(args[1])
	} else if v, ok := condition.(error); ok {
		if v != nil {
			fmt.Println(v)
			return condition
		}
	}
	return condition
}

// 将字符填满位数 str 要填充的字符  size 填充为多少位  addStr 填充什么字符
func StrSetSize(str string, size int, addStr string) string {
	var a string
	if len(str) > size {
		return str[0:size]
	} else {
		for i := 0; i < size-len(str); i++ {
			a = a + addStr
		}
		return fmt.Sprintf("%s%s", str, a)
	}
}