From 806b499e72451373facbda5255d0a42ada5ec088 Mon Sep 17 00:00:00 2001
From: farhansolodev <farhanalvi005@gmail.com>
Date: Mon, 3 Apr 2023 22:48:57 +0400
Subject: [PATCH] more explicit --rm instructions + erroring out if no --id
 provided

---
 cmd/action.go | 14 +++++++++-----
 cmd/flags.go  |  2 +-
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/cmd/action.go b/cmd/action.go
index 5564249..d338a19 100644
--- a/cmd/action.go
+++ b/cmd/action.go
@@ -92,10 +92,14 @@ var CliAction = func(ctx *cli.Context) error {
     }
 
     // Function called to stop and remove server from Docker
-    if RemoveVM != "" && ID != "" {
-        err := client.RemoveContianer(RemoveVM, ID)
-        if err != nil {
-            fmt.Print(err)
+    if RemoveVM != "" {
+        if ID == "" {
+            fmt.Println("provide container ID via --ID or --id")
+        } else {
+            err := client.RemoveContianer(RemoveVM, ID)
+            if err != nil {
+                fmt.Print(err)
+            }
         }
     }
 
@@ -178,7 +182,7 @@ var CliAction = func(ctx *cli.Context) error {
                 fmt.Println("Success")
             }
         } else {
-            fmt.Println("provide container ID")
+            fmt.Println("provide container ID via --ID or --id")
         }
 
     }
diff --git a/cmd/flags.go b/cmd/flags.go
index 35c8c5a..c2a0a3f 100644
--- a/cmd/flags.go
+++ b/cmd/flags.go
@@ -92,7 +92,7 @@ var AppConfigFlags = []cli.Flag{
 	&cli.StringFlag{
 		Name:        "RemoveVM",
 		Aliases:     []string{"rm"},
-		Usage:       "Stop and Remove Docker container",
+		Usage:       "Stop and Remove Docker container (IP:port) accompanied by container ID via --ID or --id",
 		EnvVars:     []string{"REMOVE_VM"},
 		Destination: &RemoveVM,
 	},