bouncer/internal/format/table/writer.go

package table

import (
	"io"

	"forge.cadoles.com/cadoles/bouncer/internal/format"
	"github.com/jedib0t/go-pretty/v6/table"
)

const Format format.Format = "table"

const DefaultCompactModeMaxColumnWidth = 30

func init() {
	format.Register(Format, NewWriter(DefaultCompactModeMaxColumnWidth))
}

type Writer struct {
	compactModeMaxColumnWidth int
}

// Write implements format.Writer.
func (w *Writer) Write(writer io.Writer, hints format.Hints, data ...any) error {
	t := table.NewWriter()

	t.SetOutputMirror(writer)

	var props []format.Prop

	if hints.Props != nil {
		props = hints.Props
	} else {
		if len(data) > 0 {
			props = getProps(data[0])
		} else {
			props = make([]format.Prop, 0)
		}
	}

	labels := table.Row{}

	for _, p := range props {
		labels = append(labels, p.Label())
	}

	t.AppendHeader(labels)

	isCompactMode := hints.OutputMode == format.OutputModeCompact

	for _, d := range data {
		row := table.Row{}

		for _, p := range props {
			value := getFieldValue(d, p.Name())

			compactModeMaxColumnWidth := format.PropHint(p,
				hintCompactModeMaxColumnWidth,
				w.compactModeMaxColumnWidth,
			)

			if isCompactMode && len(value) > compactModeMaxColumnWidth {
				value = value[:compactModeMaxColumnWidth] + "..."
			}

			row = append(row, value)
		}

		t.AppendRow(row)
	}

	t.Render()

	return nil
}

func NewWriter(compactModeMaxColumnWidth int) *Writer {
	return &Writer{compactModeMaxColumnWidth}
}

var _ format.Writer = &Writer{}