zwpanel/service/lib/cmn/base.go

package cmn

import (
	// "calendar-note-gin/assets"
	"crypto/md5"
	"encoding/hex"
	"fmt"
	"math/rand"
	"os"
	"path"
	"strconv"
	"strings"
	"sun-panel/assets"
	"time"
)

const (
	// 时间格式

	TimeFormatMode1         = "2006-01-02 15:04:05" // 标准格式
	TimeFormatMode4         = "2006-01-02 15:04"    // 标准格式 无秒
	TimeFormatMode2         = "Mon Jan 2 15:04:05 -0700 MST 2006"
	TimeFormatMode3         = "Mon, 2 Jan 2006 15:04:05 -0700 MST" // webdav格式
	TimeYYYY_mm_dd          = "2006-01-02"
	TIME_MODE_REMINDER_TIME = "200601021504" // 提醒定时器的执行时间格式

	// 随机码字典

	RAND_CODE_MODE1 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789" // 大写，小写，数字
	RAND_CODE_MODE2 = "abcdefghijklmnopqrstuvwxyz0123456789"                           // 小写，数字
	RAND_CODE_MODE3 = "0123456789"                                                     // 数字
)

type Version_Info struct {
	Version      string
	Version_code int
}

func GetTime() string {
	return time.Unix(time.Now().Unix(), 0).Format(TimeFormatMode1)
}

// 字符串转时间
func StrToTime(timeMode, formatTimeStr string) (t time.Time, err error) {
	loc, err := time.LoadLocation("Local")
	if err != nil {
		return
	}
	t, err = time.ParseInLocation(timeMode, formatTimeStr, loc) //使用模板在对应时区转化为time.time类型
	return
}

// md5获取
func Md5(str string) string {
	md5Byte := md5.Sum([]byte(str))
	return hex.EncodeToString(md5Byte[:])
}

func RandNum(n int) int {
	rand.Seed(time.Now().Unix())
	return rand.Intn(n)
}

// 随机生成编码
// 随机码字典内容 参考常量 RAND_CODE_MODE*
func BuildRandCode(count int, secret_content string) (code string) {
	return BuildRandCodeBySeed(count, secret_content, time.Now().UnixNano()+int64(rand.Intn(100)))
}

// 随机生成编码 参考常量 RAND_CODE_MODE*
func BuildRandCodeBySeed(count int, secret_content string, seed int64) (code string) {
	// 获取纳秒作为随机数种子
	rand.Seed(seed)
	if secret_content == "" {
		secret_content = RAND_CODE_MODE1
	}
	for i := 0; i < count; i++ {
		code += string(secret_content[rand.Intn(len(secret_content))])
	}
	return code
}

func InSlice(items []string, item string) bool {
	for _, eachItem := range items {
		if eachItem == item {
			return true
		}
	}
	return false
}

// 字符串转int
func StrToInt(str string) int {
	intStr, _ := strconv.Atoi(str)
	return intStr
}

// uint 转string
func UintToStr(c uint) string {
	return strconv.FormatUint(uint64(c), 10)
}

// uint 转string
func StrToUint(s string) uint {
	// i, _ := strconv.Atoi(s)
	u, _ := strconv.ParseUint(s, 10, 64)
	return uint(u)
}

// 获取系统信息
func GetSysVersionInfo() Version_Info {
	cBytes, _ := assets.Asset("assets/version")
	c := string(cBytes)
	info := strings.Split(c, "|")

	return Version_Info{
		Version_code: StrToInt(info[0]),
		Version:      info[1],
	}
}

// 文件是否存在
func PathExists(path string) (bool, error) {

	_, err := os.Stat(path)
	if err == nil {
		return true, nil
	}
	if os.IsNotExist(err) {
		return false, nil
	}
	return false, err
}

// 截取字符串，支持多字节字符
// start：起始下标，负数从从尾部开始，最后一个为-1
// length：截取长度，负数表示截取到末尾
func SubRuneStr(str string, start int, length int) (result string) {
	s := []rune(str)
	total := len(s)
	if total == 0 {
		return
	}
	// 允许从尾部开始计算
	if start < 0 {
		start = total + start
		if start < 0 {
			return
		}
	}
	if start > total {
		return
	}
	// 到末尾
	if length < 0 {
		length = total
	}

	end := start + length
	if end > total {
		result = string(s[start:])
	} else {
		result = string(s[start:end])
	}

	return
}

// 字符串长度
func RuneStrLen(str string) int {
	return len([]rune(str))
}

// 是否在数组中
func InStringArray(arr []string, item string) bool {
	for _, v := range arr {
		if v == item {
			return true
		}
	}
	return true
}

// 从Assets文件夹中抽取文件保存到路劲
// AssetsTakeFileToPath("config.ini", targetPath string)
func AssetsTakeFileToPath(assetsPath, targetPath string) error {
	bytes, _ := assets.Asset("assets/" + assetsPath)
	targetPathPath := path.Dir(targetPath)
	exists, err := PathExists(targetPathPath)

	if err != nil {
		return err
	}
	if !exists {
		if err := os.MkdirAll(targetPathPath, 0777); err != nil {
			fmt.Println(456)
			return err
		}
	}
	return os.WriteFile(targetPath, bytes, 0666)
}

// 密码加密
func PasswordEncryption(password string) string {
	return Md5(Md5(Md5(password)))
}