package text

import (
	"grim"
	"html"
	"strings"
)

var nonRenderTags = map[string]bool{
	"head":  true,
	"meta":  true,
	"link":  true,
	"title": true,
	"style": true,
}

func Init() grim.Transformer {
	return grim.Transformer{
		Selector: func(n *grim.Node, c *grim.CSS) bool {
			if len(strings.TrimSpace(n.InnerText())) > 0 && !grim.ChildrenHaveText(n) {
				return true
			} else {
				return false
			}
		},
		Handler: func(n *grim.Node, c *grim.CSS) *grim.Node {
			if nonRenderTags[n.TagName()] {
				return n
			}
			words := strings.Split(strings.TrimSpace(DecodeHTMLEscapes(n.InnerText())), " ")
			n.InnerText("")
			if n.GetStyle("display") == "inline" {
				n.InnerText(DecodeHTMLEscapes(words[0]))
				for i := 0; i < len(words)-1; i++ {
					// Add the words backwards because you are inserting adjacent to the parent
					a := (len(words) - 1) - i
					if len(strings.TrimSpace(words[a])) > 0 {
						el := n.CreateElement("text")
						el.InnerText(words[a])
						el.SetStyle("display", "inline")
						el.SetStyle("font-size", "1em")
						n.Parent().InsertAfter(&el, n)
					}
				}

			} else {
				for i := 0; i < len(words); i++ {
					if len(strings.TrimSpace(words[i])) > 0 {
						el := n.CreateElement("text")
						el.InnerText(words[i])
						el.SetStyle("display", "inline")
						el.SetStyle("font-size", "1em")
						n.AppendChild(&el)
					}
				}
			}
			return n
		},
	}
}

func DecodeHTMLEscapes(input string) string {
	return html.UnescapeString(input)
}