71 lines
2.1 KiB
Go
71 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/kkyr/fig"
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
type Config struct {
|
|
Timezone string `fig:"tz" default:"America/New_York"`
|
|
Server struct {
|
|
Port string `fig:"port" default:"3000"`
|
|
LocationPhotoDir string `fig:"locationPhotoDir" default:"./app/photos/locations/"`
|
|
}
|
|
Logger struct {
|
|
Level string `fig:"loglevel" default:"info"`
|
|
LoggingFile string `fig:"loggingFile" default:"./app/log/goInventorize.log"`
|
|
}
|
|
Authentication struct {
|
|
BasicAuth bool `fig:"basicauth"`
|
|
UserName string `fig:"username" default:"admin"`
|
|
Password string `fig:"password" default:"password"`
|
|
}
|
|
|
|
Development bool `fig:"development"` // Cannot set default for BOOL will default to false if nothing supplied
|
|
}
|
|
|
|
func LoadConfig(s *Server) error {
|
|
var cfg Config
|
|
err := fig.Load(&cfg, fig.File("config.yaml"), fig.Dirs("./app/config"), fig.UseEnv("GI")) // Load in config.yaml, then overwrite with ENV variables prefixed with GI
|
|
if err != nil {
|
|
return err
|
|
}
|
|
logFile, err := os.OpenFile(cfg.Logger.LoggingFile, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666) // Open/Create log file for writting to log
|
|
if err != nil {
|
|
return err
|
|
}
|
|
switch cfg.Logger.Level { // Setting our logging level
|
|
case "error":
|
|
zerolog.SetGlobalLevel(zerolog.ErrorLevel)
|
|
case "warn":
|
|
zerolog.SetGlobalLevel(zerolog.WarnLevel)
|
|
case "debug":
|
|
zerolog.SetGlobalLevel(zerolog.DebugLevel)
|
|
default:
|
|
zerolog.SetGlobalLevel(zerolog.InfoLevel)
|
|
|
|
}
|
|
loc, err := time.LoadLocation(cfg.Timezone) // Getting our timezone info from config
|
|
if err != nil {
|
|
return err
|
|
}
|
|
s.LogFile = logFile
|
|
zerolog.TimestampFunc = func() time.Time { // Making sure our logger writes in the correct tz
|
|
return time.Now().In(loc)
|
|
}
|
|
s.Log = zerolog.New(logFile).With().Timestamp().Logger()
|
|
// Creating photo directory
|
|
err = os.MkdirAll(cfg.Server.LocationPhotoDir, 0755)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
s.Log.Info().Msg("Configuration loaded successfully...")
|
|
s.Log.Debug().Msgf("%+v", cfg)
|
|
s.Config = &cfg
|
|
return nil
|
|
}
|