WIP: refactoring logging system

cmd/gotenberg/main.go

@@ -6,12 +6,11 @@ import (
 	"net/http"
 	"os"
 	"os/signal"
-	"strconv"
 	"time"
 
-	"github.com/labstack/echo/v4"
 	"github.com/thecodingmachine/gotenberg/internal/app/api"
-	"github.com/thecodingmachine/gotenberg/internal/pkg/notify"
+	conf "github.com/thecodingmachine/gotenberg/internal/pkg/config"
+	log "github.com/thecodingmachine/gotenberg/internal/pkg/logger"
 	"github.com/thecodingmachine/gotenberg/internal/pkg/pm2"
 )
 
@@ -19,116 +18,37 @@ import (
 // nolint: gochecknoglobals
 var version = "snapshot"
 
-const (
-	defaultWaitTimeoutEnvVar        = "DEFAULT_WAIT_TIMEOUT"
-	defaultListenPortEnvVar         = "DEFAULT_LISTEN_PORT"
-	disableGoogleChromeEnvVar       = "DISABLE_GOOGLE_CHROME"
-	disableUnoconvEnvVar            = "DISABLE_UNOCONV"
-	disableHealthcheckLoggingEnvVar = "DISABLE_HEALTHCHECK_LOGGING"
-	debugProcessStartup             = "DEBUG_PROCESS_STARTUP"
-)
-
-func mustParseEnvVar() *api.Options {
-	opts := api.DefaultOptions()
-	if os.Getenv(defaultWaitTimeoutEnvVar) != "" {
-		defaultWaitTimeout, err := strconv.ParseFloat(os.Getenv(defaultWaitTimeoutEnvVar), 64)
-		if err != nil {
-			notify.ErrPrint(fmt.Errorf("%s: wrong value: want float got %v", defaultWaitTimeoutEnvVar, err))
-			os.Exit(1)
-		}
-		opts.DefaultWaitTimeout = defaultWaitTimeout
-	}
-	if v, ok := os.LookupEnv(defaultListenPortEnvVar); ok {
-		defaultListener, err := strconv.ParseUint(os.Getenv(defaultListenPortEnvVar), 10, 64)
-		if err != nil {
-			notify.ErrPrint(fmt.Errorf("%s: wrong value: want uint got %v", defaultListenPortEnvVar, err))
-			os.Exit(1)
-		}
-		if defaultListener > 65535 {
-			notify.ErrPrint(fmt.Errorf("%s: wrong value: want uint < 65535 got %v", defaultListenPortEnvVar, defaultListener))
-			os.Exit(1)
-		}
-		opts.DefaultListenPort = v
-	}
-	// checkBoolEnv is a convenience function for reading
-	// an env var with a bool value where
-	// `1` is true and `0` is false.
-	checkBoolEnv := func(name string) bool {
-		if v, ok := os.LookupEnv(name); ok {
-			if v != "1" && v != "0" {
-				notify.ErrPrint(fmt.Errorf("%s: wrong value: want \"0\" or \"1\" got %v", name, v))
-				os.Exit(1)
-			}
-			return v == "1"
-		}
-		return false
-	}
-	opts.EnableChromeEndpoints = !checkBoolEnv(disableGoogleChromeEnvVar)
-	opts.EnableUnoconvEndpoints = !checkBoolEnv(disableUnoconvEnvVar)
-	opts.EnableHealthcheckLogging = !checkBoolEnv(disableHealthcheckLoggingEnvVar)
-	opts.DebugProcessStartup = checkBoolEnv(debugProcessStartup)
-	return opts
-}
-
-func mustStartProcesses(opts *api.Options) []pm2.Process {
-	var processes []pm2.Process
-	if opts.EnableChromeEndpoints {
-		processes = append(processes, pm2.NewChrome(opts.DebugProcessStartup))
-	}
-	if opts.EnableUnoconvEndpoints {
-		processes = append(processes, pm2.NewUnoconv(opts.DebugProcessStartup))
-	}
-	for _, p := range processes {
-		notify.Printf("starting %s with PM2...", p.Fullname())
-		if err := p.Start(); err != nil {
-			notify.ErrPrint(err)
-			os.Exit(1)
-		}
-	}
-	return processes
-}
-
-func mustStartAPI(srv *echo.Echo, port string) {
-	notify.Printf("http server started on port %v", port)
-	if err := srv.Start(fmt.Sprintf(":%v", port)); err != nil {
-		if err != http.ErrServerClosed {
-			notify.ErrPrint(err)
-			os.Exit(1)
-		}
-	}
-}
-
-func mustShutdownProcesses(processes []pm2.Process) {
-	for _, p := range processes {
-		notify.Printf("shutting down %s with PM2... (Ctrl+C to force)", p.Fullname())
-		if err := p.Shutdown(); err != nil {
-			notify.ErrPrint(err)
-			os.Exit(1)
-		}
-	}
-}
-
-func mustShutdownAPI(srv *echo.Echo) {
-	// create a deadline to wait for.
-	ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second)
-	defer cancel()
-	// doesn't block if no connections, but will otherwise wait
-	// until the timeout deadline.
-	notify.Print("shutting down http server... (Ctrl+C to force)")
-	if err := srv.Shutdown(ctx); err != nil {
-		notify.ErrPrint(err)
-		os.Exit(1)
-	}
-}
-
 func main() {
-	notify.Printf("Gotenberg %s", version)
-	opts := mustParseEnvVar()
-	srv := api.New(opts)
-	processes := mustStartProcesses(opts)
-	// run our API in a goroutine so that it doesn't block.s
+	config, err := conf.FromEnv()
+	systemLogger := log.New(config.LogLevel(), "system")
+	if err != nil {
+		systemLogger.Fatal(err)
+	}
+	systemLogger.Infof("Gotenberg %s", version)
+	// start PM2 processes.
+	var processes []pm2.Process
+	if config.EnableChromeEndpoints() {
+		processes = append(processes, pm2.NewChrome(systemLogger))
+	}
+	if config.EnableUnoconvEndpoints() {
+		processes = append(processes, pm2.NewUnoconv(systemLogger))
+	}
+	for _, p := range processes {
+		systemLogger.Infof("starting %s with PM2...", p.Fullname())
+		if err := p.Start(); err != nil {
+			systemLogger.Fatal(err)
+		}
+	}
+	// run our API in a goroutine so that it doesn't block.
+	// create our API.
+	srv := api.New(config)
 	go func() {
-		mustStartAPI(srv, opts.DefaultListenPort)
+		systemLogger.Infof("http server started on port %s", config.DefaultListenPort())
+		if err := srv.Start(fmt.Sprintf(":%s", config.DefaultListenPort())); err != nil {
+			if err != http.ErrServerClosed {
+				systemLogger.Fatal(err)
+			}
+		}
 	}()
 	quit := make(chan os.Signal, 1)
 	// we'll accept graceful shutdowns when quit via SIGINT (Ctrl+C)
@@ -136,8 +56,22 @@ func main() {
 	signal.Notify(quit, os.Interrupt)
 	// block until we receive our signal.
 	<-quit
-	mustShutdownAPI(srv)
-	mustShutdownProcesses(processes)
-	notify.Print("bye!")
+	// create a deadline to wait for.
+	ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second)
+	defer cancel()
+	// doesn't block if no connections, but will otherwise wait
+	// until the timeout deadline.
+	systemLogger.Info("shutting down http server...")
+	if err := srv.Shutdown(ctx); err != nil {
+		systemLogger.Fatal(err)
+	}
+	// shutdown PM2 processes.
+	for _, p := range processes {
+		systemLogger.Infof("shutting down %s with PM2...", p.Fullname())
+		if err := p.Shutdown(); err != nil {
+			systemLogger.Fatal(err)
+		}
+	}
+	systemLogger.Info("bye!")
 	os.Exit(0)
 }