implement quiet mode (#142)

goodwithtech · Sep 9, 2021 · deee6fe · deee6fe
1 parent d619c56
commit deee6fe
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 8 deletions.
diff --git a/cmd/dockle/main.go b/cmd/dockle/main.go
@@ -69,6 +69,10 @@ OPTIONS:
 			Name:  "debug, d",
 			Usage: "debug mode",
 		},
+		cli.BoolFlag{
+			Name:  "quiet, q",
+			Usage: "suppress log output",
+		},
 
 		// Registry flag
 		cli.DurationFlag{

diff --git a/pkg/log/logger.go b/pkg/log/logger.go
@@ -2,34 +2,43 @@ package log
 
 import (
 	"fmt"
-
 	"go.uber.org/zap"
 	"go.uber.org/zap/zapcore"
+	"os"
 )
 
 var (
 	Logger      *zap.SugaredLogger
 	debugOption bool
 )
 
-func InitLogger(debug bool) (err error) {
+func InitLogger(debug, quiet bool) (err error) {
 	debugOption = debug
-	Logger, err = newLogger(debug)
+	Logger, err = newLogger(debug, quiet)
 	if err != nil {
 		return fmt.Errorf("error in new logger: %w", err)
 	}
 	return nil
-
 }
 
-func newLogger(debug bool) (*zap.SugaredLogger, error) {
+func newLogger(debug, quiet bool) (*zap.SugaredLogger, error) {
 	level := zap.NewAtomicLevel()
 	if debug {
 		level.SetLevel(zapcore.DebugLevel)
 	} else {
 		level.SetLevel(zapcore.InfoLevel)
 	}
 
+	stdout := "stdout"
+	stderr := "stderr"
+	if quiet {
+		if _, err := os.Create(os.DevNull); err != nil {
+			return nil, err
+		}
+		stdout = os.DevNull
+		stderr = os.DevNull
+	}
+
 	myConfig := zap.Config{
 		Level:             level,
 		Encoding:          "console",
@@ -48,8 +57,8 @@ func newLogger(debug bool) (*zap.SugaredLogger, error) {
 			EncodeDuration: zapcore.StringDurationEncoder,
 			EncodeCaller:   zapcore.ShortCallerEncoder,
 		},
-		OutputPaths:      []string{"stdout"},
-		ErrorOutputPaths: []string{"stderr"},
+		OutputPaths:      []string{stdout},
+		ErrorOutputPaths: []string{stderr},
 	}
 	logger, err := myConfig.Build()
 	if err != nil {

diff --git a/pkg/run.go b/pkg/run.go
@@ -27,7 +27,8 @@ func Run(c *cli.Context) (err error) {
 	ctx, cancel := context.WithTimeout(context.Background(), c.Duration("timeout"))
 	defer cancel()
 	debug := c.Bool("debug")
-	if err = log.InitLogger(debug); err != nil {
+	quiet := c.Bool("quiet")
+	if err = log.InitLogger(debug, quiet); err != nil {
 		l.Fatal(err)
 	}