2021-05-02 22:49:16 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2021-05-04 12:28:48 +00:00
|
|
|
"fmt"
|
2021-06-23 18:40:51 +00:00
|
|
|
"net/url"
|
2021-05-04 12:28:48 +00:00
|
|
|
"time"
|
2021-05-02 22:49:16 +00:00
|
|
|
|
2021-05-04 12:28:48 +00:00
|
|
|
"github.com/getsentry/sentry-go"
|
2021-05-02 22:49:16 +00:00
|
|
|
log "github.com/sirupsen/logrus"
|
2021-06-16 15:29:01 +00:00
|
|
|
"goauthentik.io/internal/common"
|
2021-05-02 22:49:16 +00:00
|
|
|
"goauthentik.io/internal/config"
|
2021-05-04 12:28:48 +00:00
|
|
|
"goauthentik.io/internal/constants"
|
2021-05-02 22:49:16 +00:00
|
|
|
"goauthentik.io/internal/gounicorn"
|
2021-06-23 18:40:51 +00:00
|
|
|
"goauthentik.io/internal/outpost/ak"
|
|
|
|
"goauthentik.io/internal/outpost/proxy"
|
2021-05-02 22:49:16 +00:00
|
|
|
"goauthentik.io/internal/web"
|
|
|
|
)
|
|
|
|
|
2021-07-17 16:04:09 +00:00
|
|
|
var running = true
|
|
|
|
|
2021-05-02 22:49:16 +00:00
|
|
|
func main() {
|
2021-05-04 12:07:12 +00:00
|
|
|
log.SetLevel(log.DebugLevel)
|
2021-05-02 22:49:16 +00:00
|
|
|
config.DefaultConfig()
|
2021-05-04 12:07:12 +00:00
|
|
|
config.LoadConfig("./authentik/lib/default.yml")
|
|
|
|
config.LoadConfig("./local.env.yml")
|
2021-05-02 22:49:16 +00:00
|
|
|
config.ConfigureLogger()
|
|
|
|
|
2021-05-04 12:28:48 +00:00
|
|
|
if config.G.ErrorReporting.Enabled {
|
|
|
|
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,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2021-06-23 18:40:51 +00:00
|
|
|
ex := common.Init()
|
2021-06-16 15:29:01 +00:00
|
|
|
defer common.Defer()
|
|
|
|
|
2021-07-17 13:49:06 +00:00
|
|
|
// u, _ := url.Parse("http://localhost:8000")
|
2021-06-23 18:40:51 +00:00
|
|
|
|
2021-06-29 14:21:00 +00:00
|
|
|
g := gounicorn.NewGoUnicorn()
|
|
|
|
ws := web.NewWebServer()
|
|
|
|
defer g.Kill()
|
|
|
|
defer ws.Shutdown()
|
2021-06-23 18:40:51 +00:00
|
|
|
for {
|
2021-07-17 16:04:09 +00:00
|
|
|
go attemptStartBackend(g)
|
2021-06-29 14:21:00 +00:00
|
|
|
ws.Start()
|
2021-07-17 16:04:09 +00:00
|
|
|
// go attemptProxyStart(u)
|
2021-06-23 18:40:51 +00:00
|
|
|
|
2021-06-29 14:21:00 +00:00
|
|
|
<-ex
|
2021-07-17 16:04:09 +00:00
|
|
|
running = false
|
|
|
|
log.WithField("logger", "authentik").Info("shutting down webserver")
|
2021-07-17 15:02:24 +00:00
|
|
|
go ws.Shutdown()
|
2021-07-17 16:04:09 +00:00
|
|
|
log.WithField("logger", "authentik").Info("killing gunicorn")
|
2021-06-23 18:40:51 +00:00
|
|
|
g.Kill()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-07-17 16:04:09 +00:00
|
|
|
func attemptStartBackend(g *gounicorn.GoUnicorn) error {
|
2021-06-23 18:40:51 +00:00
|
|
|
for {
|
|
|
|
err := g.Start()
|
2021-07-17 16:04:09 +00:00
|
|
|
if !running {
|
2021-06-29 14:21:00 +00:00
|
|
|
return nil
|
|
|
|
}
|
2021-07-17 16:04:09 +00:00
|
|
|
log.WithField("logger", "authentik.g").WithError(err).Warning("gunicorn process died, restarting")
|
2021-06-23 18:40:51 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-07-17 16:04:09 +00:00
|
|
|
func attemptProxyStart(u *url.URL) error {
|
2021-06-29 14:21:00 +00:00
|
|
|
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
|
|
|
|
}
|
2021-07-17 16:04:09 +00:00
|
|
|
if !running {
|
2021-06-29 14:21:00 +00:00
|
|
|
ac.Shutdown()
|
|
|
|
return nil
|
|
|
|
}
|
2021-06-23 18:40:51 +00:00
|
|
|
}
|
2021-06-29 14:21:00 +00:00
|
|
|
return nil
|
2021-05-02 22:49:16 +00:00
|
|
|
}
|