Add ability to force colors (for Windows)

bloodydoves · Jan 20, 2021 · 61bcac9 · 61bcac9
1 parent 4e1d450
commit 61bcac9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/bta-wiki-import.exe b/bta-wiki-import.exe
diff --git a/cmd/root.go b/cmd/root.go
@@ -1,10 +1,24 @@
 package cmd
 
 import (
+	"github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
 
+var (
+	flagColor bool
+)
+
 var RootCmd = &cobra.Command{
 	Use:   "btawiki",
 	Short: "btawiki exports data from BTA3062 game files to mediawiki",
+	PersistentPreRun: func(cmd *cobra.Command, _ []string) {
+		if flagColor {
+			logrus.SetFormatter(&logrus.TextFormatter{ForceColors: true})
+		}
+	},
+}
+
+func init() {
+	RootCmd.PersistentFlags().BoolVar(&flagColor, "color", false, "enable color logging")
 }