1.4.5.2

EdgeNode/internal/nodes/system_services.go

@@ -0,0 +1,151 @@
+package nodes
+
+import (
+	"bytes"
+	"encoding/json"
+	"errors"
+	"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
+	teaconst "github.com/TeaOSLab/EdgeNode/internal/const"
+	"github.com/TeaOSLab/EdgeNode/internal/events"
+	"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
+	"github.com/TeaOSLab/EdgeNode/internal/utils"
+	executils "github.com/TeaOSLab/EdgeNode/internal/utils/exec"
+	"github.com/TeaOSLab/EdgeNode/internal/utils/goman"
+	"github.com/iwind/TeaGo/maps"
+	"os"
+	"runtime"
+	"time"
+)
+
+func init() {
+	if !teaconst.IsMain {
+		return
+	}
+
+	var manager = NewSystemServiceManager()
+	events.On(events.EventReload, func() {
+		goman.New(func() {
+			err := manager.Setup()
+			if err != nil {
+				remotelogs.Error("SYSTEM_SERVICE", "setup system services failed: "+err.Error())
+			}
+		})
+	})
+}
+
+// SystemServiceManager 系统服务管理
+type SystemServiceManager struct {
+	lastIsOn int // -1, 0, 1
+}
+
+func NewSystemServiceManager() *SystemServiceManager {
+	return &SystemServiceManager{}
+}
+
+func (this *SystemServiceManager) Setup() error {
+	if sharedNodeConfig == nil || !sharedNodeConfig.IsOn {
+		return nil
+	}
+
+	if len(sharedNodeConfig.SystemServices) == 0 {
+		return nil
+	}
+
+	systemdParams, ok := sharedNodeConfig.SystemServices[nodeconfigs.SystemServiceTypeSystemd]
+	if ok {
+		err := this.setupSystemd(systemdParams)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func (this *SystemServiceManager) setupSystemd(params maps.Map) error {
+	// 只有在Linux下运行
+	if runtime.GOOS != "linux" {
+		return nil
+	}
+
+	if params == nil {
+		params = maps.Map{}
+	}
+	data, err := json.Marshal(params)
+	if err != nil {
+		return err
+	}
+
+	var config = &nodeconfigs.SystemdServiceConfig{}
+	err = json.Unmarshal(data, config)
+	if err != nil {
+		return err
+	}
+
+	// 检查当前的service
+	systemctl, err := executils.LookPath("systemctl")
+	if err != nil {
+		return err
+	}
+	if len(systemctl) == 0 {
+		return errors.New("can not find 'systemctl' on the system")
+	}
+
+	// 记录上次状态
+	var isOnInt int
+	if config.IsOn {
+		isOnInt = 1
+	} else {
+		isOnInt = 0
+	}
+
+	if this.lastIsOn == isOnInt {
+		return nil
+	}
+	defer func() {
+		this.lastIsOn = isOnInt
+	}()
+
+	var shortName = teaconst.SystemdServiceName
+	var cmd = executils.NewTimeoutCmd(10*time.Second, systemctl, "is-enabled", shortName)
+	cmd.WithStdout()
+	err = cmd.Run()
+	var hasInstalled = err == nil
+	if config.IsOn {
+		exe, err := os.Executable()
+		if err != nil {
+			return err
+		}
+
+		// 检查文件路径是否变化
+		if hasInstalled && cmd.Stdout() == "enabled" {
+			data, err := os.ReadFile("/etc/systemd/system/" + teaconst.SystemdServiceName + ".service")
+			if err == nil && bytes.Index(data, []byte(exe)) > 0 {
+				return nil
+			}
+		}
+
+		// 安装服务
+		var manager = utils.NewServiceManager(shortName, teaconst.ProductName)
+		err = manager.Install(exe, []string{})
+		if err != nil {
+			return err
+		}
+
+		// 启动服务
+		goman.New(func() {
+			time.Sleep(5 * time.Second)
+			_ = executils.NewTimeoutCmd(30*time.Second, systemctl, "start", teaconst.SystemdServiceName).Start()
+		})
+	} else {
+		if hasInstalled {
+			var manager = utils.NewServiceManager(shortName, teaconst.ProductName)
+			err = manager.Uninstall()
+			if err != nil {
+				return err
+			}
+		}
+	}
+
+	return nil
+}