package log

import (
	"bytes"
	"os"

	"github.com/sirupsen/logrus"
)

var Logger = logrus.New()

type CustomTextFormatter struct {
    logrus.TextFormatter
}

func (f *CustomTextFormatter) Format(entry *logrus.Entry) ([]byte, error) {
	if entry.Level != logrus.InfoLevel {
		return f.TextFormatter.Format(entry)
	}

  var b bytes.Buffer
  b.WriteString(entry.Message)
  b.WriteByte('\n')
  return b.Bytes(), nil
}


func InitLogger(debug bool) {
	if debug {
		Logger.SetFormatter(&logrus.TextFormatter{
			DisableLevelTruncation: true,
			PadLevelText: true,
			DisableTimestamp: true,
		})
		Logger.SetLevel(logrus.DebugLevel)
	} else {
		Logger.SetFormatter(&CustomTextFormatter{
			logrus.TextFormatter{
				DisableLevelTruncation: true,
				PadLevelText: true,
				DisableTimestamp: true,
			},
		})
		Logger.SetLevel(logrus.InfoLevel)
	}

  Logger.SetOutput(os.Stdout)
}