refactor

tg_bot/main.go

@@ -6,17 +6,15 @@ import (
 	"time"
 
 	"github.com/NicoNex/echotron/v3"
-	"github.com/TelegramExchange/pkg/logger"
 	"github.com/TelegramExchange/tgex-backend/tg_bot/backend"
 	"github.com/TelegramExchange/tgex-backend/tg_bot/bot"
 	"github.com/TelegramExchange/tgex-backend/tg_bot/screens"
+	"github.com/rs/zerolog"
 	"github.com/rs/zerolog/log"
 )
 
 func main() {
-	logger.Init(logger.Config{
-		PrettyConsole: true,
-	})
+	initLogger()
 
 	botToken := mustEnv("TELEGRAM__TOKEN")
 	loginURL := mustEnv("LOGIN_URL")
@@ -79,6 +77,32 @@ func main() {
 	}
 }
 
+func initLogger() {
+	zerolog.TimeFieldFormat = time.RFC3339
+
+	level := zerolog.InfoLevel
+	if parsedLevel, err := zerolog.ParseLevel(os.Getenv("LOGGER__LEVEL")); err == nil {
+		level = parsedLevel
+	}
+	zerolog.SetGlobalLevel(level)
+
+	log.Logger = zerolog.New(os.Stdout).With().Timestamp().Logger().Level(level)
+
+	prettyConsole := os.Getenv("LOGGER__PRETTY_CONSOLE")
+	if prettyConsole == "" {
+		prettyConsole = "true"
+	}
+	if prettyConsole == "true" {
+		log.Logger = zerolog.New(zerolog.ConsoleWriter{Out: os.Stdout, TimeFormat: "15:04:05"}).
+			With().
+			Timestamp().
+			Logger().
+			Level(level)
+	}
+
+	log.Info().Msg("Logger initialized")
+}
+
 func mustEnv(key string) string {
 	v := os.Getenv(key)
 	if v == "" {