mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-12-10 23:18:59 +08:00
108 lines
3.1 KiB
Go
108 lines
3.1 KiB
Go
// Copyright 2013 Apcera Inc. All rights reserved.
|
|
|
|
package termtables
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"html"
|
|
"strings"
|
|
)
|
|
|
|
type titleStyle int
|
|
|
|
const (
|
|
TitleAsCaption titleStyle = iota
|
|
TitleAsThSpan
|
|
)
|
|
|
|
// htmlStyleRules defines attributes which we can use, and might be set on a
|
|
// table by accessors, to influence the type of HTML which is output.
|
|
type htmlStyleRules struct {
|
|
title titleStyle
|
|
}
|
|
|
|
// HTML returns an HTML representations of the contents of one row of a table.
|
|
func (r *Row) HTML(tag string, style *renderStyle) string {
|
|
attrs := make([]string, len(r.cells))
|
|
elems := make([]string, len(r.cells))
|
|
for i := range r.cells {
|
|
if r.cells[i].alignment != nil {
|
|
switch *r.cells[i].alignment {
|
|
case AlignLeft:
|
|
attrs[i] = " align='left'"
|
|
case AlignCenter:
|
|
attrs[i] = " align='center'"
|
|
case AlignRight:
|
|
attrs[i] = " align='right'"
|
|
}
|
|
}
|
|
elems[i] = html.EscapeString(strings.TrimSpace(r.cells[i].Render(style)))
|
|
}
|
|
// WAG as to max capacity, plus a bit
|
|
buf := bytes.NewBuffer(make([]byte, 0, 8192))
|
|
buf.WriteString("<tr>")
|
|
for i := range elems {
|
|
fmt.Fprintf(buf, "<%s%s>%s</%s>", tag, attrs[i], elems[i], tag)
|
|
}
|
|
buf.WriteString("</tr>\n")
|
|
return buf.String()
|
|
}
|
|
|
|
func generateHtmlTitleRow(title interface{}, t *Table, style *renderStyle) string {
|
|
elContent := html.EscapeString(
|
|
strings.TrimSpace(CreateCell(t.title, &CellStyle{}).Render(style)),
|
|
)
|
|
|
|
switch style.htmlRules.title {
|
|
case TitleAsCaption:
|
|
return "<caption>" + elContent + "</caption>\n"
|
|
case TitleAsThSpan:
|
|
return fmt.Sprintf("<tr><th style=\"text-align: center\" colspan=\"%d\">%s</th></tr>\n",
|
|
style.columns, elContent)
|
|
default:
|
|
return "<!-- " + elContent + " -->"
|
|
}
|
|
}
|
|
|
|
// RenderHTML returns a string representation of a the table, suitable for
|
|
// inclusion as HTML elsewhere. Primary use-case controlling layout style
|
|
// is for inclusion into Markdown documents, documenting normal table use.
|
|
// Thus we leave the padding in place to have columns align when viewed as
|
|
// plain text and rely upon HTML ignoring extra whitespace.
|
|
func (t *Table) RenderHTML() (buffer string) {
|
|
// elements is already populated with row data
|
|
|
|
// generate the runtime style
|
|
style := createRenderStyle(t)
|
|
style.PaddingLeft = 0
|
|
style.PaddingRight = 0
|
|
|
|
// TODO: control CSS styles to suppress border based upon t.Style.SkipBorder
|
|
rowsText := make([]string, 0, len(t.elements)+6)
|
|
|
|
if t.title != nil || t.headers != nil {
|
|
rowsText = append(rowsText, "<thead>\n")
|
|
if t.title != nil {
|
|
rowsText = append(rowsText, generateHtmlTitleRow(t.title, t, style))
|
|
}
|
|
if t.headers != nil {
|
|
rowsText = append(rowsText, CreateRow(t.headers).HTML("th", style))
|
|
}
|
|
rowsText = append(rowsText, "</thead>\n")
|
|
}
|
|
|
|
rowsText = append(rowsText, "<tbody>\n")
|
|
// loop over the elements and render them
|
|
for i := range t.elements {
|
|
if row, ok := t.elements[i].(*Row); ok {
|
|
rowsText = append(rowsText, row.HTML("td", style))
|
|
} else {
|
|
rowsText = append(rowsText, fmt.Sprintf("<!-- unable to render line %d, unhandled type -->\n", i))
|
|
}
|
|
}
|
|
rowsText = append(rowsText, "</tbody>\n")
|
|
|
|
return "<table class=\"termtable\">\n" + strings.Join(rowsText, "") + "</table>\n"
|
|
}
|