fix: add cmd line argument for build tags (#33)

Signed-off-by: Devrim Şahin <devrim.sahin@picussecurity.com>

Author: kubuzetto

README.md

@@ -29,6 +29,7 @@ Flags:
       --ignore strings     Will ignore packages that contains any of these strings
   -o, --output string      Location for the output file (default "coverage.txt")
   -t, --toggle             Help message for toggle
+      --tags               Build tags for go build and go test commands
 
 ```
 

cmd/root.go

@@ -60,6 +60,14 @@ GO_TEST_BINARY="gotest"
 			return err
 		}
 
+		tagsArg := ""
+		tags, err := cmd.Flags().GetStringSlice("tags")
+		if err != nil {
+			return err
+		} else if len(tags) != 0 {
+			tagsArg = "-tags="+strings.Join(tags, ",")
+		}
+
 		payload := "mode: " + mode + "\n"
 
 		var packages []string
@@ -77,7 +85,7 @@ GO_TEST_BINARY="gotest"
 
 			if len(a) > 4 && a[len(a)-4:] == "/..." {
 				var buf bytes.Buffer
-				c := exec.Command("go", "list", a)
+				c := exec.Command("go", "list", tagsArg, a)
 				c.Stdout = &buf
 				c.Stderr = &buf
 				if err := c.Run(); err != nil {
@@ -133,6 +141,7 @@ GO_TEST_BINARY="gotest"
 					"-covermode=" + mode,
 					"-coverprofile=" + files[k],
 					"-coverpkg=" + strings.Join(packages, ","),
+					tagsArg,
 				},
 				passthrough...),
 				pkg)
@@ -203,6 +212,7 @@ func init() {
 	RootCmd.Flags().StringP("output", "o", "coverage.txt", "Location for the output file")
 	RootCmd.Flags().String("covermode", "atomic", "Which code coverage mode to use")
 	RootCmd.Flags().StringSlice("ignore", []string{}, "Will ignore packages that contains any of these strings")
+	RootCmd.Flags().StringSlice("tags", []string{}, "Tags to include")
 }
 
 // initConfig reads in config file and ENV variables if set.