flags.go

package importas

import (
	"errors"
	"flag"
	"fmt"
	"strings"
)

var errWrongAlias = errors.New("import flag must be of form path:alias")

func flags(config *Config) flag.FlagSet {
	fs := flag.FlagSet{}
	fs.Var(&config.RequiredAlias, "alias", "required import alias in form path:alias")
	fs.BoolVar(&config.DisallowUnaliased, "no-unaliased", false, "do not allow unaliased imports of aliased packages")
	fs.BoolVar(&config.DisallowExtraAliases, "no-extra-aliases", false, "do not allow non-required aliases")
	return fs
}

type aliasList [][]string

func (v *aliasList) Set(val string) error {
	lastColon := strings.LastIndex(val, ":")
	if lastColon <= 1 {
		return errWrongAlias
	}
	*v = append(*v, []string{val[:lastColon], val[lastColon+1:]})
	return nil
}

func (v *aliasList) String() string {
	return fmt.Sprintf("%v", ([][]string)(*v))
}