html.go

// Copyright 2015 Frustra. All rights reserved.
// Use of this source code is governed by the MIT
// license that can be found in the LICENSE file.

package bbcode

import (
	"html"
	"net/url"
	"sort"
	"strings"
)

// HTMLTag represents a DOM node.
type HTMLTag struct {
	Name     string
	Value    string
	Attrs    map[string]string
	Children []*HTMLTag
}

// NewHTMLTag creates a new HTMLTag with string contents specified by value.
func NewHTMLTag(value string) *HTMLTag {
	return &HTMLTag{
		Value:    value,
		Attrs:    make(map[string]string),
		Children: make([]*HTMLTag, 0),
	}
}

// The html representation of the tag with unsorted arguments.
func (t *HTMLTag) String() string {
	return t.Compile(false)
}

// The html representation of the tag with or without sorted arguments.
func (t *HTMLTag) Compile(sorted bool) string {
	var value string
	if len(t.Value) > 0 {
		value = html.EscapeString(t.Value)
	}

	var attrString string
	if sorted {
		var attrKeys []string
		for key := range t.Attrs {
			attrKeys = append(attrKeys, key)
		}
		sort.Strings(attrKeys)

		for _, key := range attrKeys {
			attrString += " " + key + `="` + strings.Replace(html.EscapeString(t.Attrs[key]), "\n", "", -1) + `"`
		}
	} else {
		for key, value := range t.Attrs {
			attrString += " " + key + `="` + strings.Replace(html.EscapeString(value), "\n", "", -1) + `"`
		}
	}

	if len(t.Children) > 0 {
		var childrenString string
		for _, child := range t.Children {
			childrenString += child.Compile(sorted)
		}
		if len(t.Name) > 0 {
			return value + "<" + t.Name + attrString + ">" + childrenString + "</" + t.Name + ">"
		} else {
			return value + childrenString
		}
	} else if len(t.Name) > 0 {
		return value + "<" + t.Name + attrString + ">"
	} else {
		return value
	}
}

func (t *HTMLTag) AppendChild(child *HTMLTag) *HTMLTag {
	if child == nil {
		t.Children = append(t.Children, NewHTMLTag(""))
	} else {
		t.Children = append(t.Children, child)
	}
	return t
}

func InsertNewlines(out *HTMLTag) {
	if strings.ContainsRune(out.Value, '\n') {
		parts := strings.Split(out.Value, "\n")
		for i, part := range parts {
			if i == 0 {
				out.Value = parts[i]
			} else {
				out.AppendChild(NewlineTag())
				if len(part) > 0 {
					out.AppendChild(NewHTMLTag(part))
				}
			}
		}
	}
}

// Returns a new HTMLTag representing a line break
func NewlineTag() *HTMLTag {
	var out = NewHTMLTag("")
	out.Name = "br"
	return out
}

func ValidURL(raw string) string {
	u, err := url.Parse(raw)
	if err != nil {
		return ""
	}
	return u.String()
}