luzifer/nginx-sso
1
0
Fork 0
mirror of https://github.com/Luzifer/nginx-sso.git synced 2024-12-20 04:41:17 +00:00
Code Issues Releases Wiki Activity

Fix: Config loading after CookieStore init (#58)

Browse source 
* fix config loading after CookieStore init

* refactor according to suggestions

* fix module init on SIGHUP
This commit is contained in:
Paul TREHIOU 2020-04-08 01:11:48 +02:00 committed by GitHub
parent df6201acb8
commit f9d9c025dd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 22 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
main.go
View file

@ -81,27 +81,31 @@ func init() {
mainCfg.AuditLog.Headers = []string{"x-origin-uri"}
}
func loadConfiguration() error {
func loadConfiguration() ([]byte, error) {
yamlSource, err := ioutil.ReadFile(cfg.ConfigFile)
if err != nil {
return fmt.Errorf("Unable to read configuration file: %s", err)
return nil, errors.Wrap(err, "Unable to read configuration file")
}
if err = yaml.Unmarshal(yamlSource, &mainCfg); err != nil {
return fmt.Errorf("Unable to load configuration file: %s", err)
return nil, errors.Wrap(err, "Unable to load configuration file")
}
return yamlSource, nil
}
func initializeModules(yamlSource []byte) error {
if mainCfg.Plugins.Directory != "" {
if err = loadPlugins(mainCfg.Plugins.Directory); err != nil {
if err := loadPlugins(mainCfg.Plugins.Directory); err != nil {
return errors.Wrap(err, "Unable to load plugins")
}
}
if err = initializeAuthenticators(yamlSource); err != nil {
if err := initializeAuthenticators(yamlSource); err != nil {
return fmt.Errorf("Unable to configure authentication: %s", err)
}
if err = initializeMFAProviders(yamlSource); err != nil {
if err := initializeMFAProviders(yamlSource); err != nil {
log.WithError(err).Fatal("Unable to configure MFA providers")
}
@ -109,11 +113,16 @@ func loadConfiguration() error {
}
func main() {
yamlSource, err := loadConfiguration()
if err != nil {
log.WithError(err).Fatal("Unable to load configuration")
}
cookieStore = sessions.NewCookieStore([]byte(mainCfg.Cookie.AuthKey))
registerModules()
if err := loadConfiguration(); err != nil {
log.WithError(err).Fatal("Unable to load configuration")
if err = initializeModules(yamlSource); err != nil {
log.WithError(err).Fatal("Unable to initialize modules")
}
http.HandleFunc("/", handleRootRequest)
@ -133,8 +142,12 @@ func main() {
for sig := range sigChan {
switch sig {
case syscall.SIGHUP:
if err := loadConfiguration(); err != nil {
if yamlSource, err = loadConfiguration(); err != nil {
log.WithError(err).Error("Unable to reload configuration")
} else {
if err = initializeModules(yamlSource); err != nil {
log.WithError(err).Error("Unable to initialize modules")
}
}
default: