mirror of
https://github.com/simon-ding/polaris.git
synced 2026-02-26 21:50:48 +08:00
87 lines
2.1 KiB
Go
87 lines
2.1 KiB
Go
package utils
|
|
|
|
import (
|
|
"errors"
|
|
"regexp"
|
|
"strconv"
|
|
"strings"
|
|
"unicode"
|
|
|
|
"github.com/adrg/strutil"
|
|
"github.com/adrg/strutil/metrics"
|
|
"golang.org/x/crypto/bcrypt"
|
|
"golang.org/x/exp/rand"
|
|
)
|
|
|
|
func isASCII(s string) bool {
|
|
for _, c := range s {
|
|
if c > unicode.MaxASCII {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
// HashPassword generates a bcrypt hash for the given password.
|
|
func HashPassword(password string) (string, error) {
|
|
bytes, err := bcrypt.GenerateFromPassword([]byte(password), 14)
|
|
return string(bytes), err
|
|
}
|
|
|
|
// VerifyPassword verifies if the given password matches the stored hash.
|
|
func VerifyPassword(password, hash string) bool {
|
|
err := bcrypt.CompareHashAndPassword([]byte(hash), []byte(password))
|
|
return err == nil
|
|
}
|
|
|
|
func IsChineseChar(str string) bool {
|
|
|
|
for _, r := range str {
|
|
if unicode.Is(unicode.Han, r) || (regexp.MustCompile("[\u3002\uff1b\uff0c\uff1a\u201c\u201d\uff08\uff09\u3001\uff1f\u300a\u300b]").MatchString(string(r))) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
var letterRunes = []rune("0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
|
|
|
|
func RandString(n int) string {
|
|
b := make([]rune, n)
|
|
for i := range b {
|
|
b[i] = letterRunes[rand.Intn(len(letterRunes))]
|
|
}
|
|
return string(b)
|
|
}
|
|
|
|
func IsNameAcceptable(name1, name2 string) bool {
|
|
|
|
re := regexp.MustCompile(`[^\p{L}\w\s]`)
|
|
name1 = re.ReplaceAllString(strings.ToLower(name1), "")
|
|
name2 = re.ReplaceAllString(strings.ToLower(name2), "")
|
|
return strutil.Similarity(name1, name2, metrics.NewHamming()) > 0.1
|
|
}
|
|
|
|
func FindSeasonEpisodeNum(name string) (se int, ep int, err error) {
|
|
seRe := regexp.MustCompile(`S\d+`)
|
|
epRe := regexp.MustCompile(`E\d+`)
|
|
nameUpper := strings.ToUpper(name)
|
|
matchEp := epRe.FindAllString(nameUpper, -1)
|
|
if len(matchEp) == 0 {
|
|
err = errors.New("no episode num")
|
|
}
|
|
matchSe := seRe.FindAllString(nameUpper, -1)
|
|
if len(matchSe) == 0 {
|
|
err = errors.New("no season num")
|
|
}
|
|
if err != nil {
|
|
return 0, 0, err
|
|
}
|
|
|
|
epNum := strings.TrimPrefix(matchEp[0], "E")
|
|
epNum1, _ := strconv.Atoi(epNum)
|
|
seNum := strings.TrimPrefix(matchSe[0], "S")
|
|
seNum1, _ := strconv.Atoi(seNum)
|
|
return seNum1, epNum1, nil
|
|
}
|