139 lines
2.3 KiB
Go
139 lines
2.3 KiB
Go
package htmlx
|
|
|
|
import (
|
|
"github.com/d5/tengo/v2"
|
|
"strings"
|
|
"html"
|
|
"fmt"
|
|
)
|
|
|
|
const RawTag = "raw"
|
|
|
|
// The type implements basic
|
|
// way to structrize HTML elements.
|
|
type Element struct {
|
|
tengo.ObjectImpl
|
|
Tag string
|
|
Attr map[string] string
|
|
Children []*Element
|
|
// The value makes sense only if
|
|
// the tag is the "raw"
|
|
Content string
|
|
Final bool
|
|
}
|
|
|
|
func (el *Element) TypeName() string {
|
|
return "*HTMLElement"
|
|
}
|
|
|
|
// The method renders the element to it's
|
|
// HTML representation.
|
|
func (el *Element) String() string {
|
|
if el.Tag == RawTag {
|
|
return html.EscapeString(el.Content)
|
|
}
|
|
|
|
var b strings.Builder
|
|
|
|
fmt.Fprintf(&b, "<%s", el.Tag)
|
|
for k, v := range el.Attr {
|
|
fmt.Fprintf(&b, " %s=%q", k, v)
|
|
}
|
|
fmt.Fprint(&b, ">")
|
|
|
|
if el.Final {
|
|
return b.String()
|
|
}
|
|
|
|
for _, child := range el.Children {
|
|
if child == nil {
|
|
continue
|
|
}
|
|
fmt.Fprint(&b, child.String())
|
|
}
|
|
|
|
fmt.Fprintf(&b, "</%s>", el.Tag)
|
|
return b.String()
|
|
}
|
|
|
|
func MakeElements(args ...tengo.Object) ([]*Element, error) {
|
|
s := []*Element{}
|
|
for _, arg := range args {
|
|
el, ok := arg.(*Element)
|
|
if !ok {
|
|
str, ok := tengo.ToString(arg)
|
|
if ok {
|
|
s = append(s, &Element{
|
|
Tag: RawTag,
|
|
Content: str,
|
|
})
|
|
}
|
|
continue
|
|
}
|
|
s = append(s, el)
|
|
}
|
|
return s, nil
|
|
}
|
|
|
|
func (el *Element) SetBody(
|
|
args ...tengo.Object,
|
|
) (tengo.Object, error) {
|
|
els, err := MakeElements(args...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
el.Children = els
|
|
return el, nil
|
|
}
|
|
|
|
func (el *Element) Add(
|
|
args ...tengo.Object,
|
|
) (tengo.Object, error) {
|
|
s, err := MakeElements(args...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
el.Children = append(el.Children, s...)
|
|
return el, nil
|
|
}
|
|
|
|
func (el *Element) SetFinal(
|
|
args ...tengo.Object,
|
|
) (tengo.Object, error) {
|
|
if len(args) > 0 {
|
|
return nil, tengo.ErrWrongNumArguments
|
|
}
|
|
el.Final = true
|
|
return el, nil
|
|
}
|
|
|
|
func (el *Element) IndexGet(
|
|
index tengo.Object,
|
|
) (tengo.Object, error) {
|
|
arg, ok := tengo.ToString(index)
|
|
if !ok {
|
|
return nil, tengo.ErrInvalidIndexValueType
|
|
}
|
|
|
|
switch arg {
|
|
case "body" :
|
|
return &tengo.UserFunction{
|
|
Name: "element.body",
|
|
Value: el.SetBody,
|
|
}, nil
|
|
case "final" :
|
|
return &tengo.UserFunction{
|
|
Name: "element.final",
|
|
Value: el.SetFinal,
|
|
}, nil
|
|
case "add" :
|
|
return &tengo.UserFunction{
|
|
Name: "element.add",
|
|
Value: el.Add,
|
|
}, nil
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
|
|
|