blob: f8107104ade05d9dc20d576009581dd4649de0fb (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
package importas
import (
"errors"
"flag"
"fmt"
"strings"
)
func flags(config *Config) flag.FlagSet {
fs := flag.FlagSet{}
fs.Var(stringMap(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 stringMap map[string]string
func (v stringMap) Set(val string) error {
spl := strings.SplitN(val, ":", 2)
if len(spl) != 2 {
return errors.New("import flag must be of form path:alias")
}
v[spl[0]] = spl[1]
return nil
}
func (v stringMap) String() string {
return fmt.Sprintf("%v", (map[string]string)(v))
}
|
