wip
This commit is contained in:
106
mangezmieux-backend/cmd/root.go
Normal file
106
mangezmieux-backend/cmd/root.go
Normal file
@@ -0,0 +1,106 @@
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user