luet/vendor/github.com/jedib0t/go-pretty/table/render_markdown.go
Ettore Di Giacinto 5b54aeb822 Update vendor
2020-11-23 19:14:07 +01:00

76 lines
2.0 KiB
Go

package table
import "strings"
// RenderMarkdown renders the Table in Markdown format. Example:
// | # | First Name | Last Name | Salary | |
// | ---:| --- | --- | ---:| --- |
// | 1 | Arya | Stark | 3000 | |
// | 20 | Jon | Snow | 2000 | You know nothing, Jon Snow! |
// | 300 | Tyrion | Lannister | 5000 | |
// | | | Total | 10000 | |
func (t *Table) RenderMarkdown() string {
t.initForRender()
var out strings.Builder
if t.numColumns > 0 {
if t.title != "" {
out.WriteString("# ")
out.WriteString(t.title)
}
t.markdownRenderRows(&out, t.rowsHeader, true, false)
t.markdownRenderRows(&out, t.rows, false, false)
t.markdownRenderRows(&out, t.rowsFooter, false, true)
if t.caption != "" {
out.WriteRune('\n')
out.WriteRune('_')
out.WriteString(t.caption)
out.WriteRune('_')
}
}
return t.render(&out)
}
func (t *Table) markdownRenderRow(out *strings.Builder, row rowStr, isSeparator bool) {
if len(row) > 0 {
// when working on line number 2 or more, insert a newline first
if out.Len() > 0 {
out.WriteRune('\n')
}
// render each column up to the max. columns seen in all the rows
out.WriteRune('|')
for colIdx := 0; colIdx < t.numColumns; colIdx++ {
if isSeparator {
out.WriteString(t.getAlign(colIdx, renderHint{}).MarkdownProperty())
} else {
var colStr string
if colIdx < len(row) {
colStr = row[colIdx]
}
out.WriteRune(' ')
if strings.Contains(colStr, "|") {
colStr = strings.Replace(colStr, "|", "\\|", -1)
}
if strings.Contains(colStr, "\n") {
colStr = strings.Replace(colStr, "\n", "<br/>", -1)
}
out.WriteString(colStr)
out.WriteRune(' ')
}
out.WriteRune('|')
}
}
}
func (t *Table) markdownRenderRows(out *strings.Builder, rows []rowStr, isHeader bool, isFooter bool) {
if len(rows) > 0 {
for idx, row := range rows {
t.markdownRenderRow(out, row, false)
if idx == len(rows)-1 && isHeader {
t.markdownRenderRow(out, t.rowSeparator, true)
}
}
}
}