diff --git a/internal/config/config.go b/internal/config/config.go index 268b3e7c7..55c63fab0 100644 --- a/internal/config/config.go +++ b/internal/config/config.go @@ -79,8 +79,6 @@ func init() { func (config *configSchema) Read() { v := viper.New() - pflag.Parse() - pflag.CommandLine.AddGoFlagSet(flag.CommandLine) pflag.Parse() v.BindPFlags(pflag.CommandLine) diff --git a/main.go b/main.go index 2effd3671..f788eab83 100644 --- a/main.go +++ b/main.go @@ -12,6 +12,7 @@ import ( "github.com/cloudflare/unsee/internal/config" "github.com/cloudflare/unsee/internal/models" "github.com/cloudflare/unsee/internal/transform" + "github.com/spf13/pflag" "github.com/DeanThompson/ginpprof" "github.com/gin-contrib/gzip" @@ -109,6 +110,13 @@ func setupLogger() { } func main() { + printVersion := pflag.Bool("version", false, "Print version and exit") + pflag.Parse() + if *printVersion { + fmt.Println(version) + return + } + config.Config.Read() setupLogger()