feat(logger): update new logger

padok-team · Apr 9, 2023 · 45ff0ac · 45ff0ac
1 parent 5cecc6e
commit 45ff0ac
Show file tree

Hide file tree

Showing 18 changed files with 92 additions and 79 deletions.
diff --git a/aws/acm/getter.go b/aws/acm/getter.go
@@ -2,10 +2,10 @@ package acm
 
 import (
 	"context"
-	"fmt"
 
 	"github.com/aws/aws-sdk-go-v2/service/acm"
 	"github.com/aws/aws-sdk-go-v2/service/acm/types"
+	"github.com/padok-team/yatas-aws/logger"
 )
 
 type ACMGetObjectAPI interface {
@@ -17,7 +17,7 @@ func GetCertificates(svc ACMGetObjectAPI) []types.CertificateDetail {
 	input := &acm.ListCertificatesInput{}
 	result, err := svc.ListCertificates(context.TODO(), input)
 	if err != nil {
-		fmt.Println(err)
+		logger.Logger.Error(err.Error())
 		// Return an empty list of certificates
 		return []types.CertificateDetail{}
 	}
@@ -33,7 +33,7 @@ func GetCertificates(svc ACMGetObjectAPI) []types.CertificateDetail {
 		input.NextToken = result.NextToken
 		result, err = svc.ListCertificates(context.TODO(), input)
 		if err != nil {
-			fmt.Println(err)
+			logger.Logger.Error(err.Error())
 			return []types.CertificateDetail{}
 		}
 		for _, r := range result.CertificateSummaryList {
@@ -47,7 +47,7 @@ func GetCertificates(svc ACMGetObjectAPI) []types.CertificateDetail {
 		}
 		result, err := svc.DescribeCertificate(context.TODO(), input)
 		if err != nil {
-			fmt.Println(err)
+			logger.Logger.Error(err.Error())
 			return []types.CertificateDetail{}
 		}
 		certificates = append(certificates, *result.Certificate)

diff --git a/aws/cloudfront/getter.go b/aws/cloudfront/getter.go
@@ -2,10 +2,10 @@ package cloudfront
 
 import (
 	"context"
-	"fmt"
 
 	"github.com/aws/aws-sdk-go-v2/service/cloudfront"
 	"github.com/aws/aws-sdk-go-v2/service/cloudfront/types"
+	"github.com/padok-team/yatas-aws/logger"
 )
 
 type SummaryToConfig struct {
@@ -22,7 +22,7 @@ func GetAllCloudfront(svc CloudfrontGetObjectApi) []types.DistributionSummary {
 	input := &cloudfront.ListDistributionsInput{}
 	result, err := svc.ListDistributions(context.TODO(), input)
 	if err != nil {
-		fmt.Println(err)
+		logger.Logger.Error(err.Error())
 		// Return an empty list of certificates
 		return []types.DistributionSummary{}
 	}
@@ -37,7 +37,7 @@ func GetAllDistributionConfig(svc CloudfrontGetObjectApi, ds []types.Distributio
 		}
 		result, err := svc.GetDistributionConfig(context.TODO(), input)
 		if err != nil {
-			fmt.Println(err)
+			logger.Logger.Error(err.Error())
 			// Return an empty list of certificates
 			return []SummaryToConfig{}
 		}

diff --git a/aws/cloudtrail/getter.go b/aws/cloudtrail/getter.go
@@ -2,11 +2,11 @@ package cloudtrail
 
 import (
 	"context"
-	"fmt"
 
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/service/cloudtrail"
 	"github.com/aws/aws-sdk-go-v2/service/cloudtrail/types"
+	"github.com/padok-team/yatas-aws/logger"
 )
 
 func GetCloudtrails(s aws.Config) []types.Trail {
@@ -16,7 +16,7 @@ func GetCloudtrails(s aws.Config) []types.Trail {
 	}
 	result, err := svc.DescribeTrails(context.TODO(), input)
 	if err != nil {
-		fmt.Println(err)
+		logger.Logger.Error(err.Error())
 		// Return an empty list
 		return []types.Trail{}
 	}

diff --git a/aws/cognito/getter.go b/aws/cognito/getter.go
@@ -7,6 +7,7 @@ import (
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/service/cognitoidentity"
 	"github.com/aws/aws-sdk-go-v2/service/cognitoidentity/types"
+	"github.com/padok-team/yatas-aws/logger"
 )
 
 func GetCognitoPools(s aws.Config) []types.IdentityPoolShortDescription {
@@ -16,7 +17,7 @@ func GetCognitoPools(s aws.Config) []types.IdentityPoolShortDescription {
 	}
 	result, err := svc.ListIdentityPools(context.TODO(), cognitoInput)
 	if err != nil {
-		fmt.Println(err)
+		logger.Logger.Error(err.Error())
 		// Return an empty list of certificates
 		return []types.IdentityPoolShortDescription{}
 	}
@@ -33,7 +34,7 @@ func GetDetailedCognitoPool(s aws.Config, pools []types.IdentityPoolShortDescrip
 		}
 		result, err := svc.DescribeIdentityPool(context.TODO(), cognitoInput)
 		if err != nil {
-			fmt.Println(err)
+			logger.Logger.Error(err.Error())
 			// Return an empty list of certificates
 			return []cognitoidentity.DescribeIdentityPoolOutput{}
 		}

diff --git a/aws/dynamodb/getter.go b/aws/dynamodb/getter.go
@@ -2,19 +2,19 @@ package dynamodb
 
 import (
 	"context"
-	"fmt"
 
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/service/dynamodb"
 	"github.com/aws/aws-sdk-go-v2/service/dynamodb/types"
+	"github.com/padok-team/yatas-aws/logger"
 )
 
 func GetDynamodbs(s aws.Config) []string {
 	svc := dynamodb.NewFromConfig(s)
 	dynamodbInput := &dynamodb.ListTablesInput{}
 	result, err := svc.ListTables(context.TODO(), dynamodbInput)
 	if err != nil {
-		fmt.Println(err)
+		logger.Logger.Error(err.Error())
 		// Return an empty list
 		return []string{}
 	}
@@ -30,7 +30,7 @@ func GetTables(s aws.Config, dynamodbs []string) []*dynamodb.DescribeTableOutput
 		}
 		resp, err := svc.DescribeTable(context.TODO(), params)
 		if err != nil {
-			fmt.Println(err)
+			logger.Logger.Error(err.Error())
 			// Return an empty list of certificates
 			return []*dynamodb.DescribeTableOutput{}
 		}
@@ -54,7 +54,7 @@ func GetContinuousBackups(s aws.Config, tables []string) []TableBackups {
 		}
 		resp, err := svc.DescribeContinuousBackups(context.TODO(), params)
 		if err != nil {
-			fmt.Println(err)
+			logger.Logger.Error(err.Error())
 			// Return an empty list of certificates
 			return []TableBackups{}
 		}

diff --git a/aws/ec2/getter.go b/aws/ec2/getter.go
@@ -2,10 +2,10 @@ package ec2
 
 import (
 	"context"
-	"fmt"
 
 	"github.com/aws/aws-sdk-go-v2/service/ec2"
 	"github.com/aws/aws-sdk-go-v2/service/ec2/types"
+	"github.com/padok-team/yatas-aws/logger"
 )
 
 type EC2GetObjectAPI interface {
@@ -16,7 +16,7 @@ func GetEC2s(svc EC2GetObjectAPI) []types.Instance {
 	input := &ec2.DescribeInstancesInput{}
 	result, err := svc.DescribeInstances(context.TODO(), input)
 	if err != nil {
-		fmt.Println(err)
+		logger.Logger.Error(err.Error())
 		// Return an empty list
 		return []types.Instance{}
 	}
@@ -31,7 +31,7 @@ func GetEC2s(svc EC2GetObjectAPI) []types.Instance {
 		input.NextToken = result.NextToken
 		result, err = svc.DescribeInstances(context.TODO(), input)
 		if err != nil {
-			fmt.Println(err)
+			logger.Logger.Error(err.Error())
 			// Return an empty list of instances
 			return []types.Instance{}
 		}

diff --git a/aws/ecr/getter.go b/aws/ecr/getter.go
@@ -2,11 +2,11 @@ package ecr
 
 import (
 	"context"
-	"fmt"
 
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/service/ecr"
 	"github.com/aws/aws-sdk-go-v2/service/ecr/types"
+	"github.com/padok-team/yatas-aws/logger"
 )
 
 func GetECRs(s aws.Config) []types.Repository {
@@ -17,7 +17,7 @@ func GetECRs(s aws.Config) []types.Repository {
 	}
 	result, err := svc.DescribeRepositories(context.TODO(), input)
 	if err != nil {
-		fmt.Println(err)
+		logger.Logger.Error(err.Error())
 		// Return an empty list
 		return []types.Repository{}
 	}
@@ -27,7 +27,7 @@ func GetECRs(s aws.Config) []types.Repository {
 			input.NextToken = result.NextToken
 			result, err = svc.DescribeRepositories(context.TODO(), input)
 			if err != nil {
-				fmt.Println(err)
+				logger.Logger.Error(err.Error())
 				// Return an empty list
 				return []types.Repository{}
 			}

diff --git a/aws/eks/getter.go b/aws/eks/getter.go
@@ -2,11 +2,11 @@ package eks
 
 import (
 	"context"
-	"fmt"
 
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/service/eks"
 	"github.com/aws/aws-sdk-go-v2/service/eks/types"
+	"github.com/padok-team/yatas-aws/logger"
 )
 
 type EKSGetObjectAPI interface {
@@ -18,7 +18,7 @@ func GetClusters(svc EKSGetObjectAPI) []types.Cluster {
 	input := &eks.ListClustersInput{}
 	result, err := svc.ListClusters(context.TODO(), input)
 	if err != nil {
-		fmt.Println(err)
+		logger.Logger.Error(err.Error())
 		// Return an empty list
 		return []types.Cluster{}
 	}
@@ -34,7 +34,7 @@ func GetClusters(svc EKSGetObjectAPI) []types.Cluster {
 		input.NextToken = result.NextToken
 		result, err = svc.ListClusters(context.TODO(), input)
 		if err != nil {
-			fmt.Println(err)
+			logger.Logger.Error(err.Error())
 			// Return an empty list of instances
 			return []types.Cluster{}
 		}
@@ -49,7 +49,7 @@ func GetClusters(svc EKSGetObjectAPI) []types.Cluster {
 		}
 		result, err := svc.DescribeCluster(context.TODO(), input)
 		if err != nil {
-			fmt.Println(err)
+			logger.Logger.Error(err.Error())
 			// Return an empty list of instances
 			return []types.Cluster{}
 		}

diff --git a/aws/guardduty/getter.go b/aws/guardduty/getter.go
@@ -2,18 +2,18 @@ package guardduty
 
 import (
 	"context"
-	"fmt"
 
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/service/guardduty"
+	"github.com/padok-team/yatas-aws/logger"
 )
 
 func GetDetectors(s aws.Config) []string {
 	svc := guardduty.NewFromConfig(s)
 	input := &guardduty.ListDetectorsInput{}
 	result, err := svc.ListDetectors(context.TODO(), input)
 	if err != nil {
-		fmt.Println(err)
+		logger.Logger.Error(err.Error())
 		// Return an empty list
 		return []string{}
 	}