Merge pull request #1005 from synfinatic/move-logger

refactor the logger to be a singleton and use init()

cmd/aws-sso/logger.go

@@ -18,18 +18,10 @@ package main
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
-import (
-	"github.com/sirupsen/logrus"
-)
+import "github.com/synfinatic/aws-sso-cli/internal/logger"
 
-var log *logrus.Logger
+var log *logger.Logger
 
-/*
-func SetLogger(l *logrus.Logger) {
-	log = l
-}
-
-func GetLogger() *logrus.Logger {
-	return log
+func init() {
+	log = logger.GetLogger()
 }
-*/

cmd/aws-sso/main.go

@@ -28,17 +28,11 @@ import (
 	"github.com/posener/complete"
 
 	// "github.com/davecgh/go-spew/spew"
-	"github.com/sirupsen/logrus"
+
 	"github.com/synfinatic/aws-sso-cli/internal/config"
-	"github.com/synfinatic/aws-sso-cli/internal/ecs"
-	"github.com/synfinatic/aws-sso-cli/internal/ecs/client"
-	"github.com/synfinatic/aws-sso-cli/internal/ecs/server"
-	"github.com/synfinatic/aws-sso-cli/internal/helper"
 	"github.com/synfinatic/aws-sso-cli/internal/predictor"
 	"github.com/synfinatic/aws-sso-cli/internal/sso"
 	"github.com/synfinatic/aws-sso-cli/internal/storage"
-	"github.com/synfinatic/aws-sso-cli/internal/tags"
-	"github.com/synfinatic/aws-sso-cli/internal/url"
 	"github.com/synfinatic/aws-sso-cli/internal/utils"
 	"github.com/willabides/kongplete"
 )
@@ -158,20 +152,8 @@ func main() {
 		Auth: AUTH_UNKNOWN,
 	}
 
-	log = logrus.New()
 	override := parseArgs(&runCtx)
 
-	helper.SetLogger(log)
-	predictor.SetLogger(log)
-	sso.SetLogger(log)
-	storage.SetLogger(log)
-	tags.SetLogger(log)
-	url.SetLogger(log)
-	utils.SetLogger(log)
-	ecs.SetLogger(log)
-	server.SetLogger(log)
-	client.SetLogger(log)
-
 	if runCtx.Auth == AUTH_NO_CONFIG {
 		// side-step the rest of the setup...
 		if err = runCtx.Kctx.Run(&runCtx); err != nil {
@@ -331,12 +313,6 @@ func parseArgs(ctx *RunContext) sso.OverrideSettings {
 		Threads:    threads, // must be > 0 to override config
 	}
 
-	log.SetFormatter(&logrus.TextFormatter{
-		DisableLevelTruncation: true,
-		PadLevelText:           true,
-		DisableTimestamp:       true,
-	})
-
 	return override
 }
 

internal/ecs/client/client.go

@@ -32,9 +32,16 @@ import (
 
 	"github.com/davecgh/go-spew/spew"
 	"github.com/synfinatic/aws-sso-cli/internal/ecs"
+	"github.com/synfinatic/aws-sso-cli/internal/logger"
 	"github.com/synfinatic/aws-sso-cli/internal/storage"
 )
 
+var log *logger.Logger
+
+func init() {
+	log = logger.GetLogger()
+}
+
 type ECSClient struct {
 	server      string
 	authToken   string

internal/ecs/client_request_test.go

@@ -2,7 +2,7 @@ package ecs
 
 /*
  * AWS SSO CLI
- * Copyright (c) 2021-2022 Aaron Turner  <synfinatic at gmail dot com>
+ * Copyright (c) 2021-202 Aaron Turner  <synfinatic at gmail dot com>
  *
  * This program is free software: you can redistribute it
  * and/or modify it under the terms of the GNU General Public License as

internal/ecs/http.go

@@ -5,6 +5,7 @@ import (
 	"net/http"
 	"strconv"
 
+	"github.com/synfinatic/aws-sso-cli/internal/logger"
 	"github.com/synfinatic/aws-sso-cli/internal/storage"
 )
 
@@ -34,6 +35,7 @@ func WriteCreds(w http.ResponseWriter, creds *storage.RoleCredentials) {
 
 // JSONResponse return a JSON blob as a result
 func JSONResponse(w http.ResponseWriter, jdata interface{}) {
+	log := logger.GetLogger()
 	w.Header().Set("Content-Type", CHARSET_JSON)
 	if err := json.NewEncoder(w).Encode(jdata); err != nil {
 		log.Error(err.Error())

internal/ecs/server/server.go

@@ -28,9 +28,16 @@ import (
 
 	// "github.com/davecgh/go-spew/spew"
 	"github.com/synfinatic/aws-sso-cli/internal/ecs"
+	"github.com/synfinatic/aws-sso-cli/internal/logger"
 	"github.com/synfinatic/aws-sso-cli/internal/storage"
 )
 
+var log *logger.Logger
+
+func init() {
+	log = logger.GetLogger()
+}
+
 type EcsServer struct {
 	listener     net.Listener
 	authToken    string

internal/helper/helper.go

@@ -27,9 +27,16 @@ import (
 	"path"
 
 	"github.com/riywo/loginshell"
+	"github.com/synfinatic/aws-sso-cli/internal/logger"
 	"github.com/synfinatic/aws-sso-cli/internal/utils"
 )
 
+var log *logger.Logger
+
+func init() {
+	log = logger.GetLogger()
+}
+
 //go:embed bash_profile.sh zshrc.sh aws-sso.fish
 var embedFiles embed.FS
 

internal/ecs/client/logger.go → internal/logger/logger.go

@@ -1,4 +1,4 @@
-package client
+package logger
 
 /*
  * AWS SSO CLI
@@ -22,19 +22,29 @@ import (
 	"github.com/sirupsen/logrus"
 )
 
-var log *logrus.Logger
+var log *Logger
 
-func SetLogger(l *logrus.Logger) {
-	log = l
+type Logger struct {
+	*logrus.Logger
 }
 
-/*
-func GetLogger() *logrus.Logger {
-	return log
+func NewLogger(l *logrus.Logger) *Logger {
+	return &Logger{l}
 }
-*/
 
-// this is configured by cmd/main.go, but we have this here for unit tests
 func init() {
-	log = logrus.New()
+	log = &Logger{logrus.New()}
+	log.SetFormatter(&logrus.TextFormatter{
+		DisableLevelTruncation: true,
+		PadLevelText:           true,
+		DisableTimestamp:       true,
+	})
+}
+
+func SetLogger(l *Logger) {
+	log = l
+}
+
+func GetLogger() *Logger {
+	return log
 }

internal/predictor/predictor.go

@@ -25,10 +25,17 @@ import (
 	// "github.com/davecgh/go-spew/spew"
 	"github.com/goccy/go-yaml"
 	"github.com/posener/complete"
+	"github.com/synfinatic/aws-sso-cli/internal/logger"
 	"github.com/synfinatic/aws-sso-cli/internal/sso"
 	"github.com/synfinatic/aws-sso-cli/internal/utils"
 )
 
+var log *logger.Logger
+
+func init() {
+	log = logger.GetLogger()
+}
+
 type Predictor struct {
 	configFile string
 	accountids []string