changes per @ethanfrey comments

log/tm_logger.go

@@ -17,9 +17,14 @@ type tmLogger struct {
 	srcLogger kitlog.Logger
 }
 
+// Interface assertions
+var _ Logger = (*tmLogger)(nil)
+
 // NewTMTermLogger returns a logger that encodes msg and keyvals to the Writer
 // using go-kit's log as an underlying logger and our custom formatter. Note
 // that underlying logger could be swapped with something else.
+//
+// Default logging level is info. You can change it using SetLevel().
 func NewTMLogger(w io.Writer) Logger {
 	// Color by level value
 	colorFn := func(keyvals ...interface{}) term.FgBgColor {
@@ -58,3 +63,23 @@ func (l *tmLogger) Error(msg string, keyvals ...interface{}) error {
 	lWithLevel := level.Error(l.srcLogger)
 	return kitlog.With(lWithLevel, msgKey, msg).Log(keyvals...)
 }
+
+// With returns a new contextual logger with keyvals prepended to those passed
+// to calls to Info, Debug or Error.
+func (l *tmLogger) With(keyvals ...interface{}) Logger {
+	return &tmLogger{kitlog.With(l.srcLogger, keyvals...)}
+}
+
+// WithLevel returns a new logger with the level set to lvl.
+func (l *tmLogger) WithLevel(lvl string) Logger {
+	switch lvl {
+	case "info":
+		return &tmLogger{level.NewFilter(l.srcLogger, level.AllowInfo())}
+	case "debug":
+		return &tmLogger{level.NewFilter(l.srcLogger, level.AllowDebug())}
+	case "error":
+		return &tmLogger{level.NewFilter(l.srcLogger, level.AllowError())}
+	default:
+		panic(fmt.Sprintf("Unexpected level %v, expect either \"info\" or \"debug\" or \"error\"", lvl))
+	}
+}