142 lines
3.1 KiB
Go
142 lines
3.1 KiB
Go
package tprint
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"git.milar.in/milarin/gmath"
|
|
"git.milar.in/milarin/slices"
|
|
"github.com/fatih/color"
|
|
)
|
|
|
|
type Table struct {
|
|
head [][]string
|
|
|
|
rowSep bool
|
|
headheight int
|
|
maxcols int
|
|
|
|
colwidth []int
|
|
rowheight []int
|
|
|
|
data [][][]string
|
|
}
|
|
|
|
func NewTable(head ...string) *Table {
|
|
thead := slices.Map(head, func(s string) []string { return strings.Split(s, "\n") })
|
|
return &Table{
|
|
head: thead,
|
|
colwidth: slices.Map(head, func(s string) int { return strLen(s) }),
|
|
headheight: len(slices.Search(thead, maxLengthSlice[string])),
|
|
rowheight: []int{},
|
|
maxcols: len(thead),
|
|
}
|
|
}
|
|
|
|
func (t *Table) AddRow(row ...interface{}) {
|
|
irow := slices.Map(row, func(v interface{}) []string { return strings.Split(fmt.Sprint(v), "\n") })
|
|
t.data = append(t.data, irow)
|
|
|
|
rowheight := len(slices.Search(irow, maxLengthSlice[string]))
|
|
t.rowheight = append(t.rowheight, rowheight)
|
|
if rowheight > 1 {
|
|
t.rowSep = true
|
|
}
|
|
|
|
t.maxcols = gmath.Max(t.maxcols, len(irow))
|
|
|
|
for i, cell := range irow {
|
|
maxLineLength := strLen(slices.Search(cell, maxLengthStr))
|
|
if i < len(t.colwidth) {
|
|
t.colwidth[i] = gmath.Max(t.colwidth[i], maxLineLength)
|
|
} else {
|
|
t.colwidth = append(t.colwidth, maxLineLength)
|
|
}
|
|
|
|
if i >= len(t.head) {
|
|
headname := "unknown column"
|
|
t.head = append(t.head, []string{headname})
|
|
t.colwidth[i] = gmath.Max(t.colwidth[i], strLen(headname))
|
|
}
|
|
}
|
|
}
|
|
|
|
func (t *Table) String() string {
|
|
b := new(strings.Builder)
|
|
|
|
b.WriteRune('┌')
|
|
for i, colwidth := range t.colwidth {
|
|
b.WriteString(strings.Repeat("─", colwidth))
|
|
if i < len(t.colwidth)-1 {
|
|
b.WriteRune('┬')
|
|
}
|
|
}
|
|
b.WriteRune('┐')
|
|
b.WriteRune('\n')
|
|
|
|
bold := color.New(color.Bold)
|
|
t.printRow(b, bold.Sprint, t.head, t.headheight)
|
|
|
|
if len(t.data) > 0 {
|
|
t.addNextCellLine(b)
|
|
} else {
|
|
t.addLastCellLine(b)
|
|
return b.String()
|
|
}
|
|
|
|
for i, row := range t.data {
|
|
t.printRow(b, fmt.Sprint, row, t.rowheight[i])
|
|
if t.rowSep && i < len(t.data)-1 {
|
|
t.addNextCellLine(b)
|
|
}
|
|
}
|
|
|
|
t.addLastCellLine(b)
|
|
|
|
return b.String()
|
|
}
|
|
|
|
func (t *Table) printRow(b *strings.Builder, printFunc func(...interface{}) string, rowData [][]string, rowHeight int) {
|
|
for lineIndex := 0; lineIndex < rowHeight; lineIndex++ {
|
|
b.WriteRune('│')
|
|
for colIndex := 0; colIndex < t.maxcols; colIndex++ {
|
|
line := ""
|
|
|
|
if colIndex < len(rowData) {
|
|
cell := rowData[colIndex]
|
|
if lineIndex < len(cell) {
|
|
line = cell[lineIndex]
|
|
}
|
|
}
|
|
|
|
b.WriteString(printFunc(padStringRight(line, ' ', t.colwidth[colIndex])))
|
|
b.WriteRune('│')
|
|
}
|
|
b.WriteRune('\n')
|
|
}
|
|
}
|
|
|
|
func (t *Table) addNextCellLine(b *strings.Builder) {
|
|
b.WriteRune('├')
|
|
for i, colwidth := range t.colwidth {
|
|
b.WriteString(strings.Repeat("─", colwidth))
|
|
if i < len(t.colwidth)-1 {
|
|
b.WriteRune('┼')
|
|
}
|
|
}
|
|
b.WriteRune('┤')
|
|
b.WriteRune('\n')
|
|
}
|
|
|
|
func (t *Table) addLastCellLine(b *strings.Builder) {
|
|
b.WriteRune('└')
|
|
for i, colwidth := range t.colwidth {
|
|
b.WriteString(strings.Repeat("─", colwidth))
|
|
if i < len(t.colwidth)-1 {
|
|
b.WriteRune('┴')
|
|
}
|
|
}
|
|
b.WriteRune('┘')
|
|
b.WriteRune('\n')
|
|
}
|