helpers/formatter/runner_text_formatter.go

package formatter

import (
	"bytes"
	"fmt"
	"sort"

	"github.com/Sirupsen/logrus"
	"gitlab.com/gitlab-org/gitlab-ci-multi-runner/helpers"
)

type RunnerTextFormatter struct {
	// Force disabling colors.
	DisableColors bool

	// The fields are sorted by default for a consistent output. For applications
	// that log extremely frequently and don't use the JSON formatter this may not
	// be desired.
	DisableSorting bool
}

func (f *RunnerTextFormatter) Format(entry *logrus.Entry) ([]byte, error) {
	var keys = make([]string, 0, len(entry.Data))
	for k := range entry.Data {
		keys = append(keys, k)
	}

	if !f.DisableSorting {
		sort.Strings(keys)
	}

	b := bytes.Buffer{}
	f.printColored(&b, entry, keys)
	b.WriteByte('\n')
	return b.Bytes(), nil
}

func (f *RunnerTextFormatter) printColored(b *bytes.Buffer, entry *logrus.Entry, keys []string) {
	var levelColor, levelText string
	switch entry.Level {
	case logrus.DebugLevel:
		levelColor = helpers.ANSI_BOLD_WHITE
	case logrus.WarnLevel:
		levelColor = helpers.ANSI_YELLOW
		levelText = "WARNING: "
	case logrus.ErrorLevel:
		levelColor = helpers.ANSI_BOLD_RED
		levelText = "ERROR: "
	case logrus.FatalLevel:
		levelColor = helpers.ANSI_BOLD_RED
		levelText = "FATAL: "
	case logrus.PanicLevel:
		levelColor = helpers.ANSI_BOLD_RED
		levelText = "PANIC: "
	default:
	}

	resetColor := helpers.ANSI_RESET

	if f.DisableColors {
		levelColor = ""
		resetColor = ""
	}

	indentLength := 50 - len(levelText)

	fmt.Fprintf(b, "%s%s%-*s%s ", levelColor, levelText, indentLength, entry.Message, resetColor)
	for _, k := range keys {
		v := entry.Data[k]
		fmt.Fprintf(b, " %s%s%s=%v", levelColor, k, resetColor, v)
	}
}

func SetRunnerFormatter() {
	logrus.SetFormatter(&RunnerTextFormatter{})
}