mirror of
https://github.com/yusing/godoxy.git
synced 2026-03-17 23:03:49 +01:00
fc540ea419
Propagate critical init and entrypoint failures to halt startup and log them as fatal during config loading
116 lines
2.9 KiB
Go
Executable File
116 lines
2.9 KiB
Go
Executable File
package main
|
|
|
|
import (
|
|
"os"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/yusing/godoxy/internal/api"
|
|
"github.com/yusing/godoxy/internal/auth"
|
|
"github.com/yusing/godoxy/internal/common"
|
|
"github.com/yusing/godoxy/internal/config"
|
|
"github.com/yusing/godoxy/internal/dnsproviders"
|
|
iconlist "github.com/yusing/godoxy/internal/homepage/icons/list"
|
|
"github.com/yusing/godoxy/internal/logging"
|
|
"github.com/yusing/godoxy/internal/logging/memlogger"
|
|
"github.com/yusing/godoxy/internal/metrics/systeminfo"
|
|
"github.com/yusing/godoxy/internal/metrics/uptime"
|
|
"github.com/yusing/godoxy/internal/net/gphttp/middleware"
|
|
"github.com/yusing/godoxy/internal/route/rules"
|
|
gperr "github.com/yusing/goutils/errs"
|
|
"github.com/yusing/goutils/server"
|
|
"github.com/yusing/goutils/task"
|
|
"github.com/yusing/goutils/version"
|
|
)
|
|
|
|
func parallel(fns ...func()) {
|
|
var wg sync.WaitGroup
|
|
for _, fn := range fns {
|
|
wg.Go(fn)
|
|
}
|
|
wg.Wait()
|
|
}
|
|
|
|
func main() {
|
|
done := make(chan struct{}, 1)
|
|
go func() {
|
|
select {
|
|
case <-done:
|
|
return
|
|
case <-time.After(common.InitTimeout):
|
|
log.Fatal().Msgf("timeout waiting for initialization to complete, exiting...")
|
|
}
|
|
}()
|
|
|
|
initProfiling()
|
|
|
|
logging.InitLogger(os.Stderr, memlogger.GetMemLogger())
|
|
log.Info().Msgf("GoDoxy version %s", version.Get())
|
|
log.Trace().Msg("trace enabled")
|
|
parallel(
|
|
dnsproviders.InitProviders,
|
|
iconlist.InitCache,
|
|
systeminfo.Poller.Start,
|
|
middleware.LoadComposeFiles,
|
|
)
|
|
|
|
if common.APIJWTSecret == nil {
|
|
log.Warn().Msg("API_JWT_SECRET is not set, using random key")
|
|
common.APIJWTSecret = common.RandomJWTKey()
|
|
}
|
|
|
|
for _, dir := range common.RequiredDirectories {
|
|
prepareDirectory(dir)
|
|
}
|
|
|
|
err := config.Load()
|
|
if err != nil {
|
|
var criticalErr config.CriticalError
|
|
if errors.As(err, &criticalErr) {
|
|
gperr.LogFatal("critical error in config", criticalErr)
|
|
}
|
|
gperr.LogWarn("errors in config", err)
|
|
}
|
|
|
|
config.StartProxyServers()
|
|
|
|
if err := auth.Initialize(); err != nil {
|
|
log.Fatal().Err(err).Msg("failed to initialize authentication")
|
|
}
|
|
rules.InitAuthHandler(auth.AuthOrProceed)
|
|
|
|
// API Handler needs to start after auth is initialized.
|
|
server.StartServer(task.RootTask("api_server", false), server.Options{
|
|
Name: "api",
|
|
HTTPAddr: common.APIHTTPAddr,
|
|
Handler: api.NewHandler(true),
|
|
})
|
|
|
|
// Local API Handler is used for unauthenticated access.
|
|
if common.LocalAPIHTTPAddr != "" {
|
|
server.StartServer(task.RootTask("local_api_server", false), server.Options{
|
|
Name: "local_api",
|
|
HTTPAddr: common.LocalAPIHTTPAddr,
|
|
Handler: api.NewHandler(false),
|
|
})
|
|
}
|
|
|
|
listenDebugServer()
|
|
|
|
uptime.Poller.Start()
|
|
config.WatchChanges()
|
|
|
|
close(done)
|
|
|
|
task.WaitExit(config.Value().TimeoutShutdown)
|
|
}
|
|
|
|
func prepareDirectory(dir string) {
|
|
if _, err := os.Stat(dir); os.IsNotExist(err) {
|
|
if err = os.MkdirAll(dir, 0o755); err != nil {
|
|
log.Fatal().Msgf("failed to create directory %s: %v", dir, err)
|
|
}
|
|
}
|
|
}
|