tab.go

package printers

import (
	"context"
	"fmt"
	"io"
	"strings"
	"text/tabwriter"

	"github.com/fatih/color"

	"github.com/golangci/golangci-lint/pkg/logutils"
	"github.com/golangci/golangci-lint/pkg/result"
)

type Tab struct {
	printLinterName bool
	log             logutils.Log
}

func NewTab(printLinterName bool, log logutils.Log) *Tab {
	return &Tab{
		printLinterName: printLinterName,
		log:             log,
	}
}

func (p Tab) SprintfColored(ca color.Attribute, format string, args ...interface{}) string {
	c := color.New(ca)
	return c.Sprintf(format, args...)
}

func (p *Tab) Print(ctx context.Context, issues []result.Issue) error {
	w := tabwriter.NewWriter(logutils.StdOut, 0, 0, 2, ' ', 0)

	for i := range issues {
		p.printIssue(&issues[i], w)
		p.printSuggestedEdits(&issues[i], w)
	}

	if err := w.Flush(); err != nil {
		p.log.Warnf("Can't flush tab writer: %s", err)
	}

	return nil
}

func (p Tab) printIssue(i *result.Issue, w io.Writer) {
	text := p.SprintfColored(color.FgRed, "%s", i.Text)
	if p.printLinterName {
		text = fmt.Sprintf("%s\t%s", i.FromLinter, text)
	}

	pos := p.SprintfColored(color.Bold, "%s:%d", i.FilePath(), i.Line())
	if i.Pos.Column != 0 {
		pos += fmt.Sprintf(":%d", i.Pos.Column)
	}

	fmt.Fprintf(w, "%s\t%s\n", pos, text)
}

func (p Tab) printSuggestedEdits(i *result.Issue, w io.Writer) {
	var text string
	if len(i.SuggestedFixes) > 0 {
		for _, fix := range i.SuggestedFixes {
			text += p.SprintfColored(color.FgRed, "%s\n", strings.TrimSpace(fix.Message))
			var suggestedEdits []string
			for _, textEdit := range fix.TextEdits {
				suggestedEdits = append(suggestedEdits, strings.TrimSpace(textEdit.NewText))
			}
			text += strings.Join(suggestedEdits, "\n") + "\n"
		}
	}

	fmt.Fprintln(w, text)
}