added command delete VM

cmd/action.go

@@ -26,8 +26,16 @@ var CliAction = func(ctx *cli.Context) error {
 		p2p.PrintIpTable()
 	}
 
+	// Function called to stop and remove server from Docker
+	if RemoveVM != ""  && ID != "" {
+		err := client.RemoveContianer(RemoveVM,ID)
+		if err != nil {
+			fmt.Print(err)
+		}
+	}
+
 	//Call function to create Docker container
-	if IpAddress != "" {
+	if CreateVM != "" {
 
 		var PortsInt int
 		PortsInt = 0
@@ -37,7 +45,7 @@ var CliAction = func(ctx *cli.Context) error {
 			fmt.Sscanf(Ports, "%d", &PortsInt)
 		}
 
-		imageRes, err := client.StartContainer(IpAddress,PortsInt,GPU)
+		imageRes, err := client.StartContainer(CreateVM,PortsInt,GPU)
 
 		if err != nil {
 			fmt.Print(err)

cmd/flags.go

@@ -6,10 +6,12 @@ import (
 
 // Variables declared for CLI
 var (
-	IpAddress string
-	Ports     string
-	Mode      string
-	Specs      string
+	CreateVM    string
+	Ports       string
+	Mode        string
+	RemoveVM    string
+	ID          string
+	Specs       string
 	GPU         bool
 	ListServers bool
 )
@@ -33,13 +35,25 @@ var AppConfigFlags = []cli.Flag{
 		Name:        "CreateVM",
 		Usage:       "Creates Docker container on the selected server",
 		EnvVars: []string{"CREATE_VM"},
-		Destination: &IpAddress,
+		Destination: &CreateVM,
+	},
+	&cli.StringFlag{
+		Name:        "RemoveVM",
+		Usage:       "Stop and Remove Docker container",
+		EnvVars: []string{"REMOVE_VM"},
+		Destination: &RemoveVM,
+	},
+	&cli.StringFlag{
+		Name:        "ID",
+		Usage:       "Docker Container ID",
+		EnvVars: []string{"ID"},
+		Destination: &ID,
 	},
 	&cli.StringFlag{
 		Name:        "Ports",
 		Usage:       "Number of ports to open for the Docker Container",
 		EnvVars: []string{"NUM_PORTS"},
-		Destination: &Ports,
+		Destination: &ID,
 	},
 	&cli.BoolFlag{
 		Name:        "GPU",