added command delete VM

server/docker/docker.go

@@ -231,8 +231,33 @@ func (d *DockerVM)runContainer(dockerClient *client.Client) error{
 	return nil
 }
 
-// TODO: Implement and remove docker instance running
+// StopAndRemoveContainer TODO: Implement and remove docker instance running
+// Stop and remove a container
+// Reference (https://gist.github.com/frikky/e2efcea6c733ea8d8d015b7fe8a91bf6)
+func StopAndRemoveContainer(containername string) error {
+	ctx := context.Background()
 
+	// Gets docker information from env variables
+	client, err := client.NewClientWithOpts(client.FromEnv, client.WithAPIVersionNegotiation())
+	if err != nil {
+		return err
+	}
+
+	if err = client.ContainerStop(ctx, containername, nil); err != nil {
+		return err
+	}
+
+	removeOptions := types.ContainerRemoveOptions{
+		RemoveVolumes: true,
+		Force:         true,
+	}
+
+	if err = client.ContainerRemove(ctx, containername, removeOptions); err != nil {
+		return err
+	}
+
+	return nil
+}
 
 func print(rd io.Reader) error {
 	var lastLine string

server/server.go

@@ -94,13 +94,21 @@ func Server() error{
 		resp, err := docker.BuildRunContainer(PortsInt,GPU)
 
 		if err != nil {
-
 			c.String(http.StatusInternalServerError, fmt.Sprintf("error: %s", err))
 		}
 
 		c.JSON(http.StatusOK, resp)
 	})
 
+	//Remove container
+	r.GET("/RemoveContainer", func(c *gin.Context) {
+		ID := c.DefaultQuery("id","0")
+		if err := docker.StopAndRemoveContainer(ID); err != nil {
+			c.String(http.StatusInternalServerError, fmt.Sprintf("error: %s", err))
+		}
+		c.String(http.StatusOK, "success")
+	})
+
 	// Future feature
 	/*r.GET("/create_vm/:virtualization", func(c *gin.Context) {
 		virtualization := c.Param("virtualization")