Make a table fits your terminal width

[xqm32]

Can automatically wrap text. Not sure if this is a good way, but at least it works! 😜

package main

import (
	"os"

	"github.com/charmbracelet/lipgloss"
	"github.com/charmbracelet/lipgloss/table"
	"github.com/samber/lo"
	"golang.org/x/term"
)

// Table returns a lipgloss table with the given headers and rows.
func Table(headers []string, rows [][]string) *table.Table {
	// Get the terminal width
	termWidth, _, err := term.GetSize(int(os.Stdout.Fd()))
	if err != nil {
		termWidth = 80
	}

	// Find the widest cell in each column
	colWidths := lo.Map(headers, headerWidth)
	for _, row := range rows {
		for i, cell := range row {
			if lipgloss.Width(cell) > colWidths[i] {
				colWidths[i] = lipgloss.Width(cell)
			}
		}
	}

	// Find the widest column
	maxCol := 0
	for i, width := range colWidths {
		if width > colWidths[maxCol] {
			maxCol = i
		}
	}

	// If the table is too wide, shrink the widest column
	totalWidth := lo.Sum(colWidths) + len(headers) - 1
	if totalWidth > termWidth {
		colWidths[maxCol] -= totalWidth - termWidth
	}

	return table.New().
		Border(lipgloss.HiddenBorder()).
		BorderHeader(false).
		BorderLeft(false).BorderRight(false).
		BorderTop(false).BorderBottom(false).
		Headers(headers...).
		Rows(rows...).
		StyleFunc(func(row, col int) lipgloss.Style {
			style := lipgloss.NewStyle()
			if row == table.HeaderRow {
				return style.Bold(true)
			}
			return style.Width(colWidths[col])
		})
}

func headerWidth(header string, _ int) int {
	return lipgloss.Width(header)
}