Allow to specify build tags

jondot · Mar 23, 2019 · 6ba5957 · 6ba5957
1 parent 730fcbb
commit 6ba5957
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -17,12 +17,17 @@ var (
 
 var (
 	jsonOutput = kingpin.Flag("json", "Output json").Short('j').Bool()
+	buildTags  = kingpin.Flag("tags", "Build tags").String()
 )
 
 func main() {
 	kingpin.Version(fmt.Sprintf("%s (%s)", version, commit))
 	kingpin.Parse()
 	weight := pkg.NewGoWeight()
+	if *buildTags != "" {
+		weight.BuildArgs = "-tags " + *buildTags
+	}
+
 	work := weight.BuildCurrent()
 	modules := weight.Process(work)
 

diff --git a/pkg/goweight.go b/pkg/goweight.go
@@ -48,14 +48,15 @@ type ModuleEntry struct {
 	SizeHuman string `json:"size_human"`
 }
 type GoWeight struct {
+	BuildArgs string
 }
 
 func NewGoWeight() *GoWeight {
 	return &GoWeight{}
 }
 
 func (g *GoWeight) BuildCurrent() string {
-	return strings.Split(strings.TrimSpace(run("go build -work -a 2>&1")), "=")[1]
+	return strings.Split(strings.TrimSpace(run("go build "+g.BuildArgs+" -work -a 2>&1")), "=")[1]
 }
 func (g *GoWeight) Process(work string) []*ModuleEntry {