package h

import (
	"fmt"
	"github.com/maddalax/htmgo/framework/hx"
	"html"
	"html/template"
	"strings"
)

type CustomElement = string

var (
	CachedNodeTag        CustomElement = "htmgo_cache_node"
	CachedNodeByKeyEntry CustomElement = "htmgo_cached_node_by_key_entry"
)

/*
*
void tags are tags that cannot have children
*/
var voidTags = map[string]bool{
	"area":   true,
	"base":   true,
	"br":     true,
	"col":    true,
	"embed":  true,
	"hr":     true,
	"img":    true,
	"input":  true,
	"link":   true,
	"meta":   true,
	"source": true,
	"track":  true,
	"wbr":    true,
}

type RenderContext struct {
	builder *strings.Builder
	scripts []string
	next    any
	prev    any
}

func (ctx *RenderContext) PrevIsAttribute() bool {
	_, ok := ctx.prev.(*AttributeR)
	return ok
}

func (ctx *RenderContext) AddScript(funcName string, body string) {
	script := fmt.Sprintf(`
	<script id="%s">
		function %s(self) {
				%s
		}
	</script>`, funcName, funcName, body)
	ctx.scripts = append(ctx.scripts, script)
}

func each[T any](ctx *RenderContext, arr []T, cb func(T)) {
	for i, r := range arr {
		if i == len(arr)-1 {
			ctx.next = nil
		} else {
			ctx.next = arr[i+1]
		}
		cb(r)
	}
}

func eachAttrMap(ctx *RenderContext, m *AttributeMapOrdered, cb func(string, string)) {
	entries := m.Entries()
	for i, entry := range entries {
		if i == len(entries)-1 {
			ctx.next = nil
		} else {
			ctx.next = entries[i+1]
		}
		cb(entry.Key, entry.Value)
	}
}

func (node *Element) Render(context *RenderContext) {
	// some elements may not have a tag, such as a Fragment

	if node.tag == CachedNodeTag {
		meta := node.meta.(*CachedNode)
		meta.Render(context)
		return
	}

	if node.tag == CachedNodeByKeyEntry {
		meta := node.meta.(*ByKeyEntry)
		meta.Render(context)
		return
	}

	if node.tag != "" {
		context.builder.WriteString("<")
		context.builder.WriteString(node.tag)
		eachAttrMap(context, node.attributes, func(key string, value string) {
			NewAttribute(key, value).Render(context)
		})
	}

	totalChildren := 0
	shouldFlatten := false
	for _, child := range node.children {
		switch c := child.(type) {
		case *ChildList:
			shouldFlatten = true
			totalChildren += len(c.Children)
		default:
			totalChildren++
		}
	}

	if shouldFlatten {
		// first pass, flatten the children
		flatChildren := make([]Ren, totalChildren)
		index := 0
		for _, child := range node.children {
			switch c := child.(type) {
			case *ChildList:
				for _, ren := range c.Children {
					flatChildren[index] = ren
					index++
				}
			default:
				flatChildren[index] = child
				index++
			}
		}

		node.children = flatChildren
	}

	// second pass, render any attributes within the tag
	for _, child := range node.children {
		switch child.(type) {
		case *AttributeMapOrdered:
			child.Render(context)
		case *AttributeR:
			child.Render(context)
		case *LifeCycle:
			child.Render(context)
		}
	}

	// close the tag
	if node.tag != "" {
		if voidTags[node.tag] {
			context.builder.WriteString("/")
		}
		context.builder.WriteString(">")
	}

	// void elements do not have children
	if !voidTags[node.tag] {
		// render the children elements that are not attributes
		for _, child := range node.children {
			switch child.(type) {
			case *AttributeMapOrdered:
				continue
			case *AttributeR:
				continue
			case *LifeCycle:
				continue
			default:
				child.Render(context)
			}
		}
	}

	if node.tag != "" {
		renderScripts(context)
		if !voidTags[node.tag] {
			context.builder.WriteString("</")
			context.builder.WriteString(node.tag)
			context.builder.WriteString(">")
		}
	}
}

func renderScripts(context *RenderContext) {
	for _, script := range context.scripts {
		context.builder.WriteString(script)
		context.prev = script
	}
	context.scripts = []string{}
}

func (a *AttributeR) Render(context *RenderContext) {
	context.builder.WriteString(" ")
	context.builder.WriteString(a.Name)
	if a.Value != "" {
		context.builder.WriteString(`=`)
		context.builder.WriteString(`"`)
		context.builder.WriteString(html.EscapeString(a.Value))
		context.builder.WriteString(`"`)
	}
	context.prev = a
}

func (t *TextContent) Render(context *RenderContext) {
	context.builder.WriteString(template.HTMLEscapeString(t.Content))
	context.prev = t
}

func (r *RawContent) Render(context *RenderContext) {
	context.builder.WriteString(r.Content)
	context.prev = r
}

func (c *ChildList) Render(context *RenderContext) {
	for _, child := range c.Children {
		child.Render(context)
		context.prev = child
	}
}

func (j SimpleJsCommand) Render(context *RenderContext) {
	context.builder.WriteString(j.Command)
	context.prev = j
}

func (j ComplexJsCommand) Render(context *RenderContext) {
	context.builder.WriteString(j.Command)
	context.prev = j
}

func (p *Partial) Render(context *RenderContext) {
	p.Root.Render(context)
	context.prev = p
}

func (m *AttributeMapOrdered) Render(context *RenderContext) {
	eachAttrMap(context, m, func(key string, value string) {
		NewAttribute(key, value).Render(context)
	})
}

func (l *LifeCycle) fromAttributeMap(event string, key string, value string, context *RenderContext) {

	if key == hx.GetAttr || key == hx.PatchAttr || key == hx.PostAttr {
		HxTriggerString(hx.ToHtmxTriggerName(event)).Render(context)
	}

	Attribute(key, value).Render(context)
}

func (l *LifeCycle) Render(context *RenderContext) {
	m := make(map[string]string)

	for event, commands := range l.handlers {
		m[event] = ""
		each(context, commands, func(command Command) {
			switch c := command.(type) {
			case SimpleJsCommand:
				m[event] += fmt.Sprintf("%s;", c.Command)
			case ComplexJsCommand:
				context.AddScript(c.TempFuncName, c.Command)
				m[event] += fmt.Sprintf("%s(this);", c.TempFuncName)
			case *AttributeMapOrdered:
				eachAttrMap(context, c, func(k string, v string) {
					l.fromAttributeMap(event, k, v, context)
				})
			case *AttributeR:
				l.fromAttributeMap(event, c.Name, c.Value, context)
			}
		})
	}

	children := make([]Ren, 0)

	for event, value := range m {
		if value != "" {
			children = append(children, Attribute(event, value))
		}
	}

	if len(children) == 0 {
		return
	}

	Children(children...).Render(context)
}