增加恢复模式

cmd/edge-admin/main.go

@@ -8,20 +8,22 @@ import (
 	"github.com/TeaOSLab/EdgeAdmin/internal/nodes"
 	_ "github.com/TeaOSLab/EdgeAdmin/internal/web"
 	_ "github.com/iwind/TeaGo/bootstrap"
+	"github.com/iwind/gosock/pkg/gosock"
 )
 
 func main() {
 	app := apps.NewAppCmd().
 		Version(teaconst.Version).
 		Product(teaconst.ProductName).
-		Usage(teaconst.ProcessName+" [-v|start|stop|restart|service|daemon|reset]").
+		Usage(teaconst.ProcessName+" [-v|start|stop|restart|service|daemon|reset|recover]").
 		Option("-h", "show this help").
 		Option("-v", "show version").
 		Option("start", "start the service").
 		Option("stop", "stop the service").
 		Option("service", "register service into systemd").
 		Option("daemon", "start the service with daemon").
-		Option("reset", "reset configs")
+		Option("reset", "reset configs").
+		Option("recover", "enter recovery mode")
 
 	app.On("daemon", func() {
 		nodes.NewAdminNode().Daemon()
@@ -42,6 +44,19 @@ func main() {
 		}
 		fmt.Println("done")
 	})
+	app.On("recover", func() {
+		sock := gosock.NewTmpSock(teaconst.ProcessName)
+		if !sock.IsListening() {
+			fmt.Println("[ERROR]the service not started yet, you should start the service first")
+			return
+		}
+		_, err := sock.Send(&gosock.Command{Code: "recover"})
+		if err != nil {
+			fmt.Println("[ERROR]enter recovery mode failed: " + err.Error())
+			return
+		}
+		fmt.Println("enter recovery mode successfully")
+	})
 	app.Run(func() {
 		adminNode := nodes.NewAdminNode()
 		adminNode.Run()