package cmd

import (
	"fmt"
	"github.com/kratisto/mam-contract/configuration"
	"github.com/mitchellh/go-homedir"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
	"os"
)

const (
	// Log
	parameterLogLevel  = "loglevel"
	parameterLogFormat = "logformat"

	defaultLogLevel  = "debug"
	defaultLogFormat = "text"

	// Mock
	parameterMock = "mock"

	defaultMock = false

	// Router
	parameterPort = "port"
	defaultPort   = 8080

	// DATABASE
	parameterPostgresDBName   = "postgresdbname"
	defaultPostgresDBName     = "postgres"
	parameterPostgresDBSchema = "postgresdbschema"
	defaultPostgresDBSchema   = "workScheduler"
	parameterPostgresHost     = "postgreshost"
	defaultPostgresHost       = "database"
	parameterPostgresUser     = "postgresuser"
	defaultPostgresUser       = "postgres"
	parameterPostgresPwd      = "postgrespwd"
	defaultPostgresPwd        = "password"
)

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:     "workScheduler",
		Short:   "workScheduler",
		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 {
		fmt.Println(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/.workScheduler.yaml)")

}

// initConfig reads in config file and ENV variables if set.
func initConfig() {
	if cfgFile != "" {
		// Use config file from the flag.
		viper.SetConfigFile(cfgFile)
	} else {
		// Find home directory.
		home, err := homedir.Dir()
		if err != nil {
			fmt.Println(err)
			os.Exit(1)
		}

		// Search config in home directory with name ".workScheduler" (without extension).
		viper.AddConfigPath(home)
		viper.SetConfigName(".workScheduler")
	}

	viper.AutomaticEnv() // read in environment variables that match

	// If a config file is found, read it in.
	if err := viper.ReadInConfig(); err == nil {
		fmt.Println("Using config file:", viper.ConfigFileUsed())
	}

	config.Mock = viper.GetBool(parameterMock)

	config.Port = viper.GetInt(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

}