feat: 添加自构建+重启功能

handler/restart/restart.go

@@ -0,0 +1,147 @@
+package restart
+
+import (
+	"errors"
+	"log"
+	"os"
+	"os/exec"
+	"time"
+
+	"git.lxtend.com/qqbot/handler"
+	"git.lxtend.com/qqbot/model"
+)
+
+var hasVaildBuild = true
+
+func init() {
+	handler.RegisterHandler("/重启bot", restart)
+	handler.RegisterHandler("/构建bot", build)
+	handler.RegisterHandler("/构建重启", buildAndRestart)
+	handler.RegisterHandler("/gitpull", pullCode)
+}
+
+func restart(msg model.Message) model.Reply {
+	if !hasVaildBuild {
+		return model.Reply{
+			ReplyMsg:       "上次构建失败，请先成功构建再部署",
+			ReferOriginMsg: true,
+			FromMsg:        msg,
+		}
+	}
+	go restartProgram()
+	return model.Reply{
+		ReplyMsg:       "重启中...",
+		ReferOriginMsg: true,
+		FromMsg:        msg,
+	}
+}
+
+func build(msg model.Message) model.Reply {
+	err := buildBot()
+	if err != nil {
+		return model.Reply{
+			ReplyMsg:       "构建失败,报错如下\n" + err.Error(),
+			ReferOriginMsg: true,
+			FromMsg:        msg,
+		}
+	}
+	return model.Reply{
+		ReplyMsg:       "构建成功",
+		ReferOriginMsg: true,
+		FromMsg:        msg,
+	}
+}
+
+func buildAndRestart(msg model.Message) model.Reply {
+	err := buildBot()
+	if err != nil {
+		return model.Reply{
+			ReplyMsg:       "构建失败,报错如下\n" + err.Error(),
+			ReferOriginMsg: true,
+			FromMsg:        msg,
+		}
+	}
+	if !hasVaildBuild {
+		return model.Reply{
+			ReplyMsg:       "构建失败，请先成功构建再部署",
+			ReferOriginMsg: true,
+			FromMsg:        msg,
+		}
+	}
+	go restartProgram()
+	return model.Reply{
+		ReplyMsg:       "构建成功,重启中...",
+		ReferOriginMsg: true,
+		FromMsg:        msg,
+	}
+}
+
+func pullCode(msg model.Message) model.Reply {
+	err := gitPull()
+	if err != nil {
+		return model.Reply{
+			ReplyMsg:       "拉取代码失败,报错如下\n" + err.Error(),
+			ReferOriginMsg: true,
+			FromMsg:        msg,
+		}
+	}
+	return model.Reply{
+		ReplyMsg:       "拉取代码成功",
+		ReferOriginMsg: true,
+		FromMsg:        msg,
+	}
+}
+
+func restartProgram() error {
+	log.Println("重启程序...")
+	time.Sleep(500 * time.Millisecond)
+	path, err := os.Executable()
+	if err != nil {
+		return err
+	}
+	workDir, err := os.Getwd()
+	if err != nil {
+		return err
+	}
+	cmd := exec.Command(path)
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	cmd.Dir = workDir
+	if err := cmd.Start(); err != nil {
+		return err
+	}
+	os.Exit(0)
+	return nil
+}
+
+func buildBot() error {
+	hasVaildBuild = false
+	workDir, err := os.Getwd()
+	if err != nil {
+		return err
+	}
+	cmd := exec.Command("go", "build", "-o", "qqbot")
+	cmd.Dir = workDir
+
+	output, err := cmd.CombinedOutput()
+	if err != nil {
+		return errors.New(string(output) + err.Error())
+	}
+	hasVaildBuild = true
+	return nil
+}
+
+func gitPull() error {
+	workDir, err := os.Getwd()
+	if err != nil {
+		return err
+	}
+	cmd := exec.Command("git", "pull")
+	cmd.Dir = workDir
+
+	output, err := cmd.CombinedOutput()
+	if err != nil {
+		return errors.New(string(output) + err.Error())
+	}
+	return nil
+}