package html

import . "github.com/vektah/goparsify"

func Parse(input string) (result Node, remaining string, err error) {
	return ParseString(tag, input)
}

type Tag struct {
	Name       string
	Attributes map[string]string
	Body       []Node
}

var (
	tag Parser

	identifier = Merge(And(Range("a-z", 1, 1), Range("a-zA-Z0-9", 0)))
	text       = CharRunUntil("<>")

	element  = Any(text, &tag)
	elements = Kleene(element)
	//attr := And(identifier, equal, String())
	attr  = And(identifier, WS, "=", WS, `"test"`)
	attrs = Map(Kleene(attr, WS), func(node Node) Node {
		nodes := node.([]Node)
		attr := map[string]string{}

		for _, attrNode := range nodes {
			attrNodes := attrNode.([]Node)
			attr[attrNodes[0].(string)] = attrNodes[2].(string)
		}

		return attr
	})

	tstart = And("<", identifier, attrs, ">")
	tend   = And("</", identifier, ">")
)

func init() {
	tag = Map(And(tstart, elements, tend), func(node Node) Node {
		nodes := node.([]Node)
		openTag := nodes[0].([]Node)
		return Tag{
			Name:       openTag[1].(string),
			Attributes: openTag[2].(map[string]string),
			Body:       nodes[1].([]Node),
		}

	})
}