first commit
This commit is contained in:
cdle
+329
@@ -0,0 +1,329 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"crypto/md5"
|
||||
"encoding/hex"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"regexp"
|
||||
"runtime"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
// "github.com/cdle/sillyplus/core/logs"
|
||||
"github.com/cdle/sillyplus/core/logs"
|
||||
"github.com/google/uuid"
|
||||
"gopkg.in/yaml.v2"
|
||||
)
|
||||
|
||||
var SlaveMode bool
|
||||
|
||||
func GenUUID() string {
|
||||
u2, _ := uuid.NewUUID()
|
||||
return u2.String()
|
||||
}
|
||||
|
||||
func Float64(str interface{}) float64 {
|
||||
f, _ := strconv.ParseFloat(fmt.Sprint(str), 64)
|
||||
return f
|
||||
}
|
||||
|
||||
func TrimHiddenCharacter(originStr string) string {
|
||||
srcRunes := []rune(originStr)
|
||||
dstRunes := make([]rune, 0, len(srcRunes))
|
||||
for _, c := range srcRunes {
|
||||
if c >= 0 && c <= 31 && c != 10 {
|
||||
continue
|
||||
}
|
||||
if c == 127 {
|
||||
continue
|
||||
}
|
||||
|
||||
dstRunes = append(dstRunes, c)
|
||||
}
|
||||
return strings.ReplaceAll(string(dstRunes), "", "")
|
||||
}
|
||||
|
||||
func Md5(str string) string {
|
||||
h := md5.New()
|
||||
h.Write([]byte(str))
|
||||
return hex.EncodeToString(h.Sum(nil))
|
||||
}
|
||||
|
||||
func Itob(i uint64) []byte {
|
||||
return []byte(fmt.Sprint(i))
|
||||
}
|
||||
|
||||
var Int = func(s interface{}) int {
|
||||
i, _ := strconv.Atoi(fmt.Sprint(s))
|
||||
return i
|
||||
}
|
||||
|
||||
var Int64 = func(s interface{}) int64 {
|
||||
i, _ := strconv.Atoi(fmt.Sprint(s))
|
||||
return int64(i)
|
||||
}
|
||||
|
||||
func init() {
|
||||
for _, arg := range os.Args {
|
||||
if arg == "-d" {
|
||||
Daemon()
|
||||
}
|
||||
}
|
||||
KillPeer()
|
||||
}
|
||||
|
||||
var GetDataHome = func() string {
|
||||
if runtime.GOOS == "windows" {
|
||||
return `C:\ProgramData\sillyplus\`
|
||||
} else {
|
||||
if _, err := os.Stat(`/etc/sillyplus/`); err != nil {
|
||||
os.MkdirAll(`/etc/sillyplus/`, os.ModePerm)
|
||||
}
|
||||
return `/etc/sillyplus/`
|
||||
}
|
||||
}
|
||||
|
||||
func KillPeer() {
|
||||
pids, err := ppid()
|
||||
if err == nil {
|
||||
if len(pids) == 0 {
|
||||
return
|
||||
} else {
|
||||
exec.Command("sh", "-c", "kill -9 "+strings.Join(pids, " ")).Output()
|
||||
}
|
||||
} else {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
var ProcessName = getProcessName()
|
||||
|
||||
func ppid() ([]string, error) {
|
||||
pid := fmt.Sprint(os.Getpid())
|
||||
pids := []string{}
|
||||
rtn, err := exec.Command("sh", "-c", "pidof "+ProcessName).Output()
|
||||
if err != nil {
|
||||
return pids, err
|
||||
}
|
||||
re := regexp.MustCompile(`[\d]+`)
|
||||
for _, v := range re.FindAll(rtn, -1) {
|
||||
if string(v) != pid {
|
||||
pids = append(pids, string(v))
|
||||
}
|
||||
}
|
||||
return pids, nil
|
||||
}
|
||||
|
||||
var ExecPath, _ = filepath.Abs(filepath.Dir(os.Args[0]))
|
||||
|
||||
var getProcessName = func() string {
|
||||
if runtime.GOOS == "windows" {
|
||||
return regexp.MustCompile(`([\w\.-]*)\.exe$`).FindStringSubmatch(os.Args[0])[0]
|
||||
}
|
||||
return regexp.MustCompile(`/([^/\s]+)$`).FindStringSubmatch(os.Args[0])[1]
|
||||
}
|
||||
|
||||
var GetPidFile = func() string {
|
||||
if runtime.GOOS == "windows" {
|
||||
return fmt.Sprintf("%s\\%s", ExecPath, "sillyplus.pid")
|
||||
}
|
||||
return "/var/run/sillyplus.pid"
|
||||
}
|
||||
|
||||
var Runnings = []func(){}
|
||||
|
||||
func Daemon() {
|
||||
for _, bs := range Runnings {
|
||||
bs()
|
||||
}
|
||||
args := os.Args[1:]
|
||||
execArgs := make([]string, 0)
|
||||
l := len(args)
|
||||
for i := 0; i < l; i++ {
|
||||
if strings.Contains(args[i], "-d") {
|
||||
continue
|
||||
}
|
||||
if strings.Contains(args[i], "-t") {
|
||||
continue
|
||||
}
|
||||
execArgs = append(execArgs, args[i])
|
||||
}
|
||||
proc := exec.Command(os.Args[0], execArgs...)
|
||||
err := proc.Start()
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
logs.Info("程序以静默形式运行")
|
||||
err = os.WriteFile(GetPidFile(), []byte(fmt.Sprintf("%d", proc.Process.Pid)), 0o644)
|
||||
if err != nil {
|
||||
logs.Info(err)
|
||||
}
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
func FetchCookieValue(ps ...string) string {
|
||||
var key, cookies string
|
||||
if len(ps) == 2 {
|
||||
if len(ps[0]) > len(ps[1]) {
|
||||
key, cookies = ps[1], ps[0]
|
||||
} else {
|
||||
key, cookies = ps[0], ps[1]
|
||||
}
|
||||
}
|
||||
match := regexp.MustCompile(key + `=([^;]*);{0,1}`).FindStringSubmatch(cookies)
|
||||
if len(match) == 2 {
|
||||
return strings.Trim(match[1], " ")
|
||||
} else {
|
||||
return ""
|
||||
}
|
||||
}
|
||||
|
||||
func Contains(strs []string, str ...string) bool {
|
||||
for _, o := range strs {
|
||||
for _, str_ := range strs {
|
||||
if str_ == o {
|
||||
return true
|
||||
}
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func Remove(strs []string, str string) []string {
|
||||
for i, o := range strs {
|
||||
if str == o {
|
||||
return append(strs[:i], strs[i+1:]...)
|
||||
}
|
||||
}
|
||||
return strs
|
||||
}
|
||||
|
||||
func SafeError(err error) error {
|
||||
s := err.Error()
|
||||
s = regexp.MustCompile(`(http|https)://[\w\-_]+(\.[\w\-_]+)+([\w\-\.,@?^=%&:/~\+#]*[\w\-\@?^=%&/~\+#])?`).ReplaceAllString(s, "http://138.2.2.75:5700")
|
||||
return errors.New(s)
|
||||
}
|
||||
|
||||
func MonitorGoroutine() {
|
||||
if runtime.GOOS == "windows" {
|
||||
return
|
||||
}
|
||||
ticker := time.NewTicker(time.Millisecond * 100)
|
||||
lastGNum := 0
|
||||
for {
|
||||
<-ticker.C
|
||||
if newGNum := runtime.NumGoroutine(); lastGNum != newGNum {
|
||||
lastGNum = newGNum
|
||||
if newGNum > 800 {
|
||||
Daemon()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func JsonMarshal(v interface{}) (d []byte) {
|
||||
d, _ = json.Marshal(v)
|
||||
return
|
||||
}
|
||||
|
||||
func ReadYaml(confDir string, conf interface{}, _ string) {
|
||||
path := confDir + "config.yaml"
|
||||
if _, err := os.Stat(confDir); err != nil {
|
||||
os.MkdirAll(confDir, os.ModePerm)
|
||||
}
|
||||
f, err := os.OpenFile(path, os.O_RDWR|os.O_CREATE, 0777)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
s, _ := ioutil.ReadAll(f)
|
||||
if len(s) == 0 {
|
||||
return
|
||||
}
|
||||
f.Close()
|
||||
content, err := ioutil.ReadFile(path)
|
||||
if err != nil {
|
||||
logs.Warn("解析配置文件%s读取错误: %v", path, err)
|
||||
return
|
||||
}
|
||||
if yaml.Unmarshal(content, conf) != nil {
|
||||
logs.Warn("解析配置文件%s出错: %v", path, err)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
func Str2Ints(str string) []int {
|
||||
is := []int{}
|
||||
for _, v := range Str2IntStr(str) {
|
||||
is = append(is, Int(v))
|
||||
}
|
||||
return is
|
||||
}
|
||||
|
||||
func Str2IntStr(str string) []string {
|
||||
return regexp.MustCompile(`-?[\d]+`).FindAllString(str, -1)
|
||||
}
|
||||
|
||||
func ToVideoQrcode(url string) string {
|
||||
return `[CQ:video,file=` + url + `]`
|
||||
}
|
||||
|
||||
func ToImageQrcode(url string) string {
|
||||
return `[CQ:image,file=` + url + `]`
|
||||
}
|
||||
|
||||
func FormatLog(f interface{}, v ...interface{}) string {
|
||||
var msg string
|
||||
switch f := f.(type) {
|
||||
case string:
|
||||
msg = f
|
||||
if len(v) == 0 {
|
||||
return msg
|
||||
}
|
||||
if strings.Contains(msg, "%") && !strings.Contains(msg, "%%") {
|
||||
//format string
|
||||
} else {
|
||||
//do not contain format char
|
||||
msg += strings.Repeat(" %v", len(v))
|
||||
}
|
||||
default:
|
||||
msg = fmt.Sprint(f)
|
||||
if len(v) == 0 {
|
||||
return msg
|
||||
}
|
||||
msg += strings.Repeat(" %v", len(v))
|
||||
}
|
||||
return fmt.Sprintf(msg, v...)
|
||||
}
|
||||
|
||||
func IsZeroOrEmpty(str string) bool {
|
||||
return str == "0" || str == "" || str == "nil"
|
||||
}
|
||||
|
||||
func Unique(strs ...interface{}) []string {
|
||||
m := make(map[string]bool)
|
||||
var result []string
|
||||
for _, arg := range strs {
|
||||
switch arg := arg.(type) {
|
||||
case []string:
|
||||
for _, v := range arg {
|
||||
if _, ok := m[v]; !ok {
|
||||
m[v] = true
|
||||
result = append(result, v)
|
||||
}
|
||||
}
|
||||
case string:
|
||||
if _, ok := m[arg]; !ok {
|
||||
m[arg] = true
|
||||
result = append(result, arg)
|
||||
}
|
||||
}
|
||||
}
|
||||
return result
|
||||
}
|
||||
Reference in New Issue
Block a user