1 files changed, 37 insertions, 42 deletions

diff --git a/cmd/podman/volume_prune.go b/cmd/podman/volume_prune.go
index 41d95f9c7..1f7931aa4 100644
--- a/cmd/podman/volume_prune.go
+++ b/cmd/podman/volume_prune.go
@@ -7,73 +7,69 @@ import (
 	"os"
 	"strings"
 
-	"github.com/containers/libpod/libpod"
-	"github.com/containers/libpod/libpod/adapter"
+	"github.com/containers/libpod/cmd/podman/cliconfig"
+	"github.com/containers/libpod/pkg/adapter"
 	"github.com/pkg/errors"
 	"github.com/sirupsen/logrus"
-	"github.com/urfave/cli"
+	"github.com/spf13/cobra"
 )
 
-var volumePruneDescription = `
+var (
+	volumePruneCommand     cliconfig.VolumePruneValues
+	volumePruneDescription = `
 podman volume prune
 
 Remove all unused volumes. Will prompt for confirmation if not
 using force.
 `
+	_volumePruneCommand = &cobra.Command{
+		Use:   "prune",
+		Short: "Remove all unused volumes",
+		Long:  volumePruneDescription,
+		RunE: func(cmd *cobra.Command, args []string) error {
+			volumePruneCommand.InputArgs = args
+			volumePruneCommand.GlobalFlags = MainGlobalOpts
+			return volumePruneCmd(&volumePruneCommand)
+		},
+	}
+)
 
-var volumePruneFlags = []cli.Flag{
-	cli.BoolFlag{
-		Name:  "force, f",
-		Usage: "Do not prompt for confirmation",
-	},
-}
+func init() {
+	volumePruneCommand.Command = _volumePruneCommand
+	volumePruneCommand.SetUsageTemplate(UsageTemplate())
+	flags := volumePruneCommand.Flags()
 
-var volumePruneCommand = cli.Command{
-	Name:                   "prune",
-	Usage:                  "Remove all unused volumes",
-	Description:            volumePruneDescription,
-	Flags:                  volumePruneFlags,
-	Action:                 volumePruneCmd,
-	SkipArgReorder:         true,
-	UseShortOptionHandling: true,
+	flags.BoolVarP(&volumePruneCommand.Force, "force", "f", false, "Do not prompt for confirmation")
 }
 
 func volumePrune(runtime *adapter.LocalRuntime, ctx context.Context) error {
-	var lastError error
-
-	volumes, err := runtime.GetAllVolumes()
-	if err != nil {
-		return err
+	prunedNames, prunedErrors := runtime.PruneVolumes(ctx)
+	for _, name := range prunedNames {
+		fmt.Println(name)
+	}
+	if len(prunedErrors) == 0 {
+		return nil
 	}
+	// Grab the last error
+	lastError := prunedErrors[len(prunedErrors)-1]
+	// Remove the last error from the error slice
+	prunedErrors = prunedErrors[:len(prunedErrors)-1]
 
-	for _, vol := range volumes {
-		err = runtime.RemoveVolume(ctx, vol, false, true)
-		if err == nil {
-			fmt.Println(vol.Name())
-		} else if err != libpod.ErrVolumeBeingUsed {
-			if lastError != nil {
-				logrus.Errorf("%q", lastError)
-			}
-			lastError = errors.Wrapf(err, "failed to remove volume %q", vol.Name())
-		}
+	for _, err := range prunedErrors {
+		logrus.Errorf("%q", err)
 	}
 	return lastError
 }
 
-func volumePruneCmd(c *cli.Context) error {
-
-	if err := validateFlags(c, volumePruneFlags); err != nil {
-		return err
-	}
-
-	runtime, err := adapter.GetRuntime(c)
+func volumePruneCmd(c *cliconfig.VolumePruneValues) error {
+	runtime, err := adapter.GetRuntime(&c.PodmanCommand)
 	if err != nil {
 		return errors.Wrapf(err, "error creating libpod runtime")
 	}
 	defer runtime.Shutdown(false)
 
 	// Prompt for confirmation if --force is not set
-	if !c.Bool("force") {
+	if !c.Force {
 		reader := bufio.NewReader(os.Stdin)
 		fmt.Println("WARNING! This will remove all volumes not used by at least one container.")
 		fmt.Print("Are you sure you want to continue? [y/N] ")
@@ -85,6 +81,5 @@ func volumePruneCmd(c *cli.Context) error {
 			return nil
 		}
 	}
-
 	return volumePrune(runtime, getContext())
 }