image.go

package main

import (
	"fmt"
	"os"

	vu "github.com/dvob/vu/internal"
	"github.com/dvob/vu/internal/image"
	"github.com/spf13/cobra"
)

func newImageCmd(mgr *vu.Manager) *cobra.Command {
	var pool string
	cmd := &cobra.Command{
		Use:   "image",
		Short: "manage images",
	}
	cmd.AddCommand(
		newImageListCmd(mgr, &pool),
		newImageAddCmd(mgr, &pool),
		newImageRemoveCmd(mgr, &pool),
	)
	cmd.PersistentFlags().StringVar(&pool, "pool", "base", "Image pool")
	return cmd
}

func newImageAddCmd(mgr *vu.Manager, pool *string) *cobra.Command {
	var (
		name string
		url  string
	)
	cmd := &cobra.Command{
		Use:   "add URL [NAME]",
		Short: "Add a new image from URL",
		Long: `Adds a new image from a URL. An URL can either have a http, https or
file scheme. If no name is given the name is derived from the URL.`,
		Args: cobra.MinimumNArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			url = args[0]
			if len(args) > 1 {
				name = args[1]
			}
			_, err := image.AddFromURL(mgr.Image, *pool, name, url, os.Stdout)
			return err
		},
	}
	return cmd
}

func newImageListCmd(mgr *vu.Manager, pool *string) *cobra.Command {
	cmd := &cobra.Command{
		Use:     "list",
		Short:   "list images",
		Aliases: []string{"ls"},
		RunE: func(cmd *cobra.Command, args []string) error {
			images, err := mgr.Image.List(*pool)
			if err != nil {
				return err
			}
			for _, image := range images {
				fmt.Println(image.Name)
			}
			return nil
		},
	}
	return cmd
}

func newImageRemoveCmd(mgr *vu.Manager, pool *string) *cobra.Command {
	var (
		names []string
		errs  = []error{}
	)
	cmd := &cobra.Command{
		Use:     "remove NAME...",
		Short:   "remove images",
		Aliases: []string{"rm"},
		RunE: func(cmd *cobra.Command, args []string) error {
			names = args
			for _, name := range names {
				img, err := mgr.Image.Get(*pool, name)
				if err != nil {
					return err
				}
				err = mgr.Image.Remove(img.ID)
				if err != nil {
					errs = append(errs, err)
				}
			}
			if len(errs) > 0 {
				return errs[0]
			}
			return nil
		},
		ValidArgsFunction: completeBaseImageFunc(mgr, pool, 0),
	}
	return cmd
}

func completeBaseImageFunc(mgr *vu.Manager, pool *string, max int) func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
	return func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
		if max != 0 && len(args) >= max {
			return nil, cobra.ShellCompDirectiveNoFileComp
		}
		images, err := mgr.Image.List(*pool)
		if err != nil {
			cobra.CompErrorln(err.Error())
			return nil, cobra.ShellCompDirectiveNoFileComp
		}
		imageNames := []string{}
		for _, img := range images {
			if contains(img.Name, args) {
				continue
			}
			imageNames = append(imageNames, img.Name)
		}
		return imageNames, cobra.ShellCompDirectiveNoFileComp
	}
}