FEAT : exit

unshade · Apr 13, 2024 · 841d109 · 841d109
1 parent 3b74665
commit 841d109
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/internal/controller.go b/internal/controller.go
@@ -54,6 +54,11 @@ func handleConnection(conn net.Conn, rootCmd *cobra.Command) {
 		args := strings.Split(string(buffer), " ")
 		commandName := args[0]
 
+		if commandName == "exit" {
+			conn.Close()
+			return
+		}
+
 		cmdList := rootCmd.Commands()
 
 		for _, cmd := range cmdList {