Added restart command

EstiNet · May 11, 2018 · 041368c · 041368c
1 parent 9faf69b
commit 041368c
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 9 deletions.
diff --git a/cli b/cli
diff --git a/src/cli/main/commands.go b/src/cli/main/commands.go
@@ -13,15 +13,16 @@ import (
 
 func CommandHelp(input string) {
 	println("-----Help-----")
-	println("-h               | Get the help interface for flags.")
-	println("version          | Get the version of this instance.")
-	println("status           | Get the status of the instance")
-	println("instancestop     | Stop this instance of EstiConsole.")
-	println("list             | List all of the client servers.")
-	println("attach [process] | Switch view to another process.")
-	println("stop [process]   | Stop the process using the default stop command.")
-	println("start [process]  | Start the process.")
-	println("kill [process]   | Forcibly kill the process.")
+	println("-h                | Get the help interface for flags.")
+	println("version           | Get the version of this instance.")
+	println("status            | Get the status of the instance")
+	println("instancestop      | Stop this instance of EstiConsole.")
+	println("list              | List all of the client servers.")
+	println("attach [process]  | Switch view to another process.")
+	println("stop [process]    | Stop the process using the default stop command.")
+	println("start [process]   | Start the process.")
+	println("restart [process] | Restart the process.")
+	println("kill [process]    | Forcibly kill the process.")
 }
 
 func CommandVersion(input string) {
@@ -63,6 +64,16 @@ func CommandStart(input string) {
 	println(reply.Str)
 }
 
+func CommandRestart(input string) {
+	startCon()
+	reply, err := client.Stop(context.Background(), &pb.StringRequest{Str: input, AuthToken: token})
+	checkError(err)
+	println(reply.Str)
+	reply, err = client.Start(context.Background(), &pb.StringRequest{Str: input, AuthToken: token})
+	checkError(err)
+	println(reply.Str)
+}
+
 func CommandKill(input string) {
 	startCon()
 	reply, err := client.Kill(context.Background(), &pb.StringRequest{Str: input, AuthToken: token})

diff --git a/src/cli/main/esticli.go b/src/cli/main/esticli.go
@@ -47,6 +47,7 @@ func init() {
 	commands["instancestop"] = CommandInstanceStop
 	commands["stop"] = CommandStop
 	commands["start"] = CommandStart
+	commands["restart"] = CommandRestart
 	commands["kill"] = CommandKill
 	commands["attach"] = CommandAttach
 }