authentik/cmd/server/main.go

package main

import (
	"fmt"

	"github.com/getsentry/sentry-go"
	log "github.com/sirupsen/logrus"
	"goauthentik.io/internal/common"
	"goauthentik.io/internal/config"
	"goauthentik.io/internal/constants"
	"goauthentik.io/internal/gounicorn"
	"goauthentik.io/internal/web"
)

var running = true

func main() {
	log.SetLevel(log.DebugLevel)
	config.DefaultConfig()
	err := config.LoadConfig("./authentik/lib/default.yml")
	if err != nil {
		log.WithError(err).Warning("failed to load default config")
	}
	err = config.LoadConfig("./local.env.yml")
	if err != nil {
		log.WithError(err).Debug("failed to local config")
	}
	config.ConfigureLogger()

	if config.G.ErrorReporting.Enabled {
		err := sentry.Init(sentry.ClientOptions{
			Dsn:              "https://a579bb09306d4f8b8d8847c052d3a1d3@sentry.beryju.org/8",
			AttachStacktrace: true,
			TracesSampleRate: 0.6,
			Release:          fmt.Sprintf("authentik@%s", constants.VERSION),
			Environment:      config.G.ErrorReporting.Environment,
		})
		if err != nil {
			log.WithError(err).Warning("failed to init sentry")
		}
	}

	ex := common.Init()
	defer common.Defer()

	// u, _ := url.Parse("http://localhost:8000")

	g := gounicorn.NewGoUnicorn()
	ws := web.NewWebServer()
	defer g.Kill()
	defer ws.Shutdown()
	for {
		go attemptStartBackend(g)
		ws.Start()
		// go attemptProxyStart(u)

		<-ex
		running = false
		log.WithField("logger", "authentik").Info("shutting down webserver")
		go ws.Shutdown()
		log.WithField("logger", "authentik").Info("killing gunicorn")
		g.Kill()
	}
}

func attemptStartBackend(g *gounicorn.GoUnicorn) {
	for {
		err := g.Start()
		if !running {
			return
		}
		log.WithField("logger", "authentik.g").WithError(err).Warning("gunicorn process died, restarting")
	}
}

// func attemptProxyStart(u *url.URL) error {
// 	maxTries := 100
// 	attempt := 0
// 	for {
// 		log.WithField("logger", "authentik").Debug("attempting to init outpost")
// 		ac := ak.NewAPIController(*u, config.G.SecretKey)
// 		if ac == nil {
// 			attempt += 1
// 			time.Sleep(1 * time.Second)
// 			if attempt > maxTries {
// 				break
// 			}
// 			continue
// 		}
// 		ac.Server = proxy.NewServer(ac)
// 		err := ac.Start()
// 		log.WithField("logger", "authentik").Debug("attempting to start outpost")
// 		if err != nil {
// 			attempt += 1
// 			time.Sleep(5 * time.Second)
// 			if attempt > maxTries {
// 				break
// 			}
// 			continue
// 		}
// 		if !running {
// 			ac.Shutdown()
// 			return nil
// 		}
// 	}
// 	return nil
// }