增加service命令

internal/nodes/admin_node.go

@@ -5,12 +5,16 @@ import (
 	teaconst "github.com/TeaOSLab/EdgeAdmin/internal/const"
 	"github.com/TeaOSLab/EdgeAdmin/internal/errors"
 	"github.com/TeaOSLab/EdgeAdmin/internal/events"
+	"github.com/TeaOSLab/EdgeAdmin/internal/utils"
 	"github.com/iwind/TeaGo"
 	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/lists"
 	"github.com/iwind/TeaGo/logs"
 	"github.com/iwind/TeaGo/rands"
 	"github.com/iwind/TeaGo/sessions"
 	"io/ioutil"
+	"log"
+	"net"
 	"os"
 	"os/exec"
 	"os/signal"
@@ -31,8 +35,15 @@ func (this *AdminNode) Run() {
 	secret := this.genSecret()
 	configs.Secret = secret
 
+	// 本地Sock
+	err := this.listenSock()
+	if err != nil {
+		logs.Println("NODE" + err.Error())
+		return
+	}
+
 	// 检查server配置
-	err := this.checkServer()
+	err = this.checkServer()
 	if err != nil {
 		if err != nil {
 			logs.Println("[NODE]" + err.Error())
@@ -72,6 +83,68 @@ func (this *AdminNode) Run() {
 		Start()
 }
 
+// 实现守护进程
+func (this *AdminNode) Daemon() {
+	path := os.TempDir() + "/edge-admin.sock"
+	isDebug := lists.ContainsString(os.Args, "debug")
+	isDebug = true
+	for {
+		conn, err := net.DialTimeout("unix", path, 1*time.Second)
+		if err != nil {
+			if isDebug {
+				log.Println("[DAEMON]starting ...")
+			}
+
+			// 尝试启动
+			err = func() error {
+				exe, err := os.Executable()
+				if err != nil {
+					return err
+				}
+				cmd := exec.Command(exe)
+				err = cmd.Start()
+				if err != nil {
+					return err
+				}
+				err = cmd.Wait()
+				if err != nil {
+					return err
+				}
+				return nil
+			}()
+
+			if err != nil {
+				if isDebug {
+					log.Println("[DAEMON]", err)
+				}
+				time.Sleep(1 * time.Second)
+			} else {
+				time.Sleep(5 * time.Second)
+			}
+		} else {
+			_ = conn.Close()
+			time.Sleep(5 * time.Second)
+		}
+	}
+}
+
+// 安装系统服务
+func (this *AdminNode) InstallSystemService() error {
+	shortName := teaconst.SystemdServiceName
+
+	exe, err := os.Executable()
+	if err != nil {
+		return err
+	}
+
+	manager := utils.NewServiceManager(shortName, teaconst.ProductName)
+	err = manager.Install(exe, []string{})
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
 // 检查Server配置
 func (this *AdminNode) checkServer() error {
 	configFile := Tea.ConfigFile("server.yaml")
@@ -143,3 +216,40 @@ func (this *AdminNode) genSecret() string {
 	_ = ioutil.WriteFile(tmpFile, []byte(secret), 0666)
 	return secret
 }
+
+// 监听本地sock
+func (this *AdminNode) listenSock() error {
+	path := os.TempDir() + "/edge-admin.sock"
+
+	// 检查是否已经存在
+	_, err := os.Stat(path)
+	if err == nil {
+		conn, err := net.Dial("unix", path)
+		if err != nil {
+			_ = os.Remove(path)
+		} else {
+			_ = conn.Close()
+		}
+	}
+
+	// 新的监听任务
+	listener, err := net.Listen("unix", path)
+	if err != nil {
+		return err
+	}
+	events.On(events.EventQuit, func() {
+		logs.Println("NODE", "quit unix sock")
+		_ = listener.Close()
+	})
+
+	go func() {
+		for {
+			_, err := listener.Accept()
+			if err != nil {
+				return
+			}
+		}
+	}()
+
+	return nil
+}