47 lines
1.2 KiB
Go
47 lines
1.2 KiB
Go
package config
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/spf13/pflag"
|
|
"github.com/spf13/viper"
|
|
"log"
|
|
)
|
|
|
|
func Init() {
|
|
viper.SetConfigName("config")
|
|
viper.SetConfigType("json")
|
|
viper.AddConfigPath("/etc/lumeweb/portal/")
|
|
viper.AddConfigPath("$HOME/.lumeweb/portal/")
|
|
viper.AddConfigPath(".")
|
|
viper.SetEnvPrefix("LUME_WEB_PORTAL")
|
|
|
|
pflag.String("database.type", "sqlite", "Database type")
|
|
pflag.String("database.host", "localhost", "Database host")
|
|
pflag.Int("database.port", 3306, "Database port")
|
|
pflag.String("database.user", "root", "Database user")
|
|
pflag.String("database.password", "", "Database password")
|
|
pflag.String("database.name", "lumeweb_portal", "Database name")
|
|
pflag.String("database.path", "./db.sqlite", "Database path for SQLite")
|
|
pflag.Parse()
|
|
|
|
err := viper.BindPFlags(pflag.CommandLine)
|
|
|
|
if err != nil {
|
|
log.Fatalf("Fatal error arguments: %s \n", err)
|
|
return
|
|
}
|
|
|
|
err = viper.ReadInConfig()
|
|
if err != nil {
|
|
if errors.As(err, &viper.ConfigFileNotFoundError{}) {
|
|
// Config file not found, this is not an error.
|
|
fmt.Println("Config file not found, using default settings.")
|
|
} else {
|
|
// Other error, panic.
|
|
panic(fmt.Errorf("Fatal error config file: %s \n", err))
|
|
}
|
|
}
|
|
|
|
}
|