Merge pull request #78 from bake/badge-flags

Add badge flags
posener · Jul 7, 2019 · b8f1d5f · b8f1d5f
2 parents 5a4f9df + 6118cb5
commit b8f1d5f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cmd/goreadme/main.go b/cmd/goreadme/main.go
@@ -24,6 +24,12 @@ func init() {
 	flag.BoolVar(&cfg.Functions, "functions", false, "Write functions section.")
 	flag.BoolVar(&cfg.SkipExamples, "skip-examples", false, "Skip the examples section.")
 	flag.BoolVar(&cfg.SkipSubPackages, "skip-sub-packages", false, "Skip the sub packages section.")
+	flag.BoolVar(&cfg.Badges.Goreadme, "badge-goreadme", false, "Show Goreadme badge.")
+	flag.BoolVar(&cfg.Badges.TravicCI, "badge-travisci", false, "Show TravicCI badge.")
+	flag.BoolVar(&cfg.Badges.CodeCov, "badge-codecov", false, "Show CodeCov badge.")
+	flag.BoolVar(&cfg.Badges.GolangCI, "badge-golangci", false, "Show GolangCI badge.")
+	flag.BoolVar(&cfg.Badges.GoDoc, "badge-godoc", false, "Show GoDoc badge.")
+	flag.BoolVar(&cfg.Badges.GoReportCard, "badge-goreportcard", false, "Show GoReportCard badge.")
 	flag.Usage = func() {
 		fmt.Fprint(
 			flag.CommandLine.Output(),