Default to current connection for wsh conn reinstall (#1782)

wavetermdev · Jan 21, 2025 · 4fba9e6 · 4fba9e6
1 parent 5f65d85
commit 4fba9e6
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cmd/wsh/cmd/wshcmd-conn.go b/cmd/wsh/cmd/wshcmd-conn.go
@@ -30,7 +30,6 @@ var connStatusCmd = &cobra.Command{
 var connReinstallCmd = &cobra.Command{
 	Use:     "reinstall CONNECTION",
 	Short:   "reinstall wsh on a connection",
-	Args:    cobra.ExactArgs(1),
 	RunE:    connReinstallRun,
 	PreRunE: preRunSetupRpcClient,
 }
@@ -124,6 +123,12 @@ func connStatusRun(cmd *cobra.Command, args []string) error {
 }
 
 func connReinstallRun(cmd *cobra.Command, args []string) error {
+	if len(args) != 1 {
+		if RpcContext.Conn == "" {
+			return fmt.Errorf("no connection specified")
+		}
+		args = []string{RpcContext.Conn}
+	}
 	connName := args[0]
 	if err := validateConnectionName(connName); err != nil {
 		return err