mirror of
https://github.com/crowdsecurity/crowdsec.git
synced 2025-05-14 21:33:54 +02:00
147 lines
3.2 KiB
Go
147 lines
3.2 KiB
Go
package main
|
|
|
|
// transisional file to keep (minimal) backwards compatibility with the old table
|
|
// we can migrate the code to the new dependency later, it can already use the Writer interface
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/jedib0t/go-pretty/v6/table"
|
|
"github.com/jedib0t/go-pretty/v6/text"
|
|
)
|
|
|
|
type Table struct {
|
|
Writer table.Writer
|
|
output io.Writer
|
|
align []text.Align
|
|
alignHeader []text.Align
|
|
}
|
|
|
|
func newTable(out io.Writer) *Table {
|
|
if out == nil {
|
|
panic("newTable: out is nil")
|
|
}
|
|
|
|
t := table.NewWriter()
|
|
|
|
// colorize output, use unicode box characters
|
|
fancy := shouldWeColorize()
|
|
|
|
color := table.ColorOptions{}
|
|
|
|
if fancy {
|
|
color.Header = text.Colors{text.Italic}
|
|
color.Border = text.Colors{text.FgHiBlack}
|
|
color.Separator = text.Colors{text.FgHiBlack}
|
|
}
|
|
|
|
// no upper/lower case transformations
|
|
format := table.FormatOptions{}
|
|
|
|
box := table.StyleBoxDefault
|
|
if fancy {
|
|
box = table.StyleBoxRounded
|
|
}
|
|
|
|
style := table.Style{
|
|
Box: box,
|
|
Color: color,
|
|
Format: format,
|
|
HTML: table.DefaultHTMLOptions,
|
|
Options: table.OptionsDefault,
|
|
Title: table.TitleOptionsDefault,
|
|
}
|
|
|
|
t.SetStyle(style)
|
|
|
|
return &Table{
|
|
Writer: t,
|
|
output: out,
|
|
align: make([]text.Align, 0),
|
|
alignHeader: make([]text.Align, 0),
|
|
}
|
|
}
|
|
|
|
func newLightTable(output io.Writer) *Table {
|
|
t := newTable(output)
|
|
s := t.Writer.Style()
|
|
s.Box.Left = ""
|
|
s.Box.LeftSeparator = ""
|
|
s.Box.TopLeft = ""
|
|
s.Box.BottomLeft = ""
|
|
s.Box.Right = ""
|
|
s.Box.RightSeparator = ""
|
|
s.Box.TopRight = ""
|
|
s.Box.BottomRight = ""
|
|
s.Options.SeparateRows = false
|
|
s.Options.SeparateFooter = false
|
|
s.Options.SeparateHeader = true
|
|
s.Options.SeparateColumns = false
|
|
|
|
return t
|
|
}
|
|
|
|
//
|
|
// wrapper methods for backwards compatibility
|
|
//
|
|
|
|
// setColumnConfigs must be called right before rendering,
|
|
// to allow for setting the alignment like the old API
|
|
func (t *Table) setColumnConfigs() {
|
|
configs := []table.ColumnConfig{}
|
|
// the go-pretty table does not expose the names or number of columns
|
|
for i := 0; i < len(t.align); i++ {
|
|
configs = append(configs, table.ColumnConfig{
|
|
Number: i + 1,
|
|
AlignHeader: t.alignHeader[i],
|
|
Align: t.align[i],
|
|
WidthMax: 60,
|
|
WidthMaxEnforcer: text.WrapSoft,
|
|
})
|
|
}
|
|
t.Writer.SetColumnConfigs(configs)
|
|
}
|
|
|
|
func (t *Table) Render() {
|
|
// change default options for backwards compatibility.
|
|
// we do this late to allow changing the alignment like the old API
|
|
t.setColumnConfigs()
|
|
fmt.Fprintln(t.output, t.Writer.Render())
|
|
}
|
|
|
|
func (t *Table) SetHeaders(str ...string) {
|
|
row := table.Row{}
|
|
t.align = make([]text.Align, len(str))
|
|
t.alignHeader = make([]text.Align, len(str))
|
|
|
|
for i, v := range str {
|
|
row = append(row, v)
|
|
t.align[i] = text.AlignLeft
|
|
t.alignHeader[i] = text.AlignCenter
|
|
}
|
|
|
|
t.Writer.AppendHeader(row)
|
|
}
|
|
|
|
func (t *Table) AddRow(str ...string) {
|
|
row := table.Row{}
|
|
for _, v := range str {
|
|
row = append(row, v)
|
|
}
|
|
|
|
t.Writer.AppendRow(row)
|
|
}
|
|
|
|
func (t *Table) SetRowLines(rowLines bool) {
|
|
t.Writer.Style().Options.SeparateRows = rowLines
|
|
}
|
|
|
|
func (t *Table) SetAlignment(align ...text.Align) {
|
|
// align can be shorter than t.align, it will leave the default value
|
|
copy(t.align, align)
|
|
}
|
|
|
|
func (t *Table) SetHeaderAlignment(align ...text.Align) {
|
|
copy(t.alignHeader, align)
|
|
}
|