consul/command/cli/formatting.go

package cli

import (
	"github.com/fatih/color"
	"github.com/olekukonko/tablewriter"
)

// Functions to set up colorized output.
const (
	noColor = -1
)

func colorize(message string, uc UiColor) string {
	if uc.Code == noColor {
		return message
	}

	attr := []color.Attribute{color.Attribute(uc.Code)}
	if uc.Bold {
		attr = append(attr, color.Bold)
	}

	return color.New(attr...).SprintFunc()(message)
}

// UiColor is a posix shell color code to use.
type UiColor struct {
	Code int
	Bold bool
}

// A list of colors that are useful. These are all non-bolded by default.
var (
	UiColorNone    UiColor = UiColor{noColor, false}
	UiColorRed             = UiColor{int(color.FgHiRed), false}
	UiColorGreen           = UiColor{int(color.FgHiGreen), false}
	UiColorYellow          = UiColor{int(color.FgHiYellow), false}
	UiColorBlue            = UiColor{int(color.FgHiBlue), false}
	UiColorMagenta         = UiColor{int(color.FgHiMagenta), false}
	UiColorCyan            = UiColor{int(color.FgHiCyan), false}
)

// Functions that are useful for table output.
const (
	Yellow = "yellow"
	Green  = "green"
	Red    = "red"
)

var colorMapping = map[string]int{
	Green:  tablewriter.FgGreenColor,
	Yellow: tablewriter.FgYellowColor,
	Red:    tablewriter.FgRedColor,
}

// Table is passed to UI.Table to provide a nicely formatted table.
type Table struct {
	Headers []string
	Rows    [][]Cell
}

// Cell is a single entry for a table.
type Cell struct {
	Value string
	Color string
}

// NewTable creates a new Table structure that can be used with UI.Table.
func NewTable(headers ...string) *Table {
	return &Table{
		Headers: headers,
	}
}

// AddRow adds a row to the table.
func (t *Table) AddRow(cols []string, colors []string) {
	var row []Cell

	for i, col := range cols {
		if i < len(colors) {
			row = append(row, Cell{Value: col, Color: colors[i]})
		} else {
			row = append(row, Cell{Value: col})
		}
	}

	t.Rows = append(t.Rows, row)
}