package models import ( "flag" "fmt" "os" "git.rpjosh.de/RPJosh/go-logger" "git.rpjosh.de/ncDocConverter/pkg/utils" yaml "gopkg.in/yaml.v3" ) type WebConfig struct { Server Server `yaml:"server"` Logging Logging `yaml:"logging"` } type Server struct { Address string `yaml:"address"` Certificate string `yaml:"certificate"` OneShot bool `yaml:"oneShot"` JobFile string `yaml:"jobFile"` Version string } type Logging struct { PrintLogLevel string `yaml:"printLogLevel"` WriteLogLevel string `yaml:"writeLogLevel"` LogFilePath string `yaml:"logFilePath"` } // Parses the configuration file (.yaml file) to an WebConfiguration func ParseWebConfig(webConfig *WebConfig, file string) (*WebConfig, error) { if file == "" { return webConfig, nil } dat, err := os.ReadFile(file) if err != nil { return nil, err } if err := yaml.Unmarshal(dat, &webConfig); err != nil { return nil, err } return webConfig, nil } func getDefaultConfig() *WebConfig { return &WebConfig{ Server: Server{ Address: ":4000", JobFile: utils.GetEnvString("DATA_FILE", "./ncConverter.json"), }, Logging: Logging{ PrintLogLevel: "info", WriteLogLevel: "warning", }, } } // Applies the cli and the configuration options from the config files func SetConfig(version string) (*WebConfig, error) { configPath := "./config.yaml" // the path of the configuration file is needed first to determine the "default" values for i, arg := range os.Args { if arg == "-config" || arg == "--config" && len(os.Args) > i { configPath = os.Args[i+1] break } } webConfig := getDefaultConfig() webConfig.Server.Version = version webConfig, err := ParseWebConfig(webConfig, configPath) if err != nil { logger.Error("Unable to parse the configuration file '%s': %s", configPath, err) webConfig = getDefaultConfig() err = nil } _ = flag.String("config", "./config.yaml", "Path to the configuration file (see configs/config.yaml) for an example") address := flag.String("address", webConfig.Server.Address, "Address and port on which the api and the web server should listen to") printLogLevel := flag.String("printLogLevel", webConfig.Logging.PrintLogLevel, "Minimum log level to log (debug, info, warning, error, fatal)") oneShot := flag.Bool("oneShot", webConfig.Server.OneShot, "All jobs are executed immediately and the program exists afterwards") printVersion := flag.Bool("version", false, "Prints the version of the program") flag.Parse() webConfig.Server.Address = *address webConfig.Logging.PrintLogLevel = *printLogLevel webConfig.Server.OneShot = *oneShot if *printVersion { fmt.Println(webConfig.Server.Version) os.Exit(0) } defaultLogger := logger.GetLoggerFromEnv(&logger.Logger{ PrintLevel: logger.GetLevelByName(webConfig.Logging.PrintLogLevel), LogLevel: logger.GetLevelByName(webConfig.Logging.WriteLogLevel), LogFilePath: webConfig.Logging.LogFilePath, PrintSource: true, }) logger.SetGlobalLogger(defaultLogger) return webConfig, err }