vm_delete.go

package cmd

import (
	"fmt"
	"os"
	"path"

	"github.com/exoscale/egoscale"
	"github.com/spf13/cobra"
)

// deleteCmd represents the delete command
var vmDeleteCmd = &cobra.Command{
	Use:               "delete <vm name | id>+",
	Short:             "Delete virtual machine instance(s)",
	Aliases:           gDeleteAlias,
	ValidArgsFunction: completeVMNames,
	RunE: func(cmd *cobra.Command, args []string) error {
		if len(args) < 1 {
			return cmd.Usage()
		}

		force, err := cmd.Flags().GetBool("force")
		if err != nil {
			return err
		}

		tasks := []task{}
		vms := make([]egoscale.VirtualMachine, len(args))

		for i, arg := range args {
			vm, err := getVirtualMachineByNameOrID(arg)
			if err != nil {
				return err
			}
			vms[i] = *vm

			if ok := prepareDeleteVM(vm, force); ok {
				tasks = append(tasks, task{
					&egoscale.DestroyVirtualMachine{ID: vm.ID},
					fmt.Sprintf("Destroying %q ", vm.Name),
				})
			}
		}

		resps := asyncTasks(tasks)
		errors := filterErrors(resps)
		if len(errors) > 0 {
			for _, err := range errors {
				fmt.Fprintln(os.Stderr, err) // nolint: errcheck
			}
			return nil
		}

		for i := range resps {
			vm := vms[i]
			folder := path.Join(gConfigFolder, "instances", vm.ID.String())

			if _, err := os.Stat(folder); !os.IsNotExist(err) {
				if err := os.RemoveAll(folder); err != nil {
					return err
				}
			}
		}

		return nil
	},
}

func prepareDeleteVM(vm *egoscale.VirtualMachine, force bool) bool {
	if !force {
		if !askQuestion(fmt.Sprintf("sure you want to delete %q virtual machine", vm.Name)) {
			return false
		}
	}

	return true
}

func init() {
	vmDeleteCmd.Flags().BoolP("force", "f", false, "Attempt to remove vitual machine without prompting for confirmation")
	vmCmd.AddCommand(vmDeleteCmd)
}