Add Confirm helper

Signed-off-by: apostasie <[email protected]>

Author: apostasie

cmd/nerdctl/builder.go

@@ -25,6 +25,7 @@ import (
 	"github.com/docker/go-units"
 	"github.com/spf13/cobra"
 
+	"github.com/containerd/nerdctl/v2/cmd/nerdctl/helpers"
 	"github.com/containerd/nerdctl/v2/pkg/api/types"
 	"github.com/containerd/nerdctl/v2/pkg/cmd/builder"
 )
@@ -71,23 +72,16 @@ func builderPruneAction(cmd *cobra.Command, _ []string) error {
 	}
 
 	if !options.Force {
-		var (
-			confirm string
-			msg     string
-		)
+		var msg string
 
 		if options.All {
 			msg = "This will remove all build cache."
 		} else {
 			msg = "This will remove any dangling build cache."
 		}
-		msg += " Are you sure you want to continue? [y/N] "
 
-		fmt.Fprintf(cmd.OutOrStdout(), "WARNING! %s", msg)
-		fmt.Fscanf(cmd.InOrStdin(), "%s", &confirm)
-
-		if strings.ToLower(confirm) != "y" {
-			return nil
+		if confirmed, err := helpers.Confirm(cmd, fmt.Sprintf("WARNING! %s.", msg)); err != nil || !confirmed {
+			return err
 		}
 	}
 

cmd/nerdctl/compose_rm.go

@@ -22,6 +22,7 @@ import (
 
 	"github.com/spf13/cobra"
 
+	"github.com/containerd/nerdctl/v2/cmd/nerdctl/helpers"
 	"github.com/containerd/nerdctl/v2/pkg/clientutil"
 	"github.com/containerd/nerdctl/v2/pkg/cmd/compose"
 	"github.com/containerd/nerdctl/v2/pkg/composer"
@@ -51,18 +52,15 @@ func composeRemoveAction(cmd *cobra.Command, args []string) error {
 		return err
 	}
 	if !force {
-		var confirm string
 		services := "all"
 		if len(args) != 0 {
 			services = strings.Join(args, ",")
 		}
+
 		msg := fmt.Sprintf("This will remove all stopped containers from services: %s.", services)
-		msg += "\nAre you sure you want to continue? [y/N] "
-		fmt.Fprintf(cmd.OutOrStdout(), "WARNING! %s", msg)
-		fmt.Fscanf(cmd.InOrStdin(), "%s", &confirm)
 
-		if strings.ToLower(confirm) != "y" {
-			return nil
+		if confirmed, err := helpers.Confirm(cmd, fmt.Sprintf("WARNING! %s.", msg)); err != nil || !confirmed {
+			return err
 		}
 	}
 

cmd/nerdctl/container_prune.go

@@ -17,11 +17,9 @@
 package main
 
 import (
-	"fmt"
-	"strings"
-
 	"github.com/spf13/cobra"
 
+	"github.com/containerd/nerdctl/v2/cmd/nerdctl/helpers"
 	"github.com/containerd/nerdctl/v2/pkg/api/types"
 	"github.com/containerd/nerdctl/v2/pkg/clientutil"
 	"github.com/containerd/nerdctl/v2/pkg/cmd/container"
@@ -59,15 +57,7 @@ func grantPrunePermission(cmd *cobra.Command) (bool, error) {
 	}
 
 	if !force {
-		var confirm string
-		msg := "This will remove all stopped containers."
-		msg += "\nAre you sure you want to continue? [y/N] "
-		fmt.Fprintf(cmd.OutOrStdout(), "WARNING! %s", msg)
-		fmt.Fscanf(cmd.InOrStdin(), "%s", &confirm)
-
-		if strings.ToLower(confirm) != "y" {
-			return false, nil
-		}
+		return helpers.Confirm(cmd, "WARNING! This will remove all stopped containers.")
 	}
 	return true, nil
 }

cmd/nerdctl/helpers/prompt.go

@@ -0,0 +1,39 @@
+/*
+   Copyright The containerd Authors.
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+*/
+
+package helpers
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/spf13/cobra"
+)
+
+func Confirm(cmd *cobra.Command, message string) (bool, error) {
+	message += "\nAre you sure you want to continue? [y/N] "
+	_, err := fmt.Fprint(cmd.OutOrStdout(), message)
+	if err != nil {
+		return false, err
+	}
+
+	var confirm string
+	_, err = fmt.Fscanf(cmd.InOrStdin(), "%s", &confirm)
+	if err != nil {
+		return false, err
+	}
+	return strings.ToLower(confirm) == "y", err
+}

cmd/nerdctl/image_prune.go

@@ -18,10 +18,10 @@ package main
 
 import (
 	"fmt"
-	"strings"
 
 	"github.com/spf13/cobra"
 
+	"github.com/containerd/nerdctl/v2/cmd/nerdctl/helpers"
 	"github.com/containerd/nerdctl/v2/pkg/api/types"
 	"github.com/containerd/nerdctl/v2/pkg/clientutil"
 	"github.com/containerd/nerdctl/v2/pkg/cmd/image"
@@ -82,22 +82,15 @@ func imagePruneAction(cmd *cobra.Command, _ []string) error {
 	}
 
 	if !options.Force {
-		var (
-			confirm string
-			msg     string
-		)
+		var msg string
 		if !options.All {
 			msg = "This will remove all dangling images."
 		} else {
 			msg = "This will remove all images without at least one container associated to them."
 		}
-		msg += "\nAre you sure you want to continue? [y/N] "
 
-		fmt.Fprintf(cmd.OutOrStdout(), "WARNING! %s", msg)
-		fmt.Fscanf(cmd.InOrStdin(), "%s", &confirm)
-
-		if strings.ToLower(confirm) != "y" {
-			return nil
+		if confirmed, err := helpers.Confirm(cmd, fmt.Sprintf("WARNING! %s.", msg)); err != nil || !confirmed {
+			return err
 		}
 	}