cdle
107 lines
2.2 KiB
Go
107 lines
2.2 KiB
Go
package core
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"regexp"
|
|
"runtime"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/astaxie/beego/logs"
|
|
)
|
|
|
|
var BeforeStop = []func(){}
|
|
|
|
var pidf = "/var/run/sillyGirl.pid"
|
|
|
|
func Daemon() {
|
|
for _, bs := range BeforeStop {
|
|
bs()
|
|
}
|
|
args := os.Args[1:]
|
|
execArgs := make([]string, 0)
|
|
l := len(args)
|
|
for i := 0; i < l; i++ {
|
|
if strings.Index(args[i], "-d") == 0 {
|
|
continue
|
|
}
|
|
execArgs = append(execArgs, args[i])
|
|
}
|
|
proc := exec.Command(os.Args[0], execArgs...)
|
|
err := proc.Start()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
logs.Info(sillyGirl.Get("name", "傻妞") + "以静默形式运行")
|
|
os.WriteFile(pidf, []byte(fmt.Sprintf("%d", proc.Process.Pid)), 0o644)
|
|
os.Exit(0)
|
|
}
|
|
|
|
func GitPull(filename string) (bool, error) {
|
|
if runtime.GOOS == "darwin" {
|
|
return false, errors.New("骂你一句沙雕。")
|
|
}
|
|
rtn, err := exec.Command("sh", "-c", "cd "+ExecPath+filename+" && git stash && git pull").Output()
|
|
if err != nil {
|
|
return false, errors.New("拉取代失败：" + err.Error() + "。")
|
|
}
|
|
t := string(rtn)
|
|
if !strings.Contains(t, "changed") {
|
|
if strings.Contains(t, "Already") || strings.Contains(t, "已经是最新") {
|
|
return false, nil
|
|
} else {
|
|
return false, errors.New("拉取代失败：" + t + "。")
|
|
}
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
func CompileCode() error {
|
|
|
|
cmd := exec.Command("sh", "-c", "cd "+ExecPath+" && go build -o "+pname)
|
|
NotifyMasters(os.Args[0])
|
|
_, err := cmd.Output()
|
|
if err != nil {
|
|
return errors.New("编译失败：" + err.Error() + "。")
|
|
}
|
|
sillyGirl.Set("compiled_at", time.Now().Format("2006-01-02 15:04:05"))
|
|
return nil
|
|
}
|
|
|
|
func killp() {
|
|
// data, _ := os.ReadFile(pidf)
|
|
// pid := Int(string(data))
|
|
// if pid > 0 {
|
|
// syscall.Kill(-pid, syscall.SIGKILL)
|
|
// }
|
|
pids, err := ppid()
|
|
if err == nil {
|
|
if len(pids) == 0 {
|
|
return
|
|
} else {
|
|
exec.Command("sh", "-c", "kill -9 "+strings.Join(pids, " ")).Output()
|
|
}
|
|
} else {
|
|
return
|
|
}
|
|
}
|
|
|
|
func ppid() ([]string, error) {
|
|
pid := fmt.Sprint(os.Getpid())
|
|
pids := []string{}
|
|
rtn, err := exec.Command("sh", "-c", "pidof "+pname).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
|
|
}
|