refactor(logs): updated all logs to logrus

Author: Rihoj

cmd/docs.go

@@ -5,8 +5,7 @@ Copyright © 2022 James Ray [email protected]
 package cmd
 
 import (
-	"log"
-
+	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 	"github.com/spf13/cobra/doc"
 )
@@ -18,7 +17,7 @@ var docsCmd = &cobra.Command{
 	Run: func(cmd *cobra.Command, args []string) {
 		err := doc.GenMarkdownTree(rootCmd, "./docs")
 		if err != nil {
-			log.Fatal(err)
+			log.Fatalln(err)
 		}
 	},
 }

cmd/init.go

@@ -6,6 +6,7 @@ package cmd
 
 import (
 	"github.com/rayprogramming/toolsium/lib/config"
+	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
 
@@ -14,11 +15,16 @@ var initCmd = &cobra.Command{
 	Use:   "init",
 	Short: "Setup toolsium",
 	Run: func(cmd *cobra.Command, args []string) {
+		log.Debugf("init conf path: %v", config.GetConfigFilePath())
 		err := config.CreateConfig()
 		cobra.CheckErr(err)
 	},
 }
 
+func runAWS() {
+	log.Debugln("Done running AWS command")
+}
+
 func init() {
 	rootCmd.AddCommand(initCmd)
 

cmd/mfa.go

@@ -7,11 +7,11 @@ package cmd
 import (
 	"context"
 	"errors"
-	"log"
 
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/config"
 	"github.com/aws/aws-sdk-go-v2/service/sts"
+	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
 )
@@ -38,7 +38,7 @@ to quickly create a Cobra application.`,
 	Run: func(cmd *cobra.Command, args []string) {
 		cfg, err := config.LoadDefaultConfig(context.TODO())
 		if err != nil {
-			log.Fatal(err)
+			log.Fatalln(err)
 		}
 
 		client := sts.NewFromConfig(cfg)
@@ -48,11 +48,11 @@ to quickly create a Cobra application.`,
 		}
 		tokenOutput, err := client.GetSessionToken(context.TODO(), &sessionTokenInput)
 		if err != nil {
-			log.Fatal(err)
+			log.Fatalln(err)
 		}
 		viper.Set("session", tokenOutput)
 		if err := viper.WriteConfig(); err != nil {
-			log.Fatal(err)
+			log.Fatalln(err)
 		}
 	},
 }

cmd/papi.go

@@ -7,13 +7,13 @@ package cmd
 import (
 	"context"
 	"fmt"
-	"log"
 
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/config"
 	"github.com/aws/aws-sdk-go-v2/service/ec2"
 	"github.com/aws/aws-sdk-go-v2/service/ec2/types"
 	"github.com/rayprogramming/toolsium/lib"
+	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
 
@@ -33,7 +33,7 @@ to quickly create a Cobra application.`,
 		cfg.Credentials = aws.NewCredentialsCache(&lib.MfaProvider{})
 
 		if err != nil {
-			log.Fatal(err)
+			log.Fatalln(err)
 		}
 
 		client := ec2.NewFromConfig(cfg)

cmd/root.go

@@ -9,6 +9,7 @@ import (
 	"os"
 
 	"github.com/rayprogramming/toolsium/lib/config"
+	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 )
 
@@ -44,5 +45,6 @@ func init() {
 
 // initConfig reads in config file and ENV variables if set.
 func initConfig() {
+	log.Debugf("cfgFile: %v", cfgFile)
 	config.Configure(cfgFile)
 }

lib/config/config.go

@@ -19,6 +19,7 @@ import (
 func Configure(cfgFile string) { t.Configure(cfgFile) }
 func (t *Toolsium) Configure(cfgFile string) {
 	if cfgFile != "" {
+		log.Debugf("Using custom cfgFile %v", cfgFile)
 		// Use config file from the flag.
 		t.Config.SetConfigFile(cfgFile)
 	} else {
@@ -48,7 +49,7 @@ func (t *Toolsium) GetConfigDir() (path string) {
 func SetConfigDir(confDir string) { t.SetConfigDir(confDir) }
 func (t *Toolsium) SetConfigDir(confDir string) {
 	if _, osErr := os.Stat(confDir); os.IsNotExist(osErr) {
-		log.Infof("Provided config directory didn't exist. %v", confDir)
+		log.Debugf("Provided config directory didn't exist. %v", confDir)
 		t.ConfigDir = t.DefaultConfigDirPath()
 	}
 	t.ConfigDir = confDir
@@ -98,7 +99,7 @@ func (t *Toolsium) createConfigDirectory() {
 	if _, err := os.Stat(path); errors.Is(err, os.ErrNotExist) {
 		err := os.Mkdir(path, os.ModePerm)
 		if err != nil {
-			log.Errorln(err)
+			log.Fatalln(err)
 		}
 	}
 }