mirror of
https://github.com/Mrs4s/go-cqhttp.git
synced 2025-05-05 03:23:49 +08:00
120 lines
2.5 KiB
Go
120 lines
2.5 KiB
Go
package global
|
||
|
||
import (
|
||
"math"
|
||
"regexp"
|
||
"strconv"
|
||
"strings"
|
||
"sync"
|
||
|
||
log "github.com/sirupsen/logrus"
|
||
"github.com/tidwall/gjson"
|
||
)
|
||
|
||
// EnsureBool 判断给定的p是否可表示为合法Bool类型,否则返回defaultVal
|
||
//
|
||
// 支持的合法类型有
|
||
//
|
||
// type bool
|
||
//
|
||
// type gjson.True or gjson.False
|
||
//
|
||
// type string "true","yes","1" or "false","no","0" (case insensitive)
|
||
func EnsureBool(p interface{}, defaultVal bool) bool {
|
||
var str string
|
||
if b, ok := p.(bool); ok {
|
||
return b
|
||
}
|
||
if j, ok := p.(gjson.Result); ok {
|
||
if !j.Exists() {
|
||
return defaultVal
|
||
}
|
||
if j.Type == gjson.True {
|
||
return true
|
||
}
|
||
if j.Type == gjson.False {
|
||
return false
|
||
}
|
||
if j.Type != gjson.String {
|
||
return defaultVal
|
||
}
|
||
str = j.Str
|
||
} else if s, ok := p.(string); ok {
|
||
str = s
|
||
}
|
||
str = strings.ToLower(str)
|
||
switch str {
|
||
case "true", "yes", "1":
|
||
return true
|
||
case "false", "no", "0":
|
||
return false
|
||
default:
|
||
return defaultVal
|
||
}
|
||
}
|
||
|
||
// VersionNameCompare 检查版本名是否需要更新, 仅适用于 go-cqhttp 的版本命名规则
|
||
//
|
||
// 例: v0.9.29-fix2 == v0.9.29-fix2 -> false
|
||
//
|
||
// v0.9.29-fix1 < v0.9.29-fix2 -> true
|
||
//
|
||
// v0.9.29-fix2 > v0.9.29-fix1 -> false
|
||
//
|
||
// v0.9.29-fix2 < v0.9.30 -> true
|
||
//
|
||
// v1.0.0-alpha2 < v1.0.0-beta1 -> true
|
||
//
|
||
// v1.0.0 > v1.0.0-beta1 -> false
|
||
func VersionNameCompare(current, remote string) bool {
|
||
defer func() { // 应该不会panic， 为了保险还是加个
|
||
if err := recover(); err != nil {
|
||
log.Warn("检查更新失败！")
|
||
}
|
||
}()
|
||
sp := regexp.MustCompile(`v(\d+)\.(\d+)\.(\d+)-?(.+)?`)
|
||
cur := sp.FindStringSubmatch(current)
|
||
re := sp.FindStringSubmatch(remote)
|
||
for i := 1; i <= 3; i++ {
|
||
curSub, _ := strconv.Atoi(cur[i])
|
||
reSub, _ := strconv.Atoi(re[i])
|
||
if curSub != reSub {
|
||
return curSub < reSub
|
||
}
|
||
}
|
||
if cur[4] == "" || re[4] == "" {
|
||
return re[4] == "" && cur[4] != re[4]
|
||
}
|
||
return cur[4] < re[4]
|
||
}
|
||
|
||
var (
|
||
// once lazy compile the reg
|
||
once sync.Once
|
||
// reg is splitURL regex pattern.
|
||
reg *regexp.Regexp
|
||
)
|
||
|
||
// SplitURL 将给定URL字符串分割为两部分，用于URL预处理防止风控
|
||
func SplitURL(s string) []string {
|
||
once.Do(func() { // lazy init.
|
||
reg = regexp.MustCompile(`(?i)[a-z\d][-a-z\d]{0,62}(\.[a-z\d][-a-z\d]{0,62})+\.?`)
|
||
})
|
||
idx := reg.FindAllStringIndex(s, -1)
|
||
if len(idx) == 0 {
|
||
return []string{s}
|
||
}
|
||
var result []string
|
||
last := 0
|
||
for i := 0; i < len(idx); i++ {
|
||
if len(idx[i]) != 2 {
|
||
continue
|
||
}
|
||
m := int(math.Abs(float64(idx[i][0]-idx[i][1]))/1.5) + idx[i][0]
|
||
result = append(result, s[last:m])
|
||
last = m
|
||
}
|
||
result = append(result, s[last:])
|
||
return result
|
||
}
|