offa/main.go at main · go-oidfed/offa · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package main

import (
	"os"
	"os/signal"
	"syscall"

	"github.com/go-oidfed/lib"
	"github.com/go-oidfed/lib/jwx"
	log "github.com/sirupsen/logrus"

	"github.com/go-oidfed/offa/internal"
	"github.com/go-oidfed/offa/internal/cache"
	"github.com/go-oidfed/offa/internal/config"
	"github.com/go-oidfed/offa/internal/logger"
	"github.com/go-oidfed/offa/internal/server"
)

func main() {
	handleSignals()
	config.MustLoadConfig()
	logger.Init()
	cache.Init()
	internal.InitKeys()
	for _, c := range config.Get().Federation.TrustMarks {
		if err := c.Verify(
			config.Get().Federation.EntityID, "",
			jwx.NewTrustMarkSigner(internal.FederationSigner()),
		); err != nil {
			log.Fatal(err)
		}
	}
	if config.Get().Federation.UseResolveEndpoint {
		oidfed.DefaultMetadataResolver = oidfed.SmartRemoteMetadataResolver{}
	}
	server.Init()
	server.Start()
}

func handleSignals() {
	signals := make(chan os.Signal, 1)
	signal.Notify(signals, syscall.SIGHUP, syscall.SIGUSR1)
	go func() {
		for {
			sig := <-signals
			switch sig {
			case syscall.SIGHUP:
				reload()
			case syscall.SIGUSR1:
				reloadLogFiles()
			}
		}
	}()
}

func reload() {
	log.Info("Reloading config")
	config.MustLoadConfig()
	if config.Get().Federation.UseResolveEndpoint {
		oidfed.DefaultMetadataResolver = oidfed.SmartRemoteMetadataResolver{}
	}
	logger.SetOutput()
	logger.MustUpdateAccessLogger()
}

func reloadLogFiles() {
	log.Debug("Reloading log files")
	logger.SetOutput()
	logger.MustUpdateAccessLogger()
}