package cmd import ( "fmt" "mangezmieux-backend/configuration" "mangezmieux-backend/internal/logger" "os" "github.com/spf13/cobra" "github.com/spf13/viper" ) const ( // Log. parameterLogLevel = "loglevel" parameterLogFormat = "logformat" defaultLogLevel = "debug" defaultLogFormat = "text" // Mock. parameterMock = "mock" defaultMock = true // Router. parameterPort = "port" defaultPort = "8080" // DATABASE. parameterPostgresDBName = "postgresdbname" defaultPostgresDBName = "mangezmieux" parameterPostgresDBSchema = "postgresdbschema" defaultPostgresDBSchema = "mangezmieux" parameterPostgresHost = "postgreshost" defaultPostgresHost = "localhost" parameterPostgresUser = "postgresuser" defaultPostgresUser = "postgres" parameterPostgresPwd = "postgrespwd" defaultPostgresPwd = "mysecretpassword" ) var ( config = &configuration.Config{} cfgFile string // GITHASH : Stores the git revision to be displayed. GITHASH string // VERSION : Stores the binary version to be displayed. VERSION string // rootCmd represents the base command when called without any subcommands. rootCmd = &cobra.Command{ Use: "mangezmieux", Short: "mangezmieux", Version: fmt.Sprintf("%s (%s)", VERSION, GITHASH), } ) // Execute adds all child commands to the root command and sets flags appropriately. // This is called by main.main(). It only needs to happen once to the rootCmd. func Execute() { if err := rootCmd.Execute(); err != nil { logger.GetLogger().Error(err) os.Exit(1) } } func init() { rootCmd.AddCommand(serveCmd) // Here you will define your flags and configuration settings. // Cobra supports persistent flags, which, if defined here, // will be global for your application. rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.mangezmieux.yaml)") } // initConfig reads in config file and ENV variables if set. func initConfig() { if cfgFile != "" { // Use config file from the flag. viper.SetConfigFile(cfgFile) } viper.AutomaticEnv() // read in environment variables that match // If a config file is found, read it in. if err := viper.ReadInConfig(); err == nil { logger.GetLogger().Info("Using config file:", viper.ConfigFileUsed()) } config.Mock = viper.GetBool(parameterMock) config.Port = viper.GetString(parameterPort) config.LogLevel = viper.GetString(parameterLogLevel) config.LogFormat = viper.GetString(parameterLogFormat) config.PostgresDBName = viper.GetString(parameterPostgresDBName) config.PostgresDBSchema = viper.GetString(parameterPostgresDBSchema) config.PostgresHost = viper.GetString(parameterPostgresHost) config.PostgresUser = viper.GetString(parameterPostgresUser) config.PostgresPwd = viper.GetString(parameterPostgresPwd) config.Version = VERSION }