增加service命令

cmd/edge-node/main.go

@@ -7,22 +7,17 @@ import (
 	"github.com/TeaOSLab/EdgeNode/internal/nodes"
 	"github.com/iwind/TeaGo/Tea"
 	_ "github.com/iwind/TeaGo/bootstrap"
-	"github.com/iwind/TeaGo/lists"
 	"github.com/iwind/TeaGo/types"
 	"io/ioutil"
-	"log"
-	"net"
 	"os"
-	"os/exec"
 	"syscall"
-	"time"
 )
 
 func main() {
 	app := apps.NewAppCmd().
 		Version(teaconst.Version).
 		Product(teaconst.ProductName).
-		Usage(teaconst.ProcessName + " [-v|start|stop|restart|quit|test|daemon]")
+		Usage(teaconst.ProcessName + " [-v|start|stop|restart|quit|test|service|daemon]")
 
 	app.On("test", func() {
 		err := nodes.NewNode().Test()
@@ -31,47 +26,14 @@ func main() {
 		}
 	})
 	app.On("daemon", func() {
-		path := os.TempDir() + "/edge-node.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)
-			}
+		nodes.NewNode().Daemon()
+	})
+	app.On("service", func() {
+		err := nodes.NewNode().InstallSystemService()
+		if err != nil {
+			fmt.Println("[ERROR]install failed: " + err.Error())
 		}
+		fmt.Println("done")
 	})
 	app.On("quit", func() {
 		pidFile := Tea.Root + "/bin/pid"