-
Notifications
You must be signed in to change notification settings - Fork 459
/
utils.go
82 lines (73 loc) · 1.7 KB
/
utils.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package utils
import (
"crypto/md5"
"encoding/hex"
"errors"
"math/rand"
"net/url"
"os/exec"
"regexp"
"strings"
)
func GetFFmpegPath() (string, error) {
path, err := exec.LookPath("ffmpeg")
if errors.Is(err, exec.ErrDot) {
// put ffmpeg.exe and binary like bililive-windows-amd64.exe to the same folder is allowed
path, err = exec.LookPath("./ffmpeg")
}
return path, err
}
func IsFFmpegExist() bool {
_, err := GetFFmpegPath()
return err == nil
}
func GetMd5String(b []byte) string {
md5Obj := md5.New()
md5Obj.Write(b)
return hex.EncodeToString(md5Obj.Sum(nil))
}
var (
lowercaseRunes = []rune("abcdefghijklmnopqrstuvwxyz")
uppercaseRunes = []rune("ABCDEFGHIJKLMNOPQRSTUVWXYZ")
lettersRunes = append(lowercaseRunes, uppercaseRunes...)
digitsRunes = []rune("0123456789")
allRunes = append(lettersRunes, digitsRunes...)
)
func GenRandomName(n int) string {
b := make([]rune, n)
b[0] = lowercaseRunes[rand.Intn(len(lowercaseRunes))]
for i := 1; i < n; i++ {
b[i] = allRunes[rand.Intn(len(allRunes))]
}
return string(b)
}
func GenRandomString(length int, validChars string) string {
b := make([]string, length)
chars := strings.Split(validChars, "")
for i := range b {
b[i] = chars[rand.Intn(len(chars))]
}
return strings.Join(b, "")
}
func Match1(re, str string) string {
reg, err := regexp.Compile(re)
if err != nil {
return ""
}
match := reg.FindStringSubmatch(str)
if match == nil || len(match) < 2 {
return ""
}
return match[1]
}
func GenUrls(strs ...string) ([]*url.URL, error) {
urls := make([]*url.URL, 0, len(strs))
for _, str := range strs {
u, err := url.Parse(str)
if err != nil {
return nil, err
}
urls = append(urls, u)
}
return urls, nil
}