driftctl/logger/plugin_logger.go

package logger

import (
	"io"
	"log"

	"github.com/hashicorp/go-hclog"
	"github.com/sirupsen/logrus"
)

type terraformPluginFormatter struct {
	logrus.Formatter
}

func (f *terraformPluginFormatter) Format(entry *logrus.Entry) ([]byte, error) {
	entry.Message = "[TerraformPlugin] " + entry.Message
	return f.Formatter.Format(entry)
}

type TerraformPluginLogger struct {
	logger *logrus.Logger
}

func NewTerraformPluginLogger() TerraformPluginLogger {
	config := getConfig()
	logger := logrus.New()
	logger.SetLevel(logrus.ErrorLevel)
	logger.SetReportCaller(false)
	logger.SetFormatter(&terraformPluginFormatter{Formatter: config.Formatter})

	// Disable terraform provider log if we are not in trace level
	if config.Level == logrus.TraceLevel {
		logger.SetLevel(logrus.TraceLevel)
	}

	return TerraformPluginLogger{logger}
}

func (t TerraformPluginLogger) Trace(msg string, args ...interface{}) {
	t.logger.Trace(msg, args)
}

func (t TerraformPluginLogger) Debug(msg string, args ...interface{}) {
	t.Trace(msg, args)
}

func (t TerraformPluginLogger) Info(msg string, args ...interface{}) {
	t.Trace(msg, args)
}

func (t TerraformPluginLogger) Warn(msg string, args ...interface{}) {
	t.Trace(msg, args)
}

func (t TerraformPluginLogger) Error(msg string, args ...interface{}) {
	t.Trace(msg, args)
}

func (t TerraformPluginLogger) IsTrace() bool {
	return true
}

func (t TerraformPluginLogger) IsDebug() bool {
	return false
}

func (t TerraformPluginLogger) IsInfo() bool {
	return false
}

func (t TerraformPluginLogger) IsWarn() bool {
	return false
}

func (t TerraformPluginLogger) IsError() bool {
	return false
}

func (t TerraformPluginLogger) With(args ...interface{}) hclog.Logger {
	return t
}

func (t TerraformPluginLogger) Named(name string) hclog.Logger {
	return t
}

func (t TerraformPluginLogger) ResetNamed(name string) hclog.Logger {
	return t
}

func (t TerraformPluginLogger) SetLevel(level hclog.Level) {}

func (t TerraformPluginLogger) StandardLogger(opts *hclog.StandardLoggerOptions) *log.Logger {
	stdLogger := log.New(t.logger.Writer(), "", log.Flags())
	stdLogger.SetOutput(t.logger.Writer())
	return stdLogger
}

func (t TerraformPluginLogger) StandardWriter(opts *hclog.StandardLoggerOptions) io.Writer {
	return t.logger.Writer()
}

func (t TerraformPluginLogger) Log(level hclog.Level, msg string, args ...interface{}) {
	t.logger.Log(logrus.TraceLevel, msg, args)
}

func (t TerraformPluginLogger) ImpliedArgs() []interface{} {
	return nil
}

func (t TerraformPluginLogger) Name() string {
	return "TerraformPlugin"
}