feat: Add a sub-command for docker - container (#27)

* Add command to stop with all flag
* Add command to rename
* Add command to delete with all flag

cmd/docker/container/container.go

@@ -0,0 +1,23 @@
+package container
+
+import (
+	"log"
+
+	"github.com/spf13/cobra"
+)
+
+var (
+	runnnigContainer = []string{}
+)
+
+// containerCmd represents the container command
+var ContainerCmd = &cobra.Command{
+	Use:   "container [command] [flags]",
+	Short: "Container related commands",
+	Run: func(cmd *cobra.Command, args []string) {
+		err := cmd.Help()
+		if err != nil {
+			log.Fatal(err)
+		}
+	},
+}

cmd/docker/container/delete.go

@@ -0,0 +1,108 @@
+package container
+
+import (
+	"fmt"
+	"log"
+	"strings"
+
+	"github.com/docker/docker/api/types"
+	"github.com/manifoldco/promptui"
+	"github.com/spf13/cobra"
+)
+
+var (
+	deleteAll bool
+)
+
+// deleteContainerCmd represents the delete command
+var deleteContainerCmd = &cobra.Command{
+	Use:   "delete",
+	Short: "Delete a running container",
+	Run: func(cmd *cobra.Command, args []string) {
+		deleteContainer()
+	},
+}
+
+func deleteContainer() {
+	ctx, cli := dockerClient()
+	containerList := runnnigContainerList(cli, ctx)
+
+	if len(containerList) == 0 {
+		fmt.Println("No container running")
+		return
+	}
+
+	// If the flag --all is set, delete all containers
+	if deleteAll {
+
+		prompt := promptui.Select{
+			Label: "Are you sure you want to delete all containers?",
+			Items: []string{"Yes", "No"},
+		}
+
+		_, option, err := prompt.Run()
+		checkErr(err)
+
+		if option == "Yes" {
+
+			for _, container := range containerList {
+				fmt.Printf("Deleting container %s (%s)...\n", container.Names[0][1:], container.ID[:6])
+				err = cli.ContainerRemove(ctx, container.ID, types.ContainerRemoveOptions{Force: true})
+				checkErr(err)
+			}
+
+			fmt.Println("Deleted all containers")
+			return
+		} else {
+			fmt.Println("Containers not removed")
+			return
+		}
+
+	}
+
+	// If the flag --all is not set, delete a specific container
+	for _, container := range containerList {
+		runnnigContainer = append(runnnigContainer, container.Names[0][1:]+" - "+container.ID[:6])
+
+	}
+
+	prompt := promptui.Select{
+		Label: "Select a containe to delete",
+		Items: runnnigContainer,
+	}
+
+	_, conSelection, err := prompt.Run()
+	checkErr(err)
+
+	prompt = promptui.Select{
+		Label: "Are you sure you want to delete " + conSelection + " ?",
+		Items: []string{"Yes", "No"},
+	}
+
+	_, option, err := prompt.Run()
+	checkErr(err)
+
+	if option == "Yes" {
+		slpit := strings.Split(conSelection, " - ")
+
+		err = cli.ContainerRemove(ctx, slpit[1], types.ContainerRemoveOptions{Force: true})
+		checkErr(err)
+
+		fmt.Printf("Container %s deleted successfully\n", slpit[0])
+	} else {
+		fmt.Println("Container not removed")
+		return
+	}
+
+}
+
+func checkErr(err error) {
+	if err != nil {
+		log.Fatal(err)
+	}
+}
+
+func init() {
+	ContainerCmd.AddCommand(deleteContainerCmd)
+	deleteContainerCmd.Flags().BoolVarP(&deleteAll, "all", "a", false, "Delete all containers")
+}

cmd/docker/container/helper.go

@@ -0,0 +1,27 @@
+package container
+
+import (
+	"context"
+
+	"github.com/docker/docker/api/types"
+	"github.com/docker/docker/client"
+)
+
+// dockerClient returns a docker client
+func dockerClient() (context.Context, *client.Client) {
+	ctx := context.Background()
+	cli, err := client.NewClientWithOpts(client.FromEnv, client.WithAPIVersionNegotiation())
+	checkErr(err)
+	defer cli.Close()
+
+	return ctx, cli
+}
+
+// runnnigContainerList returns a list of running containers
+func runnnigContainerList(cli *client.Client, ctx context.Context) []types.Container {
+
+	containerList, err := cli.ContainerList(ctx, types.ContainerListOptions{})
+	checkErr(err)
+	return containerList
+
+}

cmd/docker/container/rename.go

@@ -0,0 +1,72 @@
+package container
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+	"strings"
+
+	"github.com/manifoldco/promptui"
+	"github.com/spf13/cobra"
+)
+
+// renameCmd represents the rename command
+var renameCmd = &cobra.Command{
+	Use:   "rename",
+	Short: "Rename a container",
+	Run: func(cmd *cobra.Command, args []string) {
+		renameContainer()
+	},
+}
+
+func renameContainer() {
+	ctx, cli := dockerClient()
+	containerList := runnnigContainerList(cli, ctx)
+
+	if len(containerList) == 0 {
+		fmt.Println("No container running")
+		return
+	}
+
+	for _, container := range containerList {
+		runnnigContainer = append(runnnigContainer, container.Names[0][1:]+" - "+container.ID[:6])
+
+	}
+
+	// Prompt the user to select a container
+	fmt.Println("CONTAINER NAME - CONTAINER ID")
+	prompt := promptui.Select{
+		Label: "Select a container to rename",
+		Items: runnnigContainer,
+	}
+
+	_, result, err := prompt.Run()
+	checkErr(err)
+
+	// Get new name from the command line
+	scanner := bufio.NewScanner(os.Stdin)
+	fmt.Print("Enter new name: ")
+	scanner.Scan()
+	newName := scanner.Text()
+
+	newName = strings.TrimSpace(newName)
+
+	if newName == "" {
+		fmt.Println("New name cannot be empty")
+		return
+	}
+
+	slpit := strings.Split(result, " - ")
+
+	// Rename the container
+	err = cli.ContainerRename(ctx, slpit[0], newName)
+	checkErr(err)
+
+	fmt.Printf("Container %s renamed to %s\n", slpit[0], newName)
+
+}
+
+func init() {
+	ContainerCmd.AddCommand(renameCmd)
+
+}

cmd/docker/container/stop.go

@@ -0,0 +1,119 @@
+package container
+
+import (
+	"fmt"
+	"log"
+	"strings"
+
+	containertypes "github.com/docker/docker/api/types/container"
+	"github.com/manifoldco/promptui"
+	"github.com/spf13/cobra"
+)
+
+var (
+	stopall bool
+)
+
+// deleteallcontainerCmd represents the deleteallcontainer command
+var stopAllContainersCmd = &cobra.Command{
+	Use:   "stop",
+	Short: "Stop a running container",
+	Run: func(cmd *cobra.Command, args []string) {
+		stopContainers()
+	},
+}
+
+func stopContainers() {
+	ctx, cli := dockerClient()
+	containerList := runnnigContainerList(cli, ctx)
+
+	if len(containerList) == 0 {
+		fmt.Println("No container running")
+		return
+	}
+
+	// If the flag --all is set, delete all containers
+	if stopall {
+
+		prompt := promptui.Select{
+			Label: "Are you sure you want to stop all containers?",
+			Items: []string{"Yes", "No"},
+		}
+
+		_, option, err := prompt.Run()
+
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		if option == "Yes" {
+
+			for _, container := range containerList {
+				fmt.Printf("Stopping container %s (%s)...\n", container.Names[0][1:], container.ID[:6])
+				noWaitTimeout := 5 // to not wait for the container to exit gracefully
+				err := cli.ContainerStop(ctx, container.ID, containertypes.StopOptions{Timeout: &noWaitTimeout})
+				if err != nil {
+					log.Fatal(err)
+				}
+
+			}
+			fmt.Println("Stopped all containers")
+			return
+
+		} else {
+			fmt.Println("Aborted")
+			return
+		}
+
+	}
+
+	// If the flag --all is not set, delete a specific container
+	for _, container := range containerList {
+		runnnigContainer = append(runnnigContainer, container.Names[0][1:]+" - "+container.ID[:6])
+	}
+
+	// Prompt the user to select a container
+	fmt.Println("CONTAINER NAME - CONTAINER ID")
+	prompt := promptui.Select{
+		Label: "Select a container to stop",
+		Items: runnnigContainer,
+	}
+
+	_, result, err := prompt.Run()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	prompt = promptui.Select{
+		Label: "Are you sure you want to stop " + result + "?",
+		Items: []string{"Yes", "No"},
+	}
+
+	_, option, err := prompt.Run()
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	if option == "Yes" {
+
+		split := strings.Split(result, " - ")
+
+		noWaitTimeout := 5 // to not wait for the container to exit gracefully
+		err := cli.ContainerStop(ctx, split[1], containertypes.StopOptions{Timeout: &noWaitTimeout})
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		fmt.Println("Container + " + split[0] + " stopped successfully")
+
+	}
+
+}
+
+func init() {
+
+	ContainerCmd.AddCommand(stopAllContainersCmd)
+
+	stopAllContainersCmd.Flags().BoolVarP(&stopall, "all", "a", false, "Delete all containers")
+
+}

cmd/docker/docker-file.go

@@ -48,8 +48,6 @@ func createDockerfile(lang string) {
 }
 
 func init() {
-	DockerCmd.AddCommand(dockerfileCmd)
-
 	dockerfileCmd.Flags().StringVarP(&language, "lang", "l", "", "Programming language to generate Dockerfile for.")
 	err := dockerfileCmd.MarkFlagRequired("lang")
 	checkNilErr(err)

cmd/docker/docker.go

@@ -1,6 +1,7 @@
 package docker
 
 import (
+	"github.com/Pradumnasaraf/candy/cmd/docker/container"
 	"github.com/spf13/cobra"
 )
 
@@ -13,3 +14,8 @@ var DockerCmd = &cobra.Command{
 		checkNilErr(err)
 	},
 }
+
+func init() {
+	DockerCmd.AddCommand(container.ContainerCmd)
+	DockerCmd.AddCommand(dockerfileCmd)
+}

cmd/kubernetes/kubernetes-manifest.go

@@ -54,8 +54,6 @@ func createManifestFile(filename string, obj string) {
 }
 
 func init() {
-	KubernetesCmd.AddCommand(kubernetesManifestCmd)
-
 	kubernetesManifestCmd.Flags().StringVarP(&k8Obj, "obj", "o", "", "Kubernetes object to generate manifest for.")
 	err := kubernetesManifestCmd.MarkFlagRequired("obj")
 	checkNilErr(err)

cmd/kubernetes/kubernetes.go

@@ -13,3 +13,7 @@ var KubernetesCmd = &cobra.Command{
 		checkNilErr(err)
 	},
 }
+
+func init() {
+	KubernetesCmd.AddCommand(kubernetesManifestCmd)
+}